def __init__(self):
super().__init__()
self.pretrained_model = models.resnet18(pretrained=True)
classifier = [
nn.Linear(self.pretrained_model.fc.in_features, 17)
]
self.classifier = nn.Sequential(*classifier)
self.pretrained_model.fc = self.classifier
评论列表
文章目录