Questions about using NFNet as the backbone of UNet

[taochx]

Thank you for your sharing the code.
I used some models in the 'timm' folder as the backnet of UNet, such as resnest269e, gluon_resnet_v1c, skresnext50_32x4d, etc. They are all working, but when I try to use the NFNet as the backbone, it can not run.
I don't know which part is wrong.