我按照教程构建了一个简单的网络,但遇到了以下错误:
RuntimeError: 期望类型为torch.cuda.FloatTensor的对象,但为参数#4 ‘mat1’找到了类型torch.FloatTensor
有什么帮助吗?谢谢!
回答:
TL;DR
这是修复方法
inputs = inputs.to(device)
为什么?!
torch.nn.Module.to()
和 torch.Tensor.to()
之间存在细微的差异:虽然 Module.to()
是一个原地操作符,但 Tensor.to()
不是。因此
net.to(device)
会改变 net
本身并将其移动到 device
。另一方面
inputs.to(device)
不会改变 inputs
,而是返回一个位于 device
上的 inputs
的副本。为了使用这个“在设备上”的副本,你需要将其赋值给一个变量,因此
inputs = inputs.to(device)