From e5ab942c14fd3903121c6f04bb324483cb3e3427 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 16 Sep 2019 22:42:52 +0200 Subject: [PATCH] updates --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index 527d3466..79cca577 100755 --- a/models.py +++ b/models.py @@ -79,7 +79,7 @@ def create_modules(module_defs, img_size, arc): # Initialize preceding Conv2d() bias (https://arxiv.org/pdf/1708.02002.pdf section 3.3) try: if arc == 'defaultpw': # default with positive weights - b = [-4, -4] # obj, cls + b = [-4, -3.6] # obj, cls elif arc == 'default': # default no pw (40 cls, 80 obj) b = [-5.5, -4.0] elif arc == 'uBCE': # unified BCE (80 classes)