Tune Splatfacto pose optimization hyperparameters

nerfstudio-project · Feb 9, 2024 · 2e37af2 · 2e37af2
1 parent 04149f7
commit 2e37af2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/nerfstudio/configs/method_configs.py b/nerfstudio/configs/method_configs.py
@@ -634,7 +634,12 @@
         "rotation": {"optimizer": AdamOptimizerConfig(lr=0.001, eps=1e-15), "scheduler": None},
         "camera_opt": {
             "optimizer": AdamOptimizerConfig(lr=1e-3, eps=1e-15),
-            "scheduler": ExponentialDecaySchedulerConfig(lr_final=5e-5, max_steps=30000),
+            "scheduler": ExponentialDecaySchedulerConfig(
+                lr_final=5e-5,
+                max_steps=30000,
+                warmup_steps=3000,
+                lr_pre_warmup=0,
+                ramp='step'),
         },
     },
     viewer=ViewerConfig(num_rays_per_chunk=1 << 15),