tensordiffeq / TensorDiffEq

Efficient and Scalable Physics-Informed Deep Learning and Scientific Machine Learning on top of Tensorflow for multi-worker distributed computing
http://docs.tensordiffeq.io
111 stars 38 forks source link

Plotting loss during iterations #32

Open ADS-Astral opened 2 years ago

ADS-Astral commented 2 years ago

Is there a way to generate an array of the loss printed in terminal during iterations? For the purposes of plotting the loss over the iterations?