RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
加载训练集和验证集的input时改为input.cuda()
具体在train() 与validate()函数的
enumerate后面
input=input.cuda()
RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
加载训练集和验证集的input时改为input.cuda()
具体在train() 与validate()函数的
enumerate后面
input=input.cuda()