From 4fc00128293d32ae841f04ab8b6db7a32c29e153 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 5 Apr 2020 18:03:49 -0700 Subject: [PATCH] initial batchnorm to 0.03 momentum --- models.py | 1 - 1 file changed, 1 deletion(-) diff --git a/models.py b/models.py index 3d6b3ad6..2f93ff22 100755 --- a/models.py +++ b/models.py @@ -56,7 +56,6 @@ def create_modules(module_defs, img_size): # imagenet mean and var https://pytorch.org/docs/stable/torchvision/models.html#classification modules.running_mean = torch.tensor([0.485, 0.456, 0.406]) modules.running_var = torch.tensor([0.0524, 0.0502, 0.0506]) - modules.momentum = 0.003 elif mdef['type'] == 'maxpool': size = mdef['size']