lm_word_level:语言模型单词级别 源码

词级语言建模RNN 本示例在语言建模任务上训练多层RNN(Elman,GRU或LSTM)。 默认情况下,训练脚本使用提供的Wikitext-2数据集。 然后,生成的脚本可以使用经过训练的模型来生成新文本。 python main.py --cuda --epochs 6 # Train a LSTM on Wikitext-2 with CUDA python main.py --cuda --epochs 6 --tied # Train a tied LSTM on Wikitext-2 with CUDA python main.py --cuda --ep