fix udml cfg (#2073)

2022-06-16 14:03:37 +08:00 · 2022-06-16 14:03:37 +08:00 · 2b1713b603
parent 1c3212876c
commit 2b1713b603
1 changed files with 1 additions and 0 deletions
--- 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