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。