Can I use `train.sh' to train the model on multiple GPUs?

maudzung / TTNet-Real-time-Analysis-System-for-Table-Tennis-Pytorch

Unofficial implementation of "TTNet: Real-time temporal and spatial video analysis of table tennis" (CVPR 2020)

https://arxiv.org/pdf/2004.09927.pdf

586 stars 157 forks source link

Can I use `train.sh' to train the model on multiple GPUs? #1

Closed chenzhutian closed 4 years ago

chenzhutian commented 4 years ago

It seems that the train.sh only train the model using single GPU. Many thanks!

maudzung commented 4 years ago

Hi Chenzhutian,

The train.sh bash shell script is for training the model on a single machine that has multiple GPUs. During training, you can check the usage of GPUs by executing the below command in a terminal:

nvidia-smi -l 1

chenzhutian commented 4 years ago

Cool! Thank you very much.