Skip to content

Commit e5c4bc4

Browse files
author
Rustem Galiullin
committed
rm changes from pan model
1 parent 5edc0ee commit e5c4bc4

File tree

1 file changed

+1
-4
lines changed
  • segmentation_models_pytorch/decoders/pan

1 file changed

+1
-4
lines changed

segmentation_models_pytorch/decoders/pan/model.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -58,8 +58,6 @@ def __init__(
5858
activation: Optional[Union[str, callable]] = None,
5959
upsampling: int = 4,
6060
aux_params: Optional[dict] = None,
61-
encoder_kwargs: Optional[dict] = None,
62-
encoder_depth: int = 5,
6361
):
6462
super().__init__()
6563

@@ -69,10 +67,9 @@ def __init__(
6967
self.encoder = get_encoder(
7068
encoder_name,
7169
in_channels=in_channels,
72-
depth=encoder_depth,
70+
depth=5,
7371
weights=encoder_weights,
7472
output_stride=encoder_output_stride,
75-
**({} if encoder_kwargs is None else encoder_kwargs),
7673
)
7774

7875
self.decoder = PANDecoder(

0 commit comments

Comments
 (0)