Fix dtype bug in base_dense_head

open-mmlab · Dec 13, 2021 · c9f4297 · c9f4297
1 parent 43699a2
commit c9f4297
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 = []