Skip to content

Commit 0c1fab0

Browse files
author
tripleMu
committed
Add new config for reset50
1 parent 0e2ac35 commit 0c1fab0

File tree

2 files changed

+8
-4
lines changed

2 files changed

+8
-4
lines changed

projects/classification/configs/deit_default_settings.yaml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
DATA:
22
BATCH_SIZE: 16
33
DATASET: imagenet
4-
DATA_PATH: /DATA/disk1/ImageNet/extract
4+
DATA_PATH: /media/ubuntu/Extra/oneflow_work/ImageNet
55
IMG_SIZE: 224
66
INTERPOLATION: bicubic
77
ZIP_MODE: False
@@ -27,6 +27,7 @@ TRAIN:
2727

2828
LR_SCHEDULER:
2929
NAME: cosine
30+
MILESTONES: None
3031

3132
OPTIMIZER:
3233
NAME: adamw

projects/classification/configs/resnet50_default_settings.yaml

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,15 +17,18 @@ MODEL:
1717
TRAIN:
1818
START_EPOCH: 0
1919
EPOCHS: 300
20-
WARMUP_EPOCHS: 20
21-
WEIGHT_DECAY: 5e-4
20+
WARMUP_EPOCHS: 3
21+
WARMUP_LR: 0.0001
22+
MIN_LR: 1.0e-06
23+
WEIGHT_DECAY: 2.0e-05
2224
BASE_LR: 0.01
2325
CLIP_GRAD: None
2426
AUTO_RESUME: True
2527
ACCUMULATION_STEPS: 0
2628

2729
LR_SCHEDULER:
28-
NAME: multi_step
30+
NAME: cosine
31+
MILESTONES: None
2932

3033
OPTIMIZER:
3134
NAME: sgd

0 commit comments

Comments
 (0)