Prerak_Srivastava:

So i checked, and the parameters(weights and biases) of the model before saving and after loading turns out to be same.





 Prerak_Srivastava:

and when i try to load net_1 for evaluation the data seems corrupted


Could you clarify these two statements, please?
Is the input data corrupted, since you’ve already checked, that all parameters are equal?
Could you run a constant input, e.g. torch.ones, through the model before saving and after loading the model and compare the outputs?