RuntimeError: Expected floating point type for target with class probabilities, got Int

  • Post author:
  • Post category:其他





在跑train.py文件时候,




问题描述


显示我提供的数据为整数形式,而实际上应该是float形式

于是我就在outputs和targets前面添加了float()

结果却显示

结果就是张量不能用float(target)形式。






解决方案:

直接torch修改为float形式就可以了

即target.float()



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