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

fix examples (#21)

Browse Source
This commit is contained in:
Nicola Demo
2022-07-21 13:41:59 +02:00
committed by GitHub
parent 62f203fcc3
commit e8c2f87460
14 changed files with 63 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
examples/run_stokes.py
View File

@@ -36,9 +36,8 @@ if __name__ == "__main__":
if args.s:
pinn.span_pts(200, mode_spatial='grid', locations=['gamma_top', 'gamma_bot', 'gamma_in', 'gamma_out'])
pinn.span_pts(2000, mode_spatial='random', locations=['D'])
pinn.plot_pts()
pinn.span_pts(200, 'grid', locations=['gamma_top', 'gamma_bot', 'gamma_in', 'gamma_out'])
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):
@@ -48,8 +47,8 @@ if __name__ == "__main__":
else:
pinn.load_state('pina.stokes')
plotter = Plotter()
plotter.plot(pinn, component='ux')
plotter.plot(pinn, component='uy')
plotter.plot(pinn, component='p')
plotter.plot(pinn, components='ux')
plotter.plot(pinn, components='uy')
plotter.plot(pinn, components='p')