diff --git a/ppcls/configs/ImageNet/Distillation/mv3_large_x1_0_distill_mv3_small_x1_0.yaml b/ppcls/configs/ImageNet/Distillation/mv3_large_x1_0_distill_mv3_small_x1_0.yaml
index 9612be4e61bcdda39d568d0422a5b741a96b9d20..a7265b066e1c526fbb63f59993ff68bb4ae09d8a 100644
--- a/ppcls/configs/ImageNet/Distillation/mv3_large_x1_0_distill_mv3_small_x1_0.yaml
+++ b/ppcls/configs/ImageNet/Distillation/mv3_large_x1_0_distill_mv3_small_x1_0.yaml
@@ -18,7 +18,7 @@ Global:
 # model architecture
 Arch:
   name: "DistillationModel"
-  class_num: 1000
+  class_num: &class_num 1000
   # if not null, its lengths should be same as models
   pretrained_list:
   # if not null, its lengths should be same as models
@@ -28,11 +28,13 @@ Arch:
   models:
     - Teacher:
         name: MobileNetV3_large_x1_0
+        class_num: *class_num
         pretrained: True
         use_ssld: True
         dropout_prob: null
     - Student:
         name: MobileNetV3_small_x1_0
+        class_num: *class_num
         pretrained: False
         dropout_prob: null