fix: fix batch size to 64 * 4

2021-11-04 12:52:36 +00:00 · 2021-11-04 12:52:36 +00:00 · 720c451bc4
parent 10559616ba
commit 720c451bc4
1 changed files with 1 additions and 1 deletions
--- a/ppcls/configs/ImageNet/ResNet/ResNet50_fp16.yaml
+++ b/ppcls/configs/ImageNet/ResNet/ResNet50_fp16.yaml
@ -78,7 +78,7 @@ DataLoader:

    sampler:
      name: DistributedBatchSampler
-      batch_size: 32
+      batch_size: 64
      drop_last: False
      shuffle: True
    loader: