fix cae numclass

970773de · 悟、 · Tingquan Gao · 4a72afc4 · 970773de · 970773de
2 changed file
--- a/ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml
+++ b/ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml
@@ -76,7 +76,7 @@ DataLoader:
            mixup_alpha: 0.8
            cutmix_alpha: 1.0
            switch_prob: 0.5
-            num_classes: 102
+            num_classes: 1000
      transform_ops:
        - DecodeImage:
            to_rgb: True

--- a/ppcls/configs/CAE/cae_large_patch16_224_finetune.yaml
+++ b/ppcls/configs/CAE/cae_large_patch16_224_finetune.yaml
@@ -76,7 +76,7 @@ DataLoader:
            mixup_alpha: 0.8
            cutmix_alpha: 1.0
            switch_prob: 0.5
-            num_classes: 102
+            num_classes: 1000
      transform_ops:
        - DecodeImage:
            to_rgb: True