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

updates

This commit is contained in:
Glenn Jocher 2019-12-09 13:25:34 -08:00
parent 07c1fafba8
commit 3bfbab7afd
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
models.py
View File

@ -30,6 +30,7 @@ def create_modules(module_defs, img_size, arc):
kernel_size=size, kernel_size=size,
stride=stride, stride=stride,
padding=pad, padding=pad,
groups=int(mdef['groups']) if 'groups' in mdef else 1,
bias=not bn)) bias=not bn))
if bn: if bn:
modules.add_module('BatchNorm2d', nn.BatchNorm2d(filters, momentum=0.1)) modules.add_module('BatchNorm2d', nn.BatchNorm2d(filters, momentum=0.1))