RuntimeError: Input type (torch.cuda.DoubleTensor) and weight type (torch.cuda.FloatTensor) should b

admin • • 人工智能

链接: http://t.csdn.cn/z3gkm

torch.tensor对应的是torch.DoubleTensor
torch.Tensor对应的是torch.FloatTensor

观察下 t的大小写
input = torch.Tensor(np.random.randn(4,3,256,256)).to(device)

本图文内容来源于网友网络收集整理提供，作为学习参考使用，版权属于原作者。
THE END
分享
二维码

)">
< <上一篇
下一篇>>