ashawkey / torch-ngp

A pytorch CUDA extension implementation of instant-ngp (sdf and nerf), with a GUI.
MIT License
2.11k stars 275 forks source link

The growth process of `local_step` seems to have an issue. #196

Open boyaom opened 8 months ago

boyaom commented 8 months ago

During the nerf training phase, the value of local_step is incremented by one separately in the run_cuda and train_step functions. torch-ngp/utils.py#L858 torch-ngp/renderer.py#L939 Therefore, local_step will increase by 2 in each epoch, which leads to the step_counter updating only at even positions each time.