diff --git a/nerfstudio/field_components/mlp.py b/nerfstudio/field_components/mlp.py index 7774e043a..6f167056e 100644 --- a/nerfstudio/field_components/mlp.py +++ b/nerfstudio/field_components/mlp.py @@ -93,4 +93,6 @@ class MLP(FieldComponent): x = layer(x) if self.activation: x = self.activation(x) + if self.out_activation is not None: + x = self.out_activation(x) return x