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

fix scope bug for prune

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