increase bs to avoid oom

07b597f5 · gaotingquan · cuicheng01 · caa6393c · 07b597f5
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppcls/configs/ImageNet/EfficientNet/EfficientNetB7.yaml ppcls/configs/ImageNet/EfficientNet/EfficientNetB7.yaml +1 -1

未找到文件。
--- 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: