Deeplabv3 not predicting classes

Question

I\'m trying to train deeplabv3 model with mobilenetv3_small_seg architecture. I trained the model but the predictions I get is a complete blank mask with no class prediction