configs/BigTransfer/BiT101x3.yaml

@@ -32,25 +32,25 @@ drop_remainder: True
 image_resize: 224
 hflip: 0.5
 crop_pct: 0.875
-interpolation: 'bilinear'
 
 # model config
 model: 'BiTresnet101x3'
 num_classes: 1000
 pretrained: False
 ckpt_path: ''
-keep_checkpoint_max: 30
+keep_checkpoint_max: 10
 save_checkpoint: True
 ckpt_save_dir: ''
 epoch_size: 90
 dataset_sink_mode: True
-amp_level: 'O0'
+amp_level: 'O2'
 
 # loss config
 loss: 'CE'
 label_smoothing: 0.1
 
 # lr scheduler config
+warmup_epochs: 1
 scheduler: 'multi_step_decay'
 lr: 0.12
 decay_rate: 0.5

configs/BigTransfer/test.yaml

@@ -36,7 +36,7 @@ crop_pct: 0.875
 # model config
 model: 'BiTresnet152x4'
 num_classes: 1000
-pretrained: True
+pretrained: False
 ckpt_path: ''
 keep_checkpoint_max: 10
 save_checkpoint: True