I tried to import SSDlite model. And when I proceed on doing so I am getting this error. Here’s the code snippet and the error that I am receiving.
Can you please suggest me on how to import the model and get the performance metrics
model seems to be the state_dict and not an nn.Module object.
Create the model object before and load the state_dict afterwards:
model = MyModel() # create model object
model.load_state_dict(torch.load(path))