Skip to content

Add masked-window FM processor#412

Draft
sgreenbury wants to merge 5 commits into
283/rb-fm-latent-runsfrom
283/rb-masked-window-fm
Draft

Add masked-window FM processor#412
sgreenbury wants to merge 5 commits into
283/rb-fm-latent-runsfrom
283/rb-masked-window-fm

Conversation

@sgreenbury

Copy link
Copy Markdown
Contributor

Summary

  • Contributes towards Run latest pipeline with Rayleigh-Bénard dataset #283.
  • Add a masked-window flow-matching processor that trains over the full
    input-plus-output trajectory window while clamping sampled context states.
  • Add processor configs for masked-window FM with U-Net and ViT backbones,
    including setup-time resolution of automatic input/output/channel sizes.
  • Add a Rayleigh-Benard ViT-large 4096-epoch masked-window FM submitter using
    cached LOLA latents and the LOLA-style epoch/step budget.
  • Add tests for masked-window training, inference shape handling, config setup,
    and input-channel validation.

Testing

  • Not run in this PR-prep pass.

Add a flow-matching processor variant that masks temporal windows
so RB experiments can train with the masked-window objective.
Add the first Rayleigh-Benard masked-window FM submit script so the
new processor can be launched with the comparison workflow.
Revise the masked-window submit script with the final RB launch
settings needed for the comparison run.
Clean up lint failures in the masked-window FM branch without
changing the training behavior.
Only auto-fill encoder input channels when that key exists. This
preserves configs without in_channels while keeping RB auto setup.
@sgreenbury sgreenbury mentioned this pull request Jun 5, 2026
1 task
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