compatibility fix

xiao543348405 · Dec 24, 2017 · a1033fa · a1033fa
1 parent 499486c
commit a1033fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/layers/modules/multibox_loss.py b/layers/modules/multibox_loss.py
@@ -93,7 +93,7 @@ def forward(self, predictions, priors, targets):
         loss_c = log_sum_exp(batch_conf) - batch_conf.gather(1, conf_t.view(-1,1))
 
         # Hard Negative Mining
-        loss_c[pos] = 0 # filter out pos boxes for now
+        loss_c[pos.view(-1)] = 0 # filter out pos boxes for now
         loss_c = loss_c.view(num, -1)
         _,loss_idx = loss_c.sort(1, descending=True)
         _,idx_rank = loss_idx.sort(1)