RuntimeError: Input type (torch.cuda.FloatTensor) and weight type should be the same

最新推荐文章于 2024-04-04 12:54:01 发布

炸那个炸

最新推荐文章于 2024-04-04 12:54:01 发布

阅读量2.9k

点赞数

文章标签：深度学习 pytorch 神经网络

本文链接：https://blog.csdn.net/qq_35049121/article/details/122567394

版权

做神经网络的一个报错

原因：输入数据和参数的数据格式不匹配，应该是数据传到了GPU上而，神经网络模型没有传输到GPU上。

解决：在实例化网络后，加一条传到GPU上的代码即可。

model=DANN() # DANN是你自己搭建的神经网络,实例化神经网络
device = torch.device("cuda:" + config.gpu_id if torch.cuda.is_available() else "cpu")
model=model.to(device) #传到GPU

优惠劵

炸那个炸

关注关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
打赏
3
评论
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type should be the same

做神经网络的一个报错原因：输入数据和参数的数据格式不匹配，应该是数据传到了GPU上而，神经网络模型没有传输到GPU上。解决：在实例化网络后，加一条传到GPU上的代码即可。model=DANN() # DANN是你自己搭建的神经网络,实例化神经网络device = torch.device("cuda:" + config.gpu_id if torch.cuda.is_available() else "cpu")model=model.to(device) #传到GPU...
复制链接

扫一扫