《南溪的目标检测学习笔记》——预训练微调的学习笔记

Posted 2022-02-06 songyuc

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了《南溪的目标检测学习笔记》——预训练微调的学习笔记相关的知识，希望对你有一定的参考价值。

1 ImageNet分类网络微调

可以直接修改模型的成员layer，例如：

model.fc = nn.Linear(fc_features, n)

需要对head部分进行修改；

以上是关于《南溪的目标检测学习笔记》——预训练微调的学习笔记的主要内容，如果未能解决你的问题，请参考以下文章