5 classes

[faxirabd]

Hello, I just tried using deeplabv3+ code with only 5 classes instead of 20. The training goes gut, however the evaluation loss goes to nan (infinity) and testing is the same.
Is this model works only with 20 classes?

[Barthas255]

I used it for only 2 classes and it seems to work fine