lars c09ebd2410 Merge pull request 'Add class-balanced secondary particle-type loss (gitea #44)' (#70) from fix/issue-44 into master ...

Reviewed-on: #70

2026-08-18 10:41:42 +02:00

..

legacy

Bump ruff line-length to 120 and reformat

2026-08-12 13:33:09 +02:00

test_amp.py

Add bf16 autocast to the training loop (gitea #47)

2026-08-17 15:36:12 +02:00

test_analysis_reduce.py

Offset event_id across multi-shard reference reads in giant analyze (gitea #22)

2026-08-17 15:53:02 +02:00

test_bump_dataset_version.py

Move scripts/ to giant/tools/ (issues.md Issue 9)

2026-08-13 10:31:47 +02:00

test_catalog.py

Bump ruff line-length to 120 and reformat

2026-08-12 13:33:09 +02:00

test_checkpoint_io.py

Decouple secondary-species vocabulary from conditioning.particle.emb_dim (gitea #29)

2026-08-13 16:11:14 +02:00

test_cli_new_run.py

Add per-stage init_from/freeze (gitea #42)

2026-08-17 14:23:20 +02:00

test_cli_predict.py

Extract predict/rollout's duplicated inference bootstrap into giant.checkpoint_io (issues.md Issue 5)

2026-08-12 15:31:14 +02:00

test_cli_rollout.py

Extract predict/rollout's duplicated inference bootstrap into giant.checkpoint_io (issues.md Issue 5)

2026-08-12 15:31:14 +02:00

test_cli_train_overrides.py

Add per-stage init_from/freeze (gitea #42)

2026-08-17 14:23:20 +02:00

test_cond_layout.py

Give the cond_cat/cond_cont column layout one owner (gitea #37)

2026-08-14 14:16:20 +02:00

test_condor.py

Bump ruff line-length to 120 and reformat

2026-08-12 13:33:09 +02:00

test_config_consumed_keys.py

Implement stage2_model.stage1_context = "sampled" (gitea #41)

2026-08-17 12:27:13 +02:00

test_config.py

Add class-balanced secondary particle-type loss (gitea #44)

2026-08-17 23:02:50 +02:00

test_create_root_files.py

Move scripts/ to giant/tools/ (issues.md Issue 9)

2026-08-13 10:31:47 +02:00

test_dataset.py

Move scripts/ to giant/tools/ (issues.md Issue 9)

2026-08-13 10:31:47 +02:00

test_dwarf.py

Let dwarf warm-cache take --config so it can't under-warm a config's cache keys (gitea #59)

2026-08-17 08:54:14 +02:00

test_flow.py

Pass ConditioningAxisConfig/ParticleTypeConfig themselves instead of raw dicts (gitea #38)

2026-08-14 15:03:55 +02:00

test_geometry.py

Bump ruff line-length to 120 and reformat

2026-08-12 13:33:09 +02:00

test_layers.py

Deduplicate n_sec_head/type_head MLPs into build_mlp_head (gitea #36)

2026-08-14 10:57:47 +02:00

test_loader.py

Add class-balanced secondary particle-type loss (gitea #44)

2026-08-17 23:02:50 +02:00

test_materials.py

Bump ruff line-length to 120 and reformat

2026-08-12 13:33:09 +02:00

test_migration_v02_v03.py

Unify the two v0.2->v0.3 migration surfaces (issues.md Issue 6)

2026-08-13 09:56:22 +02:00

test_model_summary.py

Add giant model summary command (gitea #46)

2026-08-17 11:48:02 +02:00

test_network.py

Pass ConditioningAxisConfig/ParticleTypeConfig themselves instead of raw dicts (gitea #38)

2026-08-14 15:03:55 +02:00

test_objectives.py

Pass ConditioningAxisConfig/ParticleTypeConfig themselves instead of raw dicts (gitea #38)

2026-08-14 15:03:55 +02:00

test_particles.py

Bump ruff line-length to 120 and reformat

2026-08-12 13:33:09 +02:00

test_phase2.py

Clip raw predicted log_mass in decode_secondaries (gitea #54)

2026-08-17 09:31:37 +02:00

test_pipeline.py

Add class-balanced secondary particle-type loss (gitea #44)

2026-08-17 23:02:50 +02:00

test_render.py

Bump ruff line-length to 120 and reformat

2026-08-12 13:33:09 +02:00

test_rollout.py

Implement n_sec.mode = "stop_token" for the AR secondary decoder (gitea #40)

2026-08-17 10:44:14 +02:00

test_router_gating.py

v0.3.0 step 3: per-stage train.py trainers + pipeline.py/cli.py rewrite

2026-08-06 11:31:49 +02:00

test_router.py

Add bf16 autocast to the training loop (gitea #47)

2026-08-17 15:36:12 +02:00

test_sample.py

Implement n_sec.mode = "stop_token" for the AR secondary decoder (gitea #40)

2026-08-17 10:44:14 +02:00

test_setup_cache.py

Add class-balanced secondary particle-type loss (gitea #44)

2026-08-17 23:02:50 +02:00

test_steps_to_parquet_parallel.py

Move scripts/ to giant/tools/ (issues.md Issue 9)

2026-08-13 10:31:47 +02:00

test_steps_to_parquet.py

Move scripts/ to giant/tools/ (issues.md Issue 9)

2026-08-13 10:31:47 +02:00

test_train.py

Add class-balanced secondary particle-type loss (gitea #44)

2026-08-17 23:02:50 +02:00

test_transforms.py

Give the cond_cat/cond_cont column layout one owner (gitea #37)

2026-08-14 14:16:20 +02:00

test_type_embedding_distance.py

Delete docs/v0.3.0-design.md and strip all references to it

2026-08-10 11:19:02 +02:00

test_validate.py

Pass ConditioningAxisConfig/ParticleTypeConfig themselves instead of raw dicts (gitea #38)

2026-08-14 15:03:55 +02:00

test_wgan.py

Pass ConditioningAxisConfig/ParticleTypeConfig themselves instead of raw dicts (gitea #38)

2026-08-14 15:03:55 +02:00