So i checked, and the parameters(weights and biases) of the model before saving and after loading turns out to be same.
I am getting nan values every time i try to train and load this ensemble type of model, basically, the second pipeline only returns nan values while the first pipeline doesn’t return nan values, although both are being saved in the same “.pt” file.
It’s happening every time when I try to save the model,