Fix some bugs in RNN training.