search

Vegetebird / MHFormer

[CVPR 2022] MHFormer: Multi-Hypothesis Transformer for 3D Human Pose Estimation
MIT License
530 stars 85 forks source link

multiple gpu? #28

Closed rambleramble closed 2 years ago

rambleramble commented 2 years ago

saw this issue and we are trying to use multiple gpus?

https://github.com/Vegetebird/MHFormer/issues/27

Any advice?

Vegetebird commented 2 years ago

No problem, you can use "model = nn.DataParallel(model)" to use multiple gpus.

anas-zafar commented 2 years ago

Hi @Vegetebird , For multiple GPUs, the GPU seems to go to sleep at intervals within the epochs over different iterations. Any solution for this?