How to load a non usual pretrained model?

vision
1

Hello,
l trained my own model and saved it . Now, l want to load this pretrained model to fine tune it with new data.

There is exist a function to load usual pretrained model but not for our own model.

model_ft = models.resnet18(pretrained=True)
num_ftrs = model_ft.fc.in_features
model_ft.fc = nn.Linear(num_ftrs, 2)

How can l do that ?

Thank you

2

torch.save and torch.load should fit your purpose.