amc
amc copied to clipboard
shared index of groups in mobilenetV2 residual block missing one line code
the code logic in the #371 line of channel_pruning_env.py is missing one line code: self.shared_idx.append(share_group) should be added after the for loop.
@grayicon 需要加吗?没看出来有啥影响啊
@grayicon 需要加吗?没看出来有啥影响啊 shared_ids.append(share_group)是在for loop下的if分支进行的,该分支为new group code block,这会导致,最后一个share_group 无法添加到shared_ids中去,因此在for loop后添加一句self.shared_idx.append(share_group) 将最后一个shared_group添加到shared_idx中去
請問Mobilenetv2在搜尋完後怎麼導出權重檔案?