Revert "swich to new pytorch api (facebookresearch#359)" (facebookres…

…earch#364) This reverts commit 6cbb3d2.
Lyears · Jan 21, 2019 · eb3b079 · eb3b079
1 parent 7510037
commit eb3b079
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,
-            reduction='sum',
+            size_average=False,
         ) / (sampled_inds.numel())
 
         objectness_loss = F.binary_cross_entropy_with_logits(