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

batchnorm momentum to 0.03

This commit is contained in:
Glenn Jocher 2020-04-02 14:10:45 -07:00
parent 9155ef3f4f
commit aa4591d7e9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -32,7 +32,7 @@ def create_modules(module_defs, img_size):
groups=mdef['groups'] if 'groups' in mdef else 1, groups=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.003, eps=1E-4)) modules.add_module('BatchNorm2d', nn.BatchNorm2d(filters, momentum=0.03, eps=1E-4))
else: else:
routs.append(i) # detection output (goes into yolo layer) routs.append(i) # detection output (goes into yolo layer)