How to launch distributed training?

ashawkey / stable-dreamfusion

Text-to-3D & Image-to-3D & Mesh Exportation with NeRF + Diffusion.

Apache License 2.0

8k stars 710 forks source link

Open XiongFenghhh opened 1 year ago

XiongFenghhh commented 1 year ago

In the origin article they trained nerf with 4 tpus. Can stable-dreamfusion launched for distributed training with multiple gpu?

ashawkey commented 1 year ago

@XiongFenghhh Hi, unfortunately, this is currently not supported.

flobotics commented 1 year ago

hi, @ashawkey is multi GPU support planned for near future to run it on cloud machines ?

asher-lab commented 1 year ago

Hi @XiongFenghhh we have the same issue and I hope they can fix this soon