When trying to train my LSTM model my cell will print out epoch 1/2 training on n samples etc and then just shut down and all variables are lost therefore I have to run all