Missing mlp output activation (#930)

This commit is contained in:
Matthew Tancik authored and GitHub committed 2022-11-09 22:54:04 -08:00
1 parent ee6b007155
commit 3833e4de56
1 file changed
+2
+2
View File
@@ -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