physics
pde
neural-operator
fluid-dynamics
combustion
benchmark
westlake-ai4s commited on
Commit
8e1d5ab
·
verified ·
1 Parent(s): fd9df56

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +138 -0
README.md ADDED
@@ -0,0 +1,138 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: cc-by-4.0
3
+ tags:
4
+ - physics
5
+ - pde
6
+ - neural-operator
7
+ - fluid-dynamics
8
+ - combustion
9
+ - benchmark
10
+ datasets:
11
+ - AI4Science-WestlakeU/RealPDEBench
12
+ ---
13
+
14
+ # RealPDEBench Model Checkpoints
15
+
16
+ Trained model checkpoints for [RealPDEBench](https://github.com/AI4Science-WestlakeU/RealPDEBench), a benchmark for evaluating neural PDE solvers on real-world experimental data.
17
+
18
+ ## Models (10 architectures)
19
+
20
+ | Model | Type | File Size (per checkpoint) |
21
+ |-------|------|---------------------------|
22
+ | DPOT-L | Transformer | 2.5-2.6 GB |
23
+ | FNO | Spectral | 385M-2.1G |
24
+ | Galerkin Transformer | Transformer | 386-642M |
25
+ | WDNO | Diffusion | 351M-1.4G |
26
+ | DPOT-S | Transformer | 118-159M |
27
+ | U-Net | CNN | 88-89M |
28
+ | CNO | Hybrid | 31M |
29
+ | MWT | Wavelet | 22M |
30
+ | Transolver | Transformer | 17M |
31
+ | DeepONet | Neural Operator | 14M |
32
+
33
+ ## Scenarios (5)
34
+
35
+ | Scenario | Description |
36
+ |----------|-------------|
37
+ | `cylinder` | Flow past a circular cylinder |
38
+ | `controlled_cylinder` | Actively controlled cylinder flow |
39
+ | `fsi` | Fluid-structure interaction |
40
+ | `foil` | Flow past an airfoil |
41
+ | `combustion` | Turbulent combustion |
42
+
43
+ ## Training Paradigms
44
+
45
+ | File | Paradigm |
46
+ |------|----------|
47
+ | `numerical.pth` | Trained on numerical simulation data only |
48
+ | `real.pth` | Trained on real experimental data only |
49
+ | `finetune.pth` | Pretrained on numerical, finetuned on real |
50
+ | `numerical_base_for_finetune.pth` | Numerical pretrain base (DPOT-S/L only) |
51
+
52
+ ## Directory Structure
53
+
54
+ ```
55
+ {scenario}/{model}/{paradigm}.pth
56
+ configs/{scenario}/{model}.yaml
57
+ ```
58
+
59
+ Example: `cylinder/fno/finetune.pth` + `configs/cylinder/fno.yaml`
60
+
61
+ ## Quick Start
62
+
63
+ ### Install
64
+
65
+ ```bash
66
+ git clone https://github.com/AI4Science-WestlakeU/RealPDEBench.git
67
+ cd RealPDEBench && pip install -e .
68
+ ```
69
+
70
+ ### Download a Single Checkpoint
71
+
72
+ ```python
73
+ from huggingface_hub import hf_hub_download
74
+
75
+ path = hf_hub_download(
76
+ repo_id="AI4Science-WestlakeU/RealPDEBench-models",
77
+ filename="cylinder/fno/finetune.pth",
78
+ )
79
+ ```
80
+
81
+ ### Download All Checkpoints for a Scenario
82
+
83
+ ```python
84
+ from huggingface_hub import snapshot_download
85
+
86
+ snapshot_download(
87
+ repo_id="AI4Science-WestlakeU/RealPDEBench-models",
88
+ allow_patterns="cylinder/**",
89
+ local_dir="./checkpoints",
90
+ )
91
+ ```
92
+
93
+ ### Evaluate
94
+
95
+ ```bash
96
+ python eval.py --config configs/cylinder/fno.yaml \
97
+ --checkpoint_path ./checkpoints/cylinder/fno/finetune.pth \
98
+ --dataset_type real --test_mode all
99
+ ```
100
+
101
+ ## Checkpoint Format
102
+
103
+ ```python
104
+ checkpoint = torch.load("cylinder/fno/finetune.pth")
105
+ # Keys: model_state_dict, train_losses, val_losses,
106
+ # iteration, best_iteration, best_val_loss
107
+ ```
108
+
109
+ ## DPOT Pretrained Weights
110
+
111
+ DPOT models require pretrained backbone weights (**not included here**).
112
+ Download via:
113
+
114
+ ```bash
115
+ # Option 1: Built-in download script
116
+ python -m realpdebench.utils.dpot_ckpts_dl
117
+
118
+ # Option 2: From HuggingFace directly
119
+ # https://huggingface.co/hzk17/DPOT
120
+ ```
121
+
122
+ ## Dataset
123
+
124
+ The corresponding dataset is hosted at: [AI4Science-WestlakeU/RealPDEBench](https://huggingface.co/datasets/AI4Science-WestlakeU/RealPDEBench)
125
+
126
+ ## Citation
127
+
128
+ ```bibtex
129
+ @article{deng2025realpdebench,
130
+ title={Can Neural Operators Always Be Trusted? A Benchmark for Real-World PDE Solving},
131
+ author={Deng, Wenhao and Wu, Tailin and Feng, Jiawei and others},
132
+ year={2025}
133
+ }
134
+ ```
135
+
136
+ ## License
137
+
138
+ CC BY 4.0