Hi @JuliousHurtado ,
Thank you for the response.
In that approach, how do you specify the model whether train should be True or False?
I mean, I never explicitly call the forward function in the inference. I simply y = model(X).
Thanks