How to save raw predictions?
I have it hard coded in my logging callback not to print them, but trainer.test still prints them.
I'm wondering the canonical way to save  them with wandb_logger or another callback.
(It's an NLP summarization task).
Thx!