当Inputtype与weighttype类型不匹配，特别是在CUDA环境中遇到错误时，可以通过将输入数据类型转换到同一设备来解决。解决方案包括使用`torch.device()`将数据移动到GPU，或者直接调用`input.cuda()`。确保在设备间正确同步以避免类型冲突。
问题原因：这是由于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()#这两种代码任选其一即可
2447
2982
1002
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
扫一扫
