未验证 提交 dab888af 编写于 作者: G GGBond8488 提交者: GitHub

replace_static.sparsity_with_incubate.asp (#1603)

上级 a35619b8
......@@ -539,8 +539,8 @@ def build_prune_program(executor,
train_program_info.program = pruned_program
elif strategy.startswith('asp'):
from paddle.static import sparsity
pruner = sparsity
from paddle.incubate import asp
pruner = asp
excluded_params_name = []
if config['prune_params_name'] is None:
config['prune_params_name'] = _get_asp_prune_params(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册