Unverified Commit 205fa4e7 authored by Francisco Massa's avatar Francisco Massa Committed by GitHub

Revert "swich to new pytorch api (#359)" (#364)

This reverts commit 6cbb3d27.
parent 6cbb3d27
......@@ -126,7 +126,7 @@ class RPNLossComputation(object):
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(
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment