How to use 2 GPUs? - Githubissues

ttengwang / PDVC

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

MIT License

200 stars 23 forks source link

Open yhd-123 opened 7 months ago

yhd-123 commented 7 months ago

python train.py --cfg_path ${cfgs/anet_c3d_pdvc.yml} But only one GPU is used!

ttengwang commented 7 months ago

This repository does not currently support data parallelism for multiple gpus. If you are interested, you can modify the code accordingly. A helpful reference is https://pytorch.org/tutorials/intermediate/model_parallel_tutorial.html

yhd-123 commented 7 months ago

OK！Thanks!