tbepler/topaz

Training on multi gpus

Closed this issue · 1 comments

nguyen14ck commented

Can we train the model on multi gpus?
I set

--device=0,1

but that did not work.

Thanks

tbepler commented

No, we don't have support for multi GPU training. Unfortunately, this wouldn't provide any speed up for topaz train anwyay, because training isn't bottlenecked by the GPU operations.