Improve the save and load model in RnnEmbTrLm