creiser / kilonerf

Code for KiloNeRF: Speeding up Neural Radiance Fields with Thousands of Tiny MLPs
471 stars 52 forks source link

Can the training run on multiple gpu to speed up training? #3

Open nitthilan opened 3 years ago

nitthilan commented 3 years ago

Hi, Is there a configuration to make the whole training procedure use multiple GPUs?

Thanking you.

Regards, K. J. Nitthilan

creiser commented 3 years ago

Thanks for your interest in our work. We did not implement Multi-GPU training, but it should be possible as usual.

yuanhaowang1213 commented 2 years ago

Will the implementation need some GPU coding or just python Dataparrallel is ok?

Could you pls give some insights?

Thank you.