ResNet相关知识

1、Softmax(dim=1)的含义

num_classes = 2 #几分类问题就写几
resnet50 = models.resnet50(weights=models.ResNet50_Weights.IMAGENET1K_V2)
fc_inputs = resnet50.fc.in_features
resnet50.fc = nn.Sequential(
    nn.Linear(fc_inputs, num_classes),    #输出变更为自己的类别数
    nn.Softmax(dim=1))          #将结果变更为每一类的概率，合计为1

dim=1，是操作第2维的数据。参考 Pytorch小记-torch.nn.Softmax(dim=1)如何理解？_nn.softmax(dim=-1)-CSDN博客