search

wanmeihuali / taichi_3d_gaussian_splatting

An unofficial implementation of paper 3D Gaussian Splatting for Real-Time Radiance Field Rendering by taichi lang.
Apache License 2.0
638 stars 58 forks source link

Speeding up training stage #130

Open hardikdava opened 10 months ago

hardikdava commented 10 months ago

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 commented 10 months ago

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 commented 8 months ago

@wanmeihuali do you still have plans for batch processing?

wanmeihuali commented 8 months ago

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

yanzhoupan commented 8 months ago

Will start working on batch training.