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