0
点赞
收藏
分享

微信扫一扫

深度学习:invalid index of a 0-dim tensor. Use `tensor.item()` in Python or `tensor.item<T>()` in C++ to

版本问题

源代码:

test_loss += F.nll_loss(out_tgt.log(), target_label, size_average=False).data[0] # sum up batch loss

修改后代码:

test_loss += F.nll_loss(out_tgt.log(), target_label, size_average=False).item()  # sum up batch loss
举报

相关推荐

0 条评论