criterion = nn.MSELoss()
criterion(a, b)
这是a的dtype=torch.float，b的dtype=torch.int64
因此，都改成float
免责声明！
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。
猜您在找
RuntimeError: reciprocal is not implemented for type torch.cuda.LongTensor
"host_softmax" not implemented for 'Long'
存在空闲CUDA前提下报错：RuntimeError: CUDA error: out of memory
[报错]-RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.cuda.FloatTensor) should be the same
Pytoch的nn.CrossEntropyLoss 报错 RuntimeError: expected scalar type Long but found Float
#报错记录#RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
pytorch遇到的问题：RuntimeError: randperm is only implemented for CPU
*** RuntimeError: CUDA error: out of memory.
RuntimeError: CUDA error: invalid device ordinal
RuntimeError: CUDA error: an illegal memory access was encountered