Skip to content

Commit

Permalink
examples: add heat
Browse files Browse the repository at this point in the history
  • Loading branch information
pkarnakov committed Jan 26, 2024
1 parent 2d10bf8 commit c52d5b1
Show file tree
Hide file tree
Showing 21 changed files with 1,674 additions and 0 deletions.
Binary file added examples/heat/heat_train_k.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added examples/heat/heat_train_u.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
89 changes: 89 additions & 0 deletions examples/heat/out_odil/args.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,89 @@
{
"CUDA_VISIBLE_DEVICES": "-1",
"Ncb": 128,
"Nci": 4096,
"Nt": 64,
"Nx": 64,
"ODIL_BACKEND": "",
"ODIL_DTYPE": "",
"ODIL_JIT": "1",
"ODIL_MT": "",
"ODIL_WARN": "",
"OMP_NUM_THREADS": "1",
"adam_beta_1": null,
"adam_beta_2": null,
"adam_epsilon": null,
"arch_k": [
5,
5
],
"arch_u": [
10,
10
],
"bfgs_m": 50,
"bfgs_maxls": 50,
"bfgs_pgtol": null,
"callback_update_state": 0,
"checkpoint": null,
"checkpoint_every": 0,
"checkpoint_train": null,
"double": 0,
"dump_data": 1,
"echo": 0,
"epoch_start": 0,
"epochs": null,
"every_factor": 2.0,
"frame_start": 0,
"frames": 10,
"history_every": 10,
"history_full": 10,
"imposed": "stripe",
"infer_k": 1,
"jac_epsilon": 1e-08,
"jac_factor": 1,
"jac_nsmp0": 50,
"jac_nsmp1": 1,
"keep_frozen": 1,
"keep_init": 1,
"kimp": 2,
"kmax": 0.1,
"ktreg": 0,
"ktregdecay": 0,
"kwreg": 0,
"kwregdecay": 0,
"kxreg": 0,
"kxregdecay": 0,
"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,
"nimp": 200,
"nlvl": 100,
"nn_initializer": "legacy",
"noise": 0,
"omega": 0.6,
"optimizer": "adam",
"outdir": "out_odil",
"plot_every": 2000,
"plot_title": 1,
"plotext": "png",
"ref_path": "ref/ref.pickle",
"report_every": 500,
"restriction": "full",
"runtime_backend": "tf",
"runtime_dtype": "float32",
"runtime_jit": true,
"seed": 1000,
"smooth_post": 2,
"smooth_pre": 2,
"solver": "odil"
}
Binary file added examples/heat/out_odil/k_00010.png
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading

0 comments on commit c52d5b1

Please sign in to comment.