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(),