I am training a custom dataset using deeplabv3 and xception_71 as backbone. I can train the model without problem but was unable to visualize it using vis.py. It seems that