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

BatchNorm2d() to EfficientDet defaults: decay=0.997 eps=1e-4

This commit is contained in:
Glenn Jocher 2020-03-14 16:54:04 -07:00
parent 9ce4ec48a7
commit ea4c26b32d
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

@ -34,7 +34,7 @@ def create_modules(module_defs, img_size, arc):
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.1)) modules.add_module('BatchNorm2d', nn.BatchNorm2d(filters, momentum=0.003, eps=1E-4))
else: else:
routs.append(i) # detection output (goes into yolo layer) routs.append(i) # detection output (goes into yolo layer)