From d623a425d9190ab71d11d8b4123d66a6e70bfdb1 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 22 Nov 2019 16:20:11 -1000 Subject: [PATCH] updates --- models.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/models.py b/models.py index 9d16ff95..9bfbb114 100755 --- a/models.py +++ b/models.py @@ -36,7 +36,8 @@ def create_modules(module_defs, img_size, arc): if mdef['activation'] == 'leaky': # TODO: activation study https://github.com/ultralytics/yolov3/issues/441 modules.add_module('activation', nn.LeakyReLU(0.1, inplace=True)) # modules.add_module('activation', nn.PReLU(num_parameters=1, init=0.10)) - # modules.add_module('activation', Swish()) + elif mdef['activation'] == 'swish': + modules.add_module('activation', Swish()) elif mdef['type'] == 'maxpool': kernel_size = int(mdef['size'])