@David_Reiss I m using following code for load model and passing model object Please look into it. is it correct way?
import torch
import torchvision
model = torch.load(“agestage250.pth”)
model.eval()
input_tensor = torch.rand(1,3,224,224)
script_model = torch.jit.trace(model,input_tensor)
script_model.save(“converted.pt”)
after run this code i m getting following error
Traceback (most recent call last):
File “convert.py”, line 6, in 
model.eval()
AttributeError: ‘collections.OrderedDict’ object has no attribute ‘eval’