From 3833e4de56f8001da53c4e02b745a47c3e9380a7 Mon Sep 17 00:00:00 2001
From: Matthew Tancik <tancik@berkeley.edu>
Date: Wed, 9 Nov 2022 22:54:04 -0800
Subject: [PATCH] Missing mlp output activation (#930)

---
 nerfstudio/field_components/mlp.py | 2 ++
 1 file changed, 2 insertions(+)

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