Backend PyTorch: Add regularizer to DeepONet

Author: vl-dud

deepxde/nn/pytorch/deeponet.py

@@ -68,6 +68,7 @@ def __init__(
         kernel_initializer,
         num_outputs=1,
         multi_output_strategy=None,
+        regularization=None,
     ):
         super().__init__()
         if isinstance(activation, dict):
@@ -76,6 +77,7 @@ def __init__(
         else:
             self.activation_branch = self.activation_trunk = activations.get(activation)
         self.kernel_initializer = kernel_initializer
+        self.regularizer = regularization
 
         self.num_outputs = num_outputs
         if self.num_outputs == 1:
@@ -190,6 +192,7 @@ def __init__(
         kernel_initializer,
         num_outputs=1,
         multi_output_strategy=None,
+        regularization=None,
     ):
         super().__init__()
         if isinstance(activation, dict):
@@ -198,6 +201,7 @@ def __init__(
         else:
             self.activation_branch = self.activation_trunk = activations.get(activation)
         self.kernel_initializer = kernel_initializer
+        self.regularizer = regularization
 
         self.num_outputs = num_outputs
         if self.num_outputs == 1:
@@ -295,7 +299,7 @@ def __init__(
         regularization=None,
     ):
         super().__init__()
-        self.regularization = regularization  # TODO: currently unused
+        self.regularizer = regularization
         self.pod_basis = torch.as_tensor(pod_basis, dtype=torch.float32)
         if isinstance(activation, dict):
             activation_branch = activation["branch"]