Hi @learningpytorch
I did see your colab code.
You have used train_loader instead of valid_loader for testing the performance. In the validation you’ll have to move the X_test, y_test to CUDA to solve the issue.