Skip to content

Commit ad9b828

Browse files
authored
fix (#1549)
1 parent 12b170c commit ad9b828

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddleslim/auto_compression/compressor.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -769,7 +769,7 @@ def single_strategy_compress(self, strategy, config, strategy_idx,
769769
train_program_info, test_program_info = self._prepare_program(
770770
inference_program, feed_target_names, fetch_targets, patterns,
771771
strategy, config, train_config)
772-
if 'unstructure' in self._strategy:
772+
if 'unstructure' in strategy:
773773
test_program_info.program._program = remove_unused_var_nodes(
774774
test_program_info.program._program)
775775
test_program_info = self._start_train(

0 commit comments

Comments
 (0)