Upload folder using huggingface_hub
Browse files
cgn-7be9d4b/SHA256SUMS
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
ebcf0c00d9d401b94935ba8027f7d45a074e2d1f2fd20cf67456eabab019b800 ./config.json
|
| 2 |
+
ff176ac4ceca983a39c3e03e546267c2d3802a9687961f575b4b413073300966 ./metrics-test_extrap.json
|
| 3 |
+
e7dce5abdb1e61c14f6e9b2cc8d1fbbfadd37be26712e83421a9fef15c9915ed ./metrics-test_interp.json
|
| 4 |
+
4950d798a774be111d7096a16a112d38385d44e7832c7577bb1fa82eb7b76060 ./model-best-096000.pt
|
| 5 |
+
bac624bd89f94b04809cfa28da67447d31e2bfe5dd62176eb12418d2333ba7d2 ./normalization_stats.npz
|
cgn-7be9d4b/config.json
ADDED
|
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"run": {
|
| 3 |
+
"benchmark": "taylor_impact_2d",
|
| 4 |
+
"seed": 1,
|
| 5 |
+
"commit": "7be9d4b"
|
| 6 |
+
},
|
| 7 |
+
"model": {
|
| 8 |
+
"family": "cgn",
|
| 9 |
+
"input_frames": 6,
|
| 10 |
+
"connectivity_radius": 1.5,
|
| 11 |
+
"hidden_dim": 128,
|
| 12 |
+
"message_passing_steps": 10,
|
| 13 |
+
"nmlp_layers": 2,
|
| 14 |
+
"particle_type_embedding_size": 9,
|
| 15 |
+
"noise_std": 0.02,
|
| 16 |
+
"dim": 2,
|
| 17 |
+
"max_neighbors": 32
|
| 18 |
+
},
|
| 19 |
+
"train": {
|
| 20 |
+
"batch_size": 8,
|
| 21 |
+
"lr_init": 0.0001,
|
| 22 |
+
"lr_decay": 0.1,
|
| 23 |
+
"lr_decay_steps": 40000,
|
| 24 |
+
"training_steps": 100000,
|
| 25 |
+
"val_every": 2000,
|
| 26 |
+
"w_pos": 1.0,
|
| 27 |
+
"w_aux": 1.0
|
| 28 |
+
},
|
| 29 |
+
"protocol": {
|
| 30 |
+
"input_frames": 6,
|
| 31 |
+
"horizon": "full",
|
| 32 |
+
"eval_times": "native",
|
| 33 |
+
"standard": true
|
| 34 |
+
},
|
| 35 |
+
"n_particle_types": 2
|
| 36 |
+
}
|
cgn-7be9d4b/metrics-test_extrap.json
ADDED
|
@@ -0,0 +1,94 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"split": "test_extrap",
|
| 3 |
+
"checkpoint": "model-best-096000.pt",
|
| 4 |
+
"input_frames": 6,
|
| 5 |
+
"protocol_standard": true,
|
| 6 |
+
"aux_field": "von_mises_stress",
|
| 7 |
+
"aux_unit": "MPa",
|
| 8 |
+
"cases": {
|
| 9 |
+
"T-20-60-200": {
|
| 10 |
+
"one_step_position_rmse": 0.004784522836366059,
|
| 11 |
+
"one_step_aux_rmse": 38.95289401988924,
|
| 12 |
+
"rollout_position_rmse": 8.559585820079924,
|
| 13 |
+
"rollout_aux_rmse": 82.86745015672713,
|
| 14 |
+
"qoi_pred": {
|
| 15 |
+
"final_length": 28.048575401306152,
|
| 16 |
+
"mushroom_width": 75.09690284729004,
|
| 17 |
+
"peak_von_mises": 236.3180252567927,
|
| 18 |
+
"t_peak_von_mises": 0.267934466078759
|
| 19 |
+
},
|
| 20 |
+
"qoi_true": {
|
| 21 |
+
"final_length": 31.300888061523438,
|
| 22 |
+
"mushroom_width": 77.65174102783203,
|
| 23 |
+
"peak_von_mises": 212.20605452537538,
|
| 24 |
+
"t_peak_von_mises": 0.05392327593292976
|
| 25 |
+
},
|
| 26 |
+
"qoi_error": {
|
| 27 |
+
"final_length": -3.252312660217285,
|
| 28 |
+
"mushroom_width": -2.554838180541992,
|
| 29 |
+
"peak_von_mises": 24.11197073141733,
|
| 30 |
+
"t_peak_von_mises": 0.21401119014582923
|
| 31 |
+
}
|
| 32 |
+
},
|
| 33 |
+
"T-20-80-200": {
|
| 34 |
+
"one_step_position_rmse": 0.004543031253782171,
|
| 35 |
+
"one_step_aux_rmse": 39.78981531026215,
|
| 36 |
+
"rollout_position_rmse": 7.523828841431291,
|
| 37 |
+
"rollout_aux_rmse": 77.70431297355131,
|
| 38 |
+
"qoi_pred": {
|
| 39 |
+
"final_length": 45.14631724357605,
|
| 40 |
+
"mushroom_width": 75.09782409667969,
|
| 41 |
+
"peak_von_mises": 217.5365364229679,
|
| 42 |
+
"t_peak_von_mises": 0.29993016864333066
|
| 43 |
+
},
|
| 44 |
+
"qoi_true": {
|
| 45 |
+
"final_length": 43.98169922828674,
|
| 46 |
+
"mushroom_width": 88.17000198364258,
|
| 47 |
+
"peak_von_mises": 203.08150714896618,
|
| 48 |
+
"t_peak_von_mises": 0.03795272988474013
|
| 49 |
+
},
|
| 50 |
+
"qoi_error": {
|
| 51 |
+
"final_length": 1.1646180152893066,
|
| 52 |
+
"mushroom_width": -13.07217788696289,
|
| 53 |
+
"peak_von_mises": 14.455029274001731,
|
| 54 |
+
"t_peak_von_mises": 0.26197743875859053
|
| 55 |
+
}
|
| 56 |
+
},
|
| 57 |
+
"T-20-100-200": {
|
| 58 |
+
"one_step_position_rmse": 0.004620682621476287,
|
| 59 |
+
"one_step_aux_rmse": 42.55197587187548,
|
| 60 |
+
"rollout_position_rmse": 6.851943743192715,
|
| 61 |
+
"rollout_aux_rmse": 77.80495413636444,
|
| 62 |
+
"qoi_pred": {
|
| 63 |
+
"final_length": 64.1743552684784,
|
| 64 |
+
"mushroom_width": 75.10093879699707,
|
| 65 |
+
"peak_von_mises": 215.60115682029723,
|
| 66 |
+
"t_peak_von_mises": 0.2499345444266295
|
| 67 |
+
},
|
| 68 |
+
"qoi_true": {
|
| 69 |
+
"final_length": 58.99770545959473,
|
| 70 |
+
"mushroom_width": 94.23845672607422,
|
| 71 |
+
"peak_von_mises": 196.53865495598316,
|
| 72 |
+
"t_peak_von_mises": 0.03795318481890357
|
| 73 |
+
},
|
| 74 |
+
"qoi_error": {
|
| 75 |
+
"final_length": 5.176649808883667,
|
| 76 |
+
"mushroom_width": -19.13751792907715,
|
| 77 |
+
"peak_von_mises": 19.062501864314072,
|
| 78 |
+
"t_peak_von_mises": 0.2119813596077259
|
| 79 |
+
}
|
| 80 |
+
}
|
| 81 |
+
},
|
| 82 |
+
"mean": {
|
| 83 |
+
"one_step_position_rmse": 0.004649412237208173,
|
| 84 |
+
"one_step_aux_rmse": 40.43156173400896,
|
| 85 |
+
"rollout_position_rmse": 7.645119468234642,
|
| 86 |
+
"rollout_aux_rmse": 79.45890575554763,
|
| 87 |
+
"qoi_abs_error": {
|
| 88 |
+
"final_length": 3.1978601614634194,
|
| 89 |
+
"mushroom_width": 11.588177998860678,
|
| 90 |
+
"peak_von_mises": 19.20983395657771,
|
| 91 |
+
"t_peak_von_mises": 0.22932332950404855
|
| 92 |
+
}
|
| 93 |
+
}
|
| 94 |
+
}
|
cgn-7be9d4b/metrics-test_interp.json
ADDED
|
@@ -0,0 +1,166 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"split": "test_interp",
|
| 3 |
+
"checkpoint": "model-best-096000.pt",
|
| 4 |
+
"input_frames": 6,
|
| 5 |
+
"protocol_standard": true,
|
| 6 |
+
"aux_field": "von_mises_stress",
|
| 7 |
+
"aux_unit": "MPa",
|
| 8 |
+
"cases": {
|
| 9 |
+
"T-20-60-130": {
|
| 10 |
+
"one_step_position_rmse": 0.0029863393295605166,
|
| 11 |
+
"one_step_aux_rmse": 33.42273237359162,
|
| 12 |
+
"rollout_position_rmse": 1.1724716982061856,
|
| 13 |
+
"rollout_aux_rmse": 53.7102300990292,
|
| 14 |
+
"qoi_pred": {
|
| 15 |
+
"final_length": 41.39955711364746,
|
| 16 |
+
"mushroom_width": 61.520240783691406,
|
| 17 |
+
"peak_von_mises": 193.53798085212708,
|
| 18 |
+
"t_peak_von_mises": 0.04393999701812402
|
| 19 |
+
},
|
| 20 |
+
"qoi_true": {
|
| 21 |
+
"final_length": 45.76681065559387,
|
| 22 |
+
"mushroom_width": 52.26179313659668,
|
| 23 |
+
"peak_von_mises": 198.2767936247587,
|
| 24 |
+
"t_peak_von_mises": 0.03395583786256341
|
| 25 |
+
},
|
| 26 |
+
"qoi_error": {
|
| 27 |
+
"final_length": -4.367253541946411,
|
| 28 |
+
"mushroom_width": 9.258447647094727,
|
| 29 |
+
"peak_von_mises": -4.738812772631633,
|
| 30 |
+
"t_peak_von_mises": 0.009984159155560605
|
| 31 |
+
}
|
| 32 |
+
},
|
| 33 |
+
"T-20-80-130": {
|
| 34 |
+
"one_step_position_rmse": 0.0027164460224023235,
|
| 35 |
+
"one_step_aux_rmse": 34.04496360425782,
|
| 36 |
+
"rollout_position_rmse": 0.7264597605049596,
|
| 37 |
+
"rollout_aux_rmse": 44.40699328871509,
|
| 38 |
+
"qoi_pred": {
|
| 39 |
+
"final_length": 61.77581262588501,
|
| 40 |
+
"mushroom_width": 61.50594902038574,
|
| 41 |
+
"peak_von_mises": 188.93121943235397,
|
| 42 |
+
"t_peak_von_mises": 0.0439445119002249
|
| 43 |
+
},
|
| 44 |
+
"qoi_true": {
|
| 45 |
+
"final_length": 60.6046097278595,
|
| 46 |
+
"mushroom_width": 58.82277488708496,
|
| 47 |
+
"peak_von_mises": 188.09792789041995,
|
| 48 |
+
"t_peak_von_mises": 0.04594452652442098
|
| 49 |
+
},
|
| 50 |
+
"qoi_error": {
|
| 51 |
+
"final_length": 1.1712028980255127,
|
| 52 |
+
"mushroom_width": 2.6831741333007812,
|
| 53 |
+
"peak_von_mises": 0.8332915419340168,
|
| 54 |
+
"t_peak_von_mises": -0.002000014624196078
|
| 55 |
+
}
|
| 56 |
+
},
|
| 57 |
+
"T-20-100-130": {
|
| 58 |
+
"one_step_position_rmse": 0.0028854622672042415,
|
| 59 |
+
"one_step_aux_rmse": 38.12394288316013,
|
| 60 |
+
"rollout_position_rmse": 0.960856562423106,
|
| 61 |
+
"rollout_aux_rmse": 56.90052162833632,
|
| 62 |
+
"qoi_pred": {
|
| 63 |
+
"final_length": 78.47338104248047,
|
| 64 |
+
"mushroom_width": 61.50519943237305,
|
| 65 |
+
"peak_von_mises": 185.71876949119567,
|
| 66 |
+
"t_peak_von_mises": 0.04394554690884283
|
| 67 |
+
},
|
| 68 |
+
"qoi_true": {
|
| 69 |
+
"final_length": 77.05013656616211,
|
| 70 |
+
"mushroom_width": 62.88201713562012,
|
| 71 |
+
"peak_von_mises": 181.20112224334477,
|
| 72 |
+
"t_peak_von_mises": 0.04394554690884283
|
| 73 |
+
},
|
| 74 |
+
"qoi_error": {
|
| 75 |
+
"final_length": 1.4232444763183594,
|
| 76 |
+
"mushroom_width": -1.3768177032470703,
|
| 77 |
+
"peak_von_mises": 4.517647247850903,
|
| 78 |
+
"t_peak_von_mises": 0.0
|
| 79 |
+
}
|
| 80 |
+
},
|
| 81 |
+
"T-20-60-170": {
|
| 82 |
+
"one_step_position_rmse": 0.0038058265096635666,
|
| 83 |
+
"one_step_aux_rmse": 35.42581067034154,
|
| 84 |
+
"rollout_position_rmse": 2.1174855869506,
|
| 85 |
+
"rollout_aux_rmse": 56.42568927607488,
|
| 86 |
+
"qoi_pred": {
|
| 87 |
+
"final_length": 29.267239332199097,
|
| 88 |
+
"mushroom_width": 74.74068450927734,
|
| 89 |
+
"peak_von_mises": 200.90682324091594,
|
| 90 |
+
"t_peak_von_mises": 0.033981678191729606
|
| 91 |
+
},
|
| 92 |
+
"qoi_true": {
|
| 93 |
+
"final_length": 37.26132607460022,
|
| 94 |
+
"mushroom_width": 65.12842559814453,
|
| 95 |
+
"peak_von_mises": 204.55431056102117,
|
| 96 |
+
"t_peak_von_mises": 0.029971853968519403
|
| 97 |
+
},
|
| 98 |
+
"qoi_error": {
|
| 99 |
+
"final_length": -7.994086742401123,
|
| 100 |
+
"mushroom_width": 9.612258911132812,
|
| 101 |
+
"peak_von_mises": -3.6474873201052276,
|
| 102 |
+
"t_peak_von_mises": 0.004009824223210202
|
| 103 |
+
}
|
| 104 |
+
},
|
| 105 |
+
"T-20-80-170": {
|
| 106 |
+
"one_step_position_rmse": 0.003541684903098537,
|
| 107 |
+
"one_step_aux_rmse": 36.54841526958318,
|
| 108 |
+
"rollout_position_rmse": 1.4009586861498804,
|
| 109 |
+
"rollout_aux_rmse": 49.493475812496264,
|
| 110 |
+
"qoi_pred": {
|
| 111 |
+
"final_length": 47.87074828147888,
|
| 112 |
+
"mushroom_width": 74.74292755126953,
|
| 113 |
+
"peak_von_mises": 197.30715097427367,
|
| 114 |
+
"t_peak_von_mises": 0.037974934765466165
|
| 115 |
+
},
|
| 116 |
+
"qoi_true": {
|
| 117 |
+
"final_length": 50.93541741371155,
|
| 118 |
+
"mushroom_width": 74.04891586303711,
|
| 119 |
+
"peak_von_mises": 196.38054361239077,
|
| 120 |
+
"t_peak_von_mises": 0.039929159434196955
|
| 121 |
+
},
|
| 122 |
+
"qoi_error": {
|
| 123 |
+
"final_length": -3.064669132232666,
|
| 124 |
+
"mushroom_width": 0.6940116882324219,
|
| 125 |
+
"peak_von_mises": 0.9266073618829012,
|
| 126 |
+
"t_peak_von_mises": -0.0019542246687307904
|
| 127 |
+
}
|
| 128 |
+
},
|
| 129 |
+
"T-20-100-170": {
|
| 130 |
+
"one_step_position_rmse": 0.0035285250304216576,
|
| 131 |
+
"one_step_aux_rmse": 38.95531082980482,
|
| 132 |
+
"rollout_position_rmse": 1.2631911881958011,
|
| 133 |
+
"rollout_aux_rmse": 54.46365565068296,
|
| 134 |
+
"qoi_pred": {
|
| 135 |
+
"final_length": 67.2752423286438,
|
| 136 |
+
"mushroom_width": 74.75165557861328,
|
| 137 |
+
"peak_von_mises": 193.17722634601594,
|
| 138 |
+
"t_peak_von_mises": 0.03196907636266972
|
| 139 |
+
},
|
| 140 |
+
"qoi_true": {
|
| 141 |
+
"final_length": 66.79958987236023,
|
| 142 |
+
"mushroom_width": 79.65279006958008,
|
| 143 |
+
"peak_von_mises": 190.65364715766907,
|
| 144 |
+
"t_peak_von_mises": 0.03797780354960969
|
| 145 |
+
},
|
| 146 |
+
"qoi_error": {
|
| 147 |
+
"final_length": 0.47565245628356934,
|
| 148 |
+
"mushroom_width": -4.901134490966797,
|
| 149 |
+
"peak_von_mises": 2.523579188346872,
|
| 150 |
+
"t_peak_von_mises": -0.006008727186939969
|
| 151 |
+
}
|
| 152 |
+
}
|
| 153 |
+
},
|
| 154 |
+
"mean": {
|
| 155 |
+
"one_step_position_rmse": 0.0032440473437251407,
|
| 156 |
+
"one_step_aux_rmse": 36.086862605123194,
|
| 157 |
+
"rollout_position_rmse": 1.2735705804050887,
|
| 158 |
+
"rollout_aux_rmse": 52.56676095922245,
|
| 159 |
+
"qoi_abs_error": {
|
| 160 |
+
"final_length": 3.082684874534607,
|
| 161 |
+
"mushroom_width": 4.7543074289957685,
|
| 162 |
+
"peak_von_mises": 2.864570905458592,
|
| 163 |
+
"t_peak_von_mises": 0.003992824976439608
|
| 164 |
+
}
|
| 165 |
+
}
|
| 166 |
+
}
|
cgn-7be9d4b/model-best-096000.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4950d798a774be111d7096a16a112d38385d44e7832c7577bb1fa82eb7b76060
|
| 3 |
+
size 6447539
|
cgn-7be9d4b/normalization_stats.npz
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bac624bd89f94b04809cfa28da67447d31e2bfe5dd62176eb12418d2333ba7d2
|
| 3 |
+
size 1640
|