This commit is contained in:
Glenn Jocher
2019-06-26 11:27:36 +02:00
parent 45540c787f
commit cbfc5a00e5
2 changed files with 1 additions and 3 deletions
-2
View File
@@ -9,8 +9,6 @@ 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()