From c9f4297b67bc4ddba9513c4d3dac14a2da8605d2 Mon Sep 17 00:00:00 2001
From: Yosuke Shinya <42844407+shinya7y@users.noreply.github.com>
Date: Sun, 12 Dec 2021 17:23:43 +0000
Subject: [PATCH] Fix dtype bug in base_dense_head

---
 mmdet/models/dense_heads/base_dense_head.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mmdet/models/dense_heads/base_dense_head.py b/mmdet/models/dense_heads/base_dense_head.py
index 503ccfc9f9c..0c7abb7b9b8 100644
--- 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 = []