From 0cd9aa5e906f6f40ee96eab0557430d3ac84470c Mon Sep 17 00:00:00 2001 From: ceci3 <592712189@qq.com> Date: Thu, 28 Nov 2019 02:32:20 +0000 Subject: [PATCH] update mobilenet --- paddleslim/nas/search_space/mobilenet_block.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddleslim/nas/search_space/mobilenet_block.py b/paddleslim/nas/search_space/mobilenet_block.py index b08a4c78..39ce4050 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)) -- GitLab