Use a custom pre-trained classifier as backbone for unet training

Hi and welcome!

The problem here is that any learner expects a function that returns a model, not the model itself. See here, someone had the same problem.

Unfortunately, I don’t know how you can then load your pretrained weights into this model, maybe somebody else can help :slight_smile: