1
0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-24 18:51:33 +00:00
ColossalAI/examples/tutorial/auto_parallel/config.py
Frank Lee e6943e2d11
[example] integrate autoparallel demo with CI ()
* [example] integrate autoparallel demo with CI

* polish code

* polish code

* polish code

* polish code
2023-01-12 16:26:42 +08:00

3 lines
31 B
Python

BATCH_SIZE = 32
NUM_EPOCHS = 2