apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updates

Browse Source
This commit is contained in:
Glenn Jocher 2020-03-09 18:03:34 -07:00
parent 204594f299
commit 207cf14df4
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
models.py
View File

@ -41,6 +41,9 @@ def create_modules(module_defs, img_size, arc):
elif mdef['activation'] == 'swish': elif mdef['activation'] == 'swish':
modules.add_module('activation', Swish()) modules.add_module('activation', Swish())
if not bn: # detection output layer
routs.append(i)
elif mdef['type'] == 'maxpool': elif mdef['type'] == 'maxpool':
size = mdef['size'] size = mdef['size']
stride = mdef['stride'] stride = mdef['stride']