From 6a849313125114d869dcafad003f70a203bb4fe0 Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Fri, 11 Jun 2021 17:20:58 +0800
Subject: [PATCH] fix distill_prune export model (#3362)

---
 ppdet/slim/__init__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ppdet/slim/__init__.py b/ppdet/slim/__init__.py
index 255a8859d..d929807c3 100644
--- a/ppdet/slim/__init__.py
+++ b/ppdet/slim/__init__.py
@@ -47,6 +47,7 @@ def build_slim_model(cfg, slim_cfg, mode='train'):
             model = pruner(model)
             load_pretrain_weight(model, weights)
         cfg['model'] = model
+        cfg['slim_type'] = cfg.slim
     else:
         load_config(slim_cfg)
         model = create(cfg.architecture)
-- 
GitLab