Training on GPUs

[ymxlzgy]

Thanks for sharing your code, I wonder that whether I can choose which GPU I can train this model on or not?

[ymxlzgy]

How can I train this framework on multi GPUs ?

[Wenwei Zhang]

Use nn.DistributedDataParallel should work.

[ymxlzgy]

Thanks!