pytorch 分类任务的教程:https://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html
主要使用的是:
_,predict = torch.max(out.data,1)
最后的predict
就是预测的类别了
pytorch 分类任务的教程:https://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html
主要使用的是:
_,predict = torch.max(out.data,1)
最后的predict
就是预测的类别了