对yolov4剪枝后微调训练的mAP下降 #518
Description
你好，我用torch-pruning对yolov4剪枝50%后，微调训练的第一个epoch测试集loss达到最低，mAP最高，随后测试集loss上升，mAP下降，训练集的loss持续下降，这是怎么回事？有可参考的微调参数吗？
原学习率Init_lr=1e-3，Min_lr=1e-5，现在调整为原来的0.1
{{ message }}
你好，我用torch-pruning对yolov4剪枝50%后，微调训练的第一个epoch测试集loss达到最低，mAP最高，随后测试集loss上升，mAP下降，训练集的loss持续下降，这是怎么回事？有可参考的微调参数吗？
原学习率Init_lr=1e-3，Min_lr=1e-5，现在调整为原来的0.1