0


报错:RuntimeError: expected scalar type Double but found Float

这个问题是深度学习,用pytorch跑的时候出现的

解决办法:这个是格式问题,希望的格式是double,但得到的是float。字面意思是这个,但是并不是非要把格式改成double,这个时候应该在出错的前面几处代码设个断点debug一下,我得到的结果是image、img_rgb都是tensor.unit8格式的,但程序所需要的是torch.float32格式的,因此在这两行后面加上.float()即可

会出上述的错误,改成:


本文转载自: https://blog.csdn.net/weixin_44570845/article/details/126507381
版权归原作者 Faster--YOLO 所有, 如有侵权,请联系我们删除。

“报错:RuntimeError: expected scalar type Double but found Float”的评论:

还没有评论