From 720c451bc4adc07900d6ce0e386bdcc699022014 Mon Sep 17 00:00:00 2001
From: gaotingquan <gaotingquan@baidu.com>
Date: Thu, 4 Nov 2021 12:52:36 +0000
Subject: [PATCH] fix: fix batch size to 64 * 4
---
ppcls/configs/ImageNet/ResNet/ResNet50_fp16.yaml | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/ppcls/configs/ImageNet/ResNet/ResNet50_fp16.yaml b/ppcls/configs/ImageNet/ResNet/ResNet50_fp16.yaml
index e58539b46..f039a60b6 100644
--- 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: