@ptrblck
Hello I am getting the same problem ,
whenever i am trying to load a saved model , both on cpu as well as on gpu , more than often i am getting nan values.
This behavior is quite inconsistent on both CPU and GPU.
Basically my model is an ensemble-based model, and it consists of 3 different models being used together at once and all the three models are saved on one single file .pt file.
The absurd thing is i am successfully able to load model 1 and model 3 from the same file and they aren’t returning nan values while model 2 is returning nan values
I am using numpy 1.19.4
Any solution to this