A softmax layer helps: class ResNet(nn.Module): def init(self): super(ResNet, self).init() resnet = models.resnet50(pretrained=True) modules ...
確定! 回上一頁