search

liuyuan-pal / NeuRay

[CVPR2022] Neural Rays for Occlusion-aware Image-based Rendering
GNU General Public License v3.0
408 stars 31 forks source link

How to apply multi-gpus training? #29

Open lifuguan opened 1 year ago

lifuguan commented 1 year ago

Thanks for your wonderful work. Is it possible to train the model using multiple gpus?

liuyuan-pal commented 1 year ago

Experiments are all conducted on a single 2080Ti or 1080Ti GPU (sadly, we don't have too many GPUs) so the current script does not support multi-GPU training. Maybe you can apply pytorch-lightning which requires rewriting the training script.