From 2b1713b603f7b3f31bbcc4b96addc4176720cfbf Mon Sep 17 00:00:00 2001
From: littletomatodonkey <dazhiningsibuqu@163.com>
Date: Thu, 16 Jun 2022 14:03:37 +0800
Subject: [PATCH] fix udml cfg (#2073)

---
 ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_udml.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_udml.yaml b/ppcls/configs/ImageNet/Distillation/PPLCNet_x2_5_udml.yaml
index 43b64888..b425b592 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
-- 
GitLab