{ "system": "passive_scalar", "paper_name": "Passive scalar", "dataset": "passive_scalar_v1", "test_gids": [ 0, 7, 17, 19, 45, 54, 71, 78, 85, 94 ], "val_gids": [ 25, 32, 43, 44, 52, 55, 58, 69, 89, 93 ], "fine_horizon_idx": 40, "x0_from_truth_bank": false, "s_star": 1, "table2": { "direct_si": "cascaded_si_S01", "edm": "edm", "cascaded_si": "cascaded_si_S01" }, "validation_energy_distance_by_S": { "1": 0.17306941650503, "2": 0.5197569839991051, "5": 0.8252763928969088, "10": 1.3195428451076956, "20": 2.128299478057891, "40": 3.269881230277261 }, "units": [ { "key": "cascaded_si_S01", "S": 1, "method": "cascaded_si", "paper_names": [ "direct_si", "cascaded_si" ], "config": "records/passive_scalar/runs/cascaded_si_S01/config.yaml", "selection": { "lr": 0.003, "sigma": 1.0, "selected_update": 6000, "validation_energy_distance": 0.17306941650503 }, "sampler": { "n_steps": 40, "variant": "noise_scale_1p0", "nfe": 40 }, "checkpoint": { "file": "passive-scalar/direct-si/model.pt", "sha256": "7ac8fbd0edb3658952fcafe1c8e10feec069bae40df67bdb73ec3b2c5fd27f3f", "bytes": 18587370 }, "expected_csfid": { "per_condition": { "0": 9.301505129800887, "7": 6.999326082120092, "17": 7.988786201364974, "19": 8.764889866887358, "45": 9.098692218227313, "54": 8.834131472642184, "71": 9.85266447519433, "78": 9.691036184475479, "85": 8.188405591947514, "94": 7.992308199325407 }, "n_gen_nonfinite": {}, "mean": 8.671174542198553, "sem": 0.279008527535713, "n": 10 }, "selection_flags": [] }, { "key": "cascaded_si_S02", "S": 2, "method": "cascaded_si", "paper_names": [], "config": "records/passive_scalar/runs/cascaded_si_S02/config.yaml", "selection": { "lr": 0.0003, "sigma": 4.0, "selected_update": 3000, "validation_energy_distance": 0.5197569839991051 }, "sampler": { "n_steps": 40, "variant": "noise_scale_1p0", "nfe": 80 }, "checkpoint": { "file": "passive-scalar/cascaded-si-S02/model.pt", "sha256": "7faa6a6466ddcae430d77e501e0f76f3b92836baa415dfd606425ed863493b9f", "bytes": 18587370 }, "expected_csfid": { "per_condition": { "0": 18.644554989597793, "7": 15.1181824349909, "17": 15.397152953385728, "19": 16.79350199533451, "45": 26.08087482672596, "54": 19.623429040131953, "71": 24.385423667695306, "78": 18.070791837513184, "85": 17.3606518007266, "94": 16.063036183189922 }, "n_gen_nonfinite": {}, "mean": 18.753759972929185, "sem": 1.1744716754985969, "n": 10 }, "selection_flags": [] }, { "key": "cascaded_si_S05", "S": 5, "method": "cascaded_si", "paper_names": [], "config": "records/passive_scalar/runs/cascaded_si_S05/config.yaml", "selection": { "lr": 0.001, "sigma": 32.0, "selected_update": 8000, "validation_energy_distance": 0.8252763928969088 }, "sampler": { "n_steps": 40, "variant": "noise_scale_1p0", "nfe": 200 }, "checkpoint": { "file": "passive-scalar/cascaded-si-S05/model.pt", "sha256": "41be255ea7bb0beaef16caa37d2bc408c6c5b525a6c26958916f5e27618e801b", "bytes": 18587434 }, "expected_csfid": { "per_condition": { "0": 33.008599859598746, "7": 25.33035308765824, "17": 27.029751435494006, "19": 36.74900160599293, "45": 40.36190972639173, "54": 34.068754837466884, "71": 40.39227733968147, "78": 32.0267967976792, "85": 29.60857688494275, "94": 31.15855901162371 }, "n_gen_nonfinite": {}, "mean": 32.973458058652966, "sem": 1.6136256279837646, "n": 10 }, "selection_flags": [] }, { "key": "cascaded_si_S10", "S": 10, "method": "cascaded_si", "paper_names": [], "config": "records/passive_scalar/runs/cascaded_si_S10/config.yaml", "selection": { "lr": 0.001, "sigma": 32.0, "selected_update": 11000, "validation_energy_distance": 1.3195428451076956 }, "sampler": { "n_steps": 80, "variant": "noise_scale_1p0", "nfe": 800 }, "checkpoint": { "file": "passive-scalar/cascaded-si-S10/model.pt", "sha256": "f730fbbc1bfd042b771cee1500b863c191bfb3e41722a19859987683fb12180b", "bytes": 18587434 }, "expected_csfid": { "per_condition": { "0": 39.40223997774541, "7": 28.391132538688794, "17": 33.66047867345088, "19": 50.70222471216181, "45": 46.69638187161013, "54": 43.83844979928182, "71": 51.80822874686169, "78": 41.560247566900614, "85": 40.191818714206875, "94": 35.157606603664476 }, "n_gen_nonfinite": {}, "mean": 41.14088092045725, "sem": 2.3611758048704354, "n": 10 }, "selection_flags": [] }, { "key": "cascaded_si_S20", "S": 20, "method": "cascaded_si", "paper_names": [], "config": "records/passive_scalar/runs/cascaded_si_S20/config.yaml", "selection": { "lr": 0.001, "sigma": 32.0, "selected_update": 17000, "validation_energy_distance": 2.128299478057891 }, "sampler": { "n_steps": 80, "variant": "noise_scale_1p0", "nfe": 1600 }, "checkpoint": { "file": "passive-scalar/cascaded-si-S20/model.pt", "sha256": "1cbcf0ed47cc2423f20d97480c8ecdba416bab1f155c3d0bdf4a9ca2f34cdbb9", "bytes": 18587562 }, "expected_csfid": { "per_condition": { "0": 55.523564819455146, "7": 41.609638471631285, "17": 52.22519174041531, "19": 69.32357767903466, "45": 67.22082543260842, "54": 61.02834986212878, "71": 65.35893570806462, "78": 55.95454194548762, "85": 47.95601314330446, "94": 53.25112783062713 }, "n_gen_nonfinite": {}, "mean": 56.945176663275745, "sem": 2.791231902897987, "n": 10 }, "selection_flags": [] }, { "key": "cascaded_si_S40", "S": 40, "method": "cascaded_si", "paper_names": [], "config": "records/passive_scalar/runs/cascaded_si_S40/config.yaml", "selection": { "lr": 0.001, "sigma": 16.0, "selected_update": 13000, "validation_energy_distance": 3.269881230277261 }, "sampler": { "n_steps": 80, "variant": "noise_scale_1p0", "nfe": 3200 }, "checkpoint": { "file": "passive-scalar/cascaded-si-S40/model.pt", "sha256": "5d3c6ff0b138298dc2c0af3c116316909d6aa10d15ae9ee85c71906aa6d5e21c", "bytes": 18587754 }, "expected_csfid": { "per_condition": { "0": 75.8534706135913, "7": 54.137818501023844, "17": 57.83665352795538, "19": 93.75504399670626, "45": 96.6819009445029, "54": 75.58561441286702, "71": 93.10959723101439, "78": 70.75124986539777, "85": 67.16704531254987, "94": 67.18434334928963 }, "n_gen_nonfinite": {}, "mean": 75.20627377548985, "sem": 4.738450881551494, "n": 10 }, "selection_flags": [] }, { "key": "edm", "S": 1, "method": "edm", "paper_names": [ "edm" ], "config": "records/passive_scalar/runs/edm/config.yaml", "selection": { "lr": 0.001, "P_mean": -1.2, "selected_update": 16000, "validation_energy_distance": 0.10606232925071438 }, "sampler": { "n_steps": 40, "variant": "default", "nfe": 79 }, "checkpoint": { "file": "passive-scalar/edm/model.pt", "sha256": "c74afc28febc20af1221ed8cc580b8214e6372a9d9f67d909b94cb8365a5c7bb", "bytes": 18587370 }, "expected_csfid": { "per_condition": { "0": 8.214064883145397, "7": 6.340900551288996, "17": 6.946395490153236, "19": 8.17481947293578, "45": 8.22899421983783, "54": 7.849720606557412, "71": 9.988332341720536, "78": 8.283878193030546, "85": 7.2782537114106844, "94": 7.506518008403248 }, "n_gen_nonfinite": {}, "mean": 7.881187747848367, "sem": 0.31040175244938206, "n": 10 }, "selection_flags": [] } ] }