Commit 6cbb3d27 authored by 103yiran's avatar 103yiran Committed by Francisco Massa

swich to new pytorch api (#359)

parent 3b27142e
......@@ -126,7 +126,7 @@ class RPNLossComputation(object):
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(
......
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