Hi, I see that the function inference is called during training. inference function however doesn't take in a new data (like a test set for example). How can I apply the finished model on new data input?

Thank you.