Fix High CPU Usage - Githubissues

kcheng1021 / GaussianPro

[ICML2024] Official code for GaussianPro: 3D Gaussian Splatting with Progressive Propagation

https://kcheng1021.github.io/gaussianpro.github.io/

MIT License

551 stars 36 forks source link

Fix High CPU Usage #49

Closed kevintsq closed 3 weeks ago

kevintsq commented 3 weeks ago

Some tensors were created on the CPU, causing high CPU usage. This PR fixes this issue by creating them directly on the GPU, lowering CPU usage and thereby improving training speed.