splatfacto multi-GPU scene reconstruction

[laszilagyi]

Hi All,

I'd like to reconstruct a large scene using multiple GPUs to improve speed, however the training hangs after initialization. I've used these command with no luck (analogous to the nerfacto multi-GPU example, removing some parameters):

export CUDA_VISIBLE_DEVICES=0,1,2,3
ns-train splatfacto-big --machine.num-devices 4 --data data/nerfstudio/aspen

Any additional parameters I need?

Another related question: How can I control the size of the model (the number of Gaussians)?

Appreciate the help!