fix bug in vgg weight_init (#140)

2021-01-15 10:23:11 +08:00 · 2021-01-15 10:23:11 +08:00 · b98b317012
parent c8109d1eca
commit b98b317012
1 changed files with 0 additions and 2 deletions
--- a/mmcls/models/backbones/vgg.py
+++ b/mmcls/models/backbones/vgg.py
@ -154,8 +154,6 @@ class VGG(BaseBackbone):
                    constant_init(m, 1)
                elif isinstance(m, nn.Linear):
                    normal_init(m, std=0.01)
-        else:
-            raise TypeError('pretrained must be a str or None')

    def forward(self, x):
        outs = []