Browse Source

更新 'configs/crossvit/crossvit.yaml'

ema3
songyw 1 month ago
parent
commit
dd98c4615b
1 changed files with 5 additions and 5 deletions
  1. +5
    -5
      configs/crossvit/crossvit.yaml

+ 5
- 5
configs/crossvit/crossvit.yaml View File

@@ -24,7 +24,7 @@ dataset: 'imagenet'
data_dir: './imagenet/'
shuffle: True
dataset_download: False
batch_size: 128
batch_size: 320
drop_remainder: True

# Augmentation config
@@ -36,7 +36,7 @@ hflip: 0.5
interpolation: 'bicubic'
auto_augment: randaug-m9-mstd0.5-inc1
re_prob: 0.25
mixup: 0.8
mixup: 0.2
cutmix: 1.0
color_jitter: 0.4
use_ema: True
@@ -48,7 +48,7 @@ pretrained: False
ckpt_path: ''
keep_checkpoint_max: 30
ckpt_save_dir: './ckpt/'
epoch_size: 600
epoch_size: 900
dataset_sink_mode: True
amp_level: 'O2'
drop_path_rate: 0.1
@@ -59,12 +59,12 @@ label_smoothing: 0.1

# lr scheduler config
scheduler: 'warmup_cosine_decay'
lr: 0.0008
lr: 0.0009
min_lr: 0.00001
warmup_epochs: 40
decay_epochs: 260
decay_rate: 0.1
num_cycles: 2
num_cycles: 3
cycle_decay: 1




Loading…
Cancel
Save