isaaccorley / torchseg

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

Standardize losses to use `ignore_index` #7

Closed isaaccorley closed 7 months ago

isaaccorley commented 7 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 7 months ago

Closed in #11