diff --git a/igs2gs/igs2gs_config.py b/igs2gs/igs2gs_config.py index 84f2073..eef6932 100644 --- 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",