From 07b597f56ea20019f6ffeaba729914b3f3848174 Mon Sep 17 00:00:00 2001
From: gaotingquan <gaotingquan@baidu.com>
Date: Wed, 31 May 2023 02:23:40 +0000
Subject: [PATCH] increase bs to avoid oom

---
 ppcls/configs/ImageNet/EfficientNet/EfficientNetB7.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppcls/configs/ImageNet/EfficientNet/EfficientNetB7.yaml b/ppcls/configs/ImageNet/EfficientNet/EfficientNetB7.yaml
index 83b36406..8f02cd27 100644
--- 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:
-- 
GitLab