When I changed my code from Keras to PyTorch, I am not able to reproduce the results that I was getting in Keras in PyTorch.
I am using the MSELoss() as my loss function with an Unet architecture. Basically, my model learns well but my inference turns out to be bad.
What could be resulting in such a problem?
I have checked my code but I am not able to reproduce the same result while keeping all the parameters the same.
Thanks