From 899047a3b37c30d762bb5c17ee3654bf719754b8 Mon Sep 17 00:00:00 2001 From: Ma Zerun Date: Mon, 26 Jul 2021 05:33:11 -0400 Subject: [PATCH] Fix duplicate `init_weights` call in ViT init function. (#373) --- mmcls/models/backbones/vision_transformer.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/mmcls/models/backbones/vision_transformer.py b/mmcls/models/backbones/vision_transformer.py index becf369b9c5..07f0a0167bd 100644 --- a/mmcls/models/backbones/vision_transformer.py +++ b/mmcls/models/backbones/vision_transformer.py @@ -455,8 +455,6 @@ def __init__(self, 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)