changed placement of regularization in fnn constructor

deepxde/nn/pytorch/fnn.py

@@ -23,6 +23,7 @@ def __init__(
             self.activation = activations.get(activation)
         initializer = initializers.get(kernel_initializer)
         initializer_zero = initializers.get("zeros")
+        self.regularizer = regularization
 
         self.linears = torch.nn.ModuleList()
         for i in range(1, len(layer_sizes)):
@@ -33,9 +34,6 @@ def __init__(
             )
             initializer(self.linears[-1].weight)
             initializer_zero(self.linears[-1].bias)
-        self.regularizer = regularization
-        # currently list with two components: regularization type, weight decay
-        # currently supports l2 regularization
 
     def forward(self, inputs):
         x = inputs