From bef506475f282e5d969366da06f2d65575f7ca4c Mon Sep 17 00:00:00 2001 From: Justin Kerr Date: Fri, 13 Oct 2023 00:03:35 -0700 Subject: [PATCH] params, remove normalization on quats --- nerfstudio/configs/method_configs.py | 6 +++--- nerfstudio/models/gaussian_splatting.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/nerfstudio/configs/method_configs.py b/nerfstudio/configs/method_configs.py index 72469180e..22448ae21 100644 --- a/nerfstudio/configs/method_configs.py +++ b/nerfstudio/configs/method_configs.py @@ -643,14 +643,14 @@ method_configs["gaussian-splatting"] = TrainerConfig( "optimizer": AdamOptimizerConfig(lr=0.005, eps=1e-15), "scheduler": ExponentialDecaySchedulerConfig( lr_final=1e-3, - max_steps=3000, + max_steps=15000, ), }, "rotation": { - "optimizer": AdamOptimizerConfig(lr=3e-4, eps=1e-15), + "optimizer": AdamOptimizerConfig(lr=.001, eps=1e-15), "scheduler": ExponentialDecaySchedulerConfig( lr_final=1e-4, - max_steps=3000, + max_steps=15000, ), }, }, diff --git a/nerfstudio/models/gaussian_splatting.py b/nerfstudio/models/gaussian_splatting.py index 35d416b33..7123c9c53 100644 --- a/nerfstudio/models/gaussian_splatting.py +++ b/nerfstudio/models/gaussian_splatting.py @@ -572,7 +572,7 @@ class GaussianSplattingModel(Model): self.means[crop_ids], torch.exp(self.scales[crop_ids]), 1, - self.quats[crop_ids]/self.quats[crop_ids].norm(dim=-1,keepdim=True), + self.quats[crop_ids],#/self.quats[crop_ids].norm(dim=-1,keepdim=True), viewmat.squeeze()[:3, :], projmat.squeeze() @ viewmat.squeeze(), camera.fx.item(),