/aligned-cross-entropy

Test implementation of "Aligned Cross Entropy for Non-Autoregressive Machine Translation" https://arxiv.org/abs/2004.01655

Primary LanguageJupyter NotebookMIT LicenseMIT

m3yrin/aligned-cross-entropy [WIP]

Test implementation of "Aligned Cross Entropy for Non-Autoregressive Machine Translation" https://arxiv.org/abs/2004.01655

Open In Colab

  • Built on PyTorch 1.5.0
  • code is in axe/axe.py

Demo

Model

Dataset

Example

Please see detail at my colaboratory notebook.
Open In Colab
https://colab.research.google.com/github/m3yrin/aligned-cross-entropy/blob/master/demo.ipynb