diff --git a/paddleslim/nas/search_space/mobilenet_block.py b/paddleslim/nas/search_space/mobilenet_block.py index b08a4c78dc364530539fa0dd01a2122e592b8d07..39ce405095b903e32462fa798521a663498b04da 100644 --- a/paddleslim/nas/search_space/mobilenet_block.py +++ b/paddleslim/nas/search_space/mobilenet_block.py @@ -68,7 +68,7 @@ class MobileNetV2BlockSpace(SearchSpaceBase): range_table_base.append(len(self.repeat)) range_table_base.append(len(self.k_size)) else: - for i in range(block_num): + for i in range(self.block_num): range_table_base.append(len(self.multiply)) range_table_base.append(len(self.filter_num)) range_table_base.append(len(self.repeat))