RuntimeError: Tensors must be CUDA and dense

EdwardLeeLPZ / PowerBEV

POWERBEV, a novel and elegant vision-based end-to-end framework that only consists of 2D convolutional layers to perform perception and forecasting of multiple objects in BEVs.

Other

82 stars 18 forks source link

RuntimeError: Tensors must be CUDA and dense #7

Closed pupu-chenyanyan closed 6 months ago

pupu-chenyanyan commented 7 months ago

Dear authors: Hi ! @EdwardLeeLPZ when i run the train.py with two gpus, I met the wrong, ie, File "XXX/lib/python3.8/site-packages/torch/nn/parallel/distributed.py", line 1334, in _distributed_broadcast_coalesced dist._broadcast_coalesced(self.process_group, tensors, buffer_size, authoritative_rank) showed “Tensors must be CUDA and dense". However,I examine the parameter of tensors, this is a list where all elements are on cuda: 0.Hence, I do not know what's wrong？Thanks！

Zzmes commented 7 months ago

You can comment these lines in trainer.py