Fix duplicate `init_weights` call in ViT init function. (#373)

2021-07-26 05:33:11 -04:00 · 2021-07-26 05:33:11 -04:00 · 899047a3b3
parent 57e9692765
commit 899047a3b3
1 changed files with 0 additions and 2 deletions
--- a/mmcls/models/backbones/vision_transformer.py
+++ b/mmcls/models/backbones/vision_transformer.py
@ -455,8 +455,6 @@ class VisionTransformer(BaseBackbone):
            norm_cfg, embed_dim, postfix=1)
        self.add_module(self.norm1_name, norm1)

-        self.init_weights()
-
    def init_weights(self):
        super(VisionTransformer, self).init_weights()
        nn.init.normal_(self.pos_embed, std=0.02)