RuntimeError: Input type (torch.cuda.DoubleTensor) and weight type (torch.DoubleTensor) should be the same
时间: 2023-12-04 19:38:37 浏览: 149
pytorch模型提示超出内存RuntimeError: CUDA out of memory.
5星 · 资源好评率100%
这个错误提示表明输入类型(torch.cuda.DoubleTensor)和权重类型(torch.DoubleTensor)不一致,需要将它们转换为相同的类型。可以使用.to()方法将输入和权重都转换为相同的类型,例如:
```
input = input.to(torch.double)
weight = weight.to(torch.double)
```
如果你使用的是GPU,还需要将它们都移动到相同的设备上,例如:
```
input = input.to(device)
weight = weight.to(device)
```
其中,device是你当前使用的设备,可以使用torch.device()方法指定。
阅读全文