mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-29 21:46:06 +00:00
11 lines
138 B
Python
11 lines
138 B
Python
from colossalai.amp import AMP_TYPE
|
|
|
|
BATCH_SIZE = 128
|
|
NUM_EPOCHS = 200
|
|
|
|
CONFIG = dict(
|
|
fp16=dict(
|
|
mode=AMP_TYPE.TORCH
|
|
)
|
|
)
|