Skip to content

Flow matching pipeline - #2671

Draft
moritzhauschulz wants to merge 7 commits into
ecmwf:jk/develop/diffusion-full-pipelinefrom
moritzhauschulz:flow-matching-pipeline
Draft

Flow matching pipeline#2671
moritzhauschulz wants to merge 7 commits into
ecmwf:jk/develop/diffusion-full-pipelinefrom
moritzhauschulz:flow-matching-pipeline

Conversation

@moritzhauschulz

Copy link
Copy Markdown
Contributor

Description

DRAFT to inspect diff

Issue Number

Is this PR a draft? Mark it as draft.

Checklist before asking for review

  • I have performed a self-review of my code
  • My changes comply with basic sanity checks:
    • I have fixed formatting issues with ./scripts/actions.sh lint
    • I have run unit tests with ./scripts/actions.sh unit-test
    • I have documented my code and I have updated the docstrings.
    • I have added unit tests, if relevant
  • I have tried my changes with data and code:
    • I have run the integration tests with ./scripts/actions.sh integration-test
    • (bigger changes) I have run a full training and I have written in the comment the run_id(s): launch-slurm.py --time 60
    • (bigger changes and experiments) I have shared a hegdedoc in the github issue with all the configurations and runs for this experiments
  • I have informed and aligned with people impacted by my change:
    • for config changes: the MatterMost channels and/or a design doc
    • for changes of dependencies: the MatterMost software development channel

moritzhauschulz and others added 7 commits July 10, 2026 17:22
The diagnostic loop only tracked x_cur (the state before each update), so
the actual returned sample at node N was never recorded and x_t appeared to
stall above x0_hat. Record the final x_next at the terminal node: for the ve
path the last step is a plain Euler step (x_next == D), so the endpoint lands
on the x0_hat curve. x0_hat/velocity are undefined at sigma=0, so they get NaN
and are omitted from the plot.

Co-Authored-By: Claude Opus 4.8 <[email protected]>
Claude-Session: https://claude.ai/code/session_01YEgxx1GYiHf2cSH9Fw7kKU
@github-actions github-actions Bot added the model Related to model training or definition (not generic infra) label Jul 22, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

model Related to model training or definition (not generic infra)

Projects

Status: No status

Development

Successfully merging this pull request may close these issues.

1 participant