It would be great to add logging of the model's predictions(text) to Weights & Biases (Wandb) during evaluation on the validation set.