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

报错内容为:输入的是cuda类型的数据,但是使用的权重类型不是,他们的类型应当一致。 解决方法: 将你的网络模型改为cuda类型即可 在使用模型之前 。 如 model class yourModelName old version: model class x new version: model class.cuda model class x ...

2021-07-18 10:04 0 142 推荐指数:

查看详情

torch.cuda.FloatTensor

Pytorch中的tensor又包括CPU上的数据类型和GPU上的数据类型,一般GPU上的Tensor是CPU上的Tensor加cuda()函数得到。 一般系统默认是torch.FloatTensor类型。例如data = torch.Tensor(2,3)是一个2*3的张量,类型 ...

Tue Dec 22 16:56:00 CST 2020 0 1141
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM