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

1 ImageNet分类网络微调

可以直接修改模型的成员layer,例如:

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

2 COCO预训练微调

需要对head部分进行修改;

版权声明:本文为CSDN博主「songyuc」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/songyuc/article/details/122481649

songyuc

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐