例子:
RuntimeError: Expected object of scalar type Long but got scalar type Float for argumenLoss =
torch.nn.CrossEntropyLoss()
loss = Loss(out, label)t #2 ‘target’
直接在label后边加上类型转换即可。
torch.nn.CrossEntropyLoss()
loss = Loss(out, label.long())t #2 ‘target’