文件名称:seq2seq:PyTorch中的Seq2Seq代码
文件大小:248KB
文件格式:ZIP
更新时间:2024-06-08 02:47:58
pytorch seq2seq neural-machine-translation Python
PyTorch中的Seq2seq代码 根据和 数据预处理: 我使用以下步骤 > config/WMT14/download.sh # download WMT14 data into raw_data/WMT14 > config/WMT14/prepare.sh # preprocess the data, and copy the files to data/WMT14 然后运行以下命令以保存在h5文件中: > python scripts/prepro_text.py 训练: 训练需要一些目录来保存模型的快照,张量板事件 > mkdir -p save events 在config.yaml中定义的参数下训练模型 > python nmt.py -c config.yaml 检查options / opts.py以获取有关选项的更多信息。 评估模型: