Skip to content

GitLab

Commit 05a1c5b6 authored by Ruilong Li's avatar Ruilong Li
Browse files

reduce ckpt size for occ grid

parent 4d595ae4
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
nerfacc/grid.py
View file @ 05a1c5b6
...@@ -169,9 +169,9 @@ class OccupancyGrid(Grid): ...@@ -169,9 +169,9 @@ class OccupancyGrid(Grid):
grid_coords = _meshgrid3d(resolution).reshape( grid_coords = _meshgrid3d(resolution).reshape(
self.num_cells, self.NUM_DIM self.num_cells, self.NUM_DIM
) )
self.register_buffer("grid_coords", grid_coords) self.register_buffer("grid_coords", grid_coords, persistent=False)
grid_indices = torch.arange(self.num_cells) grid_indices = torch.arange(self.num_cells)
self.register_buffer("grid_indices", grid_indices) self.register_buffer("grid_indices", grid_indices, persistent=False)
@torch.no_grad() @torch.no_grad()
def _get_all_cells(self) -> torch.Tensor: def _get_all_cells(self) -> torch.Tensor:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment