Commit 2c68e545 authored by Kai Chen's avatar Kai Chen
Browse files

api version fix

parent d08a9ec2
......@@ -30,9 +30,8 @@ model = dict(
target_stds=[1.0, 1.0, 1.0, 1.0]))
# training and testing settings
train_cfg = dict(
pos_iou_thr=0.5,
neg_iou_thr=0.4,
min_pos_iou=0.4,
assigner=dict(
pos_iou_thr=0.5, neg_iou_thr=0.4, min_pos_iou=0.4, ignore_iof_thr=-1),
smoothl1_beta=0.11,
gamma=2.0,
alpha=0.25,
......
......@@ -121,8 +121,8 @@ def anchor_target_single(flat_anchors,
num_valid_anchors = anchors.shape[0]
bbox_targets = torch.zeros_like(anchors)
bbox_weights = torch.zeros_like(anchors)
labels = anchors.new_zeros((num_valid_anchors, ))
label_weights = anchors.new_zeros((num_valid_anchors, ))
labels = gt_labels.new_zeros(num_valid_anchors)
label_weights = gt_labels.new_zeros(num_valid_anchors, dtype=torch.float)
pos_inds = sampling_result.pos_inds
neg_inds = sampling_result.neg_inds
......
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