Skip to content

Commit

Permalink
reduce ckpt size for occ grid
Browse files Browse the repository at this point in the history
  • Loading branch information
liruilong940607 committed Dec 30, 2022
1 parent 4d595ae commit 05a1c5b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions nerfacc/grid.py
Original file line number Diff line number Diff line change
Expand Up @@ -169,9 +169,9 @@ def __init__(
grid_coords = _meshgrid3d(resolution).reshape(
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)
self.register_buffer("grid_indices", grid_indices)
self.register_buffer("grid_indices", grid_indices, persistent=False)

@torch.no_grad()
def _get_all_cells(self) -> torch.Tensor:
Expand Down

0 comments on commit 05a1c5b

Please sign in to comment.