Multi-GPUs issue about pifu HOT 5 OPEN

Of course, but it may take time because I'm very beginner of GitHub.
If you are okay with it, I'll try :)

from pifu.

I pushed a small fix. Can you merge your multi-gpu code with it and see if it works?

from pifu.

The current code in the repository does not support multi-GPU training. The easiest way to use multi-gpu is using torch.nnDataparallel. But make sure you adjust code for gen_mesh and error evaluation.

from pifu.

After adding few codes to main file, code is well running now. However, I didn't adjust any code for gen_mesh. Do you think it will make a problem?
Also, can I ask what how you modified the code to work?

Anyway, thank you soooooooooo much for your fast and caring feed back. It really helped me a lot!

from pifu.

I meant you have to pass netG.module instead of netG if you replace the network with torch.dataparallel. You don't have to change code inside gen_mesh itself. Glad that it worked out. Would it be possible to send pull request with multi-GPU support if you don't mind? (I just haven't had time to do it by myself, thanks!)

from pifu.