deep-learning - 在 Pytorch 中微调预训练模型 MobileNet_V2

Question

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

Answer 1

执行以下操作:

import torch
model = torch.hub.load('pytorch/vision', 'mobilenet_v2', pretrained=True)
print(model.classifier)

model.classifier[1] = torch.nn.Linear(in_features=model.classifier[1].in_features, out_features=10)
print(model.classifier)

输出:

Sequential(
  (0): Dropout(p=0.2)
  (1): Linear(in_features=1280, out_features=1000, bias=True)
)
Sequential(
  (0): Dropout(p=0.2)
  (1): Linear(in_features=1280, out_features=10, bias=True)
)

备注 : 你需要 torch >= 1.1.0使用 torch.hub .