We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 16c682c commit 5612624Copy full SHA for 5612624
mmdet/models/dense_heads/base_dense_head.py
@@ -85,7 +85,7 @@ def get_bboxes(self,
85
featmap_sizes = [cls_scores[i].shape[-2:] for i in range(num_levels)]
86
mlvl_priors = self.prior_generator.grid_priors(
87
featmap_sizes,
88
- dtype=cls_scores[0].device,
+ dtype=cls_scores[0].dtype,
89
device=cls_scores[0].device)
90
91
result_list = []
0 commit comments