model training

[YiF-chen]

Hi , Teoge. I want to train it use n(n>1) GPUS? if , need to change the code.

[Teoge]

Multi-GPU training with pytorch is pretty simple. All you need to do is wrap the model with nn.DataParallel. Here are some links that might help you: https://pytorch.org/tutorials/beginner/blitz/data_parallel_tutorial.html https://pytorch.org/tutorials/beginner/former_torchies/parallelism_tutorial.html Although I haven't tried it yet, I believe it should work fine. Feel free to let me know if you encounter more problems.