From f022f0c0482d3b16018d346c774cb1c7ca191d19 Mon Sep 17 00:00:00 2001 From: bkerbl Date: Wed, 12 Jul 2023 15:38:28 +0200 Subject: [PATCH] fixed learning rate for spatial --- scene/gaussian_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scene/gaussian_model.py b/scene/gaussian_model.py index 20a81d4..699f02c 100644 --- a/scene/gaussian_model.py +++ b/scene/gaussian_model.py @@ -84,7 +84,7 @@ class GaussianModel: self.active_sh_degree += 1 def create_from_pcd(self, pcd : BasicPointCloud, spatial_lr_scale : float): - self.spatial_lr_scale = spatial_lr_scale + self.spatial_lr_scale = 5#spatial_lr_scale fused_point_cloud = torch.tensor(np.asarray(pcd.points)).float().cuda() fused_color = RGB2SH(torch.tensor(np.asarray(pcd.colors)).float().cuda()) features = torch.zeros((fused_color.shape[0], 3, (self.max_sh_degree + 1) ** 2)).float().cuda()