I’m trying to load a model that I saved with fastai (using resnet34 base architecture and save function).
model_path = '../input/resnet34-ship-clf/Resnet34_clf_256x256_1ep.pth' model = nn.Sequential(models.resnet34()) pretrained_weights = torch.load(model_path) model.load_state_dict(pretrained_weights, strict=False)
I’ve got the error message :
Pre-trained weight names are in the form :
What should I do during loading to get same names ?