About Mutil-Gpus for training?

[GreenLimeSia]

Hi, authors:

I found an issue when training the code with multi-GPUs, which is that only a single GPU is used despite inputting multiple GPUs. Can you solve this problem in your spare time?

Thanks!

[imlixinyang]

Hi! The code is expected to support multi-gpu training now with DataParallel.
Can you share which command you used?

[GreenLimeSia]

I train the model with your code and your official command, i.e., python core/train.py --config configs/celeba-hq.yaml --gpus 0,1.
However, only one single GPU is used when training. The reason for this issue may be that Class "Gen" has no function of forward(self, ..). Can you help me?

[GreenLimeSia]

A link for this issue is in here . Can you solve this problem in your spare time? @imlixinyang
Thanks!

[imlixinyang]

Try command "python core/train.py --config configs/celeba-hq.yaml --gpus 0 1".

[GreenLimeSia]

I will try it now. Thanks for your reply. I am doing a new work based on your novel work. I will cite your work.
Thanks for your hard work.

[GreenLimeSia]

It works now. Thanks again. @imlixinyang

[imlixinyang]

Glad to hear that! Gook luck for your research.