From ea4c26b32d621286610a21949eed3d506cad95c9 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 14 Mar 2020 16:54:04 -0700 Subject: [PATCH] BatchNorm2d() to EfficientDet defaults: decay=0.997 eps=1e-4 --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index 04224ac0..1d9dae0e 100755 --- a/models.py +++ b/models.py @@ -34,7 +34,7 @@ def create_modules(module_defs, img_size, arc): groups=mdef['groups'] if 'groups' in mdef else 1, bias=not 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: routs.append(i) # detection output (goes into yolo layer)