remove hardnet-39 (#6487)

* remove hardnet39 * remove hardnet39
nemonameless · Jul 22, 2022 · 00e961c · 00e961c
1 parent 49d0000
commit 00e961c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ppdet/modeling/backbones/hardnet.py b/ppdet/modeling/backbones/hardnet.py
@@ -146,7 +146,7 @@ def forward(self, x):
 class HarDNet(nn.Layer):
     def __init__(self, depth_wise=False, return_idx=[1, 3, 8, 13], arch=85):
         super(HarDNet, self).__init__()
-        assert arch in [39, 68, 85], "HarDNet-{} not support.".format(arch)
+        assert arch in [68, 85], "HarDNet-{} is not supported.".format(arch)
         if arch == 85:
             first_ch = [48, 96]
             second_kernel = 3
@@ -161,6 +161,8 @@ def __init__(self, depth_wise=False, return_idx=[1, 3, 8, 13], arch=85):
             grmul = 1.7
             gr = [14, 16, 20, 40]
             n_layers = [8, 16, 16, 16]
+        else:
+            raise ValueError("HarDNet-{} is not supported.".format(arch))
 
         self.return_idx = return_idx
         self._out_channels = [96, 214, 458, 784]