isaaccorley / torchseg

Segmentation models with pretrained backbones. PyTorch.
MIT License
104 stars 8 forks source link

Standardize losses to use `ignore_index` #7

Closed isaaccorley closed 8 months ago

isaaccorley commented 9 months ago

Loss functions seem to use one of classes or ignore_index or even both. We should support both options for all loss functions. This generates confusion when using multiple loss functions (see https://github.com/microsoft/torchgeo/issues/1896) This involves updating the following:

isaaccorley commented 8 months ago

Closed in #11