From 25ad727a3d93c12ecf3c83f9bfe49c01ab6187df Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 9 Mar 2020 18:22:42 -0700 Subject: [PATCH] updates --- models.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/models.py b/models.py index a326d7d6..0a212853 100755 --- a/models.py +++ b/models.py @@ -95,8 +95,7 @@ def create_modules(module_defs, img_size, arc): bias = module_list[-1][0].bias.view(len(mask), -1) # 255 to 3x85 bias[:, 4] += bo - bias[:, 4].mean() # obj - bias[:, 5:] += bc - bias[:, 5:].mean() # cls - module_list[-1][0].bias = torch.nn.Parameter(bias.view(-1)) # utils.print_model_biases(model) + bias[:, 5:] += bc - bias[:, 5:].mean() # cls, view with utils.print_model_biases(model) except: print('WARNING: smart bias initialization failure.')