NA-NMT
NA-NMT copied to clipboard
Non-autoregressive Neural Machine Translation (not a full version)
Non-autoregressive Neural Machine Translation
PyTorch implementation of the "Non-Autoregressive Transformer (NAT)" model described in https://arxiv.org/abs/1711.02281
Requirements: Python 3.6
pytorch > 0.2
torchtext:
open tools/text and run python setup.py install
open tools/revtok and run python setup.py install
pip install translate-toolkit
pip install tqdm
TODO
Add more description about data processing/training/testing The torchtext requirements can be removed if possible.