From e1f724b1a38f71e8194723cefefdd6627d964b32 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 19 Aug 2019 01:32:27 +0200 Subject: [PATCH] updates --- models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models.py b/models.py index c3798d5b..314382fb 100755 --- a/models.py +++ b/models.py @@ -77,7 +77,7 @@ def create_modules(module_defs, img_size): # Initialize preceding Conv2d() detection bias to -5 (https://arxiv.org/pdf/1708.02002.pdf section 3.3) bias = module_list[-1][0].bias.view(len(mask), -1) # 255 to 3x85 - bias[:, 4:] -= - 5 + bias[:, 4:] -= 5 module_list[-1][0].bias = torch.nn.Parameter(bias.view(-1)) else: