Fix dtype bug in base_dense_head

ZwwWayne · Jul 18, 2022 · 5c5431e · 5c5431e
1 parent 21a80ef
commit 5c5431e
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
@@ -98,7 +98,7 @@ def get_results(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 = []