search

Vegetebird / MHFormer

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

GPU Not being utilized within iterations (Multi GPU) #66

Closed anas-zafar closed 2 years ago

Vegetebird commented 2 years ago

I use the "model = nn.DataParallel(model)" to use multi GPU. It works well. I do not know how to solve your issue.

anas-zafar commented 2 years ago

@Vegetebird can you please point me at what line are you adding this to in the code base?

Vegetebird commented 2 years ago

I add "model = nn.DataParallel(model)" before https://github.com/Vegetebird/MHFormer/blob/fcf238631016f906477ec9c1d17582097ecf9803/main.py#L131