Update fmda_rnn_train_and_save.ipynb