Skip to content

Commit bafd4fb

Browse files
committed
fix ruff style
1 parent 14dc2a9 commit bafd4fb

File tree

1 file changed

+4
-4
lines changed
  • segmentation_models_pytorch/decoders/deeplabv3

1 file changed

+4
-4
lines changed

segmentation_models_pytorch/decoders/deeplabv3/model.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ class DeepLabV3(SegmentationModel):
4343
- dropout (float): Dropout factor in [0, 1)
4444
- activation (str): An activation function to apply "sigmoid"/"softmax"
4545
(could be **None** to return logits)
46-
kwargs: Arguments passed to the encoder class ``__init__()`` function. Applies only to ``timm`` models.
46+
kwargs: Arguments passed to the encoder class ``__init__()`` function. Applies only to ``timm`` models.
4747
Keys with ``None`` values are pruned before passing.
4848
4949
Returns:
@@ -79,7 +79,7 @@ def __init__(
7979
encoder_output_stride
8080
)
8181
)
82-
82+
8383
self.encoder = get_encoder(
8484
encoder_name,
8585
in_channels=in_channels,
@@ -91,7 +91,7 @@ def __init__(
9191

9292
if upsampling is None:
9393
if encoder_depth <= 3:
94-
scale_factor = 2 ** encoder_depth
94+
scale_factor = 2**encoder_depth
9595
else:
9696
scale_factor = encoder_output_stride
9797
else:
@@ -154,7 +154,7 @@ class DeepLabV3Plus(SegmentationModel):
154154
- dropout (float): Dropout factor in [0, 1)
155155
- activation (str): An activation function to apply "sigmoid"/"softmax"
156156
(could be **None** to return logits)
157-
kwargs: Arguments passed to the encoder class ``__init__()`` function. Applies only to ``timm`` models.
157+
kwargs: Arguments passed to the encoder class ``__init__()`` function. Applies only to ``timm`` models.
158158
Keys with ``None`` values are pruned before passing.
159159
160160
Returns:

0 commit comments

Comments
 (0)