fix yaml (#2125)

2022-07-04 12:45:18 +08:00 · 2022-07-04 12:45:18 +08:00 · b67685042d
parent a4711c3934
commit b67685042d
1 changed files with 1 additions and 0 deletions
--- 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