未验证 提交 6a849313 编写于 作者: G Guanghua Yu 提交者: GitHub

fix distill_prune export model (#3362)

上级 36b8f4a8
...@@ -47,6 +47,7 @@ def build_slim_model(cfg, slim_cfg, mode='train'): ...@@ -47,6 +47,7 @@ def build_slim_model(cfg, slim_cfg, mode='train'):
model = pruner(model) model = pruner(model)
load_pretrain_weight(model, weights) load_pretrain_weight(model, weights)
cfg['model'] = model cfg['model'] = model
cfg['slim_type'] = cfg.slim
else: else:
load_config(slim_cfg) load_config(slim_cfg)
model = create(cfg.architecture) model = create(cfg.architecture)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册