在跑train.py文件时候,
问题描述
显示我提供的数据为整数形式,而实际上应该是float形式
于是我就在outputs和targets前面添加了float()
结果却显示
结果就是张量不能用float(target)形式。
解决方案:
直接torch修改为float形式就可以了
即target.float()
版权声明:本文为weixin_57463859原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。