小编Sac*_*rma的帖子

在 Pytorch 中微调预训练模型 MobileNet_V2

我是 pyTorch 的新手,我正在尝试创建一个分类器,其中我有大约 10 种图像文件夹数据集,为此我使用的是预训练模型(MobileNet_v2),但问题是我无法更改它的 FC 层. 没有model.fc 属性。谁能帮我做到这一点。谢谢

classification deep-learning pre-trained-model pytorch mobilenet

5
推荐指数
2
解决办法
5982
查看次数