mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-04-30 04:35:17 +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
|