How to load custom encoder weights?

qubvel / segmentation_models.pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

https://smp.readthedocs.io/

MIT License

9.14k stars 1.63k forks source link

How to load custom encoder weights? #787

Closed rkinas closed 1 year ago

rkinas commented 1 year ago

Hi, Is any easy way to load custom trained (not Imagenet) weights for encoder? I have custom weights for resnet34d in file model.pth and I would like to start Unet training from this file (not Imagenet).