swich to new pytorch api (facebookresearch#359)

AnnaAraslanova · Jan 21, 2019 · 6cbb3d2 · 6cbb3d2
1 parent 3b27142
commit 6cbb3d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/maskrcnn_benchmark/modeling/rpn/loss.py b/maskrcnn_benchmark/modeling/rpn/loss.py
@@ -126,7 +126,7 @@ def __call__(self, anchors, objectness, box_regression, targets):
             box_regression[sampled_pos_inds],
             regression_targets[sampled_pos_inds],
             beta=1.0 / 9,
-            size_average=False,
+            reduction='sum',
         ) / (sampled_inds.numel())
 
         objectness_loss = F.binary_cross_entropy_with_logits(