Open zhaoyong11 opened 1 year ago
博主你好！ 你的代码很优秀，解决了我的很多问题，但是我还是有个问题想问你！ 就是，我用你的代码训练没问题，但是当我做通道剪枝的时候，需要对BN层的权重进行扩散训练。模型的反向传播中 backbone模块都有权重的梯度信息，但是到了fpn的later_conv层后，梯度信息就消失了，我就无法对fpn层以后的模型的BN进行扩散训练.会报AttributeError: 'NoneType' object has no attribute 'data'的错误，我在网上也没有找到正确的解决方法。
我没有剪枝过不太懂欸
bubbliiiing
commented
1 year ago
博主你好！ 你的代码很优秀，解决了我的很多问题，但是我还是有个问题想问你！ 就是，我用你的代码训练没问题，但是当我做通道剪枝的时候，需要对BN层的权重进行扩散训练。模型的反向传播中 backbone模块都有权重的梯度信息，但是到了fpn的later_conv层后，梯度信息就消失了，我就无法对fpn层以后的模型的BN进行扩散训练.会报AttributeError: 'NoneType' object has no attribute 'data'的错误，我在网上也没有找到正确的解决方法。