-
用于在PyTorch上训练、测试和采样自动回归递归语言模型的代码和脚本,适用RNN、GRU和LSTM层的RecurrentLM
资源介绍
复发性LM
描述
用于在具有RNN,GRU和LSTM层的PyTorch上训练,测试和采样自动回归递归语言模型的代码和脚本。
买者自负
该存储库仅用于教育和展示目的。 内容按“原样”提供,没有任何默示保证,并且对因使用该内容而造成的损失不承担任何责任。
功能性
该代码提供3种功能:训练LM,评估LM和从LM采样序列。 单词级和字符级LM都可以训练。 训练可以在CPU或GPU中进行,测试和采样在CPU上进行。
训练
培训是使用train_lm.py脚本完成的。 运行python train_lm.py --help以获得有关现有参数的详细信息。 此处提供了示例运行:
python train_lm.py --input_file data/librispeech-train-sentences.list --output_file output/final_model.pytorch --v
- 上一篇: RHCE7.0练习题
- 下一篇: bookmarks_2019_7_5.html