Pytorch backend: clear gradients (lululxvi#1035)

Charles-hit · Nov 15, 2022 · b184122 · b184122
1 parent 319e1c9
commit b184122
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/deepxde/model.py b/deepxde/model.py
@@ -265,6 +265,8 @@ def outputs(training, inputs):
                 else:
                     inputs = torch.as_tensor(inputs)
                     inputs.requires_grad_()
+            # Clear cached Jacobians and Hessians.
+            grad.clear()
             return self.net(inputs)
 
         def outputs_losses(training, inputs, targets, losses_fn):