From 8b18beb3dbd6041fd8224ee40b8d7519ba7849ab Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 2 Feb 2020 08:55:34 -0800 Subject: [PATCH] updates --- models.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/models.py b/models.py index 0b513f37..ea262375 100755 --- a/models.py +++ b/models.py @@ -18,6 +18,8 @@ def create_modules(module_defs, img_size, arc): for i, mdef in enumerate(module_defs): modules = nn.Sequential() + # if i == 0: + # modules.add_module('BatchNorm2d_0', nn.BatchNorm2d(output_filters[-1], momentum=0.1)) if mdef['type'] == 'convolutional': bn = int(mdef['batch_normalize'])