search

Ranking666 / Yolov5-Processing

Multi-backbone, Prune, Quantization, KD
152 stars 29 forks source link

上采样替换成反卷积层后剪枝报错 #4

Open zbw98 opened 2 years ago

zbw98 commented 2 years ago

image image 如图，红色划线处为修改处，不知道该如何修改

zbw98 commented 2 years ago

训练无报错，只是到了剪枝就说通道数对不上

ruofei-fan commented 2 years ago

我和你一样的问题，也是把上采样换了，只不过换的是carafe，同样是显示通道数对不上，所以想请问你这个问题解决了么？