33
33
from ._base import EncoderMixin
34
34
35
35
36
- class DPNEncorder (DPN , EncoderMixin ):
36
+ class DPNEncoder (DPN , EncoderMixin ):
37
37
def __init__ (self , stage_idxs , out_channels , depth = 5 , ** kwargs ):
38
38
super ().__init__ (** kwargs )
39
39
self ._stage_idxs = stage_idxs
@@ -75,7 +75,7 @@ def load_state_dict(self, state_dict, **kwargs):
75
75
76
76
dpn_encoders = {
77
77
"dpn68" : {
78
- "encoder" : DPNEncorder ,
78
+ "encoder" : DPNEncoder ,
79
79
"pretrained_settings" : pretrained_settings ["dpn68" ],
80
80
"params" : {
81
81
"stage_idxs" : (4 , 8 , 20 , 24 ),
@@ -91,7 +91,7 @@ def load_state_dict(self, state_dict, **kwargs):
91
91
},
92
92
},
93
93
"dpn68b" : {
94
- "encoder" : DPNEncorder ,
94
+ "encoder" : DPNEncoder ,
95
95
"pretrained_settings" : pretrained_settings ["dpn68b" ],
96
96
"params" : {
97
97
"stage_idxs" : (4 , 8 , 20 , 24 ),
@@ -108,7 +108,7 @@ def load_state_dict(self, state_dict, **kwargs):
108
108
},
109
109
},
110
110
"dpn92" : {
111
- "encoder" : DPNEncorder ,
111
+ "encoder" : DPNEncoder ,
112
112
"pretrained_settings" : pretrained_settings ["dpn92" ],
113
113
"params" : {
114
114
"stage_idxs" : (4 , 8 , 28 , 32 ),
@@ -123,7 +123,7 @@ def load_state_dict(self, state_dict, **kwargs):
123
123
},
124
124
},
125
125
"dpn98" : {
126
- "encoder" : DPNEncorder ,
126
+ "encoder" : DPNEncoder ,
127
127
"pretrained_settings" : pretrained_settings ["dpn98" ],
128
128
"params" : {
129
129
"stage_idxs" : (4 , 10 , 30 , 34 ),
@@ -138,7 +138,7 @@ def load_state_dict(self, state_dict, **kwargs):
138
138
},
139
139
},
140
140
"dpn107" : {
141
- "encoder" : DPNEncorder ,
141
+ "encoder" : DPNEncoder ,
142
142
"pretrained_settings" : pretrained_settings ["dpn107" ],
143
143
"params" : {
144
144
"stage_idxs" : (5 , 13 , 33 , 37 ),
@@ -153,7 +153,7 @@ def load_state_dict(self, state_dict, **kwargs):
153
153
},
154
154
},
155
155
"dpn131" : {
156
- "encoder" : DPNEncorder ,
156
+ "encoder" : DPNEncoder ,
157
157
"pretrained_settings" : pretrained_settings ["dpn131" ],
158
158
"params" : {
159
159
"stage_idxs" : (5 , 13 , 41 , 45 ),
0 commit comments