Skip to content

Commit 0ed43ce

Browse files
authored
fix quant export_model (PaddlePaddle#2606)
1 parent 62d8263 commit 0ed43ce

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

ppdet/slim/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,7 @@ def build_slim_model(cfg, slim_cfg, mode='train'):
5454
load_pretrain_weight(model, cfg.pretrain_weights)
5555
slim = create(cfg.slim)
5656
cfg['model'] = slim(model)
57+
cfg['slim'] = slim
5758
if mode != 'train':
5859
load_pretrain_weight(cfg['model'], cfg.weights)
5960

0 commit comments

Comments
 (0)