search

jadore801120 / attention-is-all-you-need-pytorch

A PyTorch implementation of the Transformer model in "Attention is All You Need".
MIT License
8.78k stars 1.97k forks source link

@SeoroMin #190

Open SeoroMin opened 2 years ago

SeoroMin commented 2 years ago

Added bleu.py, which is more suitable for machine translation task performance evaluation.

how to run : python bleu.py --reference ./.data/multi30k/test2016.en --candidate prediction.txt

1338199 commented 2 years ago

Sorry to bother you, does your translate.py works well? I found the generated "prediction.txt" file contains plenty of "as", although the dev accuracy is good during training. I trained the model in latest version on the master branch using commands in "train_multi30k_de_en.sh",