How to use 2 GPUs?

[yhd-123]

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

[ttengwang]

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]

OK！Thanks!