Thank you so much . I loaded the whole model and it works fine! And the serialization link you provided helped me with this . One more thing I can use the same .pth and .pkl file in Raspberry pi right? I do not have to train from the beginning if I am going to port the project to RPi?