naver / r2d2

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Does the training code support multi-gpu training?

pixelsmaker opened this issue · comments

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