Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "c2870b6e8b157ce34b5c9e09a1a4507726c73ef6"
Commit 05a1c5b6 authored by Ruilong Li's avatar Ruilong Li
Browse files

reduce ckpt size for occ grid

parent 4d595ae4
Show 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):
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:
......
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