RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
inputs = inputs.to(device)
RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
inputs = inputs.to(device)