| { |
| "system": "sqg", |
| "paper_name": "Surface quasi-geostrophy (SQG)", |
| "dataset": "sqg_v3_candidate_v2", |
| "test_gids": [ |
| 100, |
| 101, |
| 102, |
| 103, |
| 104, |
| 105, |
| 106, |
| 107, |
| 108, |
| 109 |
| ], |
| "val_gids": [ |
| 3, |
| 6, |
| 9, |
| 10, |
| 11, |
| 17, |
| 28, |
| 50, |
| 70, |
| 73 |
| ], |
| "fine_horizon_idx": 40, |
| "x0_from_truth_bank": false, |
| "s_star": 10, |
| "table2": { |
| "direct_si": "cascaded_si_S01", |
| "edm": "edm", |
| "cascaded_si": "cascaded_si_S10" |
| }, |
| "validation_energy_distance_by_S": { |
| "1": 8.574122041243665, |
| "2": 6.64245033209117, |
| "5": 4.160166093654704, |
| "10": 3.2081736575575603, |
| "20": 3.9763564860504417, |
| "40": 4.650876671501445 |
| }, |
| "units": [ |
| { |
| "key": "cascaded_si_S01", |
| "S": 1, |
| "method": "cascaded_si", |
| "paper_names": [ |
| "direct_si" |
| ], |
| "config": "records/sqg/runs/cascaded_si_S01/config.yaml", |
| "selection": { |
| "lr": 0.0001, |
| "sigma": 16.0, |
| "selected_update": 4000, |
| "validation_energy_distance": 8.574122041243665 |
| }, |
| "sampler": { |
| "n_steps": 40, |
| "variant": "noise_scale_1p0", |
| "nfe": 40 |
| }, |
| "checkpoint": { |
| "file": "sqg/direct-si/model.pt", |
| "sha256": "043edcf9b92836ab185e2400f38af9da51217ee9314513546c560139c833dd4b", |
| "bytes": 18578154 |
| }, |
| "expected_csfid": { |
| "per_condition": { |
| "100": 174.77764334659662, |
| "101": 190.22545507837197, |
| "102": 32.19350595285715, |
| "103": 259.54823436220556, |
| "104": 187.9102583014046, |
| "105": 144.12298637015994, |
| "106": 69.83315328274227, |
| "107": 180.1658685174503, |
| "108": 89.04146761014198, |
| "109": 56.93464752886899 |
| }, |
| "n_gen_nonfinite": {}, |
| "mean": 138.47532203507996, |
| "sem": 23.08644501940587, |
| "n": 10 |
| }, |
| "selection_flags": [] |
| }, |
| { |
| "key": "cascaded_si_S02", |
| "S": 2, |
| "method": "cascaded_si", |
| "paper_names": [], |
| "config": "records/sqg/runs/cascaded_si_S02/config.yaml", |
| "selection": { |
| "lr": 0.001, |
| "sigma": 2.0, |
| "selected_update": 500, |
| "validation_energy_distance": 6.64245033209117 |
| }, |
| "sampler": { |
| "n_steps": 40, |
| "variant": "noise_scale_1p0", |
| "nfe": 80 |
| }, |
| "checkpoint": { |
| "file": "sqg/cascaded-si-S02/model.pt", |
| "sha256": "7897199322040cd8e416c47d52f9c627a897ab3f61315c159c3fc21524dfbf95", |
| "bytes": 18578154 |
| }, |
| "expected_csfid": { |
| "per_condition": { |
| "100": 108.64610599112586, |
| "101": 148.213528710336, |
| "102": 12.249737372758716, |
| "103": 166.43311996183718, |
| "104": 101.97250494419791, |
| "105": 81.0061190095802, |
| "106": 55.60906851984424, |
| "107": 97.34060878782299, |
| "108": 41.70360215641068, |
| "109": 19.348780618531293 |
| }, |
| "n_gen_nonfinite": {}, |
| "mean": 83.2523176072445, |
| "sem": 16.314034007096794, |
| "n": 10 |
| }, |
| "selection_flags": [] |
| }, |
| { |
| "key": "cascaded_si_S05", |
| "S": 5, |
| "method": "cascaded_si", |
| "paper_names": [], |
| "config": "records/sqg/runs/cascaded_si_S05/config.yaml", |
| "selection": { |
| "lr": 0.0003, |
| "sigma": 1.0, |
| "selected_update": 8000, |
| "validation_energy_distance": 4.160166093654704 |
| }, |
| "sampler": { |
| "n_steps": 80, |
| "variant": "noise_scale_1p0", |
| "nfe": 400 |
| }, |
| "checkpoint": { |
| "file": "sqg/cascaded-si-S05/model.pt", |
| "sha256": "a37023ac4e537ad368207189137a505301f460b1e745288752686cd48837be3b", |
| "bytes": 18578218 |
| }, |
| "expected_csfid": { |
| "per_condition": { |
| "100": 47.44390232255884, |
| "101": 97.64966657928815, |
| "102": 13.897894548199758, |
| "103": 78.32269942558258, |
| "104": 53.961217997417634, |
| "105": 33.35838313262059, |
| "106": 23.07217369951556, |
| "107": 50.69801284998158, |
| "108": 18.393931957047016, |
| "109": 18.889878019767053 |
| }, |
| "n_gen_nonfinite": {}, |
| "mean": 43.56877605319787, |
| "sem": 8.790125902695719, |
| "n": 10 |
| }, |
| "selection_flags": [] |
| }, |
| { |
| "key": "cascaded_si_S10", |
| "S": 10, |
| "method": "cascaded_si", |
| "paper_names": [ |
| "cascaded_si" |
| ], |
| "config": "records/sqg/runs/cascaded_si_S10/config.yaml", |
| "selection": { |
| "lr": 0.001, |
| "sigma": 16.0, |
| "selected_update": 11000, |
| "validation_energy_distance": 3.2081736575575603 |
| }, |
| "sampler": { |
| "n_steps": 160, |
| "variant": "noise_scale_1p0", |
| "nfe": 1600 |
| }, |
| "checkpoint": { |
| "file": "sqg/cascaded-si-S10/model.pt", |
| "sha256": "aa94d2608a7aa21ec81cb421692ce4d8cd456d9925f428692003238f55d4cb8d", |
| "bytes": 18578282 |
| }, |
| "expected_csfid": { |
| "per_condition": { |
| "100": 64.94213015904387, |
| "101": 106.07373925577525, |
| "102": 28.498497011053587, |
| "103": 78.46324992969, |
| "104": 53.92776605817534, |
| "105": 32.34964086870488, |
| "106": 34.21413790129114, |
| "107": 57.82609257156645, |
| "108": 26.65452682986927, |
| "109": 29.50001116074702 |
| }, |
| "n_gen_nonfinite": {}, |
| "mean": 51.244979174591684, |
| "sem": 8.318097628706868, |
| "n": 10 |
| }, |
| "selection_flags": [] |
| }, |
| { |
| "key": "cascaded_si_S20", |
| "S": 20, |
| "method": "cascaded_si", |
| "paper_names": [], |
| "config": "records/sqg/runs/cascaded_si_S20/config.yaml", |
| "selection": { |
| "lr": 0.001, |
| "sigma": 16.0, |
| "selected_update": 17000, |
| "validation_energy_distance": 3.9763564860504417 |
| }, |
| "sampler": { |
| "n_steps": 40, |
| "variant": "noise_scale_1p0", |
| "nfe": 800 |
| }, |
| "checkpoint": { |
| "file": "sqg/cascaded-si-S20/model.pt", |
| "sha256": "52cb13be41c51528ca28ba9b2e1ea00c5f8fab9e86c68812d94432a009df4883", |
| "bytes": 18578346 |
| }, |
| "expected_csfid": { |
| "per_condition": { |
| "100": 106.47051789719507, |
| "101": 98.1515112149221, |
| "102": 67.8273652273723, |
| "103": 142.1986562667293, |
| "104": 94.2926867903455, |
| "105": 88.04353243050758, |
| "106": 60.465059315902636, |
| "107": 90.76624161381369, |
| "108": 70.9330081827298, |
| "109": 71.66828988572928 |
| }, |
| "n_gen_nonfinite": { |
| "101": 1 |
| }, |
| "mean": 89.08168688252472, |
| "sem": 7.557464374313741, |
| "n": 10 |
| }, |
| "selection_flags": [] |
| }, |
| { |
| "key": "cascaded_si_S40", |
| "S": 40, |
| "method": "cascaded_si", |
| "paper_names": [], |
| "config": "records/sqg/runs/cascaded_si_S40/config.yaml", |
| "selection": { |
| "lr": 0.001, |
| "sigma": 16.0, |
| "selected_update": 11000, |
| "validation_energy_distance": 4.650876671501445 |
| }, |
| "sampler": { |
| "n_steps": 640, |
| "variant": "noise_scale_1p0", |
| "nfe": 25600 |
| }, |
| "checkpoint": { |
| "file": "sqg/cascaded-si-S40/model.pt", |
| "sha256": "9add06517f1ff2a6f0f0a22bc52553323926f7102524ae13c5feac13a7a264ba", |
| "bytes": 18578538 |
| }, |
| "expected_csfid": { |
| "per_condition": { |
| "100": 103.311354260536, |
| "101": 138.43431083940442, |
| "102": 49.069336701801376, |
| "103": 151.12233778182332, |
| "104": 82.95322414622876, |
| "105": 61.61246415076869, |
| "106": 55.004062919050455, |
| "107": 116.30086955370848, |
| "108": 52.43701041972753, |
| "109": 59.53891646067143 |
| }, |
| "n_gen_nonfinite": {}, |
| "mean": 86.97838872337205, |
| "sem": 11.997151993142262, |
| "n": 10 |
| }, |
| "selection_flags": [] |
| }, |
| { |
| "key": "edm", |
| "S": 1, |
| "method": "edm", |
| "paper_names": [ |
| "edm" |
| ], |
| "config": "records/sqg/runs/edm/config.yaml", |
| "selection": { |
| "lr": 0.0003, |
| "P_mean": -3.2, |
| "selected_update": 4000, |
| "validation_energy_distance": 8.014866943636475 |
| }, |
| "sampler": { |
| "n_steps": 40, |
| "variant": "default", |
| "nfe": 79 |
| }, |
| "checkpoint": { |
| "file": "sqg/edm/model.pt", |
| "sha256": "7387034ad6413ff6073a115602533645c6f31b8005ee1a1a89ea9579e6297a99", |
| "bytes": 18578154 |
| }, |
| "expected_csfid": { |
| "per_condition": { |
| "100": 138.3704066650215, |
| "101": 177.4114321188299, |
| "102": 26.592048990966646, |
| "103": 231.37837575590652, |
| "104": 173.26949861737825, |
| "105": 126.41908743961136, |
| "106": 57.35276085330884, |
| "107": 151.24564285169487, |
| "108": 68.33837760817943, |
| "109": 35.8617353737792 |
| }, |
| "n_gen_nonfinite": {}, |
| "mean": 118.62393662746763, |
| "sem": 21.67250652363767, |
| "n": 10 |
| }, |
| "selection_flags": [] |
| } |
| ] |
| } |
|
|
