文件名称:PytorchRnnLM:在PyTorch中使用RNN的语言模型
文件大小:5KB
文件格式:ZIP
更新时间:2024-06-04 09:06:43
Python
派托克·林恩 在PyTorch中使用RNN的语言模型。 使用Wiki-Text-2长期依赖数据集: : 展示 如何使用绑定嵌入权重,如以下所述: : 如何在不使用中介填充的情况下破解PyTorch以使用PackedSequence和单词嵌入。 用法 下载数据集并将文件内容解压缩到“ wikitext-2”文件夹中。 使用“ --help”运行“ main.py”以查看可能的命令行参数。 默认参数在10分钟内在中级GPU上训练模型,以在测试集上获得135的困惑度得分。
【文件预览】:
PytorchRnnLM-master
----main.py(6KB)
----data_loader.py(2KB)
----log_timer.py(547B)
----.gitignore(37B)
----README.md(687B)
----vocab.py(2KB)