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

Merge remote-tracking branch 'origin/master'

This commit is contained in:
Glenn Jocher 2019-08-09 16:37:27 +02:00
parent 933f85f632 d41b444a15
commit 0c52cc0106
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