arabeh commited on
Commit
41bae07
Β·
1 Parent(s): d0982d3

initial commit

Browse files
README.md CHANGED
@@ -1,3 +1,67 @@
1
- ---
2
- license: cc-by-nc-4.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Time-Dependent DeepONet for FlowBench (FPO)
2
+
3
+ This repository hosts pre-trained **time-dependent DeepONet** checkpoints used in the paper:
4
+
5
+ > **Predicting Time-Dependent Flow Over Complex Geometries Using Operator Networks**
6
+
7
+ These models are trained on the **FlowBench FPO** data and learn to predict unsteady flow over complex 2D geometries.
8
+
9
+ ---
10
+
11
+ ## Associated Resources
12
+
13
+ - **Paper (arXiv):** https://arxiv.org/abs/2512.04434
14
+ - **Paper page on Hugging Face:** https://huggingface.co/papers/2512.04434
15
+ - **Dataset (FlowBench on Hugging Face):** https://huggingface.co/datasets/BGLab/FlowBench
16
+ - **Code (model implementation & training scripts):** https://github.com/baskargroup/TimeDependent-DeepONet
17
+
18
+ ---
19
+
20
+ ## Checkpoints
21
+
22
+ All checkpoints are stored under the `checkpoints/` directory:
23
+
24
+ - `time-dependent-deeponet_1in.ckpt` – model trained with input sequence length `s = 1`
25
+ - `time-dependent-deeponet_4in.ckpt` – model trained with input sequence length `s = 4`
26
+ - `time-dependent-deeponet_8in.ckpt` – model trained with input sequence length `s = 8`
27
+ - `time-dependent-deeponet_16in.ckpt` – model trained with input sequence length `s = 16`
28
+
29
+ Each checkpoint contains the weights for the time-dependent DeepONet used in the paper. For the exact architecture, data preprocessing, and training details, please refer to the GitHub repository.
30
+
31
+
32
+
33
+ You can download any checkpoint using `huggingface_hub`:
34
+
35
+ ```python
36
+ from huggingface_hub import hf_hub_download
37
+ import torch
38
+
39
+ from models.geometric_deeponet.geometric_deeponet import GeometricDeepONetTime
40
+
41
+ REPO_ID = "arabeh/DeepONet-FlowBench-FPO"
42
+ filename = "checkpoints/time-dependent-deeponet_4in.ckpt" # choose 1in / 4in / 8in / 16in
43
+
44
+ # 1) Download checkpoint file locally
45
+ ckpt_path = hf_hub_download(REPO_ID, filename)
46
+
47
+ # 2) Load the Lightning model from checkpoint
48
+ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
49
+ model = GeometricDeepONetTime.load_from_checkpoint(ckpt_path, map_location=device)
50
+ model = model.eval().to(device)
51
+ ```
52
+ For full, reproducible training and evaluation, including data loading and post-processing, please see:
53
+
54
+ > https://github.com/baskargroup/TimeDependent-DeepONet
55
+
56
+ ---
57
+
58
+ ## Citation
59
+
60
+ ```bibtex
61
+ @article{rabeh2025predicting,
62
+ title={Predicting Time-Dependent Flow Over Complex Geometries Using Operator Networks},
63
+ author={Rabeh, Ali and Murugaiyan, Suresh and Krishnamurthy, Adarsh and Ganapathysubramanian, Baskar},
64
+ journal={arXiv preprint arXiv:2512.04434},
65
+ year={2025}
66
+ }
67
+ ```
time-dependent-deeponet_16in.ckpt β†’ checkpoints/time-dependent-deeponet_16in.ckpt RENAMED
File without changes
time-dependent-deeponet_1in.ckpt β†’ checkpoints/time-dependent-deeponet_1in.ckpt RENAMED
File without changes
time-dependent-deeponet_4in.ckpt β†’ checkpoints/time-dependent-deeponet_4in.ckpt RENAMED
File without changes
time-dependent-deeponet_8in.ckpt β†’ checkpoints/time-dependent-deeponet_8in.ckpt RENAMED
File without changes