modified batch_size

modified batch_size per gpu in TinyNet_A.yaml for training with 4 gpus
PaddlePaddle · Mar 30, 2023 · a283b51 · a283b51
1 parent 4e34185
commit a283b51
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ppcls/configs/ImageNet/TinyNet/TinyNet_A.yaml b/ppcls/configs/ImageNet/TinyNet/TinyNet_A.yaml
@@ -46,7 +46,7 @@ Optimizer:
   one_dim_param_no_weight_decay: True
   lr:
     name: Step
-    learning_rate: 0.048
+    learning_rate: 0.048  # for total batch size 1024
     step_size: 2.4
     gamma: 0.97
     warmup_epoch: 3
@@ -85,7 +85,7 @@ DataLoader:
             order: ''
     sampler:
       name: DistributedBatchSampler
-      batch_size: 128
+      batch_size: 256  # for 4 gpus
       drop_last: True
       shuffle: True
     loader: