Tweak pose optimization hyperparameters

Parameters used in the Gaussian Splatting on the Move paper v1
nerfstudio-project · Apr 6, 2024 · 1384df1 · 1384df1
1 parent be7a325
commit 1384df1
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/nerfstudio/configs/method_configs.py b/nerfstudio/configs/method_configs.py
@@ -632,8 +632,13 @@
         },
         "quats": {"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),
+            "optimizer": AdamOptimizerConfig(lr=1e-4, eps=1e-15),
+            "scheduler": ExponentialDecaySchedulerConfig(
+                lr_final=5e-7,
+                max_steps=30000,
+                warmup_steps=1000,
+                lr_pre_warmup=0
+            ),
         },
     },
     viewer=ViewerConfig(num_rays_per_chunk=1 << 15),