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)

版权声明:本文为weixin_44003026原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。