diff --git a/models.py b/models.py index 089b6034..9d2538e6 100755 --- a/models.py +++ b/models.py @@ -32,7 +32,7 @@ def create_modules(module_defs, img_size): groups=mdef['groups'] if 'groups' in mdef else 1, bias=not 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: routs.append(i) # detection output (goes into yolo layer)