resnet50剪枝报错

[Wq-dd]

你好，我在使用resnet18为主干网的retinanet时，自己使用稀疏训练后的模型剪枝会报错，我的做法是：

1. 首先将训练好的模型计算bn的阈值得到每个bn层应该要剪枝的索引，并保存到一个dict里。
2. 然后循环1中的dict使用torchprunner去剪枝，会遇到前面的某些层如果剪了过多通道，后面层再剪时会出现索引越界。
3. 下面是我的部分代码。

        import torchpruner 
        # 创建ONNXGraph对象，绑定需要被剪枝的模型
        self.model.eval()
        graph = torchpruner.ONNXGraph(self.model.cpu())
        ##build ONNX静态图结构，需要指定输入的张量
        graph.build_graph(inputs=(torch.zeros(1, 3, 640, 640),))
        for i, (k, v) in enumerate(mask_dict_for_pruner.items()):
        # 获取conv1模块对应的module
            conv1_module = graph.modules[k]

            # 对前四个通道进行剪枝分析,指定对weight权重进行剪枝,剪枝前四个通道
            # weight权重out_channels对应的通道维度为0
            result = conv1_module.cut_analysis(attribute_name="weight", index=v, dim=0)

            # 剪枝执行模块执行剪枝操作，对模型完成剪枝过程.context变量提供了用于剪枝恢复的上下文
            self.model, context = torchpruner.set_cut(self.model, result)
        # 新的model即为剪枝后的模型
        print(self.model)```

请问是我的用法不对吗？还是说这种先计算剪枝的索引再调用torchpruner的方法不对呢？

[gdh1995]

每次剪枝后，model 对象变了，就都要重建 graph、重新执行 build_graph