microsoft / ProphetNet

A research project for natural language generation, containing the official implementations by MSRA NLC team.
MIT License
686 stars 109 forks source link

Selecting additional scoring methods for fine-tuning #53

Open mtsourma opened 2 years ago

mtsourma commented 2 years ago

We have started to fine-tune the ProphetNet model on a custom dataset. We are using fairseq==v0.9.0 version. Currently, only perplexity is supported while training, however, we would like also to validate the trained model on BLEU-4, METEOR, and ROUGE metrics. Can anyone provide any insights on this? Because the "--scoring" parameter in fairseq v0.9 is not supported.