From 9a12cc32f773ad124bc46a58ea7a3d06d0036343 Mon Sep 17 00:00:00 2001 From: Dom <97384583+tosemml@users.noreply.github.com> Date: Tue, 22 Aug 2023 19:10:29 -0700 Subject: [PATCH 1/2] use join --- mcunet/tinynas/elastic_nn/networks/ofa_proxyless.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/mcunet/tinynas/elastic_nn/networks/ofa_proxyless.py b/mcunet/tinynas/elastic_nn/networks/ofa_proxyless.py index a64cc3e..7c98e71 100644 --- a/mcunet/tinynas/elastic_nn/networks/ofa_proxyless.py +++ b/mcunet/tinynas/elastic_nn/networks/ofa_proxyless.py @@ -179,8 +179,7 @@ def module_str(self): for stage_id, block_idx in enumerate(self.block_group_info): depth = self.runtime_depth[stage_id] active_idx = block_idx[:depth] - for idx in active_idx: - _str += self.blocks[idx].module_str + '\n' + _str += "\n".join([self.blocks[idx].module_str for idx in active_idx]) if self.feature_mix_layer is not None: _str += self.feature_mix_layer.module_str + '\n' _str += self.classifier.module_str + '\n' From bbbf16737ee6703217b9787000024cc9885e4443 Mon Sep 17 00:00:00 2001 From: Dom <97384583+tosemml@users.noreply.github.com> Date: Tue, 5 Sep 2023 22:23:39 -0700 Subject: [PATCH 2/2] fix bug --- mcunet/tinynas/elastic_nn/networks/ofa_proxyless.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mcunet/tinynas/elastic_nn/networks/ofa_proxyless.py b/mcunet/tinynas/elastic_nn/networks/ofa_proxyless.py index 7c98e71..068be68 100644 --- a/mcunet/tinynas/elastic_nn/networks/ofa_proxyless.py +++ b/mcunet/tinynas/elastic_nn/networks/ofa_proxyless.py @@ -179,7 +179,7 @@ def module_str(self): for stage_id, block_idx in enumerate(self.block_group_info): depth = self.runtime_depth[stage_id] active_idx = block_idx[:depth] - _str += "\n".join([self.blocks[idx].module_str for idx in active_idx]) + _str += "\n".join([self.blocks[idx].module_str for idx in active_idx]) + '\n' if self.feature_mix_layer is not None: _str += self.feature_mix_layer.module_str + '\n' _str += self.classifier.module_str + '\n'