how to train the model use multi gpu on one machine?

Media-Smart / vedastr

A scene text recognition toolbox based on PyTorch

Apache License 2.0

535 stars 100 forks source link

how to train the model use multi gpu on one machine? #76

Open githublsk opened 3 years ago

githublsk commented 3 years ago

Hi @ChaseMonsterAway , @mileistone I have one question that how can I train the model use multi gpu on one machine, can you help us to solve the question? Thank you in advance.