fix (#1549)

PaddlePaddle · Nov 23, 2022 · ad9b828 · ad9b828
1 parent 12b170c
commit ad9b828
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddleslim/auto_compression/compressor.py b/paddleslim/auto_compression/compressor.py
@@ -769,7 +769,7 @@ def single_strategy_compress(self, strategy, config, strategy_idx,
             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(