Object detection in fast.ai v1

[muellerzr (Zachary Mueller)]

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…

Here