Does the training code support multi-gpu training?

[jerome-revaud]

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