Upload folder using huggingface_hub
Browse files
cgn-48046ea/SHA256SUMS
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
f1b4d0e7edb06b4f5932309d101687afb5ee9c57cd84389744db21131a079c9b ./config.json
|
| 2 |
+
33ac5608c9232bc1b851fa99933fcbaf8ec96b49abd424d085cd3cd31ecce2f5 ./metrics-test_interp.json
|
| 3 |
+
2139335fb0cb2f6cccaf9be69e69cced369deb6eda80f6970b292deeba07dc0a ./model-best-050000.pt
|
| 4 |
+
24912b1830a08ad99f96c44414d80cd8bfb71f64f96d61c18522288e058f3cf7 ./normalization_stats.npz
|
cgn-48046ea/config.json
ADDED
|
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"run": {
|
| 3 |
+
"benchmark": "wave_propagation_1d",
|
| 4 |
+
"seed": 1,
|
| 5 |
+
"commit": "48046ea"
|
| 6 |
+
},
|
| 7 |
+
"model": {
|
| 8 |
+
"family": "cgn",
|
| 9 |
+
"input_frames": 6,
|
| 10 |
+
"connectivity_radius": 4.8,
|
| 11 |
+
"hidden_dim": 128,
|
| 12 |
+
"message_passing_steps": 10,
|
| 13 |
+
"nmlp_layers": 2,
|
| 14 |
+
"particle_type_embedding_size": 9,
|
| 15 |
+
"noise_std": 0.06,
|
| 16 |
+
"dim": 2,
|
| 17 |
+
"max_neighbors": 32
|
| 18 |
+
},
|
| 19 |
+
"train": {
|
| 20 |
+
"batch_size": 32,
|
| 21 |
+
"lr_init": 0.0001,
|
| 22 |
+
"lr_decay": 0.1,
|
| 23 |
+
"lr_decay_steps": 20000,
|
| 24 |
+
"training_steps": 50000,
|
| 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-48046ea/metrics-test_interp.json
ADDED
|
@@ -0,0 +1,71 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"split": "test_interp",
|
| 3 |
+
"checkpoint": "model-best-050000.pt",
|
| 4 |
+
"checkpoint_path": "model-best-050000.pt",
|
| 5 |
+
"input_frames": 6,
|
| 6 |
+
"protocol_standard": true,
|
| 7 |
+
"aux_field": "axial_stress",
|
| 8 |
+
"aux_unit": "MPa",
|
| 9 |
+
"cases": {
|
| 10 |
+
"W1D-300-4": {
|
| 11 |
+
"one_step_position_rmse": 0.00592706467201435,
|
| 12 |
+
"one_step_aux_rmse": 0.019114366873149855,
|
| 13 |
+
"rollout_position_rmse": 0.8001023907774389,
|
| 14 |
+
"rollout_aux_rmse": 0.1944458723794315,
|
| 15 |
+
"qoi_pred": {
|
| 16 |
+
"arrival_time_25": 3.187744668160479,
|
| 17 |
+
"arrival_time_50": 2.0821058760854054,
|
| 18 |
+
"arrival_time_75": 1.0980489830044557,
|
| 19 |
+
"peak_stress": 1.7382339239120483
|
| 20 |
+
},
|
| 21 |
+
"qoi_true": {
|
| 22 |
+
"arrival_time_25": 3.0872030812875684,
|
| 23 |
+
"arrival_time_50": 2.0821058760854054,
|
| 24 |
+
"arrival_time_75": 0.997728156029563,
|
| 25 |
+
"peak_stress": 0.8598179817199707
|
| 26 |
+
},
|
| 27 |
+
"qoi_error": {
|
| 28 |
+
"arrival_time_25": 0.10054158687291048,
|
| 29 |
+
"arrival_time_50": 0.0,
|
| 30 |
+
"arrival_time_75": 0.10032082697489275,
|
| 31 |
+
"peak_stress": 0.8784159421920776
|
| 32 |
+
}
|
| 33 |
+
},
|
| 34 |
+
"W1D-400-2": {
|
| 35 |
+
"one_step_position_rmse": 0.00383679301873249,
|
| 36 |
+
"one_step_aux_rmse": 0.011818982659420788,
|
| 37 |
+
"rollout_position_rmse": 0.9499493180998838,
|
| 38 |
+
"rollout_aux_rmse": 0.14067942027664498,
|
| 39 |
+
"qoi_pred": {
|
| 40 |
+
"arrival_time_25": 4.293667717636287,
|
| 41 |
+
"arrival_time_50": 2.8808558431727835,
|
| 42 |
+
"arrival_time_75": 1.4888669092436642,
|
| 43 |
+
"peak_stress": 1.4808571338653564
|
| 44 |
+
},
|
| 45 |
+
"qoi_true": {
|
| 46 |
+
"arrival_time_25": 4.192738724862022,
|
| 47 |
+
"arrival_time_50": 2.7799598827585608,
|
| 48 |
+
"arrival_time_75": 1.3880422572290732,
|
| 49 |
+
"peak_stress": 0.42626917362213135
|
| 50 |
+
},
|
| 51 |
+
"qoi_error": {
|
| 52 |
+
"arrival_time_25": 0.10092899277426515,
|
| 53 |
+
"arrival_time_50": 0.10089596041422277,
|
| 54 |
+
"arrival_time_75": 0.10082465201459101,
|
| 55 |
+
"peak_stress": 1.054587960243225
|
| 56 |
+
}
|
| 57 |
+
}
|
| 58 |
+
},
|
| 59 |
+
"mean": {
|
| 60 |
+
"one_step_position_rmse": 0.0048819288453734195,
|
| 61 |
+
"one_step_aux_rmse": 0.015466674766285321,
|
| 62 |
+
"rollout_position_rmse": 0.8750258544386613,
|
| 63 |
+
"rollout_aux_rmse": 0.16756264632803824,
|
| 64 |
+
"qoi_abs_error": {
|
| 65 |
+
"arrival_time_25": 0.10073528982358781,
|
| 66 |
+
"arrival_time_50": 0.05044798020711139,
|
| 67 |
+
"arrival_time_75": 0.10057273949474188,
|
| 68 |
+
"peak_stress": 0.9665019512176514
|
| 69 |
+
}
|
| 70 |
+
}
|
| 71 |
+
}
|
cgn-48046ea/model-best-050000.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2139335fb0cb2f6cccaf9be69e69cced369deb6eda80f6970b292deeba07dc0a
|
| 3 |
+
size 6447027
|
cgn-48046ea/normalization_stats.npz
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:24912b1830a08ad99f96c44414d80cd8bfb71f64f96d61c18522288e058f3cf7
|
| 3 |
+
size 1640
|