jkfids/pinn-burgers

Optimize PDE Loss Function

Opened this issue · 0 comments

jkfids commented

Main performance limitation is calculating the PDE loss function over many collocation points. Method in PINN_1D.py relies on an explicit for loop. Would be of great benefit to vectorize this portion of the code.