From 7af836801679f0b41cb303f9435a13edeabe39c3 Mon Sep 17 00:00:00 2001
From: littletomatodonkey <2120160898@bit.edu.cn>
Date: Mon, 15 Feb 2021 16:56:56 +0800
Subject: [PATCH] fix mv3 batch size (#597)

---
 configs/MobileNetV3/MobileNetV3_large_x1_0.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml b/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml
index 6f9e5597..3c512599 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/"
-- 
GitLab