Skip to content

Commit 00e961c

Browse files
authored
remove hardnet-39 (#6487)
* remove hardnet39 * remove hardnet39
1 parent 49d0000 commit 00e961c

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

Diff for: ppdet/modeling/backbones/hardnet.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ def forward(self, x):
146146
class HarDNet(nn.Layer):
147147
def __init__(self, depth_wise=False, return_idx=[1, 3, 8, 13], arch=85):
148148
super(HarDNet, self).__init__()
149-
assert arch in [39, 68, 85], "HarDNet-{} not support.".format(arch)
149+
assert arch in [68, 85], "HarDNet-{} is not supported.".format(arch)
150150
if arch == 85:
151151
first_ch = [48, 96]
152152
second_kernel = 3
@@ -161,6 +161,8 @@ def __init__(self, depth_wise=False, return_idx=[1, 3, 8, 13], arch=85):
161161
grmul = 1.7
162162
gr = [14, 16, 20, 40]
163163
n_layers = [8, 16, 16, 16]
164+
else:
165+
raise ValueError("HarDNet-{} is not supported.".format(arch))
164166

165167
self.return_idx = return_idx
166168
self._out_channels = [96, 214, 458, 784]

0 commit comments

Comments
 (0)