From 18d4ebfd12fd32a8d98a6427054f9b816b0a4333 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 26 Apr 2020 16:25:46 -0700 Subject: [PATCH] add Mish() support --- models.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/models.py b/models.py index 5e1ac1b3..5a437027 100755 --- a/models.py +++ b/models.py @@ -45,9 +45,10 @@ def create_modules(module_defs, img_size): 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': modules.add_module('activation', Swish()) + elif mdef['activation'] == 'mish': + modules.add_module('activation', Mish()) elif mdef['type'] == 'BatchNorm2d': filters = output_filters[-1]