Closed
Description
# dtype = torch.device("cuda:0") # Uncomment this to run on GPU
it is supposed to be
# device = torch.device("cuda:0") # Uncomment this to run on GPU
same issue in PyTorch: Tensors and autograd
So x = torch.randn(N, D_in, device=device, dtype=dtype)
is the new recommended style to create tensors?
Metadata
Metadata
Assignees
Labels
No labels