Update configs#352
Merged
sgreenbury merged 11 commits intomainfrom Apr 26, 2026
Merged
Conversation
Add flow matching eval submitters for 25%, 50%, and 75% training checkpoints. Include both cached-latent encode-once evals and explicit ambient variants so the progress metrics do not overwrite final-checkpoint outputs.
Switch the U-Net 80M ablation config to the Azula library default ffn_factor=1, and re-budget hid_channels to [62, 124, 248, 496] to keep the canonical 1:2:4:8 doubling ratio while landing at ~81.3M params (within 0.3% of the 81.0M CRPS ViT baseline). Both backbones now use their respective Azula-canonical FFN ratios (UNet=1, ViT=4 per Vaswani 2017).
Merge the base eval defaults before generating struct-safe overrides for resolved configs, so old runs can accept newly added eval keys. This keeps explicit eval overrides working for rollout snapshots and future eval defaults.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
This pull request adds and updates several configuration files and documentation to support a new, centralized batch of planned ablation studies on the CNS (Conditioned Navier-Stokes) task. The main focus is on adding ablation configs for model architecture, loss variants, noise channels, and diffusion, as well as updating documentation to reflect the new planned batch and its implementation status. A new evaluation script for ensemble size ablation is also included.
The most important changes are:
New experiment configurations:
crps_unet_azula_80m.yaml), CRPS loss variants (FairCRPS and plain CRPS), noise channels (crps_vit_noise256.yaml), and diffusion ViT baseline (diffusion_vit_large.yaml). These configs standardize architecture, loss, and training parameters for fair comparison. [1] [2] [3] [4] [5]Documentation updates for planned CNS batch:
slurm_scripts/ablations/README.mdto describe the new planned CNS batch, list its runs, and mark the implementation status of each ablation. Added a section explaining the batch submission scripts and how each config/run fits into the overall study. [1] [2]Loss variant and ensemble size documentation:
Ablation status and evaluation scripts: