mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-04-26 17:53:08 +00:00
6 lines
101 B
Python
6 lines
101 B
Python
SEQ_LENGTH = 1024
|
|
BATCH_SIZE = 4
|
|
NUM_EPOCHS = 4
|
|
TPDEGREE = 2
|
|
USE_SHARD_INIT = False
|
|
placement = 'cpu' |