At the end of each epoch, the [Trainer] will evaluate the ROUGE metric and save the training checkpoint.