From 1d45ec84bc95617b36a38ff1ae33a4237df2fa73 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 4 Mar 2020 14:12:31 -0800 Subject: [PATCH] updates --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index 32f5176b..493229e7 100755 --- a/models.py +++ b/models.py @@ -87,7 +87,7 @@ def create_modules(module_defs, img_size, arc): # Initialize preceding Conv2d() bias (https://arxiv.org/pdf/1708.02002.pdf section 3.3) try: - p = math.log(1 / (modules.nc - 1)) # class probability -> sigmoid(p) = 1/nc + p = math.log(1 / (modules.nc - 0.99)) # class probability -> sigmoid(p) = 1/nc if arc == 'default' or arc == 'Fdefault': # default b = [-5.0, p] # obj, cls elif arc == 'uBCE': # unified BCE (80 classes)