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

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

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

报错:RuntimeError: expected scalar type Double but found Float_第1张图片

会出上述的错误,改成:

报错:RuntimeError: expected scalar type Double but found Float_第2张图片

 

你可能感兴趣的:(人工智能,计算机视觉,深度学习)