examples: add fields, infer_constant, poisson

examples/out_fields/args.json

@@ -0,0 +1,62 @@
+{
+    "CUDA_VISIBLE_DEVICES": "-1",
+    "Nx": 8,
+    "Ny": 4,
+    "ODIL_BACKEND": "",
+    "ODIL_DTYPE": "",
+    "ODIL_JIT": "",
+    "ODIL_MT": "",
+    "ODIL_WARN": "",
+    "OMP_NUM_THREADS": "1",
+    "adam_beta_1": null,
+    "adam_beta_2": null,
+    "adam_epsilon": null,
+    "bfgs_m": 50,
+    "callback_update_state": 0,
+    "checkpoint": null,
+    "checkpoint_every": 0,
+    "checkpoint_train": null,
+    "double": null,
+    "dump_data": 1,
+    "echo": 1,
+    "epoch_start": 0,
+    "epochs": null,
+    "every_factor": 1,
+    "frame_start": 0,
+    "frames": 1,
+    "history_every": 10,
+    "history_full": 0,
+    "jac_epsilon": 1e-08,
+    "jac_factor": 1,
+    "jac_nsmp0": 50,
+    "jac_nsmp1": 1,
+    "linsolver": "direct",
+    "linsolver_damp": 0,
+    "linsolver_dampdiag": 0,
+    "linsolver_history": 0,
+    "linsolver_maxiter": null,
+    "linsolver_tol": 1e-06,
+    "linsolver_verbose": 0,
+    "lr": 0.01,
+    "mg_interp": "stack",
+    "montage": 1,
+    "multigrid": 1,
+    "ndirect": 3,
+    "nlvl": 100,
+    "nn_initializer": "legacy",
+    "omega": 0.6,
+    "optimizer": "adam",
+    "outdir": "out_fields",
+    "plot": 1,
+    "plot_every": 100,
+    "plot_title": 0,
+    "plotext": "pdf",
+    "report_every": 50,
+    "restriction": "full",
+    "runtime_backend": "tf",
+    "runtime_dtype": "float32",
+    "runtime_jit": false,
+    "seed": 1000,
+    "smooth_post": 2,
+    "smooth_pre": 2
+}

examples/out_fields/train.csv

@@ -0,0 +1,12 @@
+epoch,frame,norm_uc,norm_un,norm_ufx,norm_ufy,loss,walltime,memory
+0,0,0.5385527610778809,0.5543389320373535,0.5436661839485168,0.5493248701095581,1.1946613788604736,3.628523826599121,493.46875
+10,1,0.37918606400489807,0.4001421630382538,0.38608336448669434,0.3936731517314911,0.6079347133636475,5.424388408660889,567.92578125
+20,1,0.2325356900691986,0.2588975727558136,0.24098439514636993,0.2513931393623352,0.2423727810382843,5.461084365844727,567.92578125
+30,1,0.1287868469953537,0.15771150588989258,0.13813869655132294,0.15007169544696808,0.0830627828836441,5.495050430297852,567.92578125
+40,1,0.06614639610052109,0.09221964329481125,0.07442742586135864,0.08578452467918396,0.025778234004974365,5.531328916549683,567.92578125
+50,1,0.034254491329193115,0.0525616854429245,0.03799011930823326,0.0477990098297596,0.007664095610380173,5.569957494735718,567.92578125
+60,1,0.01957051269710064,0.02929794229567051,0.020516837015748024,0.026186658069491386,0.002348056063055992,5.611165285110474,567.92578125
+70,1,0.013184972107410431,0.01646255888044834,0.01333641167730093,0.014809349551796913,0.0008420360391028225,5.649601697921753,567.92578125
+80,1,0.008949870243668556,0.00978162419050932,0.009176047518849373,0.009031431749463081,0.00034154695458710194,5.68494176864624,567.92578125
+90,1,0.0056144860573112965,0.005788223817944527,0.005701076705008745,0.0053533087484538555,0.00012618617620319128,5.720246315002441,567.92578125
+100,1,0.002965271472930908,0.0035612541250884533,0.0031611155718564987,0.0032508319709450006,4.203592106932774e-05,5.7563464641571045,567.92578125

examples/out_fields/train.log

@@ -0,0 +1,26 @@
+./fields.py
+Running AdamNative optimizer
+
+epoch=00000
+residual: uc:0.53855, un:0.55434, ufx:0.54367, ufy:0.54932
+memory: 493 MiB
+walltime: 3.629 s
+walltime+callback: 3.629 s
+grid_00000.png
+
+epoch=00050
+residual: uc:0.034254, un:0.052562, ufx:0.03799, ufy:0.047799
+memory: 567 MiB
+walltime: 5.570 s
+walltime+callback: 5.986 s
+walltime/epoch: 111.399 ms
+throughput: 0.000M cells/s
+
+epoch=00100
+residual: uc:0.0029653, un:0.0035613, ufx:0.0031611, ufy:0.0032508
+memory: 567 MiB
+walltime: 5.756 s
+walltime+callback: 6.175 s
+walltime/epoch: 3.728 ms
+throughput: 0.009M cells/s
+grid_00001.png

examples/out_infer_constant/args.json

@@ -0,0 +1,64 @@
+{
+    "CUDA_VISIBLE_DEVICES": "-1",
+    "Nt": 64,
+    "Nx": 64,
+    "ODIL_BACKEND": "",
+    "ODIL_DTYPE": "",
+    "ODIL_JIT": "",
+    "ODIL_MT": "",
+    "ODIL_WARN": "",
+    "OMP_NUM_THREADS": "1",
+    "adam_beta_1": null,
+    "adam_beta_2": null,
+    "adam_epsilon": null,
+    "bfgs_m": 50,
+    "c_diff": 0.01,
+    "c_src": 0.1,
+    "c_vel": 0.2,
+    "callback_update_state": 0,
+    "checkpoint": null,
+    "checkpoint_every": 0,
+    "checkpoint_train": null,
+    "double": 1,
+    "dump_data": 1,
+    "echo": 0,
+    "epoch_start": 0,
+    "epochs": null,
+    "every_factor": 1,
+    "frame_start": 0,
+    "frames": 3,
+    "history_every": 10,
+    "history_full": 0,
+    "jac_epsilon": 1e-08,
+    "jac_factor": 1,
+    "jac_nsmp0": 50,
+    "jac_nsmp1": 1,
+    "linsolver": "direct",
+    "linsolver_damp": 0,
+    "linsolver_dampdiag": 0,
+    "linsolver_history": 0,
+    "linsolver_maxiter": null,
+    "linsolver_tol": 1e-06,
+    "linsolver_verbose": 0,
+    "lr": 0.001,
+    "mg_interp": "stack",
+    "montage": 1,
+    "multigrid": 1,
+    "ndirect": 3,
+    "nlvl": 100,
+    "nn_initializer": "legacy",
+    "omega": 0.6,
+    "optimizer": "lbfgsb",
+    "outdir": "out_infer_constant",
+    "plot_every": 50,
+    "plot_title": 0,
+    "plotext": "pdf",
+    "report_every": 50,
+    "restriction": "full",
+    "runtime_backend": "tf",
+    "runtime_dtype": "float32",
+    "runtime_jit": false,
+    "seed": 1000,
+    "smooth_post": 2,
+    "smooth_pre": 2
+}

examples/out_infer_constant/train.csv

@@ -0,0 +1,17 @@
+epoch,frame,norm_0,loss,walltime,memory,c_diff,c_src,c_vel
+0,0,1.9361437091163396,3.748652462350777,4.59730339050293,499.109375,0.0,0.0,0.001
+10,1,0.7237421348310625,0.5238026777298239,6.259412527084351,538.85546875,0.18847641881300975,0.010401907295946927,0.003986294295223188
+20,1,0.4974617299682355,0.24746817278298966,6.322265863418579,539.73046875,0.2007515856317027,0.015716192256829305,0.004116382552140241
+30,1,0.394022922449441,0.15525406341559816,6.393953084945679,540.60546875,0.1837412852563917,0.034549217850700746,0.0036334016703370707
+40,1,0.33770953391075487,0.11404772929421927,6.473968744277954,541.60546875,0.13860065443184935,0.07794333688453957,0.002549807869257098
+50,1,0.25940883966016215,0.06729294609383173,6.55730676651001,542.35546875,0.07490925748311615,0.13312888198501682,0.0011296887065213293
+60,2,0.19354603920099261,0.037460069290392174,6.648121356964111,549.98046875,0.043936537723425845,0.09666946393386537,0.010988510872885715
+70,2,0.15574733839482838,0.02425723341707318,6.734349489212036,549.98046875,0.028286143082563647,0.08182446932758072,0.05578211490406085
+80,2,0.12060921370789769,0.014546582431237336,6.81940221786499,549.98046875,0.022376853685918695,0.10770004060811647,0.11454709441191703
+90,2,0.09001497062541292,0.00810269493669395,6.905359983444214,549.98046875,0.0178841155780708,0.09304726214938236,0.15970359490628314
+100,2,0.057636864112775986,0.0033220081047546047,6.9898834228515625,549.98046875,0.01293125587305922,0.10354454577185765,0.19582102491602532
+110,3,0.0399627193646099,0.001597018939014567,7.074782371520996,553.48046875,0.01077727097845015,0.09823175581874576,0.2005905921390552
+120,3,0.026655471047227763,0.0007105141367495975,7.15997576713562,553.48046875,0.008969291425772562,0.10072793121746351,0.20397594068265482
+130,3,0.01961922360523112,0.00038491393487205793,7.249751567840576,553.48046875,0.00813912245277859,0.10020335718654033,0.20366600276861832
+140,3,0.013284260018608707,0.0001764715642420058,7.333848237991333,553.48046875,0.0075813434863320345,0.09986841617539884,0.2053345541295283
+150,3,0.007397539355556484,5.4723588517007043e-05,7.417750597000122,553.60546875,0.007178529204905234,0.10007007006402131,0.2050820464351931

examples/out_infer_constant/train.log

@@ -0,0 +1,40 @@
+./infer_constant.py
+Running L-BFGS-B optimizer
+
+epoch=00000
+residual: 0:1.9361
+diff=0, src=0, vel=0.001
+memory: 499 MiB
+walltime: 4.597 s
+walltime+callback: 4.597 s
+u_00000.png
+
+epoch=00050
+residual: 0:0.25941
+diff=0.074909, src=0.13313, vel=0.0011297
+memory: 542 MiB
+walltime: 6.557 s
+walltime+callback: 6.918 s
+walltime/epoch: 131.146 ms
+throughput: 0.031M cells/s
+u_00001.png
+
+epoch=00100
+residual: 0:0.057637
+diff=0.012931, src=0.10354, vel=0.19582
+memory: 549 MiB
+walltime: 6.990 s
+walltime+callback: 7.643 s
+walltime/epoch: 8.652 ms
+throughput: 0.473M cells/s
+u_00002.png
+
+epoch=00150
+residual: 0:0.0073975
+diff=0.0071785, src=0.10007, vel=0.20508
+memory: 553 MiB
+walltime: 7.418 s
+walltime+callback: 8.359 s
+walltime/epoch: 8.557 ms
+throughput: 0.479M cells/s
+u_00003.png

examples/out_poisson/args.json

@@ -0,0 +1,68 @@
+{
+    "CUDA_VISIBLE_DEVICES": "-1",
+    "N": 64,
+    "ODIL_BACKEND": "",
+    "ODIL_DTYPE": "",
+    "ODIL_JIT": "",
+    "ODIL_MT": "",
+    "ODIL_WARN": "",
+    "OMP_NUM_THREADS": "1",
+    "adam_beta_1": null,
+    "adam_beta_2": null,
+    "adam_epsilon": null,
+    "bfgs_m": 50,
+    "callback_update_state": 0,
+    "cellbased": 1,
+    "checkpoint": null,
+    "checkpoint_every": 0,
+    "checkpoint_train": null,
+    "double": 1,
+    "dump_data": 1,
+    "dump_xmf": 0,
+    "echo": 1,
+    "epoch_start": 0,
+    "epochs": 1000,
+    "every_factor": 1,
+    "frame_start": 0,
+    "frames": 0,
+    "history_every": 10,
+    "history_full": 50,
+    "jac_epsilon": 1e-08,
+    "jac_factor": 1,
+    "jac_nsmp0": 50,
+    "jac_nsmp1": 1,
+    "linsolver": "direct",
+    "linsolver_damp": 0,
+    "linsolver_dampdiag": 0,
+    "linsolver_history": 0,
+    "linsolver_maxiter": null,
+    "linsolver_tol": 1e-06,
+    "linsolver_verbose": 0,
+    "lr": 0.005,
+    "mg_interp": "stack",
+    "mgloss": 0,
+    "montage": 1,
+    "multigrid": 1,
+    "ndim": 2,
+    "ndirect": 3,
+    "nlvl": 100,
+    "nn_initializer": "legacy",
+    "omega": 0.6,
+    "optimizer": "adam",
+    "osc_k": 2.0,
+    "outdir": "out_poisson",
+    "plot": 0,
+    "plot_every": 100,
+    "plot_title": 0,
+    "plotext": "pdf",
+    "ref": "osc",
+    "report_every": 100,
+    "restriction": "full",
+    "rhs": "exact",
+    "runtime_backend": "tf",
+    "runtime_dtype": "float32",
+    "runtime_jit": false,
+    "seed": 1000,
+    "smooth_post": 2,
+    "smooth_pre": 2
+}