Skip to content

Commit 5612624

Browse files
shinya7yZwwWayne
authored andcommitted
Fix dtype bug in base_dense_head
1 parent 16c682c commit 5612624

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mmdet/models/dense_heads/base_dense_head.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def get_bboxes(self,
8585
featmap_sizes = [cls_scores[i].shape[-2:] for i in range(num_levels)]
8686
mlvl_priors = self.prior_generator.grid_priors(
8787
featmap_sizes,
88-
dtype=cls_scores[0].device,
88+
dtype=cls_scores[0].dtype,
8989
device=cls_scores[0].device)
9090

9191
result_list = []

0 commit comments

Comments
 (0)