From 8b0e643f54f19f450b28f212732d293da56abc02 Mon Sep 17 00:00:00 2001
From: zengshao0622 <peter_z96@163.com>
Date: Thu, 8 Sep 2022 11:37:29 +0000
Subject: [PATCH] update batchsize

---
 ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml b/ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml
index 74be75c6..7ec1c9c4 100644
--- a/ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml
+++ b/ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml
@@ -102,7 +102,7 @@ DataLoader:
 
     sampler:
       name: DistributedBatchSampler
-      batch_size: 2
+      batch_size: 16
       drop_last: True
       shuffle: True
     loader:
@@ -129,7 +129,7 @@ DataLoader:
             order: ''
     sampler:
       name: DistributedBatchSampler
-      batch_size: 2
+      batch_size: 16
       drop_last: False
       shuffle: False
     loader:
-- 
GitLab