From cd9ffbf4767981c8310dc2f8fe2b9bcbe6ce26ad Mon Sep 17 00:00:00 2001 From: nsutezo Date: Wed, 5 Apr 2023 17:12:08 +0000 Subject: [PATCH] included versionadded: 0.5 and parameter to docstring --- torchgeo/trainers/segmentation.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/torchgeo/trainers/segmentation.py b/torchgeo/trainers/segmentation.py index 8c02a7552aa..00907b6d0a1 100644 --- 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__()