remove useless modification
parent
a8b2f93634
commit
120ef98853
|
@ -384,7 +384,7 @@ class VisionTransformer(BaseBackbone):
|
||||||
super(VisionTransformer, self).init_weights()
|
super(VisionTransformer, self).init_weights()
|
||||||
|
|
||||||
if not (isinstance(self.init_cfg, dict)
|
if not (isinstance(self.init_cfg, dict)
|
||||||
and self.init_cfg['type'] == 'Pretrained' or self.is_init):
|
and self.init_cfg['type'] == 'Pretrained'):
|
||||||
if self.pos_embed is not None:
|
if self.pos_embed is not None:
|
||||||
trunc_normal_(self.pos_embed, std=0.02)
|
trunc_normal_(self.pos_embed, std=0.02)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue