模型中输入为cuda类型的数据,调用模型时需更改
RuntimeError:Input type(torch.FloatTensor)and weight type (torch.cuda.FloatTensor)should be the same
最新推荐文章于 2024-04-13 10:06:38 发布
模型中输入为cuda类型的数据,调用模型时需更改