updates

2019-08-19 01:32:27 +02:00 · 2019-08-19 01:32:27 +02:00 · e1f724b1a3
parent 98a24c0a2f
commit e1f724b1a3
1 changed files with 1 additions and 1 deletions
--- 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: