fix cae numclass

2023-03-15 07:28:37 +00:00 · 2023-03-15 07:28:37 +00:00 · 970773de4a
parent 4a72afc4c1
commit 970773de4a
2 changed files with 2 additions and 2 deletions
--- 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