diff --git a/nerfstudio/model_components/renderers.py b/nerfstudio/model_components/renderers.py index 3d01bddea..5e488bc35 100644 --- a/nerfstudio/model_components/renderers.py +++ b/nerfstudio/model_components/renderers.py @@ -183,7 +183,7 @@ class SHRenderer(nn.Module): rgb = torch.sum(sh, dim=-1) + 0.5 # [..., num_samples, 3] if self.activation is not None: - self.activation(rgb) + rgb = self.activation(rgb) if not self.training: rgb = torch.nan_to_num(rgb)