search

lifu-tu / ENGINE

ENGINE: Energy-Based Inference Networks for Non-Autoregressive Machine Translation
Other
24 stars 5 forks source link

Pretained CMLM for IWSLT #4

Open NonvolatileMemory opened 4 years ago

NonvolatileMemory commented 4 years ago

Hi Lifu,

Thanks for you code and paper. It's very nice work but I am a little confused about the pretrained models, I didn't find the one for IWSLT, facebook only provided the wmt version.

lifu-tu commented 4 years ago

Thanks for your interest!

You can get the pretrained autoregressive model and CMLM model with the following command. These models are trained by myself. You could also train the models with the provided code here.

wget https://ttic.uchicago.edu/~lifu/ENGINE_WMT16ROEN.zip

After uncompressing the above zip file, you can get the pre-trained models and source language dictionary, and target language dictionary. "AR_Energy_DE2EN.pt" is the pretrained autoregressive model and "CMLM_DE2EN.pt" is the pretrained CMLM model.

Let me know if you have any other questions.

NonvolatileMemory commented 4 years ago

Thank you!