Merge pull request neuraloperator#197 from ziqi-ma/fixmlp

Fix mlp nonlinearity

neuralop/layers/mlp.py

@@ -48,7 +48,7 @@ def __init__(self, in_channels, out_channels=None, hidden_channels=None,
     def forward(self, x):
         for i, fc in enumerate(self.fcs):
             x = fc(x)
-            if i < self.n_layers:
+            if i < self.n_layers - 1:
                 x = self.non_linearity(x)
             if self.dropout is not None:
                 x = self.dropout(x)