From ad9b828ccc6caf918686adab2215eaf6da3b7f1f Mon Sep 17 00:00:00 2001 From: ceci3 Date: Wed, 23 Nov 2022 17:15:28 +0800 Subject: [PATCH] fix (#1549) --- paddleslim/auto_compression/compressor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddleslim/auto_compression/compressor.py b/paddleslim/auto_compression/compressor.py index e86551c7..e098830f 100644 --- a/paddleslim/auto_compression/compressor.py +++ b/paddleslim/auto_compression/compressor.py @@ -769,7 +769,7 @@ class AutoCompression: train_program_info, test_program_info = self._prepare_program( inference_program, feed_target_names, fetch_targets, patterns, 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) test_program_info = self._start_train( -- GitLab