folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update pinn loss histroy attribute to history_loss (#72)

Browse Source
This commit is contained in:
Eli Prater
2023-03-08 08:52:10 -05:00
committed by GitHub
parent 0bcaf62e59
commit c5b2596910
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/run_stokes.py
View File

@@ -40,7 +40,7 @@ if __name__ == "__main__":
pinn.span_pts(2000, 'random', locations=['D']) pinn.span_pts(2000, 'random', locations=['D'])
pinn.train(10000, 100) pinn.train(10000, 100)
with open('stokes_history_{}.txt'.format(args.id_run), 'w') as file_: 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))) file_.write('{} {}\n'.format(i, sum(losses)))
pinn.save_state('pina.stokes') pinn.save_state('pina.stokes')