超级会员免费看
在尝试运行深度学习模型时，遇到了一个RuntimeError，指出输入类型torch.cuda.FloatTensor和权重类型torch.FloatTensor应相同。这表明模型并未部署在GPU上。解决方法是将模型移动到GPU设备上，通过调用model().cuda()来实现。
return F.conv2d(input, weight, bias, self.stride,
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
这个证明model 不再gpu上面
我们将model().cuda()就可以了
采取的方法 出现了错误
return F.conv2d(input, weight, bias, self.stride,
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
这个证明model 不再gpu上面
我们将model().cuda()就可以了
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
扫一扫
