Hey yall! I’m a college student who is trying to learn how to use pytorch for my lab and I am going through the pytorch tutorial with the MNIST dataset. I was trying to move “operations” over to my GPU with both. But I think I am missing moving more that just the model over. I have checked some posts on here and stack overflow but I cant find anything that I haven’t tried yet. Any help or direction would be appreciated!
torch.set_default_device("mps")
# and the below
class NeuralNetwork(nn.Module):
# code for the class here
model = NeuralNetwork()
model.to("mps")