facebookresearch / robust-dynrf

An algorithm for reconstructing the radiance field of a dynamic scene from a casually-captured video.
MIT License
219 stars 11 forks source link

Problems about Using more than one GPU to Train #1

Closed xiazhi1 closed 1 year ago

xiazhi1 commented 1 year ago

Hi,Thanks for awesome work. I have some problem when I run it on more GPU. I want to know if your code supports it

alex04072000 commented 1 year ago

Thank you so much for being interested in our work! Currently, our implementation only supports single GPU training. However, multi-GPU training should be feasible by torch.nn.DataParallel.

WithFoxSquirrel commented 1 year ago

Thank you so much for being interested in our work! Currently, our implementation only supports single GPU training. However, multi-GPU training should be feasible by torch.nn.DataParallel.

How long does it take to train a model on V100?