yolov5修改yaml文件后,预训练权重加吗还
时间: 2024-05-30 22:08:17 浏览: 65
yolov5的5个预训练权重s-m-l-x-n,有需要的自取!
一般来说,如果你修改了YAML文件中的模型结构,那么预训练权重是需要重新训练的。因为模型结构的修改可能会导致模型参数的数量和顺序发生变化,从而使预训练权重无法直接适用于新的模型结构。
但是如果你只是修改了YAML文件中的一些超参数,例如学习率、批大小等,那么一般情况下是不需要重新训练预训练权重的。你可以继续使用原来的预训练权重,并基于此进行微调训练即可。
阅读全文