fix mv3 batch size
@@ -17,7 +17,7 @@ image_shape: [3, 224, 224]
LEARNING_RATE:
function: 'Cosine'
params:
- lr: 2.6
+ lr: 1.3
warmup_epoch: 5
OPTIMIZER:
@@ -29,7 +29,7 @@ OPTIMIZER:
factor: 0.00002
TRAIN:
- batch_size: 4096
+ batch_size: 2048
num_workers: 4
file_list: "./dataset/ILSVRC2012/train_list.txt"
data_dir: "./dataset/ILSVRC2012/"
