Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix fuse (#440) 

Fix fuse in models.py
This commit is contained in:
晨太狼 2019-08-09 18:44:47 +08:00 committed by Glenn Jocher
parent fdd5afa229
commit d41b444a15
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
models.py
View File

@ -213,6 +213,7 @@ class Darknet(nn.Module):
# Fuse Conv2d + BatchNorm2d layers throughout model # Fuse Conv2d + BatchNorm2d layers throughout model
fused_list = nn.ModuleList() fused_list = nn.ModuleList()
for a in list(self.children())[0]: for a in list(self.children())[0]:
if isinstance(a, nn.Sequential):
for i, b in enumerate(a): for i, b in enumerate(a):
if isinstance(b, nn.modules.batchnorm.BatchNorm2d): if isinstance(b, nn.modules.batchnorm.BatchNorm2d):
# fuse this bn layer with the previous conv2d layer # fuse this bn layer with the previous conv2d layer