Speeding up training stage

[hardikdava]

I am new to gaussian splatting and taichi as well. Is there quick way to speed up training process without loosing quality. Is it possible to do batch processing?

P.S. The training process takes more than 1 hour for 20k iterations on Tesla T4.

[wanmeihuali]

I think so. It's in previous TODO list. But I don't have that much time to implement it... It shall not be difficult, just need to add a new dim in sorting key(tile uv to duv), and also add the dim in rasterization/backward. As we only use less than 4GB GPU memory now, I think it shall help.

[jb-ye]

@wanmeihuali do you still have plans for batch processing?

[wanmeihuali]

@Alexma3312 @Alexma3312 @yanzhoupan Do you have time to take a look?

[yanzhoupan]

Will start working on batch training.