How to use EfficientNet with unet_learner as encoder in FastAI V2?


I am trying to train a unet_learner with EfficientNet backbone? I found an excellent online blog on integrating EfficientNet from timm into FastAI for classification task where EfficientNet is extracted and a head is added to it and then used within the Learner class.

Can it be used with the FastAI v2 unet_learner? If yes, how I can achieve this integration?

Thanks in advance.

There was a PR into my wwf resource but the docs don’t quite render yet. See here: at master · walkwithfastai/ · GitHub

Note: it’s the same notebook. Look a bit further down

Thank you @muellerzr. This is what I needed. Cool.