Skip to content
Snippets Groups Projects
Unverified Commit e6e23152 authored by Kai Chen's avatar Kai Chen Committed by GitHub
Browse files

fix inplace add in roi extractor (#1160)

parent 31bf963d
No related branches found
No related tags found
No related merge requests found
......@@ -103,5 +103,5 @@ class SingleRoIExtractor(nn.Module):
if inds.any():
rois_ = rois[inds, :]
roi_feats_t = self.roi_layers[i](feats[i], rois_)
roi_feats[inds] += roi_feats_t
roi_feats[inds] = roi_feats_t
return roi_feats
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment