warmup 3

nerfstudio-project · Oct 29, 2024 · 68289ae · 68289ae
1 parent 10bac1d
commit 68289ae
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/examples/blur_opt.py b/examples/blur_opt.py
@@ -81,10 +81,12 @@ def mask_loss(self, blur_mask: Tensor, step: int, eps: float = 1e-2):
         sparsity. During warmup, the bias is even higher to start with a sparse mask."""
         x = blur_mask.mean()
         if step <= self.num_warmup_steps:
-            a = 2
+            a = 3
+            b = 0.1
         else:
             a = 1
-        maskloss = a * (1 / (1 - x + eps) - 1) + 0.1 * (1 / (x + eps) - 1)
+            b = 0.1
+        maskloss = a * (1 / (1 - x + eps) - 1) + b * (1 / (x + eps) - 1)
         return maskloss