分类：multi-target

2021
02-21

解决pytorch下出现multi-target not supported at的一种可能原因

在使用交叉熵损失函数的时候，target的形状应该是和label的形状一致或者是只有batchsize这一个维度的。如果target是这样的【batchszie，1】就会出现上述的错误。改一下试试，用squeeze（）函数降低纬度，如果不知道squeeze怎么用的，可以参考这篇文章。pytorch下的unsqueeze和squeeze用法这只是一种可能的原因。补充：pytorch使用中遇到的问题1.load模型参数文件时，提示torch.cuda.is_available()isFalse。按照pytorch官方网页... 继续阅读 >