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.