 How to resolve this error please? Also I want to know if this model can be used to segment images outside the 22 classes mentioned in the paper. I'm planning to use it for lung nodules segmentation.