update parse_model_cfg() (#350)

Removing the two lines for adding batch_normalize key to convolutional layers causes the parsing of the model to break when parsing it in models.py
This commit is contained in:
Jeremy Hu 2019-06-27 18:35:24 -04:00 committed by Glenn Jocher
parent c1bb037cbe
commit b202baa31c
1 changed files with 2 additions and 0 deletions

View File

@ -9,6 +9,8 @@ def parse_model_cfg(path):
if line.startswith('['): # This marks the start of a new block
module_defs.append({})
module_defs[-1]['type'] = line[1:-1].rstrip()
if module_defs[-1]['type'] == 'convolutional':
module_defs[-1]['batch_normalize'] = 0
else:
key, value = line.split("=")
value = value.strip()