ColossalAI/examples/tutorial/auto_parallel/config.py
2022-12-01 18:50:58 +08:00

3 lines
33 B
Python

BATCH_SIZE = 128
NUM_EPOCHS = 10