Pre-processing for train/val/test vs predict

[agoodge]

Hi,

When I call train/val/test, the input is appropriately pre-processed in the train/val/test_step function. If I call predict after doing this, the input is also pre-processed similarly. However, if I call predict without calling train/val/test first, then the input is not pre-processed.

My question is, how should I use predict properly? I want to obtain the outputs of a pre-trained model on the test set (not only the performance metrics) but the way predict and test work seem to be totally different.