diff --git a/main/.doctrees/environment.pickle b/main/.doctrees/environment.pickle index 8a9ece2b2..d01867426 100644 Binary files a/main/.doctrees/environment.pickle and b/main/.doctrees/environment.pickle differ diff --git a/main/_modules/gsplat/strategy/mcmc.html b/main/_modules/gsplat/strategy/mcmc.html index 457bcf2e4..c96e68a18 100644 --- a/main/_modules/gsplat/strategy/mcmc.html +++ b/main/_modules/gsplat/strategy/mcmc.html @@ -376,7 +376,7 @@

Source code for gsplat.strategy.mcmc

         for n in range(n_max):
             for k in range(n + 1):
                 binoms[n, k] = math.comb(n, k)
-        return {"binoms": binoms, "n_max": n_max}
+ return {"binoms": binoms}
@@ -439,7 +439,6 @@

Source code for gsplat.strategy.mcmc

         state["binoms"] = state["binoms"].to(params["means"].device)
 
         binoms = state["binoms"]
-        n_max = state["n_max"]
 
         if (
             step < self.refine_stop_iter
@@ -447,12 +446,12 @@ 
Source code for gsplat.strategy.mcmc
             and step % self.refine_every == 0
         ):
             # teleport GSs
-            n_relocated_gs = self._relocate_gs(params, optimizers, binoms, n_max)
+            n_relocated_gs = self._relocate_gs(params, optimizers, binoms)
             if self.verbose:
                 print(f"Step {step}: Relocated {n_relocated_gs} GSs.")
 
             # add new GSs
-            n_new_gs = self._add_new_gs(params, optimizers, binoms, n_max)
+            n_new_gs = self._add_new_gs(params, optimizers, binoms)
             if self.verbose:
                 print(
                     f"Step {step}: Added {n_new_gs} GSs. "
@@ -473,7 +472,6 @@ 
Source code for gsplat.strategy.mcmc
         params: Union[Dict[str, torch.nn.Parameter], torch.nn.ParameterDict],
         optimizers: Dict[str, torch.optim.Optimizer],
         binoms: Tensor,
-        n_max: int,
     ) -> int:
         opacities = torch.sigmoid(params["opacities"])
         dead_mask = opacities <= self.min_opacity
@@ -485,7 +483,6 @@ 
Source code for gsplat.strategy.mcmc
                 state={},
                 mask=dead_mask,
                 binoms=binoms,
-                n_max=n_max,
                 min_opacity=self.min_opacity,
             )
         return n_gs
@@ -496,7 +493,6 @@ 
Source code for gsplat.strategy.mcmc
         params: Union[Dict[str, torch.nn.Parameter], torch.nn.ParameterDict],
         optimizers: Dict[str, torch.optim.Optimizer],
         binoms: Tensor,
-        n_max: int,
     ) -> int:
         current_n_points = len(params["means"])
         n_target = min(self.cap_max, int(1.05 * current_n_points))
@@ -508,7 +504,6 @@ 
Source code for gsplat.strategy.mcmc
                 state={},
                 n=n_gs,
                 binoms=binoms,
-                n_max=n_max,
                 min_opacity=self.min_opacity,
             )
         return n_gs