drno / README.md
Bobchen1980's picture
Update README.md
3a81f0f verified
|
Raw
History Blame Contribute Delete
2.79 kB
---
license: mit
---
# DRNO sanity outputs
This repository mirrors `outputs/sanity/`, the full sanity experiment output directory for Discrete-Residual Neural Operators (DRNO). The artifacts were generated by the code at <https://github.com/soundai2016/DRNO> with `configs/sanity.yaml`.
## Download
```bash
pip install --upgrade huggingface_hub
hf download soundai2016/drno --local-dir outputs/sanity
```
## Contents
```text
outputs/sanity/
data/ generated PDE datasets: .npz arrays + .json metadata
checkpoints/ PyTorch checkpoints for each task/model/seed/lambda/selector
histories/ per-epoch training logs
results/ CSV metrics, Pareto grids, uncertainty, runtime and audit ledgers
figures/ analysis figures from the plot stage
tables/ JSON tables derived from results/*.csv
pdebench_hdf5/ PDEBench-style HDF5 exports, manifest and audit summary
publication/ manuscript-ready figures, tables and source data
resolved_config.json
run_environment.json
publication_package.json
publication_figures_run.json
pdebench_manifest_summary.json
```
## Scope
The sanity run covers four PDE tasks: `darcy`, `burgers`, `schrodinger`, and `ns2d_kolmogorov`; train/val/test/OOD/mesh-transfer splits; seeds `0..4`; FNO, PINO-style residual, DRNO solver-consistent residual, MC-dropout FNO, DeepONet (Darcy/Burgers) and weak-form residual (Darcy/Burgers) baselines.
## Naming convention
```text
data/{task}_{split}.npz
histories/{task}_{model}_seed{seed}_lam{lambda}_{tag}.csv
checkpoints/{task}_{model}_seed{seed}_lam{lambda}_{tag}_{selector}.pt
```
Common `tag` values are `main` and `sweep`. Common checkpoint selectors are `best_rel`, `best_res`, `best_qoi`, `best_pareto`, and `final`. Decimal residual weights are filename-safe, e.g. `0.2 -> 0p2`.
## Reproduce or extend
```bash
git clone https://github.com/soundai2016/DRNO
cd DRNO
python -m pip install -e .
# Recreate the full suite from scratch.
bash scripts/run_sanity.sh
# Or reuse downloaded artifacts under outputs/sanity/ and rerun a stage.
python -m drno.run_all --config configs/sanity.yaml --stage audit
python -m drno.run_all --config configs/sanity.yaml --stage plot --force
```
The full pipeline is:
```text
generate -> train -> sweep -> eval -> pareto -> uncertainty -> benchmark -> runtime -> pdebench -> plot -> audit
```
## Notes
- `resolved_config.json` records the exact normalized configuration used for the run.
- `run_environment.json` records the software and hardware environment.
- `.pt` checkpoint files are PyTorch serialized artifacts; load them only in a trusted Python environment.
- CSV/JSON/HDF5 files are intended for analysis and reproduction; figures and `publication/` are derived display artifacts.