type(torch.LongTensor)报错有关不同变量分别在cpu和cuda下

变量名.type()转换tensor类型的时候注意!!!
torch.cuda.LongTensor:加载到GPU
torch.LongTensor:加载到CPU
官网其他转换类型

model.cuda()
input_iter = Variable(torch.tensor(input_iter))
input_iter = input_iter.type(torch.cuda.LongTensor)
output = model(input_iter)

你可能感兴趣的:(pytorch,深度学习,python,神经网络,人工智能)