Skip to content

Commit a73315f

Browse files
fix ONNXRuntime cuda test bug (open-mmlab#438)
1 parent 82b757e commit a73315f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

mmdeploy/codebase/mmdet/deploy/object_detection_model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -208,6 +208,7 @@ def forward(self, img: Sequence[torch.Tensor], img_metas: Sequence[dict],
208208
rescale = kwargs.get('rescale', True)
209209
for i in range(batch_size):
210210
dets, labels = batch_dets[i], batch_labels[i]
211+
dets = dets.to(device=torch.device(self.device))
211212
if rescale:
212213
scale_factor = img_metas[i]['scale_factor']
213214

0 commit comments

Comments
 (0)