Skip to content

Add workflow to train SR-GNN model from ensemble of nekRS simulations - #73

Draft
rickybalin wants to merge 10 commits into
mainfrom
srgnn_workflow
Draft

Add workflow to train SR-GNN model from ensemble of nekRS simulations#73
rickybalin wants to merge 10 commits into
mainfrom
srgnn_workflow

Conversation

@rickybalin

@rickybalin rickybalin commented May 14, 2026

Copy link
Copy Markdown

This PR adds a user-driven workflow for training the SR-GNN model within nekRS-ML on high and low fidelity simulation data. The standard approach to train the SR-GNN model runs a single nekRS simulation which projects the high-p solution (the target of the model) to low-p space (the input to the model). In some instances, the projected low-p solution is significantly different than what an actual low-p nekRS simulation would see. This workflow aims to address this discrepancy by running a series of low- and high-p nekRS simulations to generate the training data.

The workflow uses EnsembleLauncher to run an ensemble of nekRS simulation, each starting from a different .f field file produced by an initial pilot simulation. The remainder of the workflow is the same as the turbChannel_srgnn example.

To Do:

  • Sketch workflow, from pilot simulation, to ensemble, to pre-processing the data, training and inference.
  • Validate ensemble piece of the workflow runs as expected
  • Validate pre-processing and training to make sure all data is used as expected
  • Update READMEs

rickybalin and others added 9 commits May 14, 2026 18:19
Signed-off-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
However nekRS does not like restarting with lower p, so still investigating solutions
Signed-off-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Signed-off-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
@rickybalin

Copy link
Copy Markdown
Author

Two issues to fix:

  • project allocation is hard coded in setup_case
  • the case name is not obtained correctly because there are 2 par files in the directory

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant