Skip to content

P
PaddleSlim

PaddlePaddle / PaddleSlim
1 年多 前同步成功

通知 51
Star 1434
Fork 344
P
PaddleSlim
未验证 提交 ad9b828c 编写于 作者: C ceci3 提交者： GitHub
浏览文件
操作

fix (#1549)

上级 12b170c3
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
paddleslim/auto_compression/compressor.py
浏览文件 @ ad9b828c
...@@ -769,7 +769,7 @@ class AutoCompression: ...@@ -769,7 +769,7 @@ class AutoCompression:
train_program_info, test_program_info = self._prepare_program( train_program_info, test_program_info = self._prepare_program(
inference_program, feed_target_names, fetch_targets, patterns, inference_program, feed_target_names, fetch_targets, patterns,
strategy, config, train_config) strategy, config, train_config)
if 'unstructure' in self._strategy: if 'unstructure' in strategy:
test_program_info.program._program = remove_unused_var_nodes( test_program_info.program._program = remove_unused_var_nodes(
test_program_info.program._program) test_program_info.program._program)
test_program_info = self._start_train( test_program_info = self._start_train(
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或