There was a problem compiling the code from your notebook. Details: name 'train_mnist' is not defined

[bruno_ramos_martins]

I reviewed your code, and everything seems correct. Therefore, let’s try a few steps:

1. Try restarting your notebook;
2. If that doesn’t work, restart your environment following these instructions.

If none of this works, please let me know.