Does the code support multi-gpu training?

ttengwang / PDVC

End-to-End Dense Video Captioning with Parallel Decoding (ICCV 2021)

MIT License

200 stars 23 forks source link

Does the code support multi-gpu training? #44

Closed SuleBai closed 1 year ago

SuleBai commented 1 year ago

Hi, thanks for your great work! I use the command python train.py --cfg_path ${config_path} --gpu_id 0,1,2,3 to train the model, however, it seems only the first gpu is working. Does the code support multi-gpu training? Can you share the multi-GPU training command? Thanks a lot.

ttengwang commented 1 year ago

Hi, the repo doesn't support multiple GPU training.