Thanks, yeah as you said I now am able to save the model-
torch.jit.save(torch.jit.trace(model, (x)), "model.pth")
and load it like-
loaded_model = torch.jit.load("model.pth").
Though one trick I came up so that while loading I don’t have to deal with Classifier definition, by defining a load_model function inside the Classifier class, then have a three scripts structure like-

model.py (defining the Classifier only)
train.py (building and training model by importing Classifier from model.py)
load.py, and here loading model by importing Classifier from model.py and defining model like-from model import Classifier
loaded_model = Classifier.load_model('model.pth')



No bad actually