提交 7f910151 编写于 作者: J jiangjiajun

fix scope bug for prune

上级 95349984
......@@ -34,8 +34,12 @@ def sensitivity(program,
param_names,
eval_func,
sensitivities_file=None,
pruned_ratios=None):
scope = fluid.global_scope()
pruned_ratios=None,
scope=None):
if scope is None:
scope = fluid.global_scope()
else:
scope = scope
graph = GraphWrapper(program)
sensitivities = load_sensitivities(sensitivities_file)
......@@ -256,7 +260,8 @@ def cal_params_sensitivities(model, save_file, eval_dataset, batch_size=8):
prune_names,
eval_for_prune,
sensitivities_file=save_file,
pruned_ratios=list(np.arange(0.1, 1, 0.1)))
pruned_ratios=list(np.arange(0.1, 1, 0.1)),
scope=model.scope)
return sensitivitives
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册