运行代码
correct = pred_choice.eq(target).cpu().sum()
total_correct = correct.item + total_correct
在第二行代码报错:
TypeError: unsupported operand type(s) for +: 'builtin_function_or_method' and 'int'
问题出在 correct.item不是一个int型,它的类型打印出来是
correct.item: <built-in method item of Tensor object at 0x7f7eadf18d38>
修改方式是换成correct.item(),也就是把tensor转换成int的方法。
参考博客:pytorch 把tensor转换成int