From 0ed43ce60423632fb5218da476bebad5c740c17c Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Tue, 13 Apr 2021 20:01:10 +0800
Subject: [PATCH] fix quant export_model (#2606)

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

diff --git a/ppdet/slim/__init__.py b/ppdet/slim/__init__.py
index ebd64c0e5..ab286647d 100644
--- a/ppdet/slim/__init__.py
+++ b/ppdet/slim/__init__.py
@@ -54,6 +54,7 @@ def build_slim_model(cfg, slim_cfg, mode='train'):
             load_pretrain_weight(model, cfg.pretrain_weights)
         slim = create(cfg.slim)
         cfg['model'] = slim(model)
+        cfg['slim'] = slim
         if mode != 'train':
             load_pretrain_weight(cfg['model'], cfg.weights)
 
-- 
GitLab