hrnademi:

I experimented again the way you said and outputs were equal


That’s good to hear and indicates that saving and loading the model works properly.
If you are seeing bad results for the validation dataset, it points towards the overall training routine, the dataset splits etc. but not necessarily a software bug.