Skip to content

Instantly share code, notes, and snippets.

@madagra

madagra/pinn_pde_final_loss.py

Last active December 24, 2021 11:12
Show Gist options
  • Save madagra/65c553fc241a88b70f577dc9ac59405d to your computer and use it in GitHub Desktop.
Save madagra/65c553fc241a88b70f577dc9ac59405d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
pinn_pde_final_loss.py
final_loss = \
# average over all the colocation points
de_loss_vec.pow(2).mean() + \
# simply square the boundary contribution which is a single value
boundary_loss ** 2
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment