Add new config for reset50

Oneflow-Inc · May 13, 2022 · 0c1fab0 · 0c1fab0
1 parent 0e2ac35
commit 0c1fab0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/projects/classification/configs/deit_default_settings.yaml b/projects/classification/configs/deit_default_settings.yaml
@@ -1,7 +1,7 @@
 DATA:
   BATCH_SIZE: 16
   DATASET: imagenet
-  DATA_PATH: /DATA/disk1/ImageNet/extract
+  DATA_PATH: /media/ubuntu/Extra/oneflow_work/ImageNet
   IMG_SIZE: 224
   INTERPOLATION: bicubic
   ZIP_MODE: False
@@ -27,6 +27,7 @@ TRAIN:
 
   LR_SCHEDULER:
     NAME: cosine
+    MILESTONES: None
 
   OPTIMIZER:
     NAME: adamw

diff --git a/projects/classification/configs/resnet50_default_settings.yaml b/projects/classification/configs/resnet50_default_settings.yaml
@@ -17,15 +17,18 @@ MODEL:
 TRAIN:
   START_EPOCH: 0
   EPOCHS: 300
-  WARMUP_EPOCHS: 20
-  WEIGHT_DECAY: 5e-4
+  WARMUP_EPOCHS: 3
+  WARMUP_LR: 0.0001
+  MIN_LR: 1.0e-06
+  WEIGHT_DECAY: 2.0e-05
   BASE_LR: 0.01
   CLIP_GRAD: None
   AUTO_RESUME: True
   ACCUMULATION_STEPS: 0
 
   LR_SCHEDULER:
-    NAME: multi_step
+    NAME: cosine
+    MILESTONES: None
 
   OPTIMIZER:
     NAME: sgd