search

tusen-ai / SST

Code for a series of work in LiDAR perception, including SST (CVPR 22), FSD (NeurIPS 22), FSD++ (TPAMI 23), FSDv2, and CTRL (ICCV 23, oral).
Apache License 2.0
801 stars 102 forks source link

'ConfigDict' object has no attribute 'device' #76

Closed SH-Tan closed 2 years ago

SH-Tan commented 2 years ago

Hello, I tried to change the config file to kitti dataset, and get the below error message. I did not find any device cfg info in other config files, should I add the device info in config file?

Traceback (most recent call last): File "tools/train.py", line 230, in main() File "tools/train.py", line 220, in main train_model( File "/home/tan/SST/mmdet3d/apis/train.py", line 27, in train_model train_detector( File "/home/tan/anaconda3/envs/mmdet3d/lib/python3.8/site-packages/mmdet/apis/train.py", line 163, in train_detector model = build_dp(model, cfg.device, device_ids=cfg.gpu_ids) File "/home/tan/anaconda3/envs/mmdet3d/lib/python3.8/site-packages/mmcv/utils/config.py", line 48, in getattr raise ex AttributeError: 'ConfigDict' object has no attribute 'device'

SH-Tan commented 2 years ago

solved by added "device='cuda'" in config file