剪枝后的模型训练问题

bubbliiiing / faster-rcnn-pytorch

这是一个faster-rcnn的pytorch实现的库，可以利用voc数据集格式的数据进行训练。

MIT License

1.52k stars 351 forks source link

Open echo-hot opened 1 month ago

echo-hot commented 1 month ago

博主您好，我对您的这个faster rcnn模型进行了剪枝后，发现对剪枝后的模型进行训练的时候会对其进行初始化，导致其每一层的卷积核数量不匹配这个是怎么解决这个问题呢？是去掉rpn的初始化嘛