Skip to content

Commit f2af731

Browse files
committed
[BUG] Allow single_level_grid_priors to use cpu device.
Note: there is already an outstanding PR to resolve this issue: open-mmlab#2715
1 parent 3f8604b commit f2af731

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mmdeploy/codebase/mmdet/models/dense_heads/rtmdet_ins_head.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ def _mask_predict_by_feat_single(self, mask_feat, kernels, priors):
163163
batch_size = priors.shape[0]
164164
hw = mask_feat.size()[-2:]
165165
coord = self.prior_generator.single_level_grid_priors(
166-
hw, level_idx=0).to(mask_feat.device)
166+
hw, level_idx=0, device=mask_feat.device)
167167
coord = coord.unsqueeze(0).unsqueeze(0).repeat(batch_size, 1, 1, 1)
168168
priors = priors.unsqueeze(2)
169169
points = priors[..., :2]

0 commit comments

Comments
 (0)