search

ZwwWayne / mmMOT

[ICCV2019] Robust Multi-Modality Multi-Object Tracking
252 stars 25 forks source link

Training on GPUs #14

Closed ymxlzgy closed 4 years ago

ymxlzgy commented 4 years ago

Thanks for sharing your code, I wonder that whether I can choose which GPU I can train this model on or not?

ymxlzgy commented 4 years ago

How can I train this framework on multi GPUs ?

ZwwWayne commented 4 years ago

Use nn.DistributedDataParallel should work.

ymxlzgy commented 4 years ago

Thanks!