search

FangShancheng / ABINet

Read Like Humans: Autonomous, Bidirectional and Iterative Language Modeling for Scene Text Recognition
Other
420 stars 72 forks source link

Dataparallel for training on mutliple GPU #86

Open Preethse opened 1 year ago

Preethse commented 1 year ago

how to train on multiple gpu? I tried giving 4 Gpus but it only trains one 1 GPU.

Preethse commented 1 year ago

Only one of the GPU uses more than 50%, other GPU utilization is only 10-15%. Is there a way I can speed it up?