So when you are in you notebook with pytorch 1.0.1, using fastai, after defining the model your are doing: torch.save(model.state_dict(),'pretrained_svhn_ufpr_stage9.pth')
then you switch to a virtualenv or you update to pytorch 1.1, define your model and use:
model.load_state_dict(torch.load('pretrained_svhn_ufpr_stage9.pth'))
to load the model weights?
I tried it and it doesn’t work for me…
Best
Fabian