Yi19960820

关于yolo4剪通道与剪层之后模型微调的一个问题

5

你好，首先你的代码对我了解模型剪枝很有帮助，表示感谢！在剪枝剪层之后微调训练的时候出现了一个错误   error:Unable to find a valid cuDNN algorithm to run convolution 我替换了自己剪枝剪层之后的cfg文件及权重文件，出现了这个报错，在网上并没有得到解决方案，请问这是我配置环境的问题吗？显卡为rtx3080。

关于yolov4+mobilenetv3训练的问题

11

您好，一开始不清楚您写了mobilenet-yolov4-lite-pytorch的代码，我将您的yolov4 pytorch版本的cspdarknet53替换成了mobilenetv3，操作的一样的，唯一不同的是我还使用了v3里面的最后一层卷积层，输出维度为13x13x960，并改动了yolo4.py里面三次卷积，五次卷积以及上下采样的维度，我用3080训练了400个epoch： lr = 1e-3 Batch_size = 8 Init_Epoch = 0 Freeze_Epoch =200 lr = 1e-5 Batch_size = 8 Freeze_Epoch = 50 Unfreeze_Epoch = 200 无预训练权重，voc2012数据集，loss在收敛于26.3，map为34.5％，速度比yolov4快20％左右，是我的训练方法出了问题吗？