how to carry out multi-GPU with lightning-pose?

danbider / lightning-pose

Accelerated pose estimation and tracking using semi-supervised convolutional networks.

MIT License

223 stars 32 forks source link

how to carry out multi-GPU with lightning-pose? #138

Open Wulin-Tan opened 2 months ago

Wulin-Tan commented 2 months ago

Hi, lighting pose team I am very interested in this super cool tool. How to do multi-GPU with lighting pose? I can't find multi-GPU information in the tutorial.

themattinthehatt commented 2 months ago

We do not currently support multi-GPU training but it should not be difficult to implement with our current setup (at least for the supervised models). We'll be happy to look into this but it might take us a week or so, we'll keep you updated.