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.

Sign up for GitHub

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

Jump to bottom

Mixed precision (FP16) in NeRF code #138

Open
claforte opened this issue Jun 15, 2023 · 1 comment
Open

Mixed precision (FP16) in NeRF code #138

claforte opened this issue Jun 15, 2023 · 1 comment
Labels
enhancement New feature or request

Comments

@claforte
Copy link
Collaborator

Hi, it seems like we're using FP32 all the time in nerf_volume_renderer. Since raymarching seems to be the primary bottleneck (both for speed and VRAM usage), it sounds like we could get a big speed-up and increase batch size by leveraging mixed-precision. Has anyone looked into it? The NerfAcc docs don't seem to mention mixed precision (support or lack thereof) but it looks like NerfStudio supports it... so it might be relatively straightforward? (I'll try to dig a bit deeper in case it's really easy.)
@claforte claforte added the enhancement New feature or request label Jun 15, 2023
@claforte claforte mentioned this issue Jun 15, 2023
Open
@claforte
Copy link
Collaborator Author

Not easy but I might have found some folks who can get it done for us. ;-)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@claforte