It works :). It can not calculate generative metrics right now , but that’s being added in Seq2SeqTrainer.
Seq2SeqTrainer