included versionadded: 0.5 and parameter to docstring

microsoft · Apr 5, 2023 · cd9ffbf · cd9ffbf
1 parent 83c3038
commit cd9ffbf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torchgeo/trainers/segmentation.py b/torchgeo/trainers/segmentation.py
@@ -109,6 +109,8 @@ class and used with 'ce' loss
            The *segmentation_model* parameter was renamed to *model*,
            *encoder_name* renamed to *backbone*, and
            *encoder_weights* to *weights*.
+        .. versionadded: 0.5
+            The *class_weights* parameter.
         """
         super().__init__()