Skip to content

feat(p4): scale configs D=512/L=8 + experiment script + scaling notes (Quang P4)#31

Open
duckyquang wants to merge 1 commit into
mainfrom
quang/p4-scale
Open

feat(p4): scale configs D=512/L=8 + experiment script + scaling notes (Quang P4)#31
duckyquang wants to merge 1 commit into
mainfrom
quang/p4-scale

Conversation

@duckyquang

Copy link
Copy Markdown
Member

Implements P4 scaling deliverables:

  • marc/configs/train/scale.yaml — D=512, L=8 config for Stage A+B
  • scripts/run_scale_experiment.py — trains at 3 capacities (D=128/256/512)
  • results/p4_scale/scaling_notes.md — 3 data points (CPU run + GPU estimates)
  • marc/eval/metrics.py — fixes API compatibility with runner.py

⚠️ This branch merges all prior feature branches; merge after quang/p0-p3 branches land.

Closes #23

… (Quang P4)

Implements P4 scaling deliverables:
- marc/configs/train/scale.yaml — D=512, L=8 config for Stage A+B on geometry data
- scripts/run_scale_experiment.py — trains at 3 capacities (D=128/256/512)
- results/p4_scale/scaling_notes.md — 3 data points (CPU run + GPU estimates)

Model parameter counts: D=128/L=4: 886K, D=256/L=6: 5.1M, D=512/L=8: 26.6M

Closes #23

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
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.

[P4] Scale model + training configs

1 participant