diff --git a/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml b/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml index 6f9e5597f42abbe1398cb9c8a6e16ec246f26c88..3c5125999af1ed852846791c2f8b74aad0e94d9f 100644 --- a/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml +++ b/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml @@ -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/"