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

updates

This commit is contained in:
Glenn Jocher 2020-02-02 08:55:34 -08:00
parent d23f721dcf
commit 8b18beb3db
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -18,6 +18,8 @@ def create_modules(module_defs, img_size, arc):
for i, mdef in enumerate(module_defs): for i, mdef in enumerate(module_defs):
modules = nn.Sequential() modules = nn.Sequential()
# if i == 0:
# modules.add_module('BatchNorm2d_0', nn.BatchNorm2d(output_filters[-1], momentum=0.1))
if mdef['type'] == 'convolutional': if mdef['type'] == 'convolutional':
bn = int(mdef['batch_normalize']) bn = int(mdef['batch_normalize'])