From c5b25969103422b303d6a97f4a4aaa9e293a115c Mon Sep 17 00:00:00 2001 From: Eli Prater Date: Wed, 8 Mar 2023 08:52:10 -0500 Subject: [PATCH] update pinn loss histroy attribute to history_loss (#72) --- examples/run_stokes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/run_stokes.py b/examples/run_stokes.py index 9508281..9f48fa7 100644 --- a/examples/run_stokes.py +++ b/examples/run_stokes.py @@ -40,7 +40,7 @@ if __name__ == "__main__": pinn.span_pts(2000, 'random', locations=['D']) pinn.train(10000, 100) with open('stokes_history_{}.txt'.format(args.id_run), 'w') as file_: - for i, losses in enumerate(pinn.history): + for i, losses in pinn.history_loss.items(): file_.write('{} {}\n'.format(i, sum(losses))) pinn.save_state('pina.stokes')