You might need another color pallete other than ade20k. BTW you can also check the output directly to see if it's bad
```python predict = mx.nd.squeeze(mx.nd.argmax(output, 1)).asnumpy() print(predict) ``` --- [Visit Topic](https://discuss.mxnet.apache.org/t/help-with-deeplab-runing-problem-nan-output/6584/2) or reply to this email to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.mxnet.apache.org/email/unsubscribe/f15692c9935345870f9e1389fed99410b2730153d624dd816e5dd2c776e83d7a).
