博客

python代码总是出现pytorch训练过程训练集精度为0的情况的解决

  • Post author:
  • Post category:python


原因是gao的代码是pytorch3迁移过来的，很多地方使用了sum(pred_y== target)来求训练集的精度，但是没有转成numpy这里就会为0，因为torch数据本身不支持这种计算方式，所以改成sum(np.array(pred_y)== np.array(target.data.cpu()))即可使训练集正常运行


版权声明：本文为qq_27292549原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。