search

jchibane / srf

Stereo Radiance Fields (SRF) - Codebase
123 stars 18 forks source link

Multi-GPU Training #9

Closed yizhou-wang closed 2 years ago

yizhou-wang commented 2 years ago

Hi @jchibane,

Is your code capable of multi-GPU training? That would be very helpful for training on large datasets.

Thanks!

jchibane commented 2 years ago

Sure it is. Please have a look into PyTorchs multi GPU possibilities and adapt the code to your needs! Best, Julian