lars 44b0a92e67 Add WGAN-GP mode as a throwaway fast-eval experiment ...

Adds --mode wgan alongside flow/ddpm: both stages get a WGAN-GP
generator/critic pair (giant.model.wgan) instead of flow matching, so
inference is a single forward pass per stage rather than a 10-step ODE
integration — the fast-eval architecture noted in the roadmap.
predict/rollout auto-detect the mode from the checkpoint's model_config.
Best-checkpoint selection for wgan uses marginal-KL against the EMA
generators every epoch, since a critic loss isn't a monotone quality
signal. --router is not supported together with --mode wgan.

Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>

2026-07-22 11:20:28 +02:00

..

test_analysis.py

Reimplement rollout-vs-truth comparison on the streaming analysis module

2026-07-17 12:32:30 +02:00

test_bump_dataset_version.py

Fix ruff, ty, and pytest failures; apply ruff format

2026-07-02 16:59:22 +02:00

test_cli_predict.py

Add --comment option to predict, recorded in YAML sidecar

2026-07-06 09:37:48 +02:00

test_config.py

Apply ruff format and document lint/type tooling in CLAUDE.md

2026-06-18 17:40:27 +02:00

test_create_root_files.py

Support --energy-gev in dwarf make-root for the new minicalosim energy arg

2026-07-13 13:59:09 +02:00

test_dataset.py

Implement Phase 2: secondary particle prediction

2026-06-29 11:34:31 +02:00

test_dwarf.py

Unify dataset/tooling scripts into a single dwarf Typer CLI

2026-07-02 09:52:37 +02:00

test_flow.py

Implement Phase 2: secondary particle prediction

2026-06-29 11:34:31 +02:00

test_geometry.py

Add fast slab lookup for the GeometryOracle, replacing knn as the default

2026-07-08 11:42:12 +02:00

test_loader.py

Apply ruff format after merging phase2-secondary-prediction

2026-07-15 10:00:36 +02:00

test_materials.py

Condition on material/particle physical properties instead of learned embeddings

2026-07-17 15:12:54 +02:00

test_network.py

Implement Phase 2: secondary particle prediction

2026-06-29 11:34:31 +02:00

test_particles.py

Fix crashes in physical-property conditioning edge cases

2026-07-20 10:36:25 +02:00

test_phase2.py

Condition on material/particle physical properties instead of learned embeddings

2026-07-17 15:12:54 +02:00

test_rollout.py

Fix crashes in physical-property conditioning edge cases

2026-07-20 10:36:25 +02:00

test_router.py

Condition on material/particle physical properties instead of learned embeddings

2026-07-17 15:12:54 +02:00

test_steps_to_parquet_parallel.py

Fix ruff, ty, and pytest failures; apply ruff format

2026-07-02 16:59:22 +02:00

test_steps_to_parquet.py

Apply ruff format after merging phase2-secondary-prediction

2026-07-15 10:00:36 +02:00

test_transforms.py

Condition on material/particle physical properties instead of learned embeddings

2026-07-17 15:12:54 +02:00

test_validate.py

Fix crashes in physical-property conditioning edge cases

2026-07-20 10:36:25 +02:00

test_wgan.py

Add WGAN-GP mode as a throwaway fast-eval experiment

2026-07-22 11:20:28 +02:00