RuntimeError: expected backend CUDA and dtype Float but got backend CUDA and dtype Long

最新推荐文章于 2025-05-14 09:50:03 发布

最新推荐文章于 2025-05-14 09:50:03 发布 · 490 阅读

文章标签：

#python #深度学习 #开发语言

在运行Python深度学习项目时遇到RuntimeError，错误信息提示预期的backend为CUDA且dtype为Float，但实际得到的是CUDA和Long。问题出现在`btcdet/datasets/multifindbestfit.py`文件的第388行，可能的原因是数据类型不匹配。寻求大神解决，愿意付费。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

(btc) z@z2:~/wen/BtcDet-main$ python -m btcdet.datasets.multifindbestfit
ROOT_DIR /home/z/wen/BtcDet-main
iou3d torch.Size([28742, 28742])
coords_num torch.Size([28742]) tensor(0, device='cuda:0') tensor(354, device='cuda:0')
coord_inds torch.Size([12522])
best_iou_indices torch.Size([28742, 800]) pnt_thresh_best_iou_indices torch.Size([28742, 800]) 28742
pnt_thresh_best_iou_indices torch.Size([28742, 800])
Traceback (most recent call last):
File "/home/z/anaconda3/envs/btc/lib/python3.6/runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
File "/home/z/anaconda3/envs/btc/lib/python3.6/runpy.py", line 85, in _run_code
    exec(code, run_globals)
File "/home/z/wen/BtcDet-main/btcdet/datasets/multifindbestfit.py", line 494, in <module>
    find_best_match_boxpnts(all_db_infos_lst, box_dims_lst, sorted_iou, pnt_thresh_best_iou_indic