登录 注册
当前位置:主页 > 资源下载 > 50 > 用于在PyTorch上训练、测试和采样自动回归递归语言模型的代码和脚本,适用RNN、GRU和LSTM层的RecurrentLM

用于在PyTorch上训练、测试和采样自动回归递归语言模型的代码和脚本,适用RNN、GRU和LSTM层的RecurrentLM

  • 更新:2024-08-10 10:36:20
  • 大小:16KB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:其它 - 开发技术
  • 格式:ZIP

资源介绍

复发性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