Skip to content

GitLab

"pytorch/git@developer.sourcefind.cn:OpenDAS/fastmoe.git" did not exist on "3a0086fafb2790d259b4f1c25cb6c454ecce41a4"
Commit 2c68e545 authored by Kai Chen's avatar Kai Chen
Browse files

api version fix

parent d08a9ec2
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 5 deletions
configs/retinanet_r50_fpn_1x.py
View file @ 2c68e545
...@@ -30,9 +30,8 @@ model = dict( ...@@ -30,9 +30,8 @@ model = dict(
target_stds=[1.0, 1.0, 1.0, 1.0])) target_stds=[1.0, 1.0, 1.0, 1.0]))
# training and testing settings # training and testing settings
train_cfg = dict( train_cfg = dict(
pos_iou_thr=0.5, assigner=dict(
neg_iou_thr=0.4, pos_iou_thr=0.5, neg_iou_thr=0.4, min_pos_iou=0.4, ignore_iof_thr=-1),
min_pos_iou=0.4,
smoothl1_beta=0.11, smoothl1_beta=0.11,
gamma=2.0, gamma=2.0,
alpha=0.25, alpha=0.25,
......
mmdet/core/anchor/anchor_target.py
View file @ 2c68e545
...@@ -121,8 +121,8 @@ def anchor_target_single(flat_anchors, ...@@ -121,8 +121,8 @@ def anchor_target_single(flat_anchors,
num_valid_anchors = anchors.shape[0] num_valid_anchors = anchors.shape[0]
bbox_targets = torch.zeros_like(anchors) bbox_targets = torch.zeros_like(anchors)
bbox_weights = torch.zeros_like(anchors) bbox_weights = torch.zeros_like(anchors)
labels = anchors.new_zeros((num_valid_anchors, )) labels = gt_labels.new_zeros(num_valid_anchors)
label_weights = anchors.new_zeros((num_valid_anchors, )) label_weights = gt_labels.new_zeros(num_valid_anchors, dtype=torch.float)
pos_inds = sampling_result.pos_inds pos_inds = sampling_result.pos_inds
neg_inds = sampling_result.neg_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