Skip to content

GitLab

"...v2/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "7b4681a66d5d8a532a937c04714f5beb7789f017"
Unverified Commit 6250b2cb authored by Ginray's avatar Ginray Committed by GitHub
Browse files

[Fix] Fix the device mismatch error when torch version <1.11 (#2308)

parent 5d88bef6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmdet3d/models/dense_heads/centerpoint_head.py
View file @ 6250b2cb
......@@ -471,7 +471,7 @@ class CenterHead(BaseModule):
(gt_bboxes_3d.gravity_center, gt_bboxes_3d.tensor[:, 3:]),
dim=1).to(device)
max_objs = self.train_cfg['max_objs'] * self.train_cfg['dense_reg']
grid_size = torch.tensor(self.train_cfg['grid_size'])
grid_size = torch.tensor(self.train_cfg['grid_size']).to(device)
pc_range = torch.tensor(self.train_cfg['point_cloud_range'])
voxel_size = torch.tensor(self.train_cfg['voxel_size'])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment