Have you checked manually if the weights are still the same after loading of the model? At this point they should be different because you are returning different results, even though I don’t know why they are changed.