update batchsize

8b0e643f · 悟、 · 90d591e4 · 8b0e643f
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml +2 -2

未找到文件。
--- 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: