Backend pytorch: Bug fix: DeepONetCartesianProd and PODDeepONet (lulu…

…lxvi#1023)
Charles-hit · Nov 16, 2022 · 229a997 · 229a997
1 parent 9c5d331
commit 229a997
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepxde/nn/pytorch/deeponet.py b/deepxde/nn/pytorch/deeponet.py
@@ -42,7 +42,7 @@ def __init__(
             # Fully connected network
             self.branch = FNN(layer_sizes_branch, activation_branch, kernel_initializer)
         self.trunk = FNN(layer_sizes_trunk, self.activation_trunk, kernel_initializer)
-        self.b = torch.tensor(0.0, requires_grad=True)
+        self.b = torch.nn.parameter.Parameter(torch.tensor(0.0))
         self.regularizer = regularization
 
     def forward(self, inputs):
@@ -120,7 +120,7 @@ def __init__(
             self.trunk = FNN(
                 layer_sizes_trunk, self.activation_trunk, kernel_initializer
             )
-            self.b = torch.tensor(0.0, requires_grad=True)
+            self.b = torch.nn.parameter.Parameter(torch.tensor(0.0))
         self.regularizer = regularization
 
     def forward(self, inputs):