Merge pull request #144 from NeuroDiffGym/elaine-custom_loss

fixed h1_semi_norm loss function by converting gradient from list to …
NeuroDiffGym · Sep 23, 2021 · 7bb802a · 7bb802a
2 parents 2d14c3d + 72d6dfb
commit 7bb802a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/neurodiffeq/losses.py b/neurodiffeq/losses.py
@@ -22,6 +22,7 @@ def _h1_norm(residual, funcs, coords):
 
 def _h1_semi_norm(residual, funcs, coords):
     g = grad(residual, *coords)
+    g = torch.cat(g)
     return (g ** 2).mean()