How to get test predictions (and other non-scalar metrics)?
❓ Questions and Help
What is your question?
If I have a trained model and I want to test it using Trainer.test(), how do I get the actual predictions of the model on the test set?
I tried to log the predictions and writing a Callback to get the logs at test end, but it seems like I can only log scalar Tensors in the dictionary returned by my model's test_end().