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

问题出现一般是由于使用GPU训练时导致的,由于在训练时模型的weights没有转移到cuda上,data转移到了cuda上。此时一般检查以下在定义模型是是否进行如下操作:

device = torch.device('cuda:0')
model = resnet().to(device)

你可能感兴趣的:(深度学习,人工智能,python)