Skip to content

Commit 5f5f639

Browse files
JulienMailleJulien Maille
andauthored
fixed typo in DPNEncorder (#316)
Co-authored-by: Julien Maille <[email protected]>
1 parent 3ee899c commit 5f5f639

File tree

1 file changed

+7
-7
lines changed
  • segmentation_models_pytorch/encoders

1 file changed

+7
-7
lines changed

segmentation_models_pytorch/encoders/dpn.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
from ._base import EncoderMixin
3434

3535

36-
class DPNEncorder(DPN, EncoderMixin):
36+
class DPNEncoder(DPN, EncoderMixin):
3737
def __init__(self, stage_idxs, out_channels, depth=5, **kwargs):
3838
super().__init__(**kwargs)
3939
self._stage_idxs = stage_idxs
@@ -75,7 +75,7 @@ def load_state_dict(self, state_dict, **kwargs):
7575

7676
dpn_encoders = {
7777
"dpn68": {
78-
"encoder": DPNEncorder,
78+
"encoder": DPNEncoder,
7979
"pretrained_settings": pretrained_settings["dpn68"],
8080
"params": {
8181
"stage_idxs": (4, 8, 20, 24),
@@ -91,7 +91,7 @@ def load_state_dict(self, state_dict, **kwargs):
9191
},
9292
},
9393
"dpn68b": {
94-
"encoder": DPNEncorder,
94+
"encoder": DPNEncoder,
9595
"pretrained_settings": pretrained_settings["dpn68b"],
9696
"params": {
9797
"stage_idxs": (4, 8, 20, 24),
@@ -108,7 +108,7 @@ def load_state_dict(self, state_dict, **kwargs):
108108
},
109109
},
110110
"dpn92": {
111-
"encoder": DPNEncorder,
111+
"encoder": DPNEncoder,
112112
"pretrained_settings": pretrained_settings["dpn92"],
113113
"params": {
114114
"stage_idxs": (4, 8, 28, 32),
@@ -123,7 +123,7 @@ def load_state_dict(self, state_dict, **kwargs):
123123
},
124124
},
125125
"dpn98": {
126-
"encoder": DPNEncorder,
126+
"encoder": DPNEncoder,
127127
"pretrained_settings": pretrained_settings["dpn98"],
128128
"params": {
129129
"stage_idxs": (4, 10, 30, 34),
@@ -138,7 +138,7 @@ def load_state_dict(self, state_dict, **kwargs):
138138
},
139139
},
140140
"dpn107": {
141-
"encoder": DPNEncorder,
141+
"encoder": DPNEncoder,
142142
"pretrained_settings": pretrained_settings["dpn107"],
143143
"params": {
144144
"stage_idxs": (5, 13, 33, 37),
@@ -153,7 +153,7 @@ def load_state_dict(self, state_dict, **kwargs):
153153
},
154154
},
155155
"dpn131": {
156-
"encoder": DPNEncorder,
156+
"encoder": DPNEncoder,
157157
"pretrained_settings": pretrained_settings["dpn131"],
158158
"params": {
159159
"stage_idxs": (5, 13, 41, 45),

0 commit comments

Comments
 (0)