Algorithm doesn’t run on GPU even after storing model and data into GPU. What am i missing?

Question

You can find training section of my code below:

device = torch.device(‘cuda’ if torch.cuda.is_available() else ‘cpu’)

Note: I am receiving True w