naver/r2d2

Does the training code support multi-gpu training?

pixelsmaker opened this issue · 1 comments

pixelsmaker commented
Does the training code support multi-gpu training?
jerome-revaud commented

I think so. The GPU parameter can take multiple values. Anyway, with PyTorch it's pretty easy to parallelize on multi-GPU.