search

Naman-ntc / Pytorch-Human-Pose-Estimation

Implementation of various human pose estimation models in pytorch on multiple datasets (MPII & COCO) along with pretrained models
MIT License
474 stars 75 forks source link

how can I train deeppose with muti-gpu? #21

Closed wongkwan closed 4 years ago

wongkwan commented 4 years ago

I want to train deeppose with muti-gpu, what should I do? I try change gpu id to 0, 1, 2, 3, but it is not available. Thank you so much if you can help me !

Naman-ntc commented 4 years ago

Hi, the current codebase does not support multi GPU training. You can try enclosing the model with DataParallel (https://pytorch.org/docs/master/generated/torch.nn.DataParallel.html) and see if multi-gpu training works.

Feel free to open the issue again if facing any further issues.