increase bs to avoid oom

2023-05-31 02:23:40 +00:00 · 2023-05-31 02:23:40 +00:00 · 07b597f56e
parent caa6393cd4
commit 07b597f56e
1 changed files with 1 additions and 1 deletions
--- a/ppcls/configs/ImageNet/EfficientNet/EfficientNetB7.yaml
+++ b/ppcls/configs/ImageNet/EfficientNet/EfficientNetB7.yaml
@ -107,7 +107,7 @@ DataLoader:
            order: ''
    sampler:
      name: DistributedBatchSampler
-      batch_size: 128
+      batch_size: 64
      drop_last: False
      shuffle: False
    loader: