Actually,I am facing the same issue, but for me the other notebooks with (0.1 ,0.0001 learning rate with 25 epochs), have trained successfully. But the one with 0.01 have stucked at the epoch 1.

If you figured the solution, kindly tag