Umair_Javaid:

odel to gpu before loading, could that be the problem? I can not save the model, I have an already trained mo


Usually not.
1.But sometimes when we save the model after transferring it to cpu
2.then if u want to load that saved model again it’s better to load before moving into GPU