From b0e8ece9ba0d76713b86ca4da4060d41bbc86451 Mon Sep 17 00:00:00 2001
From: LRJKD <101466907+LRJKD@users.noreply.github.com>
Date: Wed, 18 Oct 2023 16:12:41 +0800
Subject: [PATCH] [Fix] Fix support for devices other than Cuda in
 FreeAnchor3DHead (#2769)

---
 mmdet3d/models/dense_heads/free_anchor3d_head.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/mmdet3d/models/dense_heads/free_anchor3d_head.py b/mmdet3d/models/dense_heads/free_anchor3d_head.py
index 94cf712aaa..21ac1e5aa6 100644
--- a/mmdet3d/models/dense_heads/free_anchor3d_head.py
+++ b/mmdet3d/models/dense_heads/free_anchor3d_head.py
@@ -2,6 +2,7 @@
 from typing import Dict, List
 
 import torch
+from mmengine.device import get_device
 from torch import Tensor
 from torch.nn import functional as F
 
@@ -79,7 +80,9 @@ def loss_by_feat(
         featmap_sizes = [featmap.size()[-2:] for featmap in cls_scores]
         assert len(featmap_sizes) == self.prior_generator.num_levels
 
-        anchor_list = self.get_anchors(featmap_sizes, batch_input_metas)
+        device = get_device()
+        anchor_list = self.get_anchors(featmap_sizes, batch_input_metas,
+                                       device)
         mlvl_anchors = [torch.cat(anchor) for anchor in anchor_list]
 
         # concatenate each level