diff --git a/ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_udml.yaml b/ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_udml.yaml
index 43b64888d7dc38affa7464fbb31bbe1f015d6337..b425b592fdabea61832a1ad3c2f50a20b62ecd6f 100644
--- a/ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_udml.yaml
+++ b/ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_udml.yaml
@@ -30,6 +30,7 @@ Arch:
   freeze_params_list:
   - False
   - False
+  infer_model_name: "Student"
   models:
     - Teacher:
         name: PPLCNet_x2_5