PyTorch Semantic Segmentation with weighted classes

Question

I\'m studying how to the semantic segmentation on images, but I would like to assign a weight to the classes. I\'m using this Semantic Segmentation PyTorch and with the actu