PyTorch RuntimeError: Tensor for argument #1 'self' is on CPU, but expected them to be on GPU

Question

I\'m using PyTorch for my Logistic Regression model but whenever I run the model summary I get an error

Ru