From cc08e09219960f70f4a70ef5ee0c13dfafe3832a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 28 Feb 2020 10:06:35 -0800 Subject: [PATCH] updates --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index 9214bfc2..fe21ad50 100755 --- a/models.py +++ b/models.py @@ -35,7 +35,7 @@ def create_modules(module_defs, img_size, arc): bias=not bn)) if bn: modules.add_module('BatchNorm2d', nn.BatchNorm2d(filters, momentum=0.1)) - if mdef['activation'] == 'leaky': # TODO: activation study https://github.com/ultralytics/yolov3/issues/441 + if mdef['activation'] == 'leaky': # 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)) elif mdef['activation'] == 'swish':