mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-04-29 12:15:39 +00:00
e6943e2d11
* [example] integrate autoparallel demo with CI * polish code * polish code * polish code * polish code
3 lines
31 B
Python
3 lines
31 B
Python
BATCH_SIZE = 32
|
|
NUM_EPOCHS = 2
|