深度学习: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

你可能感兴趣的:(实战深度学习常见问题汇总,迁移学习,人工智能,机器学习,python,图像处理)