From 0c1fab049d1438bb0a42cb7f484b67650edf2d42 Mon Sep 17 00:00:00 2001
From: tripleMu <865626@163.com>
Date: Fri, 13 May 2022 12:26:34 +0800
Subject: [PATCH] Add new config for reset50

---
 .../classification/configs/deit_default_settings.yaml    | 3 ++-
 .../configs/resnet50_default_settings.yaml               | 9 ++++++---
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/projects/classification/configs/deit_default_settings.yaml b/projects/classification/configs/deit_default_settings.yaml
index 899bd7cf..bc26bde9 100644
--- 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
index b972debc..27161aec 100644
--- 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