Upload G-PARC model weights, test data, and configs (4 models)
Browse files
README.md
CHANGED
|
@@ -11,21 +11,14 @@ tags:
|
|
| 11 |
|
| 12 |
Model weights, test data, and configuration files for the G-PARC elastoplastic simulation paper.
|
| 13 |
|
| 14 |
-
##
|
| 15 |
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
configs/ # Training configuration files
|
| 23 |
-
data/
|
| 24 |
-
normalization_stats.json # Global max normalization parameters
|
| 25 |
-
test/ # PLAID test simulations (.pt)
|
| 26 |
-
|
| 27 |
-
training_histories/ # Loss curves per epoch
|
| 28 |
-
```
|
| 29 |
|
| 30 |
## Dataset
|
| 31 |
|
|
@@ -39,10 +32,5 @@ PLAID 2D Elasto-Plasto-Dynamics benchmark — high-velocity impact on steel plat
|
|
| 39 |
|
| 40 |
```python
|
| 41 |
from huggingface_hub import hf_hub_download
|
| 42 |
-
|
| 43 |
ckpt = hf_hub_download("jacktbeerman/Gparc", "checkpoints/gparcv2_best.pth")
|
| 44 |
```
|
| 45 |
-
|
| 46 |
-
## Code
|
| 47 |
-
|
| 48 |
-
Full training and evaluation code: [GitHub repo link here]
|
|
|
|
| 11 |
|
| 12 |
Model weights, test data, and configuration files for the G-PARC elastoplastic simulation paper.
|
| 13 |
|
| 14 |
+
## Models
|
| 15 |
|
| 16 |
+
| Model | Description |
|
| 17 |
+
|-------|-------------|
|
| 18 |
+
| G-PARCv1 | Graph Physics-Aware Recurrent Convolutions — fully learned GNN operators |
|
| 19 |
+
| G-PARCv2 | MLS differential operators + numerical Euler integration |
|
| 20 |
+
| MeshGraphKAN | Kolmogorov-Arnold Network message passing with Fourier basis |
|
| 21 |
+
| MeshGraphNet | Standard encode-process-decode GNN (Pfaff et al., 2021) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 22 |
|
| 23 |
## Dataset
|
| 24 |
|
|
|
|
| 32 |
|
| 33 |
```python
|
| 34 |
from huggingface_hub import hf_hub_download
|
|
|
|
| 35 |
ckpt = hf_hub_download("jacktbeerman/Gparc", "checkpoints/gparcv2_best.pth")
|
| 36 |
```
|
|
|
|
|
|
|
|
|
|
|
|
checkpoints/meshgraphkan_best.pth
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 8109521
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e4d3fa10ae1104788402b4bb275ec2ff670e3c9c2722e22b52f88c70709f3e5f
|
| 3 |
size 8109521
|
checkpoints/meshgraphnet_best.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1cfcb1561e5c8bfc247069608156a27cf14589b7fe9221da17983274915be8ba
|
| 3 |
+
size 6250193
|
configs/meshgraphkan_config.json
CHANGED
|
@@ -26,7 +26,7 @@
|
|
| 26 |
"grad_clip_norm": 1.0,
|
| 27 |
"device": "cuda",
|
| 28 |
"num_workers": 4,
|
| 29 |
-
"output_dir": "/scratch/jtb3sud/
|
| 30 |
"resume": null,
|
| 31 |
"reset_best": false,
|
| 32 |
"fresh_scheduler": false,
|
|
|
|
| 26 |
"grad_clip_norm": 1.0,
|
| 27 |
"device": "cuda",
|
| 28 |
"num_workers": 4,
|
| 29 |
+
"output_dir": "/scratch/jtb3sud/delta/elasto",
|
| 30 |
"resume": null,
|
| 31 |
"reset_best": false,
|
| 32 |
"fresh_scheduler": false,
|
configs/meshgraphnet_config.json
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"data_dir": "/scratch/jtb3sud/processed_elasto_plastic/global_max/normalized/small",
|
| 3 |
+
"seq_len": 16,
|
| 4 |
+
"stride": 16,
|
| 5 |
+
"num_static_feats": 2,
|
| 6 |
+
"num_dynamic_feats": 2,
|
| 7 |
+
"hidden_dim": 128,
|
| 8 |
+
"num_layers": 4,
|
| 9 |
+
"epochs": 1500,
|
| 10 |
+
"lr": 0.0001,
|
| 11 |
+
"weight_decay": 0.0005,
|
| 12 |
+
"grad_clip_norm": 1.0,
|
| 13 |
+
"scheduler": "cosine",
|
| 14 |
+
"scheduler_step": 100,
|
| 15 |
+
"scheduler_gamma": 0.9,
|
| 16 |
+
"device": "cuda",
|
| 17 |
+
"num_workers": 4,
|
| 18 |
+
"checkpoint_dir": "/scratch/jtb3sud/meshgraphnet/elasto/run1",
|
| 19 |
+
"val_every": 10,
|
| 20 |
+
"save_every": 100
|
| 21 |
+
}
|
training_histories/meshgraphkan_history.json
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|