RuntimeError: Expected object of scalar type Long but got scalar type Int for argument #2 ‘target’ in call to _thnn_nll_loss_forward
原因:第二个参数的类型为Int,期望的是Long
方法:将改变量强制转换为Long
label_total = torch.tensor(label_total, dtype=torch.long)
RuntimeError: Expected object of scalar type Long but got scalar type Int for argument #2 ‘target’ in call to _thnn_nll_loss_forward
原因:第二个参数的类型为Int,期望的是Long
方法:将改变量强制转换为Long
label_total = torch.tensor(label_total, dtype=torch.long)