diff --git a/ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml b/ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml
index 0efe62655f3c7b6a98a03adf0f5506f8d6975aca..83bdc1840a92e1b2a7c5bc70c7f226c10c7a50c4 100644
--- a/ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml
+++ b/ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml
@@ -17,7 +17,7 @@ Global:
 # model architecture
 Arch:
   name: cae_base_patch16_224
-  class_num: 102
+  class_num: 1000
   drop_rate: 0.0
   drop_path_rate: 0.1
   attn_drop_rate: 0.0
diff --git a/ppcls/configs/CAE/cae_large_patch16_224_finetune.yaml b/ppcls/configs/CAE/cae_large_patch16_224_finetune.yaml
index a8a3469129afc11d68aa5cc7f23cadf20f3615dc..d97b4aaf661bd96a92dafa8632e9ff75079c6289 100644
--- a/ppcls/configs/CAE/cae_large_patch16_224_finetune.yaml
+++ b/ppcls/configs/CAE/cae_large_patch16_224_finetune.yaml
@@ -17,7 +17,7 @@ Global:
 # model architecture
 Arch:
   name: cae_large_patch16_224
-  class_num: 102
+  class_num: 1000
   drop_rate: 0.0
   drop_path_rate: 0.2
   attn_drop_rate: 0.0