How can I load pre-trained weights from a different task into fastai models?

[muellerzr (Zachary Mueller)]

See this post here, should help:

Loading pretrained weights that are not from ImageNet

This method does not work. Setting strict=False does not solve the problem of the mismatch of weight sizes. When strict=False PyTorch loads weights that have the same names and same sizes. A shape mismatch error is thrown for weights with same names but different sizes. I have a slightly different workaround: I wrote a function similar to learn.load and closer in vein to how PyTorch implements model.load_state_dict(). def load_diff_pretrained(learn, name:Union[Path,str], device:torch.device=No…