ashawkey / torch-ngp

A pytorch CUDA extension implementation of instant-ngp (sdf and nerf), with a GUI.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Supporting bfloat16

MiZhenxing opened this issue · comments

Hi, thanks for your excellent work!
I would like to ask whether torch-ngp supports the bfloat16 format. bfloat16 can avoid NaN problems in very challenging scenes. I think it would be very useful to support it.
Thanks a lot!