Backend pytorch: Model.predict operator clear gradient (lululxvi#1037)

Charles-hit · Nov 15, 2022 · 38f4899 · 38f4899
1 parent b184122
commit 38f4899
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/deepxde/model.py b/deepxde/model.py
@@ -846,6 +846,8 @@ def op(inputs):
                     "Model.predict() with auxiliary variable hasn't been implemented "
                     "for backend pytorch."
                 )
+            # Clear cached Jacobians and Hessians.
+            grad.clear()
             y = utils.to_numpy(y)
         elif backend_name == "paddle":
             self.net.eval()