diff --git a/ppcls/configs/ImageNet/ResNet/ResNet50_fp16.yaml b/ppcls/configs/ImageNet/ResNet/ResNet50_fp16.yaml index e58539b466f3b59d637f4f6a8e7188f6a9b8b876..f039a60b6d0ff9e1d0cc2f33e662c5856481bfbf 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: