fix udml cfg (#2073)

2b1713b6 · littletomatodonkey · GitHub · 1c321287 · 2b1713b6
显示空白变更内容
内联并排

Showing with 1 addition and 0 deletion

ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_udml.yaml ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_udml.yaml +1 -0

未找到文件。
--- 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