Hello, how can I train this model using multiple GPUs?

[mjz0110]

Hello, how can I train this model using multiple GPUs?

[usert5432]

Hi @mjz0110,

That depends on which model you would like to train:

1. If you would like to pre-train an image inpainting model, then simply run the pre-training on a machine with several GPUs and set CUDA_VISIBLE_DEVICES environment variable to "comma separated list of GPU indices" . The pre-training script will automatically use data parallel training.
2. If you would like to train a style transfer model, then unfortunately, there is currently no way to train it on multiple GPUs.

[mjz0110]

Thanks for your help!!!!