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

initial batchnorm to 0.03 momentum

This commit is contained in:
Glenn Jocher 2020-04-05 18:03:49 -07:00
parent 26fc4fb018
commit 4fc0012829
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
models.py
View File

@ -56,7 +56,6 @@ def create_modules(module_defs, img_size):
# imagenet mean and var https://pytorch.org/docs/stable/torchvision/models.html#classification # 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_mean = torch.tensor([0.485, 0.456, 0.406])
modules.running_var = torch.tensor([0.0524, 0.0502, 0.0506]) modules.running_var = torch.tensor([0.0524, 0.0502, 0.0506])
modules.momentum = 0.003
elif mdef['type'] == 'maxpool': elif mdef['type'] == 'maxpool':
size = mdef['size'] size = mdef['size']