tobyperrett / trx

Temporal-Relational CrossTransformers (CVPR 2021)
107 stars 23 forks source link

Multi-GPU inference problem. #20

Closed jiazheng-xing closed 4 months ago

jiazheng-xing commented 4 months ago

File "/slurm-files/ock/fsar_llava/fsar_llava/videollava/model/llava_arch.py", line 207, in prepare_inputs_labels_for_multimodal video_features_minibatch = self.encode_videos(videos_minibatch) # fake list [mini_b, t, l, c] RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:3 and cuda:0! (when checking argument for argument mat2 in method wrapper_CUDA_mm)