Fix dtype bug in base_dense_head

open-mmlab · Dec 15, 2021 · 5612624 · 5612624
1 parent 16c682c
commit 5612624
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmdet/models/dense_heads/base_dense_head.py b/mmdet/models/dense_heads/base_dense_head.py
@@ -85,7 +85,7 @@ def get_bboxes(self,
         featmap_sizes = [cls_scores[i].shape[-2:] for i in range(num_levels)]
         mlvl_priors = self.prior_generator.grid_priors(
             featmap_sizes,
-            dtype=cls_scores[0].device,
+            dtype=cls_scores[0].dtype,
             device=cls_scores[0].device)
 
         result_list = []