initial batchnorm to 0.03 momentum

2020-04-05 18:03:49 -07:00 · 2020-04-05 18:03:49 -07:00 · 4fc0012829
parent 26fc4fb018
commit 4fc0012829
1 changed files with 0 additions and 1 deletions
--- 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']