你好,感谢开源,有个部分代码没有看明白,想咨询一下
def set_group(self, model):
bottleneck_index = [2, 4, 6]
self.groups = [[f'model[{i}].m[{n}].cv2.conv' for n in
range(len(model.module[i].m if hasattr(model, 'module') else model[i].m))] + [
f'model[{i}].cv1.conv'] for i in bottleneck_index]
在sensitivity.py中这个函数的group是有什么意思, bottleneck_index 表示是要剪枝的层在model中的索引吗?模型在剪枝时时只剪枝这些层吗
你好,感谢开源,有个部分代码没有看明白,想咨询一下 def set_group(self, model): bottleneck_index = [2, 4, 6] self.groups = [[f'model[{i}].m[{n}].cv2.conv' for n in range(len(model.module[i].m if hasattr(model, 'module') else model[i].m))] + [ f'model[{i}].cv1.conv'] for i in bottleneck_index] 在sensitivity.py中这个函数的group是有什么意思, bottleneck_index 表示是要剪枝的层在model中的索引吗?模型在剪枝时时只剪枝这些层吗