github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-11 13:59:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

[workflow] refactored the example check workflow (#2411)

Browse Source 
* [workflow] refactored the example check workflow

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code
This commit is contained in:
Frank Lee
2023-01-10 11:26:19 +08:00
committed by GitHub
parent 8de8de9fa3
commit 8327932d2c
10 changed files with 113 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/tutorial/hybrid_parallel/config.py
View File

@@ -6,8 +6,8 @@ from colossalai.amp import AMP_TYPE
BATCH_SIZE = 256
LEARNING_RATE = 3e-3
WEIGHT_DECAY = 0.3
NUM_EPOCHS = 10
WARMUP_EPOCHS = 3
NUM_EPOCHS = 2
WARMUP_EPOCHS = 1
# model config
IMG_SIZE = 224