FrancescoSaverioZuppichini / glasses

High-quality Neural Networks for Computer Vision 😎
https://francescosaveriozuppichini.github.io/glasses-webapp/
MIT License
436 stars 37 forks source link

Load state dict silently fail #297

Closed FrancescoSaverioZuppichini closed 2 years ago

FrancescoSaverioZuppichini commented 2 years ago

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