{ "format": "wai-int-hyper-set/1", "capability": "wai.neural.int_hyper", "about": "A wai.neural.int_hyper parameter set, its identity and where it came from (tools/wai_int_hyper_train.py writes and checks this record).", "set": { "entropy.json": { "file": "leaves_entropy.json", "sha256": "361c72272daeffbb84f7f189f000b9b5b826ea07281bfcfb61e3da4e4c560f42" }, "g_s.whm": { "file": "leaves_g_s.bin", "sha256": "2c34831d14048a9d534ba76e60d9c764592ff711fca895f3961cd398ea23a71d" }, "h_s.whm": { "file": "leaves_h_s.bin", "sha256": "8eb5fae08f21204b6627f45331f12ccc711d0c1145f1adb6543dc20db2740036" } }, "listing_sha256": "ab796bb19cb202115400ea2ce16724617c3da035a2f4429d6266d419ab705ac0", "licence": "Apache-2.0", "provenance": { "init": { "from": "seed", "seed": 20261110 }, "training": { "content": "generated", "generator": "tools/wai_int_synth_real.py", "input": "the dead-leaves pictures the tool generates from seed 20261110, streams 1 to 4096", "seed": 20261110, "streams": [ 1, 4096 ] }, "coded": { "content": "generated", "generator": "tools/wai_int_synth_real.py", "input": "the dead-leaves picture the tool generates from seed 20261110, stream 0", "seed": 20261110, "stream": 0 }, "recipe": { "architecture": "mean-scale hyperprior (the exporter's topology)", "n": 48, "m": 64, "side": 128, "steps": 30000, "batch": 16, "lr": 0.001, "aux_lr": 0.001, "lr_drop": "to a tenth for the last quarter of the steps", "lambda": 0.0483, "loss": "lambda * 255^2 * MSE + bits per pixel; gradient norm clipped to 1", "order_seed": 20261110, "threads": 8, "device": "cpu", "deterministic_algorithms": true, "python": "3.11.17", "torch": "2.14.0+cpu", "numpy": "1.26.4", "compressai": "1.2.8" }, "commit": "40db6ee579dd2976042a2cf58965a6a4c7465364", "tools": { "tools/wai_int_hyper_train.py": "834acf427c2c71f34fdb3e8bd2d8a104e10e5592", "tools/wai_int_synth_real.py": "b556ac49a9fea4657c155ea126d526e56d9b1d9a", "tools/wai_sota_int_synth_export.py": "f99c3cfe717acd4365e6990e174058ecb5d9e306" }, "float_model_sha256": "2ce350db09c35f88feee2d37c96cc8149443b884e3aca16e567573acde91996f", "weights_sha256": "8fef2f4842a40da60c4384fc2096570f221a55b82f90febe5aed090965d03ff5", "log": [ { "step": 0, "bpp": 0.1131, "psnr_db": 5.76, "aux": 1978.87 }, { "step": 500, "bpp": 0.7048, "psnr_db": 18.0, "aux": 1110.83 }, { "step": 1000, "bpp": 0.8488, "psnr_db": 19.84, "aux": 263.18 }, { "step": 1500, "bpp": 0.9139, "psnr_db": 20.34, "aux": 743.64 }, { "step": 2000, "bpp": 0.9474, "psnr_db": 20.53, "aux": 1048.78 }, { "step": 2500, "bpp": 0.9903, "psnr_db": 20.99, "aux": 1445.04 }, { "step": 3000, "bpp": 0.9954, "psnr_db": 21.02, "aux": 1760.83 }, { "step": 3500, "bpp": 1.0094, "psnr_db": 21.72, "aux": 1954.02 }, { "step": 4000, "bpp": 1.0244, "psnr_db": 21.83, "aux": 1992.7 }, { "step": 4500, "bpp": 1.0124, "psnr_db": 22.32, "aux": 1852.61 }, { "step": 5000, "bpp": 1.0098, "psnr_db": 22.02, "aux": 1584.05 }, { "step": 5500, "bpp": 1.033, "psnr_db": 22.07, "aux": 1516.93 }, { "step": 6000, "bpp": 1.0251, "psnr_db": 22.18, "aux": 1234.77 }, { "step": 6500, "bpp": 1.0449, "psnr_db": 22.07, "aux": 1118.59 }, { "step": 7000, "bpp": 1.0417, "psnr_db": 22.17, "aux": 897.72 }, { "step": 7500, "bpp": 1.051, "psnr_db": 22.18, "aux": 628.53 }, { "step": 8000, "bpp": 1.0536, "psnr_db": 22.35, "aux": 535.15 }, { "step": 8500, "bpp": 1.0383, "psnr_db": 22.58, "aux": 219.34 }, { "step": 9000, "bpp": 1.0624, "psnr_db": 22.37, "aux": 141.62 }, { "step": 9500, "bpp": 1.0579, "psnr_db": 22.31, "aux": 106.97 }, { "step": 10000, "bpp": 1.062, "psnr_db": 22.42, "aux": 88.2 }, { "step": 10500, "bpp": 1.0633, "psnr_db": 22.25, "aux": 134.02 }, { "step": 11000, "bpp": 1.0503, "psnr_db": 22.67, "aux": 126.42 }, { "step": 11500, "bpp": 1.0504, "psnr_db": 22.92, "aux": 136.32 }, { "step": 12000, "bpp": 1.0592, "psnr_db": 22.55, "aux": 122.25 }, { "step": 12500, "bpp": 1.0572, "psnr_db": 22.68, "aux": 86.15 }, { "step": 13000, "bpp": 1.0611, "psnr_db": 22.71, "aux": 67.67 }, { "step": 13500, "bpp": 1.0624, "psnr_db": 22.91, "aux": 62.73 }, { "step": 14000, "bpp": 1.0699, "psnr_db": 22.74, "aux": 61.88 }, { "step": 14500, "bpp": 1.0635, "psnr_db": 22.72, "aux": 61.25 }, { "step": 15000, "bpp": 1.0648, "psnr_db": 22.73, "aux": 91.96 }, { "step": 15500, "bpp": 1.0524, "psnr_db": 22.89, "aux": 49.48 }, { "step": 16000, "bpp": 1.0685, "psnr_db": 22.82, "aux": 57.12 }, { "step": 16500, "bpp": 1.0575, "psnr_db": 23.18, "aux": 52.27 }, { "step": 17000, "bpp": 1.0448, "psnr_db": 23.1, "aux": 55.83 }, { "step": 17500, "bpp": 1.062, "psnr_db": 22.89, "aux": 51.7 }, { "step": 18000, "bpp": 1.0638, "psnr_db": 23.01, "aux": 70.66 }, { "step": 18500, "bpp": 1.0573, "psnr_db": 22.88, "aux": 43.44 }, { "step": 19000, "bpp": 1.0513, "psnr_db": 23.07, "aux": 54.32 }, { "step": 19500, "bpp": 1.0498, "psnr_db": 22.96, "aux": 70.56 }, { "step": 20000, "bpp": 1.0519, "psnr_db": 22.96, "aux": 58.63 }, { "step": 20500, "bpp": 1.058, "psnr_db": 23.14, "aux": 81.23 }, { "step": 21000, "bpp": 1.0586, "psnr_db": 23.06, "aux": 52.94 }, { "step": 21500, "bpp": 1.0617, "psnr_db": 22.95, "aux": 67.94 }, { "step": 22000, "bpp": 1.0623, "psnr_db": 22.85, "aux": 55.07 }, { "step": 22500, "bpp": 1.0539, "psnr_db": 23.06, "aux": 45.55 }, { "step": 23000, "bpp": 1.0522, "psnr_db": 23.3, "aux": 41.73 }, { "step": 23500, "bpp": 1.0575, "psnr_db": 23.29, "aux": 1.87 }, { "step": 24000, "bpp": 1.0519, "psnr_db": 23.7, "aux": 1.71 }, { "step": 24500, "bpp": 1.0715, "psnr_db": 23.03, "aux": 1.59 }, { "step": 25000, "bpp": 1.0598, "psnr_db": 23.28, "aux": 1.46 }, { "step": 25500, "bpp": 1.0572, "psnr_db": 23.32, "aux": 1.31 }, { "step": 26000, "bpp": 1.0495, "psnr_db": 23.29, "aux": 1.49 }, { "step": 26500, "bpp": 1.0536, "psnr_db": 23.22, "aux": 1.46 }, { "step": 27000, "bpp": 1.0704, "psnr_db": 23.15, "aux": 2.04 }, { "step": 27500, "bpp": 1.0601, "psnr_db": 23.43, "aux": 1.52 }, { "step": 28000, "bpp": 1.0661, "psnr_db": 23.16, "aux": 1.76 }, { "step": 28500, "bpp": 1.0742, "psnr_db": 23.14, "aux": 1.68 }, { "step": 29000, "bpp": 1.0708, "psnr_db": 23.04, "aux": 1.78 }, { "step": 29500, "bpp": 1.0454, "psnr_db": 23.56, "aux": 1.82 }, { "step": 29999, "bpp": 1.0691, "psnr_db": 23.05, "aux": 1.84 } ] }, "decode": { "width": 128, "height": 128, "rgb_file": "leaves_rgb.bin", "rgb_sha256": "3ea9b4c3026e7c81683f3845f8256b1f0e1e8268284d81f3f16f290a5dde1374", "rgb_fnv1a": "0b06431d1ee2d45d", "payload_rans_bytes": 2240, "bpp": 1.0938, "psnr_db": 22.94 } }