欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

RuntimeError: Expected object of scalar type Long but got scalar type Float for argument #2 'target'

程序员文章站 2022-03-09 13:35:55
...

pytorch报错

RuntimeError: Expected object of scalar type Long but got scalar type Float for argument #2 'target'

原因 使用了loss = criterion(output,target)

loss = criterion(output,target)
# 而output 和 target的类型都是

'torch.cuda.FloatTensor'
'torch.cuda.FloatTensor'

output.shape :
torch.Size([64, 10])
target.shape :
torch.Size([64, 1, 28, 28])

解决办法

target1 = torch.LongTensor(target)
loss = criterion(output,target1)

当然这里是我自己比较错了,应该比较label的

相关标签: GAN