Skip to content

Commit

Permalink
Fix versionadded
Browse files Browse the repository at this point in the history
  • Loading branch information
adamjstewart committed Aug 30, 2024
1 parent 232cf4b commit e861754
Show file tree
Hide file tree
Showing 5 changed files with 9 additions and 8 deletions.
5 changes: 4 additions & 1 deletion torchgeo/datamodules/digital_typhoon.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,10 @@


class DigitalTyphoonDataModule(NonGeoDataModule):
"""Digital Typhoon Data Module."""
"""Digital Typhoon Data Module.
.. versionadded:: 0.6
"""

valid_split_types = ('time', 'typhoon_id')

Expand Down
2 changes: 1 addition & 1 deletion torchgeo/datamodules/geonrw.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ class GeoNRWDataModule(NonGeoDataModule):
Implements 80/20 train/val splits based on city locations.
See :func:`setup` for more details.
.. versionadded: 0.6
.. versionadded:: 0.6
"""

def __init__(
Expand Down
2 changes: 0 additions & 2 deletions torchgeo/datasets/levircd.py
Original file line number Diff line number Diff line change
Expand Up @@ -329,8 +329,6 @@ class LEVIRCDPlus(LEVIRCDBase):
If you use this dataset in your research, please cite the following paper:
* https://arxiv.org/abs/2107.09244
.. versionchanged:: 0.6
"""

url = 'https://drive.google.com/file/d/1JamSsxiytXdzAIk6VDVWfc-OsX-81U81'
Expand Down
4 changes: 2 additions & 2 deletions torchgeo/trainers/byol.py
Original file line number Diff line number Diff line change
Expand Up @@ -122,8 +122,8 @@ class BackboneWrapper(nn.Module):
* The output of the encoding layer is passed through the projection head
* The forward call returns the output of the projection head
.. versionchanged 0.4: Name changed from *EncoderWrapper* to
*BackboneWrapper*.
.. versionchanged:: 0.4
Name changed from *EncoderWrapper* to *BackboneWrapper*.
"""

def __init__(
Expand Down
4 changes: 2 additions & 2 deletions torchgeo/trainers/segmentation.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,15 +77,15 @@ class and used with 'ce' loss.
were renamed to *model*, *backbone*, and *weights*.
.. versionadded:: 0.5
The *class_weights*, *freeze_backbone*, and *freeze_decoder* parameters.
The *class_weights*, *freeze_backbone*, and *freeze_decoder* parameters.
.. versionchanged:: 0.5
The *weights* parameter now supports WeightEnums and checkpoint paths.
*learning_rate* and *learning_rate_schedule_patience* were renamed to
*lr* and *patience*.
.. versionchanged:: 0.6
The *ignore_index* parameter now works for jaccard loss.
The *ignore_index* parameter now works for jaccard loss.
"""
self.weights = weights
super().__init__(ignore='weights')
Expand Down

0 comments on commit e861754

Please sign in to comment.