成功解决RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be

发布时间:2023-12-18 19:30

\"\" 错误代码

问题原因:这是由于Input type 与weight type的类型不符合,可以注意到Input type是没有cuda的,而weight type是在cuda上也就是有gpu的加载的。(错误代码后的or可以不用管)

解决办法:既然是类型不符导致的错误,那么就将类型转为一致,代码详见下面:

1.
device = torch.device(\'cuda:0\' if torch.cuda.is_available() else \'cpu\')
input=input.to(device)
2.
input=input.cuda()#这两种代码任选其一即可

ItVuer - 免责声明 - 关于我们 - 联系我们

本网站信息来源于互联网,如有侵权请联系:561261067@qq.com

桂ICP备16001015号