Change dummy variable in Grid to tensor instead of parameter. (#83)

107d7466 · Yuanchen Guo · GitHub · f9be5ad3 · 107d7466
Unverified Commit 107d7466 authored Oct 25, 2022 by Yuanchen Guo Committed by GitHub Oct 24, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

nerfacc/grid.py nerfacc/grid.py +1 -1

No files found.
--- a/nerfacc/grid.py
+++ b/nerfacc/grid.py
@@ -66,7 +66,7 @@ class Grid(nn.Module):

    def __init__(self, *args, **kwargs):
        super().__init__()
-        self._dummy = torch.nn.Parameter(torch.empty(0))
+        self.register_buffer("_dummy", torch.empty(0), persistent=False)

    @property
    def device(self) -> torch.device: