RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same python-3.x python-3.x