/NA-NMT

Non-autoregressive Neural Machine Translation (not a full version)

Primary LanguagePython

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.