Skip to content

Commit

Permalink
Revert "fix ruff style"
Browse files Browse the repository at this point in the history
This reverts commit 79d5568.
  • Loading branch information
brianhou0208 committed Nov 29, 2024
1 parent 79d5568 commit c97d43a
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions segmentation_models_pytorch/decoders/deeplabv3/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ class DeepLabV3(SegmentationModel):
- dropout (float): Dropout factor in [0, 1)
- activation (str): An activation function to apply "sigmoid"/"softmax"
(could be **None** to return logits)
kwargs: Arguments passed to the encoder class ``__init__()`` function. Applies only to ``timm`` models.
kwargs: Arguments passed to the encoder class ``__init__()`` function. Applies only to ``timm`` models.
Keys with ``None`` values are pruned before passing.
Returns:
Expand Down Expand Up @@ -79,7 +79,7 @@ def __init__(
encoder_output_stride
)
)

self.encoder = get_encoder(
encoder_name,
in_channels=in_channels,
Expand All @@ -91,7 +91,7 @@ def __init__(

if upsampling is None:
if encoder_depth <= 3:
scale_factor = 2**encoder_depth
scale_factor = 2 ** encoder_depth
else:
scale_factor = encoder_output_stride
else:
Expand Down Expand Up @@ -153,7 +153,7 @@ class DeepLabV3Plus(SegmentationModel):
- dropout (float): Dropout factor in [0, 1)
- activation (str): An activation function to apply "sigmoid"/"softmax"
(could be **None** to return logits)
kwargs: Arguments passed to the encoder class ``__init__()`` function. Applies only to ``timm`` models.
kwargs: Arguments passed to the encoder class ``__init__()`` function. Applies only to ``timm`` models.
Keys with ``None`` values are pruned before passing.
Returns:
Expand Down

0 comments on commit c97d43a

Please sign in to comment.