Closed FrancescoSaverioZuppichini closed 2 years ago
It would be cool if we have a way to import a state dict but to not fail when one dict of a layering mismatch. This is useful if we want to load pretrained weights but we have a different number of classes
Enhancement ✨
It would be cool if we have a way to import a state dict but to not fail when one dict of a layering mismatch. This is useful if we want to load pretrained weights but we have a different number of classes