Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

MipNeRF360 Training Time #15

Open
Han230104 opened this issue May 7, 2024 · 5 comments
Open

MipNeRF360 Training Time #15

Han230104 opened this issue May 7, 2024 · 5 comments
Labels
enhancement New feature or request

Comments

@Han230104
Copy link

Excellent Work!
I'm evaluating the MipNeRF360 dataset on one RTX4090 GPU.
The average training time on the 9 scenes is 29 min (Already move grid_x grid_y to GPU)
I wonder if this training time is normal or not since you didn't report the training time on your paper.
Thank you!

@hbb1
Copy link
Owner

hbb1 commented May 7, 2024

This is a great boost!! Before moving grid_x and grid_y to GPU, the average training time on my RTX3090 GPU is about 50 min.

@Han230104
Copy link
Author

I implemented an unofficial version of 2DGS before.
I made some modifications : the way to get the gradient of the projected 2D center , adjust parameters like prune_interval(1500), opacity_cull(0.2)
The average training time now is about 20 min while maintaining the accuracy.
If you are interested, please check it out. Thank you !

@hbb1
Copy link
Owner

hbb1 commented May 7, 2024

Great work! But It seems like that the link is broken and I cannot access. I am highly appreciated if you would be able to PR or outline the changes. I will perform testing on it then. Thank you for your great contribution!

@hbb1
Copy link
Owner

hbb1 commented May 7, 2024

Great. I found the correct link in your github profile. I will check it later.

@Han230104
Copy link
Author

I implemented my version referencing your python demo, so our cuda implementation is actually quite different.
I made some modifications on the official version based on my experience but it didn't improve the speed like I thought. I may need more time to check the differences.

@hbb1 hbb1 added the enhancement New feature or request label May 8, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants