From b23aea1a7cf6badb7cda52fda6bda045ccd0f825 Mon Sep 17 00:00:00 2001 From: ceci3 Date: Thu, 24 Nov 2022 14:27:24 +0800 Subject: [PATCH] fix export unstruct (#1548) --- 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