how to train the model on multiple GPUs?

[anchoret2009]

thanks.

[NoahSchiro]

Hi I am not the project maintainer but this question is more oriented towards a question about how PyTorch works. PyTorch has great documentation on multi-gpu training which you can find here https://pytorch.org/tutorials/beginner/ddp_series_intro.html

You will need to modify some of the variables in the train() function to do match the above documentation. It believe it is fairly straightforward as long as you are familiar with pytorch

[anchoret2009]

thanks~