mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Fix broken test, repvgg block doesn't have attn_last attr.

Browse Source
This commit is contained in:
Ross Wightman 2021-05-27 18:12:22 -07:00
parent 742c2d5247
commit f615474be3
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
timm/models/byobnet.py
View File

@ -573,9 +573,8 @@ class RepVggBlock(nn.Module):
if isinstance(m, nn.BatchNorm2d): if isinstance(m, nn.BatchNorm2d):
nn.init.normal_(m.weight, .1, .1) nn.init.normal_(m.weight, .1, .1)
nn.init.normal_(m.bias, 0, .1) nn.init.normal_(m.bias, 0, .1)
for attn in (self.attn, self.attn_last): if hasattr(self.attn, 'reset_parameters'):
if hasattr(attn, 'reset_parameters'): self.attn.reset_parameters()
attn.reset_parameters()
def forward(self, x): def forward(self, x):
if self.identity is None: if self.identity is None: