updates
This commit is contained in:
parent
f43170817c
commit
1ddde51b80
|
@ -16,7 +16,7 @@ def create_modules(module_defs):
|
||||||
module_list = nn.ModuleList()
|
module_list = nn.ModuleList()
|
||||||
yolo_index = -1
|
yolo_index = -1
|
||||||
|
|
||||||
for i, mdef in enumerate(module_defs):
|
for mdef in module_defs:
|
||||||
modules = nn.Sequential()
|
modules = nn.Sequential()
|
||||||
|
|
||||||
if mdef['type'] == 'convolutional':
|
if mdef['type'] == 'convolutional':
|
||||||
|
|
Loading…
Reference in New Issue