仅作为记录,大佬请跳过。
用.add_module
'''
6.场景1:微调ConvNet
'''
model_ft = models.inception_v3(pretrained=True)
num_ftrs = model_ft.fc.in_features
model_ft.fc = nn.Linear(num_ftrs, 1024)
model_ft.add_module('fc2', nn.Linear(1024, 2))
参考
仅作为记录,大佬请跳过。
用.add_module
'''
6.场景1:微调ConvNet
'''
model_ft = models.inception_v3(pretrained=True)
num_ftrs = model_ft.fc.in_features
model_ft.fc = nn.Linear(num_ftrs, 1024)
model_ft.add_module('fc2', nn.Linear(1024, 2))
参考