Fix for splatfacto updates

Updating config for splatfacto updates in optimizers
cvachha · Apr 4, 2024 · 5890bab · 5890bab
1 parent 5cb2902
commit 5890bab
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/igs2gs/igs2gs_config.py b/igs2gs/igs2gs_config.py
@@ -52,7 +52,7 @@
             model=InstructGS2GSModelConfig(),
         ),
     optimizers={
-        "xyz": {
+        "means": {
             "optimizer": AdamOptimizerConfig(lr=1.6e-4, eps=1e-15),
             "scheduler": ExponentialDecaySchedulerConfig(
                 lr_final=1.6e-6,
@@ -67,19 +67,20 @@
             "optimizer": AdamOptimizerConfig(lr=0.0025 / 20, eps=1e-15),
             "scheduler": None,
         },
-        "opacity": {
+        "opacities": {
             "optimizer": AdamOptimizerConfig(lr=0.05, eps=1e-15),
             "scheduler": None,
         },
-        "scaling": {
+        "scales": {
             "optimizer": AdamOptimizerConfig(lr=0.005, eps=1e-15),
             "scheduler": None,
         },
-        "rotation": {"optimizer": AdamOptimizerConfig(lr=0.001, eps=1e-15), "scheduler": None},
+        "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),
         },
+
     },
         viewer=ViewerConfig(num_rays_per_chunk=1 << 15),
         vis="viewer",