fix yaml (#2125)

PaddlePaddle · Jul 4, 2022 · b676850 · b676850
1 parent a4711c3
commit b676850
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ppcls/configs/PULC/vehicle_attribute/PPLCNet_x1_0_distillation.yaml b/ppcls/configs/PULC/vehicle_attribute/PPLCNet_x1_0_distillation.yaml
@@ -22,6 +22,7 @@ Arch:
   # if not null, its lengths should be same as models
   pretrained_list:
   # if not null, its lengths should be same as models
+  infer_model_name: "Student"
   freeze_params_list:
   - True
   - False