update mobilenet

0cd9aa5e · ceci3 · f97daa3d · 0cd9aa5e
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddleslim/nas/search_space/mobilenet_block.py paddleslim/nas/search_space/mobilenet_block.py +1 -1

未找到文件。
--- 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))