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博客