updates

2020-02-02 08:55:34 -08:00 · 2020-02-02 08:55:34 -08:00 · 8b18beb3db
parent d23f721dcf
commit 8b18beb3db
1 changed files with 2 additions and 0 deletions
--- 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'])