license: apache-2.0
language:
- en
datasets:
- BGLab/FlowBench
tags:
- OneScience
- computational-fluid-dynamics
- neural-operator
- lid-driven-cavity
FlowBench-FNO-LDC2D-Harmonics
This model reproduces the LDC_NS_2D/128x128/harmonics FNO baseline from FlowBench. It maps a three-channel input to a four-channel flow field.
Download
hf download OneScience-Group/FlowBench-FNO-LDC2D-Harmonics --local-dir ./FlowBench-FNO-LDC2D-Harmonics
cd FlowBench-FNO-LDC2D-Harmonics
hf download BGLab/FlowBench \
LDC_NS_2D/128x128/harmonics_lid_driven_cavity_X.npz \
LDC_NS_2D/128x128/harmonics_lid_driven_cavity_Y.npz \
--repo-type dataset --local-dir ./data
The two original NPZ files are also available as a provenance-preserving subset on ModelScope. The original dataset is CC BY-NC 4.0. No training data is included in this model repository.
Run
Install PyTorch, NumPy, PyYAML and Matplotlib in a compatible GPU/DCU environment, then run:
python scripts/inference.py
python scripts/result.py
# To repeat the configured training run:
python scripts/train.py
The package includes weight/best_model.pth, trained on 800 examples and selected on 200 validation examples. Recorded validation MSE is 0.0012336337 and relative L2 is 0.0434478655. Weight SHA256: 13caed2d5d34db87cf728b1669601782642d2ab83a9c8b683fe05e2a80ab5905. See results/ for metrics and figures.
The model code is distributed under Apache 2.0. The original FlowBench data retains its separate CC BY-NC 4.0 terms.