diff --git a/ppcls/configs/ImageNet/CvT/CvT_w24_384.yaml b/ppcls/configs/ImageNet/CvT/CvT_W24_384.yaml similarity index 98% rename from ppcls/configs/ImageNet/CvT/CvT_w24_384.yaml rename to ppcls/configs/ImageNet/CvT/CvT_W24_384.yaml index f5b33d21b084a1474fd45e0ff949367d163a4773..a30bd297381106c166082160436a079563aef68c 100644 --- a/ppcls/configs/ImageNet/CvT/CvT_w24_384.yaml +++ b/ppcls/configs/ImageNet/CvT/CvT_W24_384.yaml @@ -94,7 +94,7 @@ DataLoader: prob: 0.5 sampler: name: DistributedBatchSampler - batch_size: 64 + batch_size: 16 drop_last: True shuffle: True loader: @@ -122,7 +122,7 @@ DataLoader: order: '' sampler: name: DistributedBatchSampler - batch_size: 64 + batch_size: 32 drop_last: False shuffle: False loader: