Upload folder using huggingface_hub
Browse files- scheduler/COMPARISON.md +48 -0
- scheduler/cosine_760/REPORT.md +38 -0
- scheduler/cosine_760/config.toml +89 -0
- scheduler/cosine_760/eval1.log +212 -0
- scheduler/cosine_760/eval2.log +54 -0
- scheduler/cosine_760/final/chat_template.jinja +7 -0
- scheduler/cosine_760/final/config.json +32 -0
- scheduler/cosine_760/final/generation_config.json +13 -0
- scheduler/cosine_760/final/model.safetensors +3 -0
- scheduler/cosine_760/final/tokenizer.json +0 -0
- scheduler/cosine_760/final/tokenizer_config.json +11 -0
- scheduler/cosine_760/trainer_state.json +663 -0
- scheduler/greedy/REPORT.md +42 -0
- scheduler/greedy/config.toml +93 -0
- scheduler/greedy/eval1.log +213 -0
- scheduler/greedy/eval2.log +54 -0
- scheduler/greedy/final/chat_template.jinja +7 -0
- scheduler/greedy/final/config.json +32 -0
- scheduler/greedy/final/generation_config.json +13 -0
- scheduler/greedy/final/model.safetensors +3 -0
- scheduler/greedy/final/tokenizer.json +0 -0
- scheduler/greedy/final/tokenizer_config.json +11 -0
- scheduler/greedy/trainer_state.json +886 -0
- scheduler/model.toml +13 -0
- scheduler/wsd_10/REPORT.md +34 -0
- scheduler/wsd_10/config.toml +89 -0
- scheduler/wsd_10/eval1.log +212 -0
- scheduler/wsd_10/eval2.log +54 -0
- scheduler/wsd_10/final/chat_template.jinja +7 -0
- scheduler/wsd_10/final/config.json +32 -0
- scheduler/wsd_10/final/generation_config.json +13 -0
- scheduler/wsd_10/final/model.safetensors +3 -0
- scheduler/wsd_10/final/tokenizer.json +0 -0
- scheduler/wsd_10/final/tokenizer_config.json +11 -0
- scheduler/wsd_10/trainer_state.json +656 -0
- scheduler/wsd_20/REPORT.md +38 -0
- scheduler/wsd_20/config.toml +89 -0
- scheduler/wsd_20/eval1.log +205 -0
- scheduler/wsd_20/eval2.log +54 -0
- scheduler/wsd_20/final/chat_template.jinja +7 -0
- scheduler/wsd_20/final/config.json +32 -0
- scheduler/wsd_20/final/generation_config.json +13 -0
- scheduler/wsd_20/final/model.safetensors +3 -0
- scheduler/wsd_20/final/tokenizer.json +0 -0
- scheduler/wsd_20/final/tokenizer_config.json +11 -0
- scheduler/wsd_20/trainer_state.json +649 -0
scheduler/COMPARISON.md
ADDED
|
@@ -0,0 +1,48 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# LR-Scheduler Comparison — MuonQ @ 8e-3 on the best recipe, Llama 77M, 60 min each
|
| 2 |
+
|
| 3 |
+
**Hardware:** RX 9070 XT 16GB, ROCm 7.2, torch 2.13, transformers 5.14.1
|
| 4 |
+
**Common recipe:** sdpa attention, fp32 master + bf16 autocast, **muonq** (lr 8e-3, wd 0.1, warmup 200), effective batch 256×1024 tok, seed 42, same data order, same dataset. Only the LR schedule varies. Per-folder details in each `*/REPORT.md`.
|
| 5 |
+
|
| 6 |
+
**Series prerequisite — the horizon fix:** every earlier run had `max_steps = -1`, giving schedulers a 75,210-step (full-epoch) horizon, so no decay ever actually happened inside a 1-hour run. `base_train.py` now accepts `max_steps` from config and all runs here pin it to **760** (the achievable steps/hour), so each schedule genuinely completes. The constant-LR arm is `../lr/lr_8e-3/` unchanged — that run *was* constant-after-warmup by accident, making it the free reference.
|
| 7 |
+
|
| 8 |
+
## Headline table (final-model metrics from evaluate2 on `final/`)
|
| 9 |
+
|
| 10 |
+
| schedule | steps/60min | final eval loss / ppl | held-out bpb | chat bpb | logic | bake /100 | greedy loop | verdict |
|
| 11 |
+
|---|---|---|---|---|---|---|---|---|
|
| 12 |
+
| constant after warmup (`../lr/lr_8e-3/`) | 759 | 3.8583 / 47.4 | 1.3114 | 1.1914 | 0.600 (+0.018) | 46.8 | 59.8w | reference — was the project best |
|
| 13 |
+
| WSD, 10% linear decay → 0 | 740 | 3.8641¹ / 47.7 | 1.2821 | 1.1608 | 0.625 (+0.012) | 52.7 | 33.8w | big win, dominated by wsd_20 |
|
| 14 |
+
| **WSD, 20% linear decay → 0** | 730 | 3.8023¹ / 44.8 | **1.2769** | **1.1571** | **0.650 (+0.024)** | **55.6 — record** | **23.8w** | **winner — quality AND hygiene** |
|
| 15 |
+
| cosine → 5% floor (`cosine_760/`) | 750 | **3.7494 / 42.5** | 1.2775 | 1.1623 | 0.650 (**+0.040**) | 51.7 | 66.2w | best raw loss, worst hygiene |
|
| 16 |
+
| GreedyLR adaptive (`greedy/`) | 760 | 3.9267² / 50.7 | 1.3422 | 1.2188 | 0.575 (+0.009) | 44.0 | 45.5w | loses (no anneal) — but found peak LR on its own |
|
| 17 |
+
|
| 18 |
+
¹ WSD logged evals land mid-decay (wall-clock stop cut the decays 20–26% short; final saved models sit at LR 2.2e-3 / 1.6e-3, not 0) — the trainer eval-loss column undersells them; bpb/bake columns are the honest final-model numbers.
|
| 19 |
+
² greedy evaluated on a 1024-sample subset (its 20-step eval cadence needs cheap evals) — noisier and not the same subset as the others' 5000.
|
| 20 |
+
|
| 21 |
+
All runs stable, zero non-finite grad norms, peak GPU 7.2 GB, clean period boundaries. The fixed-schedule runs share the identical 9.45 early grad spike (same seed/data → identical trajectory until schedules diverge after warmup); greedy's cold 1e-3 start was the calmest (max 5.51).
|
| 22 |
+
|
| 23 |
+
## Eval loss at matched wall-clock (5-min marks, 5000-sample evals)
|
| 24 |
+
|
| 25 |
+
| ~step | constant | wsd_10 | wsd_20 | cosine_760 |
|
| 26 |
+
|---|---|---|---|---|
|
| 27 |
+
| ~67 | 5.769 | 5.704 | 5.733 | 5.704 |
|
| 28 |
+
| ~192 | 4.582 | 4.569 | 4.596 | 4.578 |
|
| 29 |
+
| ~318 | 4.181 | 4.182 | 4.206 | 4.163 |
|
| 30 |
+
| ~444 | 4.015 | 4.025 | 4.034 | **3.961** |
|
| 31 |
+
| ~570 | 3.920 | 3.929 | 3.934 | **3.822** |
|
| 32 |
+
| ~690 | 3.858 | 3.864 | 3.802 | **3.749** |
|
| 33 |
+
|
| 34 |
+
Until the decays begin (~step 450 for cosine's steepening, ~610/684 for WSD) all four are statistically the same curve — as they must be, since they train identically at peak LR. Everything the schedules buy, they buy in the decay phase: wsd_20's last interval alone is −0.09.
|
| 35 |
+
|
| 36 |
+
## What we learned
|
| 37 |
+
|
| 38 |
+
1. **The anneal is the single biggest quality lever found in this project so far.** Fixing the horizon so *any* decay completes in-window is worth +4.9 to +8.8 bake points and −0.06 to −0.11 eval loss over the identical constant-LR run — more than the entire 1e-3 → 8e-3 LR tuning gained (+3.7 bake).
|
| 39 |
+
2. **WSD-20% > WSD-10% at this horizon, and WSD cures the hot-LR hygiene tax.** Greedy-loop length returns to cool-LR levels (23.8w vs 52–66w). Trading 76 stable-at-peak steps for a longer anneal was clearly worth it; 10% decay is too short here.
|
| 40 |
+
3. **Cosine wins raw loss but not the composite.** Best ppl (42.5) and logic margin (+0.040) of the project, yet the *worst* hygiene of the series (66.2w). It's not "more time at hot LR" — cosine's mid-run LR is *below* WSD's. Suspects: its 5% floor never fully anneals (WSD goes to 0), or greedy-decode-probe noise. A `min_lr_rate = 0` re-run would settle it.
|
| 41 |
+
4. **GreedyLR is a validator, not a competitor.** Left alone with only the eval-loss signal, it climbed from 1e-3 and plateaued at 6.6–7.4e-3 — independently rediscovering our hand-tuned 8e-3 peak. But it has no concept of a budget ending, finished the run hot, and lost to everything that decays. Adaptive stable phase + forced terminal decay could combine both, but plain WSD already captures most of that.
|
| 42 |
+
5. **Metric-driven schedulers need plumbing:** `metric_for_best_model` (now a base_train config key), a fast eval cadence, and note that `warmup_steps` is silently ignored for them.
|
| 43 |
+
|
| 44 |
+
## Recommendation
|
| 45 |
+
|
| 46 |
+
- **New best recipe: WSD with 20% linear decay to 0** (`wsd_20/config.toml`) on top of muonq @ 8e-3 — bake 55.6/100, bpb 1.2769, chat bpb 1.1571, logic 0.650, hygiene restored. This replaces the constant-LR 8e-3 recipe as the reference for future experiments.
|
| 47 |
+
- Follow-ups, roughly in value order: (a) set the horizon to the *actually achievable* ~730 steps so the decay finishes (wsd_20 set the record with 20% of its decay missing); (b) try `decay_type = "1-sqrt"` (what the WSD literature recommends over linear); (c) probe 30% decay — the 10% → 20% trend hasn't turned; (d) cosine with `min_lr_rate = 0` to isolate the hygiene puzzle.
|
| 48 |
+
- For the real 19.6B-token run: the WSD structure is also *operationally* ideal — the stable phase can run for days and be annealed from any checkpoint, so the decay fraction becomes a checkpoint-time decision, not a launch-time commitment. Re-verify 10 vs 20% at a 2–3h horizon first (decay-fraction optima shrink as budgets grow).
|
scheduler/cosine_760/REPORT.md
ADDED
|
@@ -0,0 +1,38 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Experiment: cosine_760 — cosine_with_min_lr with a real horizon
|
| 2 |
+
|
| 3 |
+
**Wall-clock:** 60 min · **Recipe:** best known (sdpa + bf16 mixed + muonq @ lr 8e-3), schedule: warmup 200 → cosine decay to 5% floor over `max_steps = 760` — the anneal every previous run *thought* it was doing (they all had a 75,210-step horizon and never decayed)
|
| 4 |
+
|
| 5 |
+
## Result: best loss/ppl of the entire project — but the worst greedy-loop hygiene
|
| 6 |
+
|
| 7 |
+
| metric | cosine_760 | wsd_20 (series winner) | constant-LR reference |
|
| 8 |
+
|---|---|---|---|
|
| 9 |
+
| optimizer steps in 60 min | 750 | 730 | 759 |
|
| 10 |
+
| final eval loss / ppl | **3.7494 / 42.5** (step 697, LR ≈ 4.7e-4 — decay essentially complete) | 3.8023 / 44.8 (mid-decay) | 3.8583 / 47.4 |
|
| 11 |
+
| bake score | 51.74/100 (DOUGH) | **55.61** | 46.83 |
|
| 12 |
+
| held-out bits/byte | 1.2775 | 1.2769 | 1.3114 |
|
| 13 |
+
| chat bits/byte | 1.1623 | **1.1571** | 1.1914 |
|
| 14 |
+
| logic accuracy | 0.650, **margin +0.0400** (best) | 0.650, +0.0242 | 0.600, +0.018 |
|
| 15 |
+
| hygiene (greedy loop) | **42.1/100 (loop 66.2w) — series worst** | 79.2 (23.8w) | 47.7 (59.8w) |
|
| 16 |
+
| grad norms | 0.07–9.45, 0 non-finite | 0.06–9.45 | 0.09–9.45 |
|
| 17 |
+
| peak GPU | 7.2 GB | 7.2 GB | 7.2 GB |
|
| 18 |
+
|
| 19 |
+
Fixing the horizon alone is worth −0.11 eval loss (3.8583 → 3.7494) and +4.9 bake points over the identical-LR constant run — confirmation that the LR series' biggest handicap was the broken anneal, not the LR values. On raw language-model quality cosine and wsd_20 are a near-tie (bpb 1.2775 vs 1.2769); cosine takes ppl and logic margin.
|
| 20 |
+
|
| 21 |
+
**The puzzle is hygiene:** 66.2w greedy loops, worse even than constant LR (59.8w), while wsd_20 sits at 23.8w. Mechanistically this is *not* "more time at hot LR" — cosine's mid-run LR is *lower* than WSD's (WSD holds peak until its decay). Two candidate explanations: (a) the 5% floor (4e-4) never fully anneals, while WSD's decay-to-0 does; (b) the greedy-loop probe is a small, noisy sample. Worth a re-test with `min_lr_rate = 0` before trusting either story.
|
| 22 |
+
|
| 23 |
+
## Eval loss curve (5-min intervals)
|
| 24 |
+
|
| 25 |
+
| step | eval loss | | step | eval loss |
|
| 26 |
+
|---|---|---|---|---|
|
| 27 |
+
| 67 | 5.7043 | | 444 | 3.9611 |
|
| 28 |
+
| 129 | 4.9559 | | 507 | 3.8829 |
|
| 29 |
+
| 192 | 4.5780 | | 570 | 3.8217 |
|
| 30 |
+
| 255 | 4.3263 | | 634 | 3.7765 |
|
| 31 |
+
| 318 | 4.1631 | | 697 | 3.7494 |
|
| 32 |
+
| 381 | 4.0505 | | | |
|
| 33 |
+
|
| 34 |
+
Unlike the WSD runs, the logged curve here is representative — the decay is continuous, so the step-697 eval reflects a nearly-finished anneal.
|
| 35 |
+
|
| 36 |
+
## Verdict
|
| 37 |
+
|
| 38 |
+
**Best perplexity in the project (42.5) and proof the horizon fix matters more than any LR choice** — but the composite loses to wsd_20 on hygiene. If greedy-decoding quality doesn't matter for the use case, this is the loss-optimal schedule; if it does, WSD-20% wins. See `../COMPARISON.md`.
|
scheduler/cosine_760/config.toml
ADDED
|
@@ -0,0 +1,89 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Scheduler series — new best recipe (sdpa + bf16 mixed + muonq @ lr 8e-3), only the LR schedule varies.
|
| 2 |
+
# This run: cosine_with_min_lr with a REAL horizon (max_steps 760) — the anneal the lr series thought it was running.
|
| 3 |
+
model_config = "../model.toml"
|
| 4 |
+
|
| 5 |
+
[data]
|
| 6 |
+
# Data files (supports glob patterns or lists of files)
|
| 7 |
+
train_files = ["~/Dev/vintage-LLM/training/data/train_*.bin"]
|
| 8 |
+
valid_files = ["~/Dev/vintage-LLM/training/data/valid_*.bin"]
|
| 9 |
+
|
| 10 |
+
# Sequence length for each dataset sample
|
| 11 |
+
# Must be <= model.max_position_embeddings
|
| 12 |
+
max_seq_length = 1024
|
| 13 |
+
|
| 14 |
+
# Tokenizer (must match the one used to create .bin files)
|
| 15 |
+
tokenizer = "~/Dev/vintage-LLM/tokenizers/t-v3/"
|
| 16 |
+
|
| 17 |
+
[training]
|
| 18 |
+
# Training duration
|
| 19 |
+
num_train_epochs = 1
|
| 20 |
+
max_steps = 760 # scheduler horizon = achievable steps/hour (was -1: full epoch, so decay never happened)
|
| 21 |
+
|
| 22 |
+
# Optional: Stop training after this many minutes of wall-clock time.
|
| 23 |
+
max_train_minutes = 60
|
| 24 |
+
|
| 25 |
+
# Output directories
|
| 26 |
+
output_dir = "./"
|
| 27 |
+
final_model_dir = "./final"
|
| 28 |
+
|
| 29 |
+
# Batch sizes
|
| 30 |
+
# Effective batch size = num_devices × per_device_train_batch_size × gradient_accumulation_steps
|
| 31 |
+
per_device_train_batch_size = 8 # batch size per GPU (8x1024 tokens fits 16GB with headroom for eval)
|
| 32 |
+
per_device_eval_batch_size = 8
|
| 33 |
+
gradient_accumulation_steps = 32 # accumulate gradients over multiple mini-batches before updating the params
|
| 34 |
+
|
| 35 |
+
# Optimizer: MuonQ (4-bit quantized Muon, ../../MuonQ repo) — wired through
|
| 36 |
+
# optimizer_cls_and_kwargs in base_train.py. 2D hidden weights get Muon
|
| 37 |
+
# (Newton-Schulz 5, 4-bit mu-law-companded momentum, rank-16 residual
|
| 38 |
+
# factorization); embeddings/head/norms get its internal AdamW backup.
|
| 39 |
+
# Muon/quantization knobs (muon_*, muonq_*) default to the repo's own
|
| 40 |
+
# llama-60m "muonq" recipe in base_train.py.
|
| 41 |
+
optim = "muonq"
|
| 42 |
+
|
| 43 |
+
# Optimizer settings — lr/wd from MuonQ's llama-60m recipe (Muon takes
|
| 44 |
+
# spectral-norm-scaled steps; do not copy the AdamW 5e-4 here)
|
| 45 |
+
learning_rate = 8e-3
|
| 46 |
+
weight_decay = 0.1
|
| 47 |
+
adam_beta1 = 0.9
|
| 48 |
+
adam_beta2 = 0.95
|
| 49 |
+
max_grad_norm = 1.0
|
| 50 |
+
|
| 51 |
+
# Learning rate scheduler
|
| 52 |
+
lr_scheduler_type = "cosine_with_min_lr" # cosine decay to a non-zero floor — now actually completes in-hour
|
| 53 |
+
lr_scheduler_kwargs = { min_lr_rate = 0.05 } # final LR = 5% of peak
|
| 54 |
+
warmup_steps = 200 # 1-hour series standard (muonq-v2's 400 was a 2-hour recipe)
|
| 55 |
+
|
| 56 |
+
# Precision (choose based on your GPU)
|
| 57 |
+
bf16 = true # bfloat16 autocast + fp32 master weights (the dtype-series safe recipe)
|
| 58 |
+
fp16 = false
|
| 59 |
+
|
| 60 |
+
# Performance
|
| 61 |
+
torch_compile = false
|
| 62 |
+
gradient_checkpointing = false # trade compute for memory (enables larger models)
|
| 63 |
+
neftune_noise_alpha = 0.0 # add noise during training to improve model performance (fine-tuning)
|
| 64 |
+
|
| 65 |
+
# Checkpointing
|
| 66 |
+
# save_strategy can be "steps", "epoch", or "minutes" (wall-clock time).
|
| 67 |
+
save_strategy = "minutes"
|
| 68 |
+
save_steps = 5 # checkpoint every X minutes
|
| 69 |
+
save_total_limit = 3 # keep only X most recent checkpoints
|
| 70 |
+
|
| 71 |
+
# Evaluation
|
| 72 |
+
# eval_strategy can be "steps", "epoch", or "minutes" (wall-clock time).
|
| 73 |
+
eval_strategy = "minutes"
|
| 74 |
+
eval_steps = 5 # evaluate every X minutes
|
| 75 |
+
# Evaluate on a fixed random subset of the validation set (0 = use all)
|
| 76 |
+
max_eval_samples = 5000
|
| 77 |
+
|
| 78 |
+
# Logging
|
| 79 |
+
logging_strategy = "steps"
|
| 80 |
+
logging_steps = 10 # log every 10 steps
|
| 81 |
+
logging_first_step = true # log the first step
|
| 82 |
+
|
| 83 |
+
# Reproducibility
|
| 84 |
+
seed = 42
|
| 85 |
+
|
| 86 |
+
# Performance
|
| 87 |
+
dataloader_num_workers = 2 # number of data loading threads
|
| 88 |
+
dataloader_prefetch_factor = 2 # batches prefetched per worker
|
| 89 |
+
dataloader_pin_memory = true # pin memory for faster GPU transfer
|
scheduler/cosine_760/eval1.log
ADDED
|
@@ -0,0 +1,212 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
W0814 15:07:33.287000 58734 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
|
| 2 |
+
W0814 15:07:33.311000 58734 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
|
| 3 |
+
|
| 4 |
+
==============================================================================
|
| 5 |
+
CHECKPOINT INSPECTION
|
| 6 |
+
==============================================================================
|
| 7 |
+
checkpoint : final
|
| 8 |
+
tokenizer : final
|
| 9 |
+
device : cuda dtype: torch.bfloat16 seed: 1337
|
| 10 |
+
|
| 11 |
+
|
| 12 |
+
==============================================================================
|
| 13 |
+
SUITE: MODEL INFO AND TRAINING LINEAGE
|
| 14 |
+
==============================================================================
|
| 15 |
+
checkpoint : final
|
| 16 |
+
model_type : llama
|
| 17 |
+
architecture : LlamaForCausalLM
|
| 18 |
+
device : cuda
|
| 19 |
+
dtype : torch.bfloat16
|
| 20 |
+
parameters : 77,083,392 (77.08M)
|
| 21 |
+
disk size : 308 MB
|
| 22 |
+
layers / hidden / heads : 8 / 768 / 8 (KV heads: 4)
|
| 23 |
+
vocab (cfg / tokenizer) : 32768 / 32752
|
| 24 |
+
context length : 1024
|
| 25 |
+
chat template : True
|
| 26 |
+
training lineage : base pretraining, ctx 1024
|
| 27 |
+
embedding mean norm : 2.8554 <- typical row length of the embedding matrix; very small or huge values signal trouble
|
| 28 |
+
embedding mean cosine : 0.1174 <- avg cosine between random embedding rows. Descriptive only: it starts near 0.0
|
| 29 |
+
because RANDOM init is near-orthogonal, and it GROWS as the model trains.
|
| 30 |
+
Near 0 therefore does NOT mean "healthy" - it can mean "barely trained".
|
| 31 |
+
(note: config vocab padded by 16 unused rows - harmless)
|
| 32 |
+
|
| 33 |
+
==============================================================================
|
| 34 |
+
SUITE: PERIOD FIDELITY ON FIXED PROBE SENTENCES
|
| 35 |
+
==============================================================================
|
| 36 |
+
Scoring 20 sentences (10 historical, 10 modern)
|
| 37 |
+
The MODERN/HISTORICAL ratio is the output that means something. The absolute
|
| 38 |
+
numbers come from ~20 sentences: too noisy to rank checkpoints (use evaluate2.py).
|
| 39 |
+
|
| 40 |
+
OVERALL (270 tokens)
|
| 41 |
+
perplexity : 266.37 <- 'average branching factor'; lower = less surprised = better fit to this text
|
| 42 |
+
mean token prob : 0.0839 <- average confidence on the actual next word
|
| 43 |
+
median token prob : 0.0081 <- typical confidence (robust to a few very bad tokens)
|
| 44 |
+
worst-token prob (min): 1.7e-08 <- the single most surprising word; tiny values = the model really didn't expect it
|
| 45 |
+
10th-percentile prob : 1.4e-05 <- confidence on the model's worst 10% of guesses
|
| 46 |
+
low-confidence tokens : 52.96% <- share of tokens predicted with <1% probability (red flags)
|
| 47 |
+
mean entropy : 4.642 nats <- how spread out the predictions are; ~0 = very certain, higher = hedging
|
| 48 |
+
|
| 49 |
+
GROUP COMPARISON
|
| 50 |
+
group sentences perplexity bits/byte
|
| 51 |
+
historical 10 80.92 1.196
|
| 52 |
+
modern 10 721.86 1.963
|
| 53 |
+
modern/historical ratio: 8.92 <- >1 means period text is easier for the model than modern text (desired for a vintage model with a ~1900 knowledge cutoff)
|
| 54 |
+
(bits/byte is the tokenizer-independent version - the only column here you may
|
| 55 |
+
compare between models that use DIFFERENT tokenizers.)
|
| 56 |
+
|
| 57 |
+
PER-SENTENCE DETAIL (sorted worst-first)
|
| 58 |
+
[modern ] ppl= 3611.77 minp=6.9e-08 The engineering team rewrote the whole backend micro
|
| 59 |
+
[modern ] ppl= 2068.20 minp=1.6e-06 Robots on the automated assembly line handle the ent
|
| 60 |
+
[modern ] ppl= 1281.63 minp=7.1e-07 Small retail startups now run almost all of their bu
|
| 61 |
+
[modern ] ppl= 1160.26 minp=6.4e-08 Millions of shoppers queue outside the store because
|
| 62 |
+
[modern ] ppl= 1000.82 minp=4.1e-07 Three hours stuck in traffic in the pouring rain mad
|
| 63 |
+
[modern ] ppl= 900.90 minp=6.7e-06 She is finishing a graduate degree in machine learni
|
| 64 |
+
[modern ] ppl= 559.87 minp=1.7e-08 Members of the European Union parliament voted on th
|
| 65 |
+
[modern ] ppl= 353.74 minp=4.0e-07 She gave up her whole weekend to help me move apartm
|
| 66 |
+
[historical] ppl= 233.49 minp=1.7e-05 The keeper fed the great serpent which the naturalis
|
| 67 |
+
[modern ] ppl= 226.90 minp=4.3e-06 After years of hiding it from everyone at work, he t
|
| 68 |
+
[modern ] ppl= 165.54 minp=7.8e-06 I switched to a plant-based diet last year and I no
|
| 69 |
+
[historical] ppl= 128.03 minp=8.2e-05 He drew a distinction so fine and over-scrupulous th
|
| 70 |
+
[historical] ppl= 115.04 minp=1.0e-05 The mountain rose above the valley in a silence sole
|
| 71 |
+
[historical] ppl= 98.38 minp=6.7e-06 He gave up his fortune to the patient study of natur
|
| 72 |
+
[historical] ppl= 88.79 minp=1.2e-04 The merchants of the port have grown rich upon their
|
| 73 |
+
[historical] ppl= 85.65 minp=5.2e-05 The Lord provideth for all his creatures, giving the
|
| 74 |
+
[historical] ppl= 74.17 minp=3.0e-05 The weavers at their looms are employed in the wooll
|
| 75 |
+
[historical] ppl= 66.11 minp=1.4e-05 The labouring poor of this parish are reduced to gre
|
| 76 |
+
[historical] ppl= 38.43 minp=1.1e-04 The ballroom was filled with dancing and laughter, a
|
| 77 |
+
[historical] ppl= 19.84 minp=7.7e-04 Her Majesty was pleased to summon the Lords and Comm
|
| 78 |
+
|
| 79 |
+
==============================================================================
|
| 80 |
+
SUITE: DIACHRONIC WORD-SENSE SEPARATION
|
| 81 |
+
==============================================================================
|
| 82 |
+
Extracting contextual embeddings for 10 shifted words in period AND modern sentences.
|
| 83 |
+
Every probe word sits LAST in its sentence, so the causal model has actually read
|
| 84 |
+
the disambiguating context before it represents the word.
|
| 85 |
+
|
| 86 |
+
SENSE SEPARATION (cosine similarity of the SAME word, period vs modern sentence)
|
| 87 |
+
lower value = the model represents the two senses differently (good sign for a
|
| 88 |
+
period model); ~1.0 = it treats them as the same word regardless of context
|
| 89 |
+
want : +0.297
|
| 90 |
+
python : +0.411
|
| 91 |
+
awful : +0.574
|
| 92 |
+
meat : +0.594
|
| 93 |
+
parliament : +0.609
|
| 94 |
+
nice : +0.677
|
| 95 |
+
manufacture : +0.700
|
| 96 |
+
gay : +0.728
|
| 97 |
+
commerce : +0.773
|
| 98 |
+
science : +0.806
|
| 99 |
+
MEAN : +0.617 <- near 1.0 = model ignores period vs modern usage; lower = senses differ
|
| 100 |
+
|
| 101 |
+
PAIRWISE COSINE SIMILARITY (period sentences, first 5 words)
|
| 102 |
+
these are CONTEXTUAL vectors, so they reflect the sentences as much as the words;
|
| 103 |
+
read the spread, not individual pairs. High mean = representations collapsed.
|
| 104 |
+
gay awful nice meat want
|
| 105 |
+
gay 1.000 0.388 0.376 0.329 0.317
|
| 106 |
+
awful 1.000 0.320 0.196 0.208
|
| 107 |
+
nice 1.000 0.306 0.333
|
| 108 |
+
meat 1.000 0.477
|
| 109 |
+
want 1.000
|
| 110 |
+
all 45 pairs: mean=+0.331 std=0.127 <- healthy spread is good
|
| 111 |
+
|
| 112 |
+
NEAREST NEIGHBOURS inside the probe-word set (period sentences)
|
| 113 |
+
gay : awful (+0.39), nice (+0.38), science (+0.38), commerce (+0.33), meat (+0.33)
|
| 114 |
+
awful : gay (+0.39), nice (+0.32), parliament (+0.26), science (+0.26), commerce (+0.21)
|
| 115 |
+
nice : science (+0.40), gay (+0.38), want (+0.33), awful (+0.32), meat (+0.31)
|
| 116 |
+
meat : want (+0.48), science (+0.47), commerce (+0.46), parliament (+0.44), manufacture (+0.41)
|
| 117 |
+
want : manufacture (+0.49), meat (+0.48), commerce (+0.47), science (+0.44), parliament (+0.36)
|
| 118 |
+
|
| 119 |
+
==============================================================================
|
| 120 |
+
SUITE: GENERATION PROBES
|
| 121 |
+
==============================================================================
|
| 122 |
+
|
| 123 |
+
--- Probe 1 ---
|
| 124 |
+
prompt : The history of the world is
|
| 125 |
+
continuation: too brief for the reader's purpose, and will be a long and weary one to read it.
|
| 126 |
+
|
| 127 |
+
If I am not mistaken, I will write to the Duke of Wellington, who was the son of the Marquis of Lansdowne, a man of high birth, and who was a little
|
| 128 |
+
stats: 60 tokens | distinct-1 0.69 | distinct-2 0.98 | echo 0.08
|
| 129 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 130 |
+
|
| 131 |
+
--- Probe 2 ---
|
| 132 |
+
prompt : What is God? God is
|
| 133 |
+
continuation: a man, a man, a being, a God, a God, a God. God is God. It is God that God is man.
|
| 134 |
+
|
| 135 |
+
God is a man, and God is a God. Man is God, God only is God; God is God. God is God.
|
| 136 |
+
stats: 60 tokens | distinct-1 0.32 | distinct-2 0.65 | echo 0.45
|
| 137 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 138 |
+
|
| 139 |
+
--- Probe 3 ---
|
| 140 |
+
prompt : In the year of our Lord eighteen hundred and
|
| 141 |
+
continuation: eighty-four in the year of our Lord one hundred and eight in the year of our Lord one thousand nine hundred and ninety-nine in the year of our Lord one thousand three hundred and eighteen in the year of our Lord one thousand seven hundred and eighteen in the year of our Lord one thousand seven
|
| 142 |
+
stats: 60 tokens | distinct-1 0.30 | distinct-2 0.40 | echo 0.00
|
| 143 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 144 |
+
|
| 145 |
+
--- Probe 4 ---
|
| 146 |
+
prompt : The manufacture of cotton
|
| 147 |
+
continuation: -wood, and the manufacture of silk and linen, are very inferior, especially those of silk, from the silk-wood and cotton, which are of great value, and are manufactured in the United States, and the manufacture of silk, and the manufacture of silk, silk, and silk, is
|
| 148 |
+
stats: 60 tokens | distinct-1 0.52 | distinct-2 0.80 | echo 0.07
|
| 149 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 150 |
+
|
| 151 |
+
--- Probe 5 ---
|
| 152 |
+
prompt : The steam engine
|
| 153 |
+
continuation: is the main engine, and the engine is the main engine. A large cylinder is the largest engine, and is operated on the engine and the engine, which are the most important engines.
|
| 154 |
+
|
| 155 |
+
The engine is the engine of the engine, and the engine is the engine of the engine. The
|
| 156 |
+
stats: 60 tokens | distinct-1 0.39 | distinct-2 0.64 | echo 0.20
|
| 157 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 158 |
+
|
| 159 |
+
--- Probe 6 ---
|
| 160 |
+
prompt : The telegraph
|
| 161 |
+
continuation: ic and telegraphic telegraphic lines; and, in connection with the following letters, we are pleased to receive the same kind of news from the United States, which, after a short interval, will appear as the news of our departure.
|
| 162 |
+
|
| 163 |
+
We also have the pleasure of communicating the following news
|
| 164 |
+
stats: 60 tokens | distinct-1 0.78 | distinct-2 0.98 | echo 0.04
|
| 165 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 166 |
+
|
| 167 |
+
--- Probe 7 ---
|
| 168 |
+
prompt : My dearest sister,
|
| 169 |
+
continuation: I am not able to speak of myself as a friend, and I know no more about my brother's life than of my brother.
|
| 170 |
+
|
| 171 |
+
I do not know, my dear Mrs. Fothergill, how to live with you. I am so much delighted with you, and so much
|
| 172 |
+
stats: 60 tokens | distinct-1 0.72 | distinct-2 0.96 | echo 0.00
|
| 173 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 174 |
+
|
| 175 |
+
--- Probe 8 ---
|
| 176 |
+
prompt : Greetings, my friend
|
| 177 |
+
continuation: , and the others, to make a public profession of the Christian faith.
|
| 178 |
+
|
| 179 |
+
I have not the remotest idea of the extent of the benefits of our religion. In the first place, I believe in the fact of the Divine presence, as well as in the second, in the third person.
|
| 180 |
+
stats: 60 tokens | distinct-1 0.67 | distinct-2 0.90 | echo 0.16
|
| 181 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 182 |
+
|
| 183 |
+
--- Probe 9 ---
|
| 184 |
+
prompt : Her Majesty the Queen
|
| 185 |
+
continuation: of
|
| 186 |
+
|
| 187 |
+
Bristol. Her Majesty's Subjects, her Majesty's Subjects, and the Government of the United States, are to be sent to this Court, to be sent to them by the
|
| 188 |
+
|
| 189 |
+
King of France, and to all the Provinces of the dominions of the Great Powers of
|
| 190 |
+
stats: 60 tokens | distinct-1 0.58 | distinct-2 0.86 | echo 0.22
|
| 191 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 192 |
+
|
| 193 |
+
--- Probe 10 ---
|
| 194 |
+
prompt : Chapter I.
|
| 195 |
+
continuation: The reader will be pleased to know that I am in a condition to take the liberty of addressing you to the same effect which has been made on the
|
| 196 |
+
|
| 197 |
+
American System of Political Economy. If this doctrine should be so generally understood, it is a great evil, and, therefore, a very
|
| 198 |
+
stats: 60 tokens | distinct-1 0.85 | distinct-2 1.00 | echo 0.00
|
| 199 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 200 |
+
|
| 201 |
+
AGGREGATE over 10 probes: 600 tokens in 0.9s (667.0 tok/s)
|
| 202 |
+
mean distinct-1 0.582 | mean distinct-2 0.817 | mean echo 0.122 | worst probe: distinct-2 0.400 / echo 0.455
|
| 203 |
+
NOTE: at least one probe looped badly even though the average looks fine - read the probes above.
|
| 204 |
+
(sampling hides loops that greedy decoding exposes; evaluate2.py measures the greedy loop length.)
|
| 205 |
+
|
| 206 |
+
Full numeric results written to /home/cro/Dev/vintage-LLM/training/eval_results/evaluate-final.json
|
| 207 |
+
|
| 208 |
+
==============================================================================
|
| 209 |
+
INSPECTION COMPLETE
|
| 210 |
+
==============================================================================
|
| 211 |
+
This covered WHAT the checkpoint is, its lineage, and whether it is still vintage.
|
| 212 |
+
For "is it baked enough", run evaluate2.py on the same path.
|
scheduler/cosine_760/eval2.log
ADDED
|
@@ -0,0 +1,54 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
evaluating 1 checkpoint(s) on cuda (torch.bfloat16)
|
| 2 |
+
tokenizer: final
|
| 3 |
+
final ...W0814 15:07:39.942000 58837 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
|
| 4 |
+
W0814 15:07:39.958000 58837 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
|
| 5 |
+
|
| 6 |
+
bake 52/100 bpb 1.2775 logic 0.65 (15.3s)
|
| 7 |
+
|
| 8 |
+
report: eval2.md
|
| 9 |
+
json: eval2.json
|
| 10 |
+
|
| 11 |
+
# Evaluation: final
|
| 12 |
+
|
| 13 |
+
- **Parameters:** 77M
|
| 14 |
+
- **Note:** no trainer_state.json in the checkpoint - tokens seen unknown
|
| 15 |
+
|
| 16 |
+
## Verdict: DOUGH (bake score 52/100)
|
| 17 |
+
|
| 18 |
+
Structure is forming - real words, some grammar - but this is not a usable language model yet. It needs several times more training tokens.
|
| 19 |
+
|
| 20 |
+
## Where it sits (held-out bits/byte, lower = better)
|
| 21 |
+
|
| 22 |
+
```
|
| 23 |
+
3.50 untrained model (uniform noise)
|
| 24 |
+
2.00 word-salad
|
| 25 |
+
1.50 broken prose
|
| 26 |
+
1.33 500M model, ~0.7B tokens (early training)
|
| 27 |
+
--> 1.278 YOUR MODEL
|
| 28 |
+
1.19 500M model, ~4.7B tokens (undertrained but solid)
|
| 29 |
+
1.10 best sub-1B measured on this data (341M)
|
| 30 |
+
0.95 estimated sub-1B ceiling on this corpus
|
| 31 |
+
```
|
| 32 |
+
|
| 33 |
+
## Scores
|
| 34 |
+
|
| 35 |
+
| what | raw value | points /100 | weight | plain English |
|
| 36 |
+
|---|---|---|---|---|
|
| 37 |
+
| Held-out loss | 1.2775 bits/byte | 61 | 0.5 | how cheaply it predicts period text it never saw — the single best training signal |
|
| 38 |
+
| Logic | 0.650 acc, margin 0.040 | 36 | 0.25 | picks the *sensible* continuation over matched nonsense; 0.50 = coin-flip. normal for a good sub-1B model (Vintage1 scores 0.70-0.78) |
|
| 39 |
+
| Chat readiness | 1.1623 bits/byte | 55 | 0.15 | how cheap well-formed period dialogue already is — predicts fine-tuning ease |
|
| 40 |
+
| Hygiene | loop 66.2w, punct 0.00/100w | 42 | 0.1 | greedy-decoding loop length and broken punctuation |
|
| 41 |
+
|
| 42 |
+
## Period boundary
|
| 43 |
+
|
| 44 |
+
Clean. All 6 post-1900 trap words cost the model more than their period twins (mean shock +1.41 bits/byte, weakest pair 'atomic bomb' at +0.93). No sign of modern text in training.
|
| 45 |
+
|
| 46 |
+
## See for yourself (sampled, t=0.8)
|
| 47 |
+
|
| 48 |
+
> **LONDON, Tuesday. — The committee appointed to inquire into the condition of the** United States in the matter of the payment of dues of the United States government for the term of three years is hereby instructed to inquire into the manner of the sale of the money of the United States government in the United States. The Secretary of the Treasury has the right to make such application to the Treasurer of the United States as may seem best adapted for the purpose in view of the
|
| 49 |
+
>
|
| 50 |
+
> **A melancholy accident occurred on Thursday last at the works of Messrs. Harding and** Co. The above-mentioned event occurred on the morning of Monday the 8th instant. The ship was to proceed to the port of Pekin in the port of Pekin, where the Cruelon was to proceed. The ship was towed into the water, and the ship to land. The ship was to be delivered up to the captain of the ship for the purpose of going to a private ship. The British fleet was to sail, the remainder of the crew b
|
| 51 |
+
>
|
| 52 |
+
> **Brethren, the text which I have chosen for our consideration this morning is taken from** the chapter on the New Testament. We have not yet done anything for the Christian world. Our church is to be considered the best of the world, and our most important work is to do the good of the world. We have a right to demand the aid of other men, who would fain take up our abode in the world. We have the same right to demand a church, if it is not right, because a church is to be found outside
|
| 53 |
+
>
|
| 54 |
+
|
scheduler/cosine_760/final/chat_template.jinja
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{%- for message in messages -%}
|
| 2 |
+
{{- bos_token + '<|' + message['role'] + '|>\n' + message['content'].strip() + eos_token + '\n' -}}
|
| 3 |
+
{%- endfor -%}
|
| 4 |
+
|
| 5 |
+
{%- if add_generation_prompt -%}
|
| 6 |
+
{{- bos_token + '<|assistant|>\n' -}}
|
| 7 |
+
{%- endif -%}
|
scheduler/cosine_760/final/config.json
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"LlamaForCausalLM"
|
| 4 |
+
],
|
| 5 |
+
"attention_bias": false,
|
| 6 |
+
"attention_dropout": 0.0,
|
| 7 |
+
"bos_token_id": 3,
|
| 8 |
+
"dtype": "float32",
|
| 9 |
+
"eos_token_id": 4,
|
| 10 |
+
"head_dim": 96,
|
| 11 |
+
"hidden_act": "silu",
|
| 12 |
+
"hidden_size": 768,
|
| 13 |
+
"initializer_range": 0.02,
|
| 14 |
+
"intermediate_size": 2048,
|
| 15 |
+
"max_position_embeddings": 1024,
|
| 16 |
+
"mlp_bias": false,
|
| 17 |
+
"model_type": "llama",
|
| 18 |
+
"num_attention_heads": 8,
|
| 19 |
+
"num_hidden_layers": 8,
|
| 20 |
+
"num_key_value_heads": 4,
|
| 21 |
+
"pad_token_id": 0,
|
| 22 |
+
"pretraining_tp": 1,
|
| 23 |
+
"rms_norm_eps": 1e-06,
|
| 24 |
+
"rope_parameters": {
|
| 25 |
+
"rope_theta": 10000.0,
|
| 26 |
+
"rope_type": "default"
|
| 27 |
+
},
|
| 28 |
+
"tie_word_embeddings": true,
|
| 29 |
+
"transformers_version": "5.14.1",
|
| 30 |
+
"use_cache": false,
|
| 31 |
+
"vocab_size": 32768
|
| 32 |
+
}
|
scheduler/cosine_760/final/generation_config.json
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_from_model_config": true,
|
| 3 |
+
"bos_token_id": 3,
|
| 4 |
+
"eos_token_id": [
|
| 5 |
+
4,
|
| 6 |
+
2
|
| 7 |
+
],
|
| 8 |
+
"output_attentions": false,
|
| 9 |
+
"output_hidden_states": false,
|
| 10 |
+
"pad_token_id": 0,
|
| 11 |
+
"transformers_version": "5.14.1",
|
| 12 |
+
"use_cache": false
|
| 13 |
+
}
|
scheduler/cosine_760/final/model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bc693836047b8f4a3c04f80222a56f8b7bb924077e204df865bb366f824c64fe
|
| 3 |
+
size 308341784
|
scheduler/cosine_760/final/tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
scheduler/cosine_760/final/tokenizer_config.json
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"backend": "tokenizers",
|
| 3 |
+
"bos_token": "<|bos|>",
|
| 4 |
+
"eos_token": "<|eos|>",
|
| 5 |
+
"is_local": true,
|
| 6 |
+
"local_files_only": false,
|
| 7 |
+
"model_max_length": 1000000000000000019884624838656,
|
| 8 |
+
"pad_token": "<|pad|>",
|
| 9 |
+
"tokenizer_class": "TokenizersBackend",
|
| 10 |
+
"unk_token": "<|unk|>"
|
| 11 |
+
}
|
scheduler/cosine_760/trainer_state.json
ADDED
|
@@ -0,0 +1,663 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_global_step": null,
|
| 3 |
+
"best_metric": null,
|
| 4 |
+
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 0.01006518436771828,
|
| 6 |
+
"eval_steps": 5,
|
| 7 |
+
"global_step": 757,
|
| 8 |
+
"is_hyper_param_search": false,
|
| 9 |
+
"is_local_process_zero": true,
|
| 10 |
+
"is_world_process_zero": true,
|
| 11 |
+
"log_history": [
|
| 12 |
+
{
|
| 13 |
+
"epoch": 1.3296148438201165e-05,
|
| 14 |
+
"grad_norm": 5.513442516326904,
|
| 15 |
+
"learning_rate": 0.0,
|
| 16 |
+
"loss": 10.538199424743652,
|
| 17 |
+
"step": 1
|
| 18 |
+
},
|
| 19 |
+
{
|
| 20 |
+
"epoch": 0.00013296148438201165,
|
| 21 |
+
"grad_norm": 5.268533706665039,
|
| 22 |
+
"learning_rate": 0.00035999999999999997,
|
| 23 |
+
"loss": 10.136458502875435,
|
| 24 |
+
"step": 10
|
| 25 |
+
},
|
| 26 |
+
{
|
| 27 |
+
"epoch": 0.0002659229687640233,
|
| 28 |
+
"grad_norm": 1.034209132194519,
|
| 29 |
+
"learning_rate": 0.00076,
|
| 30 |
+
"loss": 8.0193359375,
|
| 31 |
+
"step": 20
|
| 32 |
+
},
|
| 33 |
+
{
|
| 34 |
+
"epoch": 0.0003988844531460349,
|
| 35 |
+
"grad_norm": 9.450255393981934,
|
| 36 |
+
"learning_rate": 0.00116,
|
| 37 |
+
"loss": 6.9860687255859375,
|
| 38 |
+
"step": 30
|
| 39 |
+
},
|
| 40 |
+
{
|
| 41 |
+
"epoch": 0.0005318459375280466,
|
| 42 |
+
"grad_norm": 5.685102462768555,
|
| 43 |
+
"learning_rate": 0.0015600000000000002,
|
| 44 |
+
"loss": 6.707894897460937,
|
| 45 |
+
"step": 40
|
| 46 |
+
},
|
| 47 |
+
{
|
| 48 |
+
"epoch": 0.0006648074219100582,
|
| 49 |
+
"grad_norm": 2.160360097885132,
|
| 50 |
+
"learning_rate": 0.00196,
|
| 51 |
+
"loss": 6.287691116333008,
|
| 52 |
+
"step": 50
|
| 53 |
+
},
|
| 54 |
+
{
|
| 55 |
+
"epoch": 0.0007977689062920698,
|
| 56 |
+
"grad_norm": 1.6388428211212158,
|
| 57 |
+
"learning_rate": 0.00236,
|
| 58 |
+
"loss": 5.944025039672852,
|
| 59 |
+
"step": 60
|
| 60 |
+
},
|
| 61 |
+
{
|
| 62 |
+
"epoch": 0.000890841945359478,
|
| 63 |
+
"eval_loss": 5.704253673553467,
|
| 64 |
+
"eval_runtime": 24.3899,
|
| 65 |
+
"eval_samples_per_second": 205.003,
|
| 66 |
+
"eval_steps_per_second": 25.625,
|
| 67 |
+
"step": 67
|
| 68 |
+
},
|
| 69 |
+
{
|
| 70 |
+
"epoch": 0.0009307303906740815,
|
| 71 |
+
"grad_norm": 1.2074735164642334,
|
| 72 |
+
"learning_rate": 0.00276,
|
| 73 |
+
"loss": 5.747131729125977,
|
| 74 |
+
"step": 70
|
| 75 |
+
},
|
| 76 |
+
{
|
| 77 |
+
"epoch": 0.0010636918750560932,
|
| 78 |
+
"grad_norm": 1.132364273071289,
|
| 79 |
+
"learning_rate": 0.00316,
|
| 80 |
+
"loss": 5.60839729309082,
|
| 81 |
+
"step": 80
|
| 82 |
+
},
|
| 83 |
+
{
|
| 84 |
+
"epoch": 0.0011966533594381047,
|
| 85 |
+
"grad_norm": 1.0884277820587158,
|
| 86 |
+
"learning_rate": 0.0035600000000000002,
|
| 87 |
+
"loss": 5.502901077270508,
|
| 88 |
+
"step": 90
|
| 89 |
+
},
|
| 90 |
+
{
|
| 91 |
+
"epoch": 0.0013296148438201164,
|
| 92 |
+
"grad_norm": 1.3878921270370483,
|
| 93 |
+
"learning_rate": 0.00396,
|
| 94 |
+
"loss": 5.371665573120117,
|
| 95 |
+
"step": 100
|
| 96 |
+
},
|
| 97 |
+
{
|
| 98 |
+
"epoch": 0.001462576328202128,
|
| 99 |
+
"grad_norm": 1.2677963972091675,
|
| 100 |
+
"learning_rate": 0.00436,
|
| 101 |
+
"loss": 5.2512672424316404,
|
| 102 |
+
"step": 110
|
| 103 |
+
},
|
| 104 |
+
{
|
| 105 |
+
"epoch": 0.0015955378125841397,
|
| 106 |
+
"grad_norm": 0.9181460738182068,
|
| 107 |
+
"learning_rate": 0.0047599999999999995,
|
| 108 |
+
"loss": 5.126389694213867,
|
| 109 |
+
"step": 120
|
| 110 |
+
},
|
| 111 |
+
{
|
| 112 |
+
"epoch": 0.0017152031485279501,
|
| 113 |
+
"eval_loss": 4.9558868408203125,
|
| 114 |
+
"eval_runtime": 23.6019,
|
| 115 |
+
"eval_samples_per_second": 211.847,
|
| 116 |
+
"eval_steps_per_second": 26.481,
|
| 117 |
+
"step": 129
|
| 118 |
+
},
|
| 119 |
+
{
|
| 120 |
+
"epoch": 0.0017284992969661514,
|
| 121 |
+
"grad_norm": 0.7834004163742065,
|
| 122 |
+
"learning_rate": 0.0051600000000000005,
|
| 123 |
+
"loss": 5.010536575317383,
|
| 124 |
+
"step": 130
|
| 125 |
+
},
|
| 126 |
+
{
|
| 127 |
+
"epoch": 0.001861460781348163,
|
| 128 |
+
"grad_norm": 0.7623927593231201,
|
| 129 |
+
"learning_rate": 0.00556,
|
| 130 |
+
"loss": 4.908084106445313,
|
| 131 |
+
"step": 140
|
| 132 |
+
},
|
| 133 |
+
{
|
| 134 |
+
"epoch": 0.0019944222657301744,
|
| 135 |
+
"grad_norm": 0.5423034429550171,
|
| 136 |
+
"learning_rate": 0.00596,
|
| 137 |
+
"loss": 4.820485305786133,
|
| 138 |
+
"step": 150
|
| 139 |
+
},
|
| 140 |
+
{
|
| 141 |
+
"epoch": 0.0021273837501121864,
|
| 142 |
+
"grad_norm": 0.5046894550323486,
|
| 143 |
+
"learning_rate": 0.00636,
|
| 144 |
+
"loss": 4.753760528564453,
|
| 145 |
+
"step": 160
|
| 146 |
+
},
|
| 147 |
+
{
|
| 148 |
+
"epoch": 0.002260345234494198,
|
| 149 |
+
"grad_norm": 0.3807763159275055,
|
| 150 |
+
"learning_rate": 0.0067599999999999995,
|
| 151 |
+
"loss": 4.694861602783203,
|
| 152 |
+
"step": 170
|
| 153 |
+
},
|
| 154 |
+
{
|
| 155 |
+
"epoch": 0.0023933067188762094,
|
| 156 |
+
"grad_norm": 0.37175047397613525,
|
| 157 |
+
"learning_rate": 0.0071600000000000006,
|
| 158 |
+
"loss": 4.652392578125,
|
| 159 |
+
"step": 180
|
| 160 |
+
},
|
| 161 |
+
{
|
| 162 |
+
"epoch": 0.0025262682032582214,
|
| 163 |
+
"grad_norm": 0.3488306999206543,
|
| 164 |
+
"learning_rate": 0.00756,
|
| 165 |
+
"loss": 4.597007369995117,
|
| 166 |
+
"step": 190
|
| 167 |
+
},
|
| 168 |
+
{
|
| 169 |
+
"epoch": 0.0025528605001346235,
|
| 170 |
+
"eval_loss": 4.578042030334473,
|
| 171 |
+
"eval_runtime": 23.0783,
|
| 172 |
+
"eval_samples_per_second": 216.654,
|
| 173 |
+
"eval_steps_per_second": 27.082,
|
| 174 |
+
"step": 192
|
| 175 |
+
},
|
| 176 |
+
{
|
| 177 |
+
"epoch": 0.002659229687640233,
|
| 178 |
+
"grad_norm": 0.3351574242115021,
|
| 179 |
+
"learning_rate": 0.00796,
|
| 180 |
+
"loss": 4.566041946411133,
|
| 181 |
+
"step": 200
|
| 182 |
+
},
|
| 183 |
+
{
|
| 184 |
+
"epoch": 0.0027921911720222444,
|
| 185 |
+
"grad_norm": 0.32959529757499695,
|
| 186 |
+
"learning_rate": 0.007995157495314521,
|
| 187 |
+
"loss": 4.521368408203125,
|
| 188 |
+
"step": 210
|
| 189 |
+
},
|
| 190 |
+
{
|
| 191 |
+
"epoch": 0.002925152656404256,
|
| 192 |
+
"grad_norm": 0.26094311475753784,
|
| 193 |
+
"learning_rate": 0.007978433817629256,
|
| 194 |
+
"loss": 4.468058013916016,
|
| 195 |
+
"step": 220
|
| 196 |
+
},
|
| 197 |
+
{
|
| 198 |
+
"epoch": 0.003058114140786268,
|
| 199 |
+
"grad_norm": 0.24577029049396515,
|
| 200 |
+
"learning_rate": 0.007949821789267853,
|
| 201 |
+
"loss": 4.430326843261719,
|
| 202 |
+
"step": 230
|
| 203 |
+
},
|
| 204 |
+
{
|
| 205 |
+
"epoch": 0.0031910756251682794,
|
| 206 |
+
"grad_norm": 0.2677822709083557,
|
| 207 |
+
"learning_rate": 0.007909411434256969,
|
| 208 |
+
"loss": 4.39563102722168,
|
| 209 |
+
"step": 240
|
| 210 |
+
},
|
| 211 |
+
{
|
| 212 |
+
"epoch": 0.003324037109550291,
|
| 213 |
+
"grad_norm": 0.27217116951942444,
|
| 214 |
+
"learning_rate": 0.00785732989852386,
|
| 215 |
+
"loss": 4.363616561889648,
|
| 216 |
+
"step": 250
|
| 217 |
+
},
|
| 218 |
+
{
|
| 219 |
+
"epoch": 0.003390517851741297,
|
| 220 |
+
"eval_loss": 4.3263163566589355,
|
| 221 |
+
"eval_runtime": 22.8668,
|
| 222 |
+
"eval_samples_per_second": 218.657,
|
| 223 |
+
"eval_steps_per_second": 27.332,
|
| 224 |
+
"step": 255
|
| 225 |
+
},
|
| 226 |
+
{
|
| 227 |
+
"epoch": 0.003456998593932303,
|
| 228 |
+
"grad_norm": 0.22698307037353516,
|
| 229 |
+
"learning_rate": 0.007793741049848255,
|
| 230 |
+
"loss": 4.329174041748047,
|
| 231 |
+
"step": 260
|
| 232 |
+
},
|
| 233 |
+
{
|
| 234 |
+
"epoch": 0.0035899600783143143,
|
| 235 |
+
"grad_norm": 0.2033531218767166,
|
| 236 |
+
"learning_rate": 0.007718844962273695,
|
| 237 |
+
"loss": 4.300073623657227,
|
| 238 |
+
"step": 270
|
| 239 |
+
},
|
| 240 |
+
{
|
| 241 |
+
"epoch": 0.003722921562696326,
|
| 242 |
+
"grad_norm": 0.18403764069080353,
|
| 243 |
+
"learning_rate": 0.007632877286600573,
|
| 244 |
+
"loss": 4.262511444091797,
|
| 245 |
+
"step": 280
|
| 246 |
+
},
|
| 247 |
+
{
|
| 248 |
+
"epoch": 0.0038558830470783374,
|
| 249 |
+
"grad_norm": 0.23616866767406464,
|
| 250 |
+
"learning_rate": 0.007536108508941533,
|
| 251 |
+
"loss": 4.247257232666016,
|
| 252 |
+
"step": 290
|
| 253 |
+
},
|
| 254 |
+
{
|
| 255 |
+
"epoch": 0.003988844531460349,
|
| 256 |
+
"grad_norm": 0.2175689935684204,
|
| 257 |
+
"learning_rate": 0.007428843099672093,
|
| 258 |
+
"loss": 4.220326995849609,
|
| 259 |
+
"step": 300
|
| 260 |
+
},
|
| 261 |
+
{
|
| 262 |
+
"epoch": 0.004121806015842361,
|
| 263 |
+
"grad_norm": 0.16665969789028168,
|
| 264 |
+
"learning_rate": 0.007311418555454189,
|
| 265 |
+
"loss": 4.190158462524414,
|
| 266 |
+
"step": 310
|
| 267 |
+
},
|
| 268 |
+
{
|
| 269 |
+
"epoch": 0.00422817520334797,
|
| 270 |
+
"eval_loss": 4.1631269454956055,
|
| 271 |
+
"eval_runtime": 22.8668,
|
| 272 |
+
"eval_samples_per_second": 218.658,
|
| 273 |
+
"eval_steps_per_second": 27.332,
|
| 274 |
+
"step": 318
|
| 275 |
+
},
|
| 276 |
+
{
|
| 277 |
+
"epoch": 0.004254767500224373,
|
| 278 |
+
"grad_norm": 0.183854341506958,
|
| 279 |
+
"learning_rate": 0.007184204337346831,
|
| 280 |
+
"loss": 4.169167709350586,
|
| 281 |
+
"step": 320
|
| 282 |
+
},
|
| 283 |
+
{
|
| 284 |
+
"epoch": 0.004387728984606384,
|
| 285 |
+
"grad_norm": 0.15049099922180176,
|
| 286 |
+
"learning_rate": 0.007047600708344912,
|
| 287 |
+
"loss": 4.153661727905273,
|
| 288 |
+
"step": 330
|
| 289 |
+
},
|
| 290 |
+
{
|
| 291 |
+
"epoch": 0.004520690468988396,
|
| 292 |
+
"grad_norm": 0.18056534230709076,
|
| 293 |
+
"learning_rate": 0.006902037474003746,
|
| 294 |
+
"loss": 4.125600051879883,
|
| 295 |
+
"step": 340
|
| 296 |
+
},
|
| 297 |
+
{
|
| 298 |
+
"epoch": 0.004653651953370408,
|
| 299 |
+
"grad_norm": 0.18494488298892975,
|
| 300 |
+
"learning_rate": 0.00674797263011178,
|
| 301 |
+
"loss": 4.109599685668945,
|
| 302 |
+
"step": 350
|
| 303 |
+
},
|
| 304 |
+
{
|
| 305 |
+
"epoch": 0.004786613437752419,
|
| 306 |
+
"grad_norm": 0.14171533286571503,
|
| 307 |
+
"learning_rate": 0.006585890921666403,
|
| 308 |
+
"loss": 4.102255249023438,
|
| 309 |
+
"step": 360
|
| 310 |
+
},
|
| 311 |
+
{
|
| 312 |
+
"epoch": 0.004919574922134431,
|
| 313 |
+
"grad_norm": 0.14613653719425201,
|
| 314 |
+
"learning_rate": 0.00641630231768684,
|
| 315 |
+
"loss": 4.084948348999023,
|
| 316 |
+
"step": 370
|
| 317 |
+
},
|
| 318 |
+
{
|
| 319 |
+
"epoch": 0.005052536406516443,
|
| 320 |
+
"grad_norm": 0.13041161000728607,
|
| 321 |
+
"learning_rate": 0.0062397404066629495,
|
| 322 |
+
"loss": 4.060393905639648,
|
| 323 |
+
"step": 380
|
| 324 |
+
},
|
| 325 |
+
{
|
| 326 |
+
"epoch": 0.005065832554954643,
|
| 327 |
+
"eval_loss": 4.050504684448242,
|
| 328 |
+
"eval_runtime": 22.8575,
|
| 329 |
+
"eval_samples_per_second": 218.746,
|
| 330 |
+
"eval_steps_per_second": 27.343,
|
| 331 |
+
"step": 381
|
| 332 |
+
},
|
| 333 |
+
{
|
| 334 |
+
"epoch": 0.005185497890898454,
|
| 335 |
+
"grad_norm": 0.14684684574604034,
|
| 336 |
+
"learning_rate": 0.006056760717688429,
|
| 337 |
+
"loss": 4.043218612670898,
|
| 338 |
+
"step": 390
|
| 339 |
+
},
|
| 340 |
+
{
|
| 341 |
+
"epoch": 0.005318459375280466,
|
| 342 |
+
"grad_norm": 0.1349787712097168,
|
| 343 |
+
"learning_rate": 0.005867938972560758,
|
| 344 |
+
"loss": 4.035343933105469,
|
| 345 |
+
"step": 400
|
| 346 |
+
},
|
| 347 |
+
{
|
| 348 |
+
"epoch": 0.005451420859662477,
|
| 349 |
+
"grad_norm": 0.11209040135145187,
|
| 350 |
+
"learning_rate": 0.005673869274347425,
|
| 351 |
+
"loss": 4.007895278930664,
|
| 352 |
+
"step": 410
|
| 353 |
+
},
|
| 354 |
+
{
|
| 355 |
+
"epoch": 0.005584382344044489,
|
| 356 |
+
"grad_norm": 0.13373538851737976,
|
| 357 |
+
"learning_rate": 0.00547516223811788,
|
| 358 |
+
"loss": 3.9965129852294923,
|
| 359 |
+
"step": 420
|
| 360 |
+
},
|
| 361 |
+
{
|
| 362 |
+
"epoch": 0.005717343828426501,
|
| 363 |
+
"grad_norm": 0.12187955528497696,
|
| 364 |
+
"learning_rate": 0.005272443069722628,
|
| 365 |
+
"loss": 3.9868213653564455,
|
| 366 |
+
"step": 430
|
| 367 |
+
},
|
| 368 |
+
{
|
| 369 |
+
"epoch": 0.005850305312808512,
|
| 370 |
+
"grad_norm": 0.1229710504412651,
|
| 371 |
+
"learning_rate": 0.005066349598664307,
|
| 372 |
+
"loss": 3.9734012603759767,
|
| 373 |
+
"step": 440
|
| 374 |
+
},
|
| 375 |
+
{
|
| 376 |
+
"epoch": 0.005903489906561317,
|
| 377 |
+
"eval_loss": 3.9610915184020996,
|
| 378 |
+
"eval_runtime": 22.8662,
|
| 379 |
+
"eval_samples_per_second": 218.663,
|
| 380 |
+
"eval_steps_per_second": 27.333,
|
| 381 |
+
"step": 444
|
| 382 |
+
},
|
| 383 |
+
{
|
| 384 |
+
"epoch": 0.005983266797190524,
|
| 385 |
+
"grad_norm": 0.12179016321897507,
|
| 386 |
+
"learning_rate": 0.004857530271250106,
|
| 387 |
+
"loss": 3.962753677368164,
|
| 388 |
+
"step": 450
|
| 389 |
+
},
|
| 390 |
+
{
|
| 391 |
+
"epoch": 0.006116228281572536,
|
| 392 |
+
"grad_norm": 0.11934109032154083,
|
| 393 |
+
"learning_rate": 0.004646642110339784,
|
| 394 |
+
"loss": 3.939118576049805,
|
| 395 |
+
"step": 460
|
| 396 |
+
},
|
| 397 |
+
{
|
| 398 |
+
"epoch": 0.006249189765954547,
|
| 399 |
+
"grad_norm": 0.11726455390453339,
|
| 400 |
+
"learning_rate": 0.004434348648108665,
|
| 401 |
+
"loss": 3.93060188293457,
|
| 402 |
+
"step": 470
|
| 403 |
+
},
|
| 404 |
+
{
|
| 405 |
+
"epoch": 0.006382151250336559,
|
| 406 |
+
"grad_norm": 0.09656476974487305,
|
| 407 |
+
"learning_rate": 0.0042213178383299485,
|
| 408 |
+
"loss": 3.921614074707031,
|
| 409 |
+
"step": 480
|
| 410 |
+
},
|
| 411 |
+
{
|
| 412 |
+
"epoch": 0.006515112734718571,
|
| 413 |
+
"grad_norm": 0.09665072709321976,
|
| 414 |
+
"learning_rate": 0.004008219954745005,
|
| 415 |
+
"loss": 3.910228729248047,
|
| 416 |
+
"step": 490
|
| 417 |
+
},
|
| 418 |
+
{
|
| 419 |
+
"epoch": 0.006648074219100582,
|
| 420 |
+
"grad_norm": 0.09524396061897278,
|
| 421 |
+
"learning_rate": 0.0037957254821342165,
|
| 422 |
+
"loss": 3.9034732818603515,
|
| 423 |
+
"step": 500
|
| 424 |
+
},
|
| 425 |
+
{
|
| 426 |
+
"epoch": 0.00674114725816799,
|
| 427 |
+
"eval_loss": 3.8828694820404053,
|
| 428 |
+
"eval_runtime": 22.8816,
|
| 429 |
+
"eval_samples_per_second": 218.517,
|
| 430 |
+
"eval_steps_per_second": 27.315,
|
| 431 |
+
"step": 507
|
| 432 |
+
},
|
| 433 |
+
{
|
| 434 |
+
"epoch": 0.006781035703482594,
|
| 435 |
+
"grad_norm": 0.09537574648857117,
|
| 436 |
+
"learning_rate": 0.0035845030067237996,
|
| 437 |
+
"loss": 3.8968971252441404,
|
| 438 |
+
"step": 510
|
| 439 |
+
},
|
| 440 |
+
{
|
| 441 |
+
"epoch": 0.006913997187864606,
|
| 442 |
+
"grad_norm": 0.09708276391029358,
|
| 443 |
+
"learning_rate": 0.003375217112566186,
|
| 444 |
+
"loss": 3.87552490234375,
|
| 445 |
+
"step": 520
|
| 446 |
+
},
|
| 447 |
+
{
|
| 448 |
+
"epoch": 0.007046958672246617,
|
| 449 |
+
"grad_norm": 0.10611076653003693,
|
| 450 |
+
"learning_rate": 0.003168526290512718,
|
| 451 |
+
"loss": 3.861489105224609,
|
| 452 |
+
"step": 530
|
| 453 |
+
},
|
| 454 |
+
{
|
| 455 |
+
"epoch": 0.007179920156628629,
|
| 456 |
+
"grad_norm": 0.09043016284704208,
|
| 457 |
+
"learning_rate": 0.002965080866357815,
|
| 458 |
+
"loss": 3.8511241912841796,
|
| 459 |
+
"step": 540
|
| 460 |
+
},
|
| 461 |
+
{
|
| 462 |
+
"epoch": 0.007312881641010641,
|
| 463 |
+
"grad_norm": 0.08922528475522995,
|
| 464 |
+
"learning_rate": 0.002765520954673425,
|
| 465 |
+
"loss": 3.846691131591797,
|
| 466 |
+
"step": 550
|
| 467 |
+
},
|
| 468 |
+
{
|
| 469 |
+
"epoch": 0.007445843125392652,
|
| 470 |
+
"grad_norm": 0.08538611978292465,
|
| 471 |
+
"learning_rate": 0.002570474444771763,
|
| 472 |
+
"loss": 3.846059799194336,
|
| 473 |
+
"step": 560
|
| 474 |
+
},
|
| 475 |
+
{
|
| 476 |
+
"epoch": 0.007578804609774664,
|
| 477 |
+
"grad_norm": 0.09066513925790787,
|
| 478 |
+
"learning_rate": 0.0023805550251332418,
|
| 479 |
+
"loss": 3.826742172241211,
|
| 480 |
+
"step": 570
|
| 481 |
+
},
|
| 482 |
+
{
|
| 483 |
+
"epoch": 0.007578804609774664,
|
| 484 |
+
"eval_loss": 3.8216655254364014,
|
| 485 |
+
"eval_runtime": 22.9116,
|
| 486 |
+
"eval_samples_per_second": 218.23,
|
| 487 |
+
"eval_steps_per_second": 27.279,
|
| 488 |
+
"step": 570
|
| 489 |
+
},
|
| 490 |
+
{
|
| 491 |
+
"epoch": 0.007711766094156675,
|
| 492 |
+
"grad_norm": 0.0921698808670044,
|
| 493 |
+
"learning_rate": 0.002196360252515458,
|
| 494 |
+
"loss": 3.811664581298828,
|
| 495 |
+
"step": 580
|
| 496 |
+
},
|
| 497 |
+
{
|
| 498 |
+
"epoch": 0.007844727578538687,
|
| 499 |
+
"grad_norm": 0.0789179727435112,
|
| 500 |
+
"learning_rate": 0.0020184696718185723,
|
| 501 |
+
"loss": 3.8138885498046875,
|
| 502 |
+
"step": 590
|
| 503 |
+
},
|
| 504 |
+
{
|
| 505 |
+
"epoch": 0.007977689062920698,
|
| 506 |
+
"grad_norm": 0.07974446564912796,
|
| 507 |
+
"learning_rate": 0.0018474429926226303,
|
| 508 |
+
"loss": 3.7982933044433596,
|
| 509 |
+
"step": 600
|
| 510 |
+
},
|
| 511 |
+
{
|
| 512 |
+
"epoch": 0.00811065054730271,
|
| 513 |
+
"grad_norm": 0.08164919912815094,
|
| 514 |
+
"learning_rate": 0.0016838183281341638,
|
| 515 |
+
"loss": 3.796610641479492,
|
| 516 |
+
"step": 610
|
| 517 |
+
},
|
| 518 |
+
{
|
| 519 |
+
"epoch": 0.008243612031684722,
|
| 520 |
+
"grad_norm": 0.08382760733366013,
|
| 521 |
+
"learning_rate": 0.001528110502082944,
|
| 522 |
+
"loss": 3.7793228149414064,
|
| 523 |
+
"step": 620
|
| 524 |
+
},
|
| 525 |
+
{
|
| 526 |
+
"epoch": 0.008376573516066733,
|
| 527 |
+
"grad_norm": 0.07584033906459808,
|
| 528 |
+
"learning_rate": 0.0013808094288960784,
|
| 529 |
+
"loss": 3.775922393798828,
|
| 530 |
+
"step": 630
|
| 531 |
+
},
|
| 532 |
+
{
|
| 533 |
+
"epoch": 0.008429758109819538,
|
| 534 |
+
"eval_loss": 3.7764945030212402,
|
| 535 |
+
"eval_runtime": 22.9168,
|
| 536 |
+
"eval_samples_per_second": 218.181,
|
| 537 |
+
"eval_steps_per_second": 27.273,
|
| 538 |
+
"step": 634
|
| 539 |
+
},
|
| 540 |
+
{
|
| 541 |
+
"epoch": 0.008509535000448746,
|
| 542 |
+
"grad_norm": 0.07796747237443924,
|
| 543 |
+
"learning_rate": 0.0012423785722459621,
|
| 544 |
+
"loss": 3.7791465759277343,
|
| 545 |
+
"step": 640
|
| 546 |
+
},
|
| 547 |
+
{
|
| 548 |
+
"epoch": 0.008642496484830757,
|
| 549 |
+
"grad_norm": 0.07175382971763611,
|
| 550 |
+
"learning_rate": 0.0011132534868220968,
|
| 551 |
+
"loss": 3.7700267791748048,
|
| 552 |
+
"step": 650
|
| 553 |
+
},
|
| 554 |
+
{
|
| 555 |
+
"epoch": 0.008775457969212768,
|
| 556 |
+
"grad_norm": 0.07658028602600098,
|
| 557 |
+
"learning_rate": 0.0009938404479149102,
|
| 558 |
+
"loss": 3.7676132202148436,
|
| 559 |
+
"step": 660
|
| 560 |
+
},
|
| 561 |
+
{
|
| 562 |
+
"epoch": 0.00890841945359478,
|
| 563 |
+
"grad_norm": 0.07290265709161758,
|
| 564 |
+
"learning_rate": 0.0008845151731233707,
|
| 565 |
+
"loss": 3.757847213745117,
|
| 566 |
+
"step": 670
|
| 567 |
+
},
|
| 568 |
+
{
|
| 569 |
+
"epoch": 0.009041380937976792,
|
| 570 |
+
"grad_norm": 0.07412706315517426,
|
| 571 |
+
"learning_rate": 0.0007856216402084315,
|
| 572 |
+
"loss": 3.7475379943847655,
|
| 573 |
+
"step": 680
|
| 574 |
+
},
|
| 575 |
+
{
|
| 576 |
+
"epoch": 0.009174342422358803,
|
| 577 |
+
"grad_norm": 0.07130566239356995,
|
| 578 |
+
"learning_rate": 0.0006974710048117232,
|
| 579 |
+
"loss": 3.7532440185546876,
|
| 580 |
+
"step": 690
|
| 581 |
+
},
|
| 582 |
+
{
|
| 583 |
+
"epoch": 0.00926741546142621,
|
| 584 |
+
"eval_loss": 3.749366044998169,
|
| 585 |
+
"eval_runtime": 22.9896,
|
| 586 |
+
"eval_samples_per_second": 217.49,
|
| 587 |
+
"eval_steps_per_second": 27.186,
|
| 588 |
+
"step": 697
|
| 589 |
+
},
|
| 590 |
+
{
|
| 591 |
+
"epoch": 0.009307303906740816,
|
| 592 |
+
"grad_norm": 0.06786452233791351,
|
| 593 |
+
"learning_rate": 0.0006203406214448,
|
| 594 |
+
"loss": 3.74536018371582,
|
| 595 |
+
"step": 700
|
| 596 |
+
},
|
| 597 |
+
{
|
| 598 |
+
"epoch": 0.009440265391122827,
|
| 599 |
+
"grad_norm": 0.07180707901716232,
|
| 600 |
+
"learning_rate": 0.0005544731708292279,
|
| 601 |
+
"loss": 3.743540954589844,
|
| 602 |
+
"step": 710
|
| 603 |
+
},
|
| 604 |
+
{
|
| 605 |
+
"epoch": 0.009573226875504838,
|
| 606 |
+
"grad_norm": 0.06881807744503021,
|
| 607 |
+
"learning_rate": 0.0005000758963332508,
|
| 608 |
+
"loss": 3.741817855834961,
|
| 609 |
+
"step": 720
|
| 610 |
+
},
|
| 611 |
+
{
|
| 612 |
+
"epoch": 0.00970618835988685,
|
| 613 |
+
"grad_norm": 0.07049064338207245,
|
| 614 |
+
"learning_rate": 0.000457319951907489,
|
| 615 |
+
"loss": 3.7454132080078124,
|
| 616 |
+
"step": 730
|
| 617 |
+
},
|
| 618 |
+
{
|
| 619 |
+
"epoch": 0.009839149844268862,
|
| 620 |
+
"grad_norm": 0.06726963818073273,
|
| 621 |
+
"learning_rate": 0.0004263398635712801,
|
| 622 |
+
"loss": 3.745738983154297,
|
| 623 |
+
"step": 740
|
| 624 |
+
},
|
| 625 |
+
{
|
| 626 |
+
"epoch": 0.009972111328650873,
|
| 627 |
+
"grad_norm": 0.06847967952489853,
|
| 628 |
+
"learning_rate": 0.00040723310614405915,
|
| 629 |
+
"loss": 3.730845260620117,
|
| 630 |
+
"step": 750
|
| 631 |
+
},
|
| 632 |
+
{
|
| 633 |
+
"epoch": 0.01006518436771828,
|
| 634 |
+
"step": 757,
|
| 635 |
+
"total_flos": 6.181607017178726e+16,
|
| 636 |
+
"train_loss": 4.4425654001878465,
|
| 637 |
+
"train_runtime": 3601.9689,
|
| 638 |
+
"train_samples_per_second": 54.015,
|
| 639 |
+
"train_steps_per_second": 0.211
|
| 640 |
+
}
|
| 641 |
+
],
|
| 642 |
+
"logging_steps": 10,
|
| 643 |
+
"max_steps": 760,
|
| 644 |
+
"num_input_tokens_seen": 0,
|
| 645 |
+
"num_train_epochs": 1,
|
| 646 |
+
"save_steps": 5,
|
| 647 |
+
"stateful_callbacks": {
|
| 648 |
+
"TrainerControl": {
|
| 649 |
+
"args": {
|
| 650 |
+
"should_epoch_stop": false,
|
| 651 |
+
"should_evaluate": false,
|
| 652 |
+
"should_log": false,
|
| 653 |
+
"should_save": true,
|
| 654 |
+
"should_training_stop": true
|
| 655 |
+
},
|
| 656 |
+
"attributes": {}
|
| 657 |
+
}
|
| 658 |
+
},
|
| 659 |
+
"total_flos": 6.181607017178726e+16,
|
| 660 |
+
"train_batch_size": 8,
|
| 661 |
+
"trial_name": null,
|
| 662 |
+
"trial_params": null
|
| 663 |
+
}
|
scheduler/greedy/REPORT.md
ADDED
|
@@ -0,0 +1,42 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Experiment: greedy — GreedyLR adaptive scheduler
|
| 2 |
+
|
| 3 |
+
**Wall-clock:** 59 min (hit the 760-step horizon just before the time stop) · **Recipe:** best known (sdpa + bf16 mixed + muonq), adaptive LR: start 1e-3, GreedyLR raises on eval-loss improvement / lowers on plateau (factor 0.9, patience 0, threshold 3e-3 abs, bounds 4e-4…1.6e-2), evals every 20 steps on 1024 samples
|
| 4 |
+
|
| 5 |
+
## Plumbing this run needed (first metric-driven scheduler in the project)
|
| 6 |
+
|
| 7 |
+
1. `metric_for_best_model = "loss"` — required; the Trainer steps greedy/plateau schedulers on this eval metric and crashes without it. Wired through `base_train.py` as a new config key.
|
| 8 |
+
2. **`warmup_steps` is silently ignored** — `get_scheduler` passes only `lr_scheduler_kwargs` to greedy. The low 1e-3 starting LR + raise-on-improvement acted as the de-facto warmup (cold start was stable: grad max 5.51, the calmest of any hot run).
|
| 9 |
+
3. Eval cadence 20 steps / 1024 samples (vs 5 min / 5000 elsewhere) — greedy only gets to act when an eval runs; 38 decisions this run vs ~11 on the standard cadence.
|
| 10 |
+
|
| 11 |
+
## Result: the adaptive LR *found* the tuned peak on its own — then lost for lack of an anneal
|
| 12 |
+
|
| 13 |
+
| metric | greedy | constant 8e-3 reference | wsd_20 (series winner) |
|
| 14 |
+
|---|---|---|---|
|
| 15 |
+
| optimizer steps | 760 | 759 | 730 |
|
| 16 |
+
| final eval loss / ppl | 3.9267 / 50.7 (1024-sample eval — noisier subset) | 3.8583 / 47.4 | 3.8023 / 44.8 |
|
| 17 |
+
| bake score | 44.01/100 (DOUGH) | 46.83 | **55.61** |
|
| 18 |
+
| held-out bits/byte | 1.3422 | 1.3114 | **1.2769** |
|
| 19 |
+
| chat bits/byte | 1.2188 | 1.1914 | **1.1571** |
|
| 20 |
+
| logic accuracy | 0.575 (+0.0085) | 0.600 (+0.018) | **0.650 (+0.0242)** |
|
| 21 |
+
| hygiene (greedy loop) | 56.2/100 (45.5w) | 47.7 (59.8w) | **79.2 (23.8w)** |
|
| 22 |
+
| grad norms | **0.25–5.51** (calmest hot run) | 0.09–9.45 | 0.06–9.45 |
|
| 23 |
+
| peak GPU | 7.2 GB | 7.2 GB | 7.2 GB |
|
| 24 |
+
|
| 25 |
+
**The LR trajectory is the interesting part:** 1e-3 → steady climb → oscillating plateau at **6.6–7.4e-3** from ~step 410 onward, never touching the 1.6e-2 cap, ending at 7.4e-3. With zero knowledge of our LR series, the eval-loss signal alone steered it to within ~10% of our hand-tuned 8e-3 peak — independent confirmation that 8e-3-ish is the right constant-phase LR for this setup. It also spent ~350 steps getting there (a slow de-facto warmup), and 300+ steps holding near 7e-3 where improvements hovered around the 3e-3 threshold.
|
| 26 |
+
|
| 27 |
+
**Why it still lost:** GreedyLR is a *find-the-workable-LR* policy with no concept of a budget ending — it finished the run hot at 7.4e-3. The scheduler series' central finding is that the final anneal is worth +5–9 bake points, and greedy never anneals. It beat nothing that decays, and even trails the constant-8e-3 run (it averaged a lower LR for most of the hour while paying the same no-anneal penalty).
|
| 28 |
+
|
| 29 |
+
## Eval loss curve (every 60 steps; 1024-sample evals — not directly comparable to the 5000-sample curves of other runs)
|
| 30 |
+
|
| 31 |
+
| step | eval loss | | step | eval loss |
|
| 32 |
+
|---|---|---|---|---|
|
| 33 |
+
| 60 | 5.6145 | | 480 | 4.1681 |
|
| 34 |
+
| 120 | 4.9678 | | 540 | 4.0644 |
|
| 35 |
+
| 180 | 4.6068 | | 600 | 4.0330 |
|
| 36 |
+
| 240 | 4.4325 | | 660 | 3.9626 |
|
| 37 |
+
| 300 | 4.3343 | | 720 | 3.9412 |
|
| 38 |
+
| 360 | 4.3033 | | 760 | 3.9267 |
|
| 39 |
+
|
| 40 |
+
## Verdict
|
| 41 |
+
|
| 42 |
+
**A successful experiment about an unsuccessful scheduler.** As a quality recipe it loses to every decaying schedule; as an instrument it independently validated our tuned peak LR (plateaued at 6.6–7.4e-3 vs our 8e-3). If revisited: pair greedy's adaptive stable phase with a forced terminal decay (e.g. greedy until 80% of budget, then linear-to-0) — but wsd_20 already does the simple version of that. See `../COMPARISON.md`.
|
scheduler/greedy/config.toml
ADDED
|
@@ -0,0 +1,93 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Scheduler series — new best recipe (sdpa + bf16 mixed + muonq), adaptive-LR arm.
|
| 2 |
+
# This run: GreedyLR — metric-driven; LR steps ONLY on evaluation, so evals run every 20 steps (1024 samples).
|
| 3 |
+
# NOTE: warmup_steps is IGNORED by greedy (get_scheduler passes only lr_scheduler_kwargs); the low starting
|
| 4 |
+
# learning_rate (1e-3) + greedy raising on improvement act as the de-facto warmup (sim: reaches 8e-3 ~step 400).
|
| 5 |
+
model_config = "../model.toml"
|
| 6 |
+
|
| 7 |
+
[data]
|
| 8 |
+
# Data files (supports glob patterns or lists of files)
|
| 9 |
+
train_files = ["~/Dev/vintage-LLM/training/data/train_*.bin"]
|
| 10 |
+
valid_files = ["~/Dev/vintage-LLM/training/data/valid_*.bin"]
|
| 11 |
+
|
| 12 |
+
# Sequence length for each dataset sample
|
| 13 |
+
# Must be <= model.max_position_embeddings
|
| 14 |
+
max_seq_length = 1024
|
| 15 |
+
|
| 16 |
+
# Tokenizer (must match the one used to create .bin files)
|
| 17 |
+
tokenizer = "~/Dev/vintage-LLM/tokenizers/t-v3/"
|
| 18 |
+
|
| 19 |
+
[training]
|
| 20 |
+
# Training duration
|
| 21 |
+
num_train_epochs = 1
|
| 22 |
+
max_steps = 760 # scheduler horizon = achievable steps/hour (was -1: full epoch, so decay never happened)
|
| 23 |
+
|
| 24 |
+
# Optional: Stop training after this many minutes of wall-clock time.
|
| 25 |
+
max_train_minutes = 60
|
| 26 |
+
|
| 27 |
+
# Output directories
|
| 28 |
+
output_dir = "./"
|
| 29 |
+
final_model_dir = "./final"
|
| 30 |
+
|
| 31 |
+
# Batch sizes
|
| 32 |
+
# Effective batch size = num_devices × per_device_train_batch_size × gradient_accumulation_steps
|
| 33 |
+
per_device_train_batch_size = 8 # batch size per GPU (8x1024 tokens fits 16GB with headroom for eval)
|
| 34 |
+
per_device_eval_batch_size = 8
|
| 35 |
+
gradient_accumulation_steps = 32 # accumulate gradients over multiple mini-batches before updating the params
|
| 36 |
+
|
| 37 |
+
# Optimizer: MuonQ (4-bit quantized Muon, ../../MuonQ repo) — wired through
|
| 38 |
+
# optimizer_cls_and_kwargs in base_train.py. 2D hidden weights get Muon
|
| 39 |
+
# (Newton-Schulz 5, 4-bit mu-law-companded momentum, rank-16 residual
|
| 40 |
+
# factorization); embeddings/head/norms get its internal AdamW backup.
|
| 41 |
+
# Muon/quantization knobs (muon_*, muonq_*) default to the repo's own
|
| 42 |
+
# llama-60m "muonq" recipe in base_train.py.
|
| 43 |
+
optim = "muonq"
|
| 44 |
+
|
| 45 |
+
# Optimizer settings — lr/wd from MuonQ's llama-60m recipe (Muon takes
|
| 46 |
+
# spectral-norm-scaled steps; do not copy the AdamW 5e-4 here)
|
| 47 |
+
learning_rate = 1e-3 # starting LR only — greedy climbs it (bounded by max_lr)
|
| 48 |
+
weight_decay = 0.1
|
| 49 |
+
adam_beta1 = 0.9
|
| 50 |
+
adam_beta2 = 0.95
|
| 51 |
+
max_grad_norm = 1.0
|
| 52 |
+
|
| 53 |
+
# Learning rate scheduler
|
| 54 |
+
lr_scheduler_type = "greedy" # GreedyLR: raise LR while eval loss improves, lower on plateau
|
| 55 |
+
# patience 0 = act on every eval; threshold = min improvement (abs) to count; bounds 4e-4..1.6e-2
|
| 56 |
+
lr_scheduler_kwargs = { factor = 0.9, patience = 0, threshold = 3e-3, threshold_mode = "abs", min_lr = 4e-4, max_lr = 1.6e-2, verbose = true }
|
| 57 |
+
metric_for_best_model = "loss" # greedy steps on eval_loss (wired through base_train.py)
|
| 58 |
+
warmup_steps = 200 # 1-hour series standard (muonq-v2's 400 was a 2-hour recipe)
|
| 59 |
+
|
| 60 |
+
# Precision (choose based on your GPU)
|
| 61 |
+
bf16 = true # bfloat16 autocast + fp32 master weights (the dtype-series safe recipe)
|
| 62 |
+
fp16 = false
|
| 63 |
+
|
| 64 |
+
# Performance
|
| 65 |
+
torch_compile = false
|
| 66 |
+
gradient_checkpointing = false # trade compute for memory (enables larger models)
|
| 67 |
+
neftune_noise_alpha = 0.0 # add noise during training to improve model performance (fine-tuning)
|
| 68 |
+
|
| 69 |
+
# Checkpointing
|
| 70 |
+
# save_strategy can be "steps", "epoch", or "minutes" (wall-clock time).
|
| 71 |
+
save_strategy = "minutes"
|
| 72 |
+
save_steps = 5 # checkpoint every X minutes
|
| 73 |
+
save_total_limit = 3 # keep only X most recent checkpoints
|
| 74 |
+
|
| 75 |
+
# Evaluation
|
| 76 |
+
# eval_strategy can be "steps", "epoch", or "minutes" (wall-clock time).
|
| 77 |
+
eval_strategy = "steps"
|
| 78 |
+
eval_steps = 20 # greedy only adjusts LR on eval -> needs frequent evals
|
| 79 |
+
# Evaluate on a fixed random subset of the validation set (0 = use all)
|
| 80 |
+
max_eval_samples = 1024 # keep frequent evals cheap (~4s); noisier curve, final evals unaffected
|
| 81 |
+
|
| 82 |
+
# Logging
|
| 83 |
+
logging_strategy = "steps"
|
| 84 |
+
logging_steps = 10 # log every 10 steps
|
| 85 |
+
logging_first_step = true # log the first step
|
| 86 |
+
|
| 87 |
+
# Reproducibility
|
| 88 |
+
seed = 42
|
| 89 |
+
|
| 90 |
+
# Performance
|
| 91 |
+
dataloader_num_workers = 2 # number of data loading threads
|
| 92 |
+
dataloader_prefetch_factor = 2 # batches prefetched per worker
|
| 93 |
+
dataloader_pin_memory = true # pin memory for faster GPU transfer
|
scheduler/greedy/eval1.log
ADDED
|
@@ -0,0 +1,213 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
W0814 16:07:22.252000 71987 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
|
| 2 |
+
W0814 16:07:22.274000 71987 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
|
| 3 |
+
|
| 4 |
+
==============================================================================
|
| 5 |
+
CHECKPOINT INSPECTION
|
| 6 |
+
==============================================================================
|
| 7 |
+
checkpoint : final
|
| 8 |
+
tokenizer : final
|
| 9 |
+
device : cuda dtype: torch.bfloat16 seed: 1337
|
| 10 |
+
|
| 11 |
+
|
| 12 |
+
==============================================================================
|
| 13 |
+
SUITE: MODEL INFO AND TRAINING LINEAGE
|
| 14 |
+
==============================================================================
|
| 15 |
+
checkpoint : final
|
| 16 |
+
model_type : llama
|
| 17 |
+
architecture : LlamaForCausalLM
|
| 18 |
+
device : cuda
|
| 19 |
+
dtype : torch.bfloat16
|
| 20 |
+
parameters : 77,083,392 (77.08M)
|
| 21 |
+
disk size : 308 MB
|
| 22 |
+
layers / hidden / heads : 8 / 768 / 8 (KV heads: 4)
|
| 23 |
+
vocab (cfg / tokenizer) : 32768 / 32752
|
| 24 |
+
context length : 1024
|
| 25 |
+
chat template : True
|
| 26 |
+
training lineage : base pretraining, ctx 1024
|
| 27 |
+
embedding mean norm : 3.2496 <- typical row length of the embedding matrix; very small or huge values signal trouble
|
| 28 |
+
embedding mean cosine : 0.1120 <- avg cosine between random embedding rows. Descriptive only: it starts near 0.0
|
| 29 |
+
because RANDOM init is near-orthogonal, and it GROWS as the model trains.
|
| 30 |
+
Near 0 therefore does NOT mean "healthy" - it can mean "barely trained".
|
| 31 |
+
(note: config vocab padded by 16 unused rows - harmless)
|
| 32 |
+
|
| 33 |
+
==============================================================================
|
| 34 |
+
SUITE: PERIOD FIDELITY ON FIXED PROBE SENTENCES
|
| 35 |
+
==============================================================================
|
| 36 |
+
Scoring 20 sentences (10 historical, 10 modern)
|
| 37 |
+
The MODERN/HISTORICAL ratio is the output that means something. The absolute
|
| 38 |
+
numbers come from ~20 sentences: too noisy to rank checkpoints (use evaluate2.py).
|
| 39 |
+
|
| 40 |
+
OVERALL (270 tokens)
|
| 41 |
+
perplexity : 399.51 <- 'average branching factor'; lower = less surprised = better fit to this text
|
| 42 |
+
mean token prob : 0.0563 <- average confidence on the actual next word
|
| 43 |
+
median token prob : 0.0041 <- typical confidence (robust to a few very bad tokens)
|
| 44 |
+
worst-token prob (min): 7.0e-08 <- the single most surprising word; tiny values = the model really didn't expect it
|
| 45 |
+
10th-percentile prob : 1.2e-05 <- confidence on the model's worst 10% of guesses
|
| 46 |
+
low-confidence tokens : 60.00% <- share of tokens predicted with <1% probability (red flags)
|
| 47 |
+
mean entropy : 5.001 nats <- how spread out the predictions are; ~0 = very certain, higher = hedging
|
| 48 |
+
|
| 49 |
+
GROUP COMPARISON
|
| 50 |
+
group sentences perplexity bits/byte
|
| 51 |
+
historical 10 141.24 1.347
|
| 52 |
+
modern 10 953.62 2.046
|
| 53 |
+
modern/historical ratio: 6.75 <- >1 means period text is easier for the model than modern text (desired for a vintage model with a ~1900 knowledge cutoff)
|
| 54 |
+
(bits/byte is the tokenizer-independent version - the only column here you may
|
| 55 |
+
compare between models that use DIFFERENT tokenizers.)
|
| 56 |
+
|
| 57 |
+
PER-SENTENCE DETAIL (sorted worst-first)
|
| 58 |
+
[modern ] ppl= 7408.34 minp=2.0e-07 The engineering team rewrote the whole backend micro
|
| 59 |
+
[modern ] ppl= 2671.39 minp=1.5e-06 Robots on the automated assembly line handle the ent
|
| 60 |
+
[modern ] ppl= 2036.79 minp=2.0e-06 Small retail startups now run almost all of their bu
|
| 61 |
+
[modern ] ppl= 1301.80 minp=3.2e-06 She is finishing a graduate degree in machine learni
|
| 62 |
+
[modern ] ppl= 1092.45 minp=9.4e-07 Three hours stuck in traffic in the pouring rain mad
|
| 63 |
+
[modern ] ppl= 978.99 minp=3.2e-07 Members of the European Union parliament voted on th
|
| 64 |
+
[modern ] ppl= 974.77 minp=7.0e-08 Millions of shoppers queue outside the store because
|
| 65 |
+
[modern ] ppl= 396.50 minp=1.0e-06 She gave up her whole weekend to help me move apartm
|
| 66 |
+
[historical] ppl= 330.91 minp=1.3e-05 The keeper fed the great serpent which the naturalis
|
| 67 |
+
[modern ] ppl= 310.68 minp=3.0e-06 After years of hiding it from everyone at work, he t
|
| 68 |
+
[historical] ppl= 210.57 minp=7.7e-06 The Lord provideth for all his creatures, giving the
|
| 69 |
+
[modern ] ppl= 205.29 minp=1.9e-05 I switched to a plant-based diet last year and I no
|
| 70 |
+
[historical] ppl= 177.84 minp=1.2e-05 The labouring poor of this parish are reduced to gre
|
| 71 |
+
[historical] ppl= 168.82 minp=2.7e-05 The mountain rose above the valley in a silence sole
|
| 72 |
+
[historical] ppl= 154.45 minp=6.2e-05 The weavers at their looms are employed in the wooll
|
| 73 |
+
[historical] ppl= 153.93 minp=1.2e-04 The merchants of the port have grown rich upon their
|
| 74 |
+
[historical] ppl= 141.63 minp=3.2e-06 He gave up his fortune to the patient study of natur
|
| 75 |
+
[historical] ppl= 116.44 minp=1.3e-04 He drew a distinction so fine and over-scrupulous th
|
| 76 |
+
[historical] ppl= 74.65 minp=1.9e-04 The ballroom was filled with dancing and laughter, a
|
| 77 |
+
[historical] ppl= 52.16 minp=1.5e-04 Her Majesty was pleased to summon the Lords and Comm
|
| 78 |
+
|
| 79 |
+
==============================================================================
|
| 80 |
+
SUITE: DIACHRONIC WORD-SENSE SEPARATION
|
| 81 |
+
==============================================================================
|
| 82 |
+
Extracting contextual embeddings for 10 shifted words in period AND modern sentences.
|
| 83 |
+
Every probe word sits LAST in its sentence, so the causal model has actually read
|
| 84 |
+
the disambiguating context before it represents the word.
|
| 85 |
+
|
| 86 |
+
SENSE SEPARATION (cosine similarity of the SAME word, period vs modern sentence)
|
| 87 |
+
lower value = the model represents the two senses differently (good sign for a
|
| 88 |
+
period model); ~1.0 = it treats them as the same word regardless of context
|
| 89 |
+
python : +0.415
|
| 90 |
+
want : +0.482
|
| 91 |
+
awful : +0.587
|
| 92 |
+
nice : +0.676
|
| 93 |
+
manufacture : +0.708
|
| 94 |
+
gay : +0.725
|
| 95 |
+
meat : +0.740
|
| 96 |
+
parliament : +0.745
|
| 97 |
+
science : +0.790
|
| 98 |
+
commerce : +0.809
|
| 99 |
+
MEAN : +0.668 <- near 1.0 = model ignores period vs modern usage; lower = senses differ
|
| 100 |
+
|
| 101 |
+
PAIRWISE COSINE SIMILARITY (period sentences, first 5 words)
|
| 102 |
+
these are CONTEXTUAL vectors, so they reflect the sentences as much as the words;
|
| 103 |
+
read the spread, not individual pairs. High mean = representations collapsed.
|
| 104 |
+
gay awful nice meat want
|
| 105 |
+
gay 1.000 0.367 0.440 0.355 0.345
|
| 106 |
+
awful 1.000 0.368 0.277 0.262
|
| 107 |
+
nice 1.000 0.354 0.369
|
| 108 |
+
meat 1.000 0.467
|
| 109 |
+
want 1.000
|
| 110 |
+
all 45 pairs: mean=+0.381 std=0.109 <- healthy spread is good
|
| 111 |
+
|
| 112 |
+
NEAREST NEIGHBOURS inside the probe-word set (period sentences)
|
| 113 |
+
gay : nice (+0.44), science (+0.44), commerce (+0.42), parliament (+0.42), awful (+0.37)
|
| 114 |
+
awful : nice (+0.37), gay (+0.37), parliament (+0.32), science (+0.31), meat (+0.28)
|
| 115 |
+
nice : gay (+0.44), science (+0.42), want (+0.37), awful (+0.37), commerce (+0.37)
|
| 116 |
+
meat : parliament (+0.50), science (+0.48), commerce (+0.48), want (+0.47), manufacture (+0.40)
|
| 117 |
+
want : commerce (+0.57), manufacture (+0.49), science (+0.49), parliament (+0.48), meat (+0.47)
|
| 118 |
+
|
| 119 |
+
==============================================================================
|
| 120 |
+
SUITE: GENERATION PROBES
|
| 121 |
+
==============================================================================
|
| 122 |
+
|
| 123 |
+
--- Probe 1 ---
|
| 124 |
+
prompt : The history of the world is
|
| 125 |
+
continuation: too often the subject of a few words; and the first of these is a little more than a mere coincidence of history. It has been said that the present age is the most remarkable of the present age. But what is to be said of the future is, that the present age is a little
|
| 126 |
+
stats: 60 tokens | distinct-1 0.61 | distinct-2 0.85 | echo 0.07
|
| 127 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 128 |
+
|
| 129 |
+
--- Probe 2 ---
|
| 130 |
+
prompt : What is God? God is
|
| 131 |
+
continuation: a man, he is a man. He is a man, but it is a man.
|
| 132 |
+
|
| 133 |
+
We are always ready for the use of God. We must have a right to be able to do the will of God.
|
| 134 |
+
|
| 135 |
+
We are always ready to give God's will, to make
|
| 136 |
+
stats: 60 tokens | distinct-1 0.60 | distinct-2 0.81 | echo 0.10
|
| 137 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 138 |
+
|
| 139 |
+
--- Probe 3 ---
|
| 140 |
+
prompt : In the year of our Lord eighteen hundred and
|
| 141 |
+
continuation: eighty, in the year of his life, we may hope that his labors in this field will ever continue to increase.
|
| 142 |
+
|
| 143 |
+
The great work of the world's missionary enterprise has been done in a manner that will be a great blessing to those who have not a desire to know of its work.
|
| 144 |
+
stats: 60 tokens | distinct-1 0.75 | distinct-2 1.00 | echo 0.00
|
| 145 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 146 |
+
|
| 147 |
+
--- Probe 4 ---
|
| 148 |
+
prompt : The manufacture of cotton
|
| 149 |
+
continuation: , and the manufacture of the article of export.
|
| 150 |
+
|
| 151 |
+
The manufacture of the cotton from this country is to a considerable extent in the hands of the public. The price of the produce is about the same, and the produce exported to any country is not so much as the price of its produce,
|
| 152 |
+
stats: 60 tokens | distinct-1 0.59 | distinct-2 0.85 | echo 0.11
|
| 153 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 154 |
+
|
| 155 |
+
--- Probe 5 ---
|
| 156 |
+
prompt : The steam engine
|
| 157 |
+
continuation: is suspended from the water, and, in the opinion, it is a matter of the highest moment, whether to the engineer, or the engineer, to the engineer.
|
| 158 |
+
|
| 159 |
+
If, on the contrary, the steam boat was a very good conductor, and had the advantage of the machinery, it
|
| 160 |
+
stats: 60 tokens | distinct-1 0.68 | distinct-2 0.93 | echo 0.15
|
| 161 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 162 |
+
|
| 163 |
+
--- Probe 6 ---
|
| 164 |
+
prompt : The telegraph
|
| 165 |
+
continuation: ic and telegraphic telegraphic.
|
| 166 |
+
|
| 167 |
+
The United States' Government, with the exception of a very few, is not, I think, without reason, without a knowledge of the United States.
|
| 168 |
+
|
| 169 |
+
The United States, with the exception of the United States, have not yet arrived at a
|
| 170 |
+
stats: 60 tokens | distinct-1 0.66 | distinct-2 0.84 | echo 0.07
|
| 171 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 172 |
+
|
| 173 |
+
--- Probe 7 ---
|
| 174 |
+
prompt : My dearest sister,
|
| 175 |
+
continuation: a great and beautiful woman.
|
| 176 |
+
|
| 177 |
+
I have been talking with her, but I think that I am as much of a poet as any woman in the world. I have a mind to write, but I am afraid there is something, which I would rather write for you. It was so very
|
| 178 |
+
stats: 60 tokens | distinct-1 0.79 | distinct-2 0.94 | echo 0.02
|
| 179 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 180 |
+
|
| 181 |
+
--- Probe 8 ---
|
| 182 |
+
prompt : Greetings, my friend
|
| 183 |
+
continuation: , the Prince of Wales, I am not so much the friend of the Prince of Wales as he is, but, like myself, the most intimate friend of the Prince of Wales, I should not feel much interest in the Prince's attentions to his Majesty, as I am inclined to believe that the
|
| 184 |
+
stats: 60 tokens | distinct-1 0.60 | distinct-2 0.83 | echo 0.06
|
| 185 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 186 |
+
|
| 187 |
+
--- Probe 9 ---
|
| 188 |
+
prompt : Her Majesty the Queen
|
| 189 |
+
continuation: 's
|
| 190 |
+
|
| 191 |
+
Majesty's Government's Council is, that, in case of any serious disturbance of the Government, they will be, in consequence of the Government's being, a very strong and respectable member, and of course, that he will be able to make an immediate appeal to them.
|
| 192 |
+
stats: 60 tokens | distinct-1 0.80 | distinct-2 0.98 | echo 0.04
|
| 193 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 194 |
+
|
| 195 |
+
--- Probe 10 ---
|
| 196 |
+
prompt : Chapter I.
|
| 197 |
+
continuation: The reader will perhaps ask whether the author would give him a more satisfactory account of the nature and history of the people than it is possible for a writer to convey.
|
| 198 |
+
|
| 199 |
+
The whole book is as yet incomplete as it is incomplete, and we cannot but hope that the reader, who has not the
|
| 200 |
+
stats: 60 tokens | distinct-1 0.78 | distinct-2 0.96 | echo 0.02
|
| 201 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 202 |
+
|
| 203 |
+
AGGREGATE over 10 probes: 600 tokens in 0.8s (732.7 tok/s)
|
| 204 |
+
mean distinct-1 0.687 | mean distinct-2 0.899 | mean echo 0.064 | worst probe: distinct-2 0.809 / echo 0.149
|
| 205 |
+
(sampling hides loops that greedy decoding exposes; evaluate2.py measures the greedy loop length.)
|
| 206 |
+
|
| 207 |
+
Full numeric results written to /home/cro/Dev/vintage-LLM/training/eval_results/evaluate-final.json
|
| 208 |
+
|
| 209 |
+
==============================================================================
|
| 210 |
+
INSPECTION COMPLETE
|
| 211 |
+
==============================================================================
|
| 212 |
+
This covered WHAT the checkpoint is, its lineage, and whether it is still vintage.
|
| 213 |
+
For "is it baked enough", run evaluate2.py on the same path.
|
scheduler/greedy/eval2.log
ADDED
|
@@ -0,0 +1,54 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
evaluating 1 checkpoint(s) on cuda (torch.bfloat16)
|
| 2 |
+
tokenizer: final
|
| 3 |
+
final ...W0814 16:07:28.267000 72081 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
|
| 4 |
+
W0814 16:07:28.283000 72081 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
|
| 5 |
+
|
| 6 |
+
bake 44/100 bpb 1.3422 logic 0.57 (15.2s)
|
| 7 |
+
|
| 8 |
+
report: eval2.md
|
| 9 |
+
json: eval2.json
|
| 10 |
+
|
| 11 |
+
# Evaluation: final
|
| 12 |
+
|
| 13 |
+
- **Parameters:** 77M
|
| 14 |
+
- **Note:** no trainer_state.json in the checkpoint - tokens seen unknown
|
| 15 |
+
|
| 16 |
+
## Verdict: DOUGH (bake score 44/100)
|
| 17 |
+
|
| 18 |
+
Structure is forming - real words, some grammar - but this is not a usable language model yet. It needs several times more training tokens.
|
| 19 |
+
|
| 20 |
+
## Where it sits (held-out bits/byte, lower = better)
|
| 21 |
+
|
| 22 |
+
```
|
| 23 |
+
3.50 untrained model (uniform noise)
|
| 24 |
+
2.00 word-salad
|
| 25 |
+
1.50 broken prose
|
| 26 |
+
--> 1.342 YOUR MODEL
|
| 27 |
+
1.33 500M model, ~0.7B tokens (early training)
|
| 28 |
+
1.19 500M model, ~4.7B tokens (undertrained but solid)
|
| 29 |
+
1.10 best sub-1B measured on this data (341M)
|
| 30 |
+
0.95 estimated sub-1B ceiling on this corpus
|
| 31 |
+
```
|
| 32 |
+
|
| 33 |
+
## Scores
|
| 34 |
+
|
| 35 |
+
| what | raw value | points /100 | weight | plain English |
|
| 36 |
+
|---|---|---|---|---|
|
| 37 |
+
| Held-out loss | 1.3422 bits/byte | 54 | 0.5 | how cheaply it predicts period text it never saw — the single best training signal |
|
| 38 |
+
| Logic | 0.575 acc, margin 0.009 | 18 | 0.25 | picks the *sensible* continuation over matched nonsense; 0.50 = coin-flip. weak - style without understanding |
|
| 39 |
+
| Chat readiness | 1.2188 bits/byte | 46 | 0.15 | how cheap well-formed period dialogue already is — predicts fine-tuning ease |
|
| 40 |
+
| Hygiene | loop 45.5w, punct 0.13/100w | 56 | 0.1 | greedy-decoding loop length and broken punctuation |
|
| 41 |
+
|
| 42 |
+
## Period boundary
|
| 43 |
+
|
| 44 |
+
Clean. All 6 post-1900 trap words cost the model more than their period twins (mean shock +1.63 bits/byte, weakest pair 'atomic bomb' at +1.04). No sign of modern text in training.
|
| 45 |
+
|
| 46 |
+
## See for yourself (sampled, t=0.8)
|
| 47 |
+
|
| 48 |
+
> **LONDON, Tuesday. — The committee appointed to inquire into the condition of the** work, and see if it could have been necessary to do so. The committee has no doubt the great benefit of having the enterprise a good one, and it is not only a matter of great importance, but a matter of great national interest, and it is quite certain that the project would have to be prosecuted without the slightest delay. A great many of the members of the Committee, and with so much reason, hav
|
| 49 |
+
>
|
| 50 |
+
> **A melancholy accident occurred on Thursday last at the works of Messrs. Harding and** Co. and the above mentioned. We have not an allusion to the work on the same side of the river, as we might expect to hear from the papers. It was a slight effort to ascertain the nature of the work. The last part of the work is to be done in a manner equally appropriate to the reader. A great part of it is the work of the writer; but it is not the work of the architect. A short description of the
|
| 51 |
+
>
|
| 52 |
+
> **Brethren, the text which I have chosen for our consideration this morning is taken from** the work of the school and from the work which the teacher takes to be the work of the Christian Church, and to whom it is a great joy, when in the morning the preacher takes leave to make the journey. The day for the morning has been the day for the preacher's sake. The work of the Church has been done by him, and the world is aflame with his glory. There is a time when the minister's work is don
|
| 53 |
+
>
|
| 54 |
+
|
scheduler/greedy/final/chat_template.jinja
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{%- for message in messages -%}
|
| 2 |
+
{{- bos_token + '<|' + message['role'] + '|>\n' + message['content'].strip() + eos_token + '\n' -}}
|
| 3 |
+
{%- endfor -%}
|
| 4 |
+
|
| 5 |
+
{%- if add_generation_prompt -%}
|
| 6 |
+
{{- bos_token + '<|assistant|>\n' -}}
|
| 7 |
+
{%- endif -%}
|
scheduler/greedy/final/config.json
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"LlamaForCausalLM"
|
| 4 |
+
],
|
| 5 |
+
"attention_bias": false,
|
| 6 |
+
"attention_dropout": 0.0,
|
| 7 |
+
"bos_token_id": 3,
|
| 8 |
+
"dtype": "float32",
|
| 9 |
+
"eos_token_id": 4,
|
| 10 |
+
"head_dim": 96,
|
| 11 |
+
"hidden_act": "silu",
|
| 12 |
+
"hidden_size": 768,
|
| 13 |
+
"initializer_range": 0.02,
|
| 14 |
+
"intermediate_size": 2048,
|
| 15 |
+
"max_position_embeddings": 1024,
|
| 16 |
+
"mlp_bias": false,
|
| 17 |
+
"model_type": "llama",
|
| 18 |
+
"num_attention_heads": 8,
|
| 19 |
+
"num_hidden_layers": 8,
|
| 20 |
+
"num_key_value_heads": 4,
|
| 21 |
+
"pad_token_id": 0,
|
| 22 |
+
"pretraining_tp": 1,
|
| 23 |
+
"rms_norm_eps": 1e-06,
|
| 24 |
+
"rope_parameters": {
|
| 25 |
+
"rope_theta": 10000.0,
|
| 26 |
+
"rope_type": "default"
|
| 27 |
+
},
|
| 28 |
+
"tie_word_embeddings": true,
|
| 29 |
+
"transformers_version": "5.14.1",
|
| 30 |
+
"use_cache": false,
|
| 31 |
+
"vocab_size": 32768
|
| 32 |
+
}
|
scheduler/greedy/final/generation_config.json
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_from_model_config": true,
|
| 3 |
+
"bos_token_id": 3,
|
| 4 |
+
"eos_token_id": [
|
| 5 |
+
4,
|
| 6 |
+
2
|
| 7 |
+
],
|
| 8 |
+
"output_attentions": false,
|
| 9 |
+
"output_hidden_states": false,
|
| 10 |
+
"pad_token_id": 0,
|
| 11 |
+
"transformers_version": "5.14.1",
|
| 12 |
+
"use_cache": false
|
| 13 |
+
}
|
scheduler/greedy/final/model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:797f00aeaf1c511832b19bffb1f1ff6fb014d7dede1c0ac24b770b6994c420b3
|
| 3 |
+
size 308341784
|
scheduler/greedy/final/tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
scheduler/greedy/final/tokenizer_config.json
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"backend": "tokenizers",
|
| 3 |
+
"bos_token": "<|bos|>",
|
| 4 |
+
"eos_token": "<|eos|>",
|
| 5 |
+
"is_local": true,
|
| 6 |
+
"local_files_only": false,
|
| 7 |
+
"model_max_length": 1000000000000000019884624838656,
|
| 8 |
+
"pad_token": "<|pad|>",
|
| 9 |
+
"tokenizer_class": "TokenizersBackend",
|
| 10 |
+
"unk_token": "<|unk|>"
|
| 11 |
+
}
|
scheduler/greedy/trainer_state.json
ADDED
|
@@ -0,0 +1,886 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_global_step": null,
|
| 3 |
+
"best_metric": 3.9166951179504395,
|
| 4 |
+
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 0.010105072813032885,
|
| 6 |
+
"eval_steps": 20,
|
| 7 |
+
"global_step": 760,
|
| 8 |
+
"is_hyper_param_search": false,
|
| 9 |
+
"is_local_process_zero": true,
|
| 10 |
+
"is_world_process_zero": true,
|
| 11 |
+
"log_history": [
|
| 12 |
+
{
|
| 13 |
+
"epoch": 1.3296148438201165e-05,
|
| 14 |
+
"grad_norm": 5.513442516326904,
|
| 15 |
+
"learning_rate": 0.001,
|
| 16 |
+
"loss": 10.538199424743652,
|
| 17 |
+
"step": 1
|
| 18 |
+
},
|
| 19 |
+
{
|
| 20 |
+
"epoch": 0.00013296148438201165,
|
| 21 |
+
"grad_norm": 2.141404390335083,
|
| 22 |
+
"learning_rate": 0.001,
|
| 23 |
+
"loss": 7.99212646484375,
|
| 24 |
+
"step": 10
|
| 25 |
+
},
|
| 26 |
+
{
|
| 27 |
+
"epoch": 0.0002659229687640233,
|
| 28 |
+
"grad_norm": 5.429156303405762,
|
| 29 |
+
"learning_rate": 0.001,
|
| 30 |
+
"loss": 6.921406555175781,
|
| 31 |
+
"step": 20
|
| 32 |
+
},
|
| 33 |
+
{
|
| 34 |
+
"epoch": 0.0002659229687640233,
|
| 35 |
+
"eval_loss": 6.734507083892822,
|
| 36 |
+
"eval_runtime": 4.8368,
|
| 37 |
+
"eval_samples_per_second": 211.711,
|
| 38 |
+
"eval_steps_per_second": 26.464,
|
| 39 |
+
"step": 20
|
| 40 |
+
},
|
| 41 |
+
{
|
| 42 |
+
"epoch": 0.0003988844531460349,
|
| 43 |
+
"grad_norm": 2.39795184135437,
|
| 44 |
+
"learning_rate": 0.0011111111111111111,
|
| 45 |
+
"loss": 6.502817535400391,
|
| 46 |
+
"step": 30
|
| 47 |
+
},
|
| 48 |
+
{
|
| 49 |
+
"epoch": 0.0005318459375280466,
|
| 50 |
+
"grad_norm": 1.780358076095581,
|
| 51 |
+
"learning_rate": 0.0011111111111111111,
|
| 52 |
+
"loss": 6.094605255126953,
|
| 53 |
+
"step": 40
|
| 54 |
+
},
|
| 55 |
+
{
|
| 56 |
+
"epoch": 0.0005318459375280466,
|
| 57 |
+
"eval_loss": 5.933877944946289,
|
| 58 |
+
"eval_runtime": 4.721,
|
| 59 |
+
"eval_samples_per_second": 216.905,
|
| 60 |
+
"eval_steps_per_second": 27.113,
|
| 61 |
+
"step": 40
|
| 62 |
+
},
|
| 63 |
+
{
|
| 64 |
+
"epoch": 0.0006648074219100582,
|
| 65 |
+
"grad_norm": 1.0763862133026123,
|
| 66 |
+
"learning_rate": 0.0012345679012345679,
|
| 67 |
+
"loss": 5.835469436645508,
|
| 68 |
+
"step": 50
|
| 69 |
+
},
|
| 70 |
+
{
|
| 71 |
+
"epoch": 0.0007977689062920698,
|
| 72 |
+
"grad_norm": 1.1550487279891968,
|
| 73 |
+
"learning_rate": 0.0012345679012345679,
|
| 74 |
+
"loss": 5.669637298583984,
|
| 75 |
+
"step": 60
|
| 76 |
+
},
|
| 77 |
+
{
|
| 78 |
+
"epoch": 0.0007977689062920698,
|
| 79 |
+
"eval_loss": 5.614484786987305,
|
| 80 |
+
"eval_runtime": 4.7235,
|
| 81 |
+
"eval_samples_per_second": 216.79,
|
| 82 |
+
"eval_steps_per_second": 27.099,
|
| 83 |
+
"step": 60
|
| 84 |
+
},
|
| 85 |
+
{
|
| 86 |
+
"epoch": 0.0009307303906740815,
|
| 87 |
+
"grad_norm": 1.2857813835144043,
|
| 88 |
+
"learning_rate": 0.0013717421124828531,
|
| 89 |
+
"loss": 5.549149703979492,
|
| 90 |
+
"step": 70
|
| 91 |
+
},
|
| 92 |
+
{
|
| 93 |
+
"epoch": 0.0010636918750560932,
|
| 94 |
+
"grad_norm": 0.9777137637138367,
|
| 95 |
+
"learning_rate": 0.0013717421124828531,
|
| 96 |
+
"loss": 5.450737762451172,
|
| 97 |
+
"step": 80
|
| 98 |
+
},
|
| 99 |
+
{
|
| 100 |
+
"epoch": 0.0010636918750560932,
|
| 101 |
+
"eval_loss": 5.396044731140137,
|
| 102 |
+
"eval_runtime": 4.7268,
|
| 103 |
+
"eval_samples_per_second": 216.636,
|
| 104 |
+
"eval_steps_per_second": 27.079,
|
| 105 |
+
"step": 80
|
| 106 |
+
},
|
| 107 |
+
{
|
| 108 |
+
"epoch": 0.0011966533594381047,
|
| 109 |
+
"grad_norm": 1.4468932151794434,
|
| 110 |
+
"learning_rate": 0.0015241579027587256,
|
| 111 |
+
"loss": 5.346249389648437,
|
| 112 |
+
"step": 90
|
| 113 |
+
},
|
| 114 |
+
{
|
| 115 |
+
"epoch": 0.0013296148438201164,
|
| 116 |
+
"grad_norm": 1.385213017463684,
|
| 117 |
+
"learning_rate": 0.0015241579027587256,
|
| 118 |
+
"loss": 5.2223976135253904,
|
| 119 |
+
"step": 100
|
| 120 |
+
},
|
| 121 |
+
{
|
| 122 |
+
"epoch": 0.0013296148438201164,
|
| 123 |
+
"eval_loss": 5.167303085327148,
|
| 124 |
+
"eval_runtime": 4.7507,
|
| 125 |
+
"eval_samples_per_second": 215.546,
|
| 126 |
+
"eval_steps_per_second": 26.943,
|
| 127 |
+
"step": 100
|
| 128 |
+
},
|
| 129 |
+
{
|
| 130 |
+
"epoch": 0.001462576328202128,
|
| 131 |
+
"grad_norm": 1.344064712524414,
|
| 132 |
+
"learning_rate": 0.0016935087808430283,
|
| 133 |
+
"loss": 5.121601867675781,
|
| 134 |
+
"step": 110
|
| 135 |
+
},
|
| 136 |
+
{
|
| 137 |
+
"epoch": 0.0015955378125841397,
|
| 138 |
+
"grad_norm": 1.2178468704223633,
|
| 139 |
+
"learning_rate": 0.0016935087808430283,
|
| 140 |
+
"loss": 5.022267913818359,
|
| 141 |
+
"step": 120
|
| 142 |
+
},
|
| 143 |
+
{
|
| 144 |
+
"epoch": 0.0015955378125841397,
|
| 145 |
+
"eval_loss": 4.967755317687988,
|
| 146 |
+
"eval_runtime": 4.7177,
|
| 147 |
+
"eval_samples_per_second": 217.055,
|
| 148 |
+
"eval_steps_per_second": 27.132,
|
| 149 |
+
"step": 120
|
| 150 |
+
},
|
| 151 |
+
{
|
| 152 |
+
"epoch": 0.0017284992969661514,
|
| 153 |
+
"grad_norm": 1.7606405019760132,
|
| 154 |
+
"learning_rate": 0.0018816764231589203,
|
| 155 |
+
"loss": 4.9363243103027346,
|
| 156 |
+
"step": 130
|
| 157 |
+
},
|
| 158 |
+
{
|
| 159 |
+
"epoch": 0.001861460781348163,
|
| 160 |
+
"grad_norm": 1.4632689952850342,
|
| 161 |
+
"learning_rate": 0.0018816764231589203,
|
| 162 |
+
"loss": 4.849981307983398,
|
| 163 |
+
"step": 140
|
| 164 |
+
},
|
| 165 |
+
{
|
| 166 |
+
"epoch": 0.001861460781348163,
|
| 167 |
+
"eval_loss": 4.82466459274292,
|
| 168 |
+
"eval_runtime": 4.7613,
|
| 169 |
+
"eval_samples_per_second": 215.068,
|
| 170 |
+
"eval_steps_per_second": 26.883,
|
| 171 |
+
"step": 140
|
| 172 |
+
},
|
| 173 |
+
{
|
| 174 |
+
"epoch": 0.0019944222657301744,
|
| 175 |
+
"grad_norm": 1.3040728569030762,
|
| 176 |
+
"learning_rate": 0.002090751581287689,
|
| 177 |
+
"loss": 4.788387298583984,
|
| 178 |
+
"step": 150
|
| 179 |
+
},
|
| 180 |
+
{
|
| 181 |
+
"epoch": 0.0021273837501121864,
|
| 182 |
+
"grad_norm": 1.1895735263824463,
|
| 183 |
+
"learning_rate": 0.002090751581287689,
|
| 184 |
+
"loss": 4.7253562927246096,
|
| 185 |
+
"step": 160
|
| 186 |
+
},
|
| 187 |
+
{
|
| 188 |
+
"epoch": 0.0021273837501121864,
|
| 189 |
+
"eval_loss": 4.692420959472656,
|
| 190 |
+
"eval_runtime": 4.7224,
|
| 191 |
+
"eval_samples_per_second": 216.838,
|
| 192 |
+
"eval_steps_per_second": 27.105,
|
| 193 |
+
"step": 160
|
| 194 |
+
},
|
| 195 |
+
{
|
| 196 |
+
"epoch": 0.002260345234494198,
|
| 197 |
+
"grad_norm": 1.218579888343811,
|
| 198 |
+
"learning_rate": 0.002323057312541877,
|
| 199 |
+
"loss": 4.674822616577148,
|
| 200 |
+
"step": 170
|
| 201 |
+
},
|
| 202 |
+
{
|
| 203 |
+
"epoch": 0.0023933067188762094,
|
| 204 |
+
"grad_norm": 1.7899105548858643,
|
| 205 |
+
"learning_rate": 0.002323057312541877,
|
| 206 |
+
"loss": 4.62569580078125,
|
| 207 |
+
"step": 180
|
| 208 |
+
},
|
| 209 |
+
{
|
| 210 |
+
"epoch": 0.0023933067188762094,
|
| 211 |
+
"eval_loss": 4.606845855712891,
|
| 212 |
+
"eval_runtime": 4.7684,
|
| 213 |
+
"eval_samples_per_second": 214.747,
|
| 214 |
+
"eval_steps_per_second": 26.843,
|
| 215 |
+
"step": 180
|
| 216 |
+
},
|
| 217 |
+
{
|
| 218 |
+
"epoch": 0.0025262682032582214,
|
| 219 |
+
"grad_norm": 1.1465339660644531,
|
| 220 |
+
"learning_rate": 0.0025811747917131966,
|
| 221 |
+
"loss": 4.587751007080078,
|
| 222 |
+
"step": 190
|
| 223 |
+
},
|
| 224 |
+
{
|
| 225 |
+
"epoch": 0.002659229687640233,
|
| 226 |
+
"grad_norm": 1.4619438648223877,
|
| 227 |
+
"learning_rate": 0.0025811747917131966,
|
| 228 |
+
"loss": 4.551251220703125,
|
| 229 |
+
"step": 200
|
| 230 |
+
},
|
| 231 |
+
{
|
| 232 |
+
"epoch": 0.002659229687640233,
|
| 233 |
+
"eval_loss": 4.531779766082764,
|
| 234 |
+
"eval_runtime": 4.7447,
|
| 235 |
+
"eval_samples_per_second": 215.821,
|
| 236 |
+
"eval_steps_per_second": 26.978,
|
| 237 |
+
"step": 200
|
| 238 |
+
},
|
| 239 |
+
{
|
| 240 |
+
"epoch": 0.0027921911720222444,
|
| 241 |
+
"grad_norm": 1.514225721359253,
|
| 242 |
+
"learning_rate": 0.0028679719907924407,
|
| 243 |
+
"loss": 4.521094131469726,
|
| 244 |
+
"step": 210
|
| 245 |
+
},
|
| 246 |
+
{
|
| 247 |
+
"epoch": 0.002925152656404256,
|
| 248 |
+
"grad_norm": 1.3342633247375488,
|
| 249 |
+
"learning_rate": 0.0028679719907924407,
|
| 250 |
+
"loss": 4.481899261474609,
|
| 251 |
+
"step": 220
|
| 252 |
+
},
|
| 253 |
+
{
|
| 254 |
+
"epoch": 0.002925152656404256,
|
| 255 |
+
"eval_loss": 4.473923206329346,
|
| 256 |
+
"eval_runtime": 4.7271,
|
| 257 |
+
"eval_samples_per_second": 216.625,
|
| 258 |
+
"eval_steps_per_second": 27.078,
|
| 259 |
+
"step": 220
|
| 260 |
+
},
|
| 261 |
+
{
|
| 262 |
+
"epoch": 0.003058114140786268,
|
| 263 |
+
"grad_norm": 1.1252554655075073,
|
| 264 |
+
"learning_rate": 0.003186635545324934,
|
| 265 |
+
"loss": 4.454861068725586,
|
| 266 |
+
"step": 230
|
| 267 |
+
},
|
| 268 |
+
{
|
| 269 |
+
"epoch": 0.0031910756251682794,
|
| 270 |
+
"grad_norm": 0.9204785227775574,
|
| 271 |
+
"learning_rate": 0.003186635545324934,
|
| 272 |
+
"loss": 4.433399200439453,
|
| 273 |
+
"step": 240
|
| 274 |
+
},
|
| 275 |
+
{
|
| 276 |
+
"epoch": 0.0031910756251682794,
|
| 277 |
+
"eval_loss": 4.432496070861816,
|
| 278 |
+
"eval_runtime": 4.7255,
|
| 279 |
+
"eval_samples_per_second": 216.696,
|
| 280 |
+
"eval_steps_per_second": 27.087,
|
| 281 |
+
"step": 240
|
| 282 |
+
},
|
| 283 |
+
{
|
| 284 |
+
"epoch": 0.003324037109550291,
|
| 285 |
+
"grad_norm": 0.9875030517578125,
|
| 286 |
+
"learning_rate": 0.0035407061614721484,
|
| 287 |
+
"loss": 4.423421859741211,
|
| 288 |
+
"step": 250
|
| 289 |
+
},
|
| 290 |
+
{
|
| 291 |
+
"epoch": 0.003456998593932303,
|
| 292 |
+
"grad_norm": 1.1480664014816284,
|
| 293 |
+
"learning_rate": 0.0035407061614721484,
|
| 294 |
+
"loss": 4.397797393798828,
|
| 295 |
+
"step": 260
|
| 296 |
+
},
|
| 297 |
+
{
|
| 298 |
+
"epoch": 0.003456998593932303,
|
| 299 |
+
"eval_loss": 4.395366191864014,
|
| 300 |
+
"eval_runtime": 4.7188,
|
| 301 |
+
"eval_samples_per_second": 217.005,
|
| 302 |
+
"eval_steps_per_second": 27.126,
|
| 303 |
+
"step": 260
|
| 304 |
+
},
|
| 305 |
+
{
|
| 306 |
+
"epoch": 0.0035899600783143143,
|
| 307 |
+
"grad_norm": 1.0977789163589478,
|
| 308 |
+
"learning_rate": 0.003934117957191276,
|
| 309 |
+
"loss": 4.393301773071289,
|
| 310 |
+
"step": 270
|
| 311 |
+
},
|
| 312 |
+
{
|
| 313 |
+
"epoch": 0.003722921562696326,
|
| 314 |
+
"grad_norm": 1.0433765649795532,
|
| 315 |
+
"learning_rate": 0.003934117957191276,
|
| 316 |
+
"loss": 4.361746215820313,
|
| 317 |
+
"step": 280
|
| 318 |
+
},
|
| 319 |
+
{
|
| 320 |
+
"epoch": 0.003722921562696326,
|
| 321 |
+
"eval_loss": 4.3616042137146,
|
| 322 |
+
"eval_runtime": 4.7244,
|
| 323 |
+
"eval_samples_per_second": 216.749,
|
| 324 |
+
"eval_steps_per_second": 27.094,
|
| 325 |
+
"step": 280
|
| 326 |
+
},
|
| 327 |
+
{
|
| 328 |
+
"epoch": 0.0038558830470783374,
|
| 329 |
+
"grad_norm": 1.0764110088348389,
|
| 330 |
+
"learning_rate": 0.004371242174656973,
|
| 331 |
+
"loss": 4.3703865051269535,
|
| 332 |
+
"step": 290
|
| 333 |
+
},
|
| 334 |
+
{
|
| 335 |
+
"epoch": 0.003988844531460349,
|
| 336 |
+
"grad_norm": 1.165968894958496,
|
| 337 |
+
"learning_rate": 0.004371242174656973,
|
| 338 |
+
"loss": 4.350434875488281,
|
| 339 |
+
"step": 300
|
| 340 |
+
},
|
| 341 |
+
{
|
| 342 |
+
"epoch": 0.003988844531460349,
|
| 343 |
+
"eval_loss": 4.334297180175781,
|
| 344 |
+
"eval_runtime": 4.7184,
|
| 345 |
+
"eval_samples_per_second": 217.024,
|
| 346 |
+
"eval_steps_per_second": 27.128,
|
| 347 |
+
"step": 300
|
| 348 |
+
},
|
| 349 |
+
{
|
| 350 |
+
"epoch": 0.004121806015842361,
|
| 351 |
+
"grad_norm": 0.9461768269538879,
|
| 352 |
+
"learning_rate": 0.004856935749618859,
|
| 353 |
+
"loss": 4.3365730285644535,
|
| 354 |
+
"step": 310
|
| 355 |
+
},
|
| 356 |
+
{
|
| 357 |
+
"epoch": 0.004254767500224373,
|
| 358 |
+
"grad_norm": 0.8112942576408386,
|
| 359 |
+
"learning_rate": 0.004856935749618859,
|
| 360 |
+
"loss": 4.318856430053711,
|
| 361 |
+
"step": 320
|
| 362 |
+
},
|
| 363 |
+
{
|
| 364 |
+
"epoch": 0.004254767500224373,
|
| 365 |
+
"eval_loss": 4.319216251373291,
|
| 366 |
+
"eval_runtime": 4.7521,
|
| 367 |
+
"eval_samples_per_second": 215.484,
|
| 368 |
+
"eval_steps_per_second": 26.935,
|
| 369 |
+
"step": 320
|
| 370 |
+
},
|
| 371 |
+
{
|
| 372 |
+
"epoch": 0.004387728984606384,
|
| 373 |
+
"grad_norm": 0.8098408579826355,
|
| 374 |
+
"learning_rate": 0.005396595277354288,
|
| 375 |
+
"loss": 4.325001525878906,
|
| 376 |
+
"step": 330
|
| 377 |
+
},
|
| 378 |
+
{
|
| 379 |
+
"epoch": 0.004520690468988396,
|
| 380 |
+
"grad_norm": 0.9229351282119751,
|
| 381 |
+
"learning_rate": 0.005396595277354288,
|
| 382 |
+
"loss": 4.310536956787109,
|
| 383 |
+
"step": 340
|
| 384 |
+
},
|
| 385 |
+
{
|
| 386 |
+
"epoch": 0.004520690468988396,
|
| 387 |
+
"eval_loss": 4.303993225097656,
|
| 388 |
+
"eval_runtime": 4.7235,
|
| 389 |
+
"eval_samples_per_second": 216.787,
|
| 390 |
+
"eval_steps_per_second": 27.098,
|
| 391 |
+
"step": 340
|
| 392 |
+
},
|
| 393 |
+
{
|
| 394 |
+
"epoch": 0.004653651953370408,
|
| 395 |
+
"grad_norm": 0.7983461022377014,
|
| 396 |
+
"learning_rate": 0.005996216974838097,
|
| 397 |
+
"loss": 4.303889465332031,
|
| 398 |
+
"step": 350
|
| 399 |
+
},
|
| 400 |
+
{
|
| 401 |
+
"epoch": 0.004786613437752419,
|
| 402 |
+
"grad_norm": 0.9175242185592651,
|
| 403 |
+
"learning_rate": 0.005996216974838097,
|
| 404 |
+
"loss": 4.306822204589844,
|
| 405 |
+
"step": 360
|
| 406 |
+
},
|
| 407 |
+
{
|
| 408 |
+
"epoch": 0.004786613437752419,
|
| 409 |
+
"eval_loss": 4.303272247314453,
|
| 410 |
+
"eval_runtime": 4.7189,
|
| 411 |
+
"eval_samples_per_second": 217.0,
|
| 412 |
+
"eval_steps_per_second": 27.125,
|
| 413 |
+
"step": 360
|
| 414 |
+
},
|
| 415 |
+
{
|
| 416 |
+
"epoch": 0.004919574922134431,
|
| 417 |
+
"grad_norm": 0.6384010314941406,
|
| 418 |
+
"learning_rate": 0.005396595277354288,
|
| 419 |
+
"loss": 4.271321868896484,
|
| 420 |
+
"step": 370
|
| 421 |
+
},
|
| 422 |
+
{
|
| 423 |
+
"epoch": 0.005052536406516443,
|
| 424 |
+
"grad_norm": 0.6614411473274231,
|
| 425 |
+
"learning_rate": 0.005396595277354288,
|
| 426 |
+
"loss": 4.2241767883300785,
|
| 427 |
+
"step": 380
|
| 428 |
+
},
|
| 429 |
+
{
|
| 430 |
+
"epoch": 0.005052536406516443,
|
| 431 |
+
"eval_loss": 4.230008602142334,
|
| 432 |
+
"eval_runtime": 4.7416,
|
| 433 |
+
"eval_samples_per_second": 215.959,
|
| 434 |
+
"eval_steps_per_second": 26.995,
|
| 435 |
+
"step": 380
|
| 436 |
+
},
|
| 437 |
+
{
|
| 438 |
+
"epoch": 0.005185497890898454,
|
| 439 |
+
"grad_norm": 0.7242400050163269,
|
| 440 |
+
"learning_rate": 0.005996216974838097,
|
| 441 |
+
"loss": 4.2259784698486325,
|
| 442 |
+
"step": 390
|
| 443 |
+
},
|
| 444 |
+
{
|
| 445 |
+
"epoch": 0.005318459375280466,
|
| 446 |
+
"grad_norm": 0.8357770442962646,
|
| 447 |
+
"learning_rate": 0.005996216974838097,
|
| 448 |
+
"loss": 4.2215126037597654,
|
| 449 |
+
"step": 400
|
| 450 |
+
},
|
| 451 |
+
{
|
| 452 |
+
"epoch": 0.005318459375280466,
|
| 453 |
+
"eval_loss": 4.221831321716309,
|
| 454 |
+
"eval_runtime": 4.754,
|
| 455 |
+
"eval_samples_per_second": 215.399,
|
| 456 |
+
"eval_steps_per_second": 26.925,
|
| 457 |
+
"step": 400
|
| 458 |
+
},
|
| 459 |
+
{
|
| 460 |
+
"epoch": 0.005451420859662477,
|
| 461 |
+
"grad_norm": 0.7000677585601807,
|
| 462 |
+
"learning_rate": 0.006662463305375663,
|
| 463 |
+
"loss": 4.216830444335938,
|
| 464 |
+
"step": 410
|
| 465 |
+
},
|
| 466 |
+
{
|
| 467 |
+
"epoch": 0.005584382344044489,
|
| 468 |
+
"grad_norm": 0.5402432084083557,
|
| 469 |
+
"learning_rate": 0.006662463305375663,
|
| 470 |
+
"loss": 4.207019805908203,
|
| 471 |
+
"step": 420
|
| 472 |
+
},
|
| 473 |
+
{
|
| 474 |
+
"epoch": 0.005584382344044489,
|
| 475 |
+
"eval_loss": 4.203587532043457,
|
| 476 |
+
"eval_runtime": 4.7245,
|
| 477 |
+
"eval_samples_per_second": 216.743,
|
| 478 |
+
"eval_steps_per_second": 27.093,
|
| 479 |
+
"step": 420
|
| 480 |
+
},
|
| 481 |
+
{
|
| 482 |
+
"epoch": 0.005717343828426501,
|
| 483 |
+
"grad_norm": 0.6724000573158264,
|
| 484 |
+
"learning_rate": 0.007402737005972959,
|
| 485 |
+
"loss": 4.2224266052246096,
|
| 486 |
+
"step": 430
|
| 487 |
+
},
|
| 488 |
+
{
|
| 489 |
+
"epoch": 0.005850305312808512,
|
| 490 |
+
"grad_norm": 0.5229582786560059,
|
| 491 |
+
"learning_rate": 0.007402737005972959,
|
| 492 |
+
"loss": 4.224952316284179,
|
| 493 |
+
"step": 440
|
| 494 |
+
},
|
| 495 |
+
{
|
| 496 |
+
"epoch": 0.005850305312808512,
|
| 497 |
+
"eval_loss": 4.210740089416504,
|
| 498 |
+
"eval_runtime": 4.7257,
|
| 499 |
+
"eval_samples_per_second": 216.685,
|
| 500 |
+
"eval_steps_per_second": 27.086,
|
| 501 |
+
"step": 440
|
| 502 |
+
},
|
| 503 |
+
{
|
| 504 |
+
"epoch": 0.005983266797190524,
|
| 505 |
+
"grad_norm": 0.558103084564209,
|
| 506 |
+
"learning_rate": 0.006662463305375663,
|
| 507 |
+
"loss": 4.192767333984375,
|
| 508 |
+
"step": 450
|
| 509 |
+
},
|
| 510 |
+
{
|
| 511 |
+
"epoch": 0.006116228281572536,
|
| 512 |
+
"grad_norm": 0.5505457520484924,
|
| 513 |
+
"learning_rate": 0.006662463305375663,
|
| 514 |
+
"loss": 4.158684921264649,
|
| 515 |
+
"step": 460
|
| 516 |
+
},
|
| 517 |
+
{
|
| 518 |
+
"epoch": 0.006116228281572536,
|
| 519 |
+
"eval_loss": 4.15805196762085,
|
| 520 |
+
"eval_runtime": 4.7412,
|
| 521 |
+
"eval_samples_per_second": 215.98,
|
| 522 |
+
"eval_steps_per_second": 26.998,
|
| 523 |
+
"step": 460
|
| 524 |
+
},
|
| 525 |
+
{
|
| 526 |
+
"epoch": 0.006249189765954547,
|
| 527 |
+
"grad_norm": 0.3989739716053009,
|
| 528 |
+
"learning_rate": 0.007402737005972959,
|
| 529 |
+
"loss": 4.163018035888672,
|
| 530 |
+
"step": 470
|
| 531 |
+
},
|
| 532 |
+
{
|
| 533 |
+
"epoch": 0.006382151250336559,
|
| 534 |
+
"grad_norm": 0.5627162456512451,
|
| 535 |
+
"learning_rate": 0.007402737005972959,
|
| 536 |
+
"loss": 4.164209365844727,
|
| 537 |
+
"step": 480
|
| 538 |
+
},
|
| 539 |
+
{
|
| 540 |
+
"epoch": 0.006382151250336559,
|
| 541 |
+
"eval_loss": 4.168121337890625,
|
| 542 |
+
"eval_runtime": 4.7243,
|
| 543 |
+
"eval_samples_per_second": 216.751,
|
| 544 |
+
"eval_steps_per_second": 27.094,
|
| 545 |
+
"step": 480
|
| 546 |
+
},
|
| 547 |
+
{
|
| 548 |
+
"epoch": 0.006515112734718571,
|
| 549 |
+
"grad_norm": 0.4144296944141388,
|
| 550 |
+
"learning_rate": 0.006662463305375663,
|
| 551 |
+
"loss": 4.1303466796875,
|
| 552 |
+
"step": 490
|
| 553 |
+
},
|
| 554 |
+
{
|
| 555 |
+
"epoch": 0.006648074219100582,
|
| 556 |
+
"grad_norm": 0.4982388913631439,
|
| 557 |
+
"learning_rate": 0.006662463305375663,
|
| 558 |
+
"loss": 4.116038513183594,
|
| 559 |
+
"step": 500
|
| 560 |
+
},
|
| 561 |
+
{
|
| 562 |
+
"epoch": 0.006648074219100582,
|
| 563 |
+
"eval_loss": 4.106968879699707,
|
| 564 |
+
"eval_runtime": 4.7284,
|
| 565 |
+
"eval_samples_per_second": 216.562,
|
| 566 |
+
"eval_steps_per_second": 27.07,
|
| 567 |
+
"step": 500
|
| 568 |
+
},
|
| 569 |
+
{
|
| 570 |
+
"epoch": 0.006781035703482594,
|
| 571 |
+
"grad_norm": 0.48199519515037537,
|
| 572 |
+
"learning_rate": 0.007402737005972959,
|
| 573 |
+
"loss": 4.121062850952148,
|
| 574 |
+
"step": 510
|
| 575 |
+
},
|
| 576 |
+
{
|
| 577 |
+
"epoch": 0.006913997187864606,
|
| 578 |
+
"grad_norm": 0.5215345025062561,
|
| 579 |
+
"learning_rate": 0.007402737005972959,
|
| 580 |
+
"loss": 4.111361694335938,
|
| 581 |
+
"step": 520
|
| 582 |
+
},
|
| 583 |
+
{
|
| 584 |
+
"epoch": 0.006913997187864606,
|
| 585 |
+
"eval_loss": 4.113387107849121,
|
| 586 |
+
"eval_runtime": 4.7279,
|
| 587 |
+
"eval_samples_per_second": 216.586,
|
| 588 |
+
"eval_steps_per_second": 27.073,
|
| 589 |
+
"step": 520
|
| 590 |
+
},
|
| 591 |
+
{
|
| 592 |
+
"epoch": 0.007046958672246617,
|
| 593 |
+
"grad_norm": 0.3848040997982025,
|
| 594 |
+
"learning_rate": 0.006662463305375663,
|
| 595 |
+
"loss": 4.077674484252929,
|
| 596 |
+
"step": 530
|
| 597 |
+
},
|
| 598 |
+
{
|
| 599 |
+
"epoch": 0.007179920156628629,
|
| 600 |
+
"grad_norm": 0.4323752224445343,
|
| 601 |
+
"learning_rate": 0.006662463305375663,
|
| 602 |
+
"loss": 4.059872817993164,
|
| 603 |
+
"step": 540
|
| 604 |
+
},
|
| 605 |
+
{
|
| 606 |
+
"epoch": 0.007179920156628629,
|
| 607 |
+
"eval_loss": 4.064393520355225,
|
| 608 |
+
"eval_runtime": 4.7217,
|
| 609 |
+
"eval_samples_per_second": 216.87,
|
| 610 |
+
"eval_steps_per_second": 27.109,
|
| 611 |
+
"step": 540
|
| 612 |
+
},
|
| 613 |
+
{
|
| 614 |
+
"epoch": 0.007312881641010641,
|
| 615 |
+
"grad_norm": 0.4613271653652191,
|
| 616 |
+
"learning_rate": 0.007402737005972959,
|
| 617 |
+
"loss": 4.067990493774414,
|
| 618 |
+
"step": 550
|
| 619 |
+
},
|
| 620 |
+
{
|
| 621 |
+
"epoch": 0.007445843125392652,
|
| 622 |
+
"grad_norm": 0.3521135449409485,
|
| 623 |
+
"learning_rate": 0.007402737005972959,
|
| 624 |
+
"loss": 4.075962066650391,
|
| 625 |
+
"step": 560
|
| 626 |
+
},
|
| 627 |
+
{
|
| 628 |
+
"epoch": 0.007445843125392652,
|
| 629 |
+
"eval_loss": 4.064351558685303,
|
| 630 |
+
"eval_runtime": 4.7464,
|
| 631 |
+
"eval_samples_per_second": 215.745,
|
| 632 |
+
"eval_steps_per_second": 26.968,
|
| 633 |
+
"step": 560
|
| 634 |
+
},
|
| 635 |
+
{
|
| 636 |
+
"epoch": 0.007578804609774664,
|
| 637 |
+
"grad_norm": 0.3451947569847107,
|
| 638 |
+
"learning_rate": 0.006662463305375663,
|
| 639 |
+
"loss": 4.04071159362793,
|
| 640 |
+
"step": 570
|
| 641 |
+
},
|
| 642 |
+
{
|
| 643 |
+
"epoch": 0.007711766094156675,
|
| 644 |
+
"grad_norm": 0.2876155972480774,
|
| 645 |
+
"learning_rate": 0.006662463305375663,
|
| 646 |
+
"loss": 4.019502258300781,
|
| 647 |
+
"step": 580
|
| 648 |
+
},
|
| 649 |
+
{
|
| 650 |
+
"epoch": 0.007711766094156675,
|
| 651 |
+
"eval_loss": 4.021025657653809,
|
| 652 |
+
"eval_runtime": 4.725,
|
| 653 |
+
"eval_samples_per_second": 216.718,
|
| 654 |
+
"eval_steps_per_second": 27.09,
|
| 655 |
+
"step": 580
|
| 656 |
+
},
|
| 657 |
+
{
|
| 658 |
+
"epoch": 0.007844727578538687,
|
| 659 |
+
"grad_norm": 0.47599413990974426,
|
| 660 |
+
"learning_rate": 0.007402737005972959,
|
| 661 |
+
"loss": 4.03144416809082,
|
| 662 |
+
"step": 590
|
| 663 |
+
},
|
| 664 |
+
{
|
| 665 |
+
"epoch": 0.007977689062920698,
|
| 666 |
+
"grad_norm": 0.3505311608314514,
|
| 667 |
+
"learning_rate": 0.007402737005972959,
|
| 668 |
+
"loss": 4.022784805297851,
|
| 669 |
+
"step": 600
|
| 670 |
+
},
|
| 671 |
+
{
|
| 672 |
+
"epoch": 0.007977689062920698,
|
| 673 |
+
"eval_loss": 4.033039093017578,
|
| 674 |
+
"eval_runtime": 4.7522,
|
| 675 |
+
"eval_samples_per_second": 215.479,
|
| 676 |
+
"eval_steps_per_second": 26.935,
|
| 677 |
+
"step": 600
|
| 678 |
+
},
|
| 679 |
+
{
|
| 680 |
+
"epoch": 0.00811065054730271,
|
| 681 |
+
"grad_norm": 0.35564446449279785,
|
| 682 |
+
"learning_rate": 0.006662463305375663,
|
| 683 |
+
"loss": 4.009813690185547,
|
| 684 |
+
"step": 610
|
| 685 |
+
},
|
| 686 |
+
{
|
| 687 |
+
"epoch": 0.008243612031684722,
|
| 688 |
+
"grad_norm": 0.3954601287841797,
|
| 689 |
+
"learning_rate": 0.006662463305375663,
|
| 690 |
+
"loss": 3.9844127655029298,
|
| 691 |
+
"step": 620
|
| 692 |
+
},
|
| 693 |
+
{
|
| 694 |
+
"epoch": 0.008243612031684722,
|
| 695 |
+
"eval_loss": 3.992824077606201,
|
| 696 |
+
"eval_runtime": 4.7343,
|
| 697 |
+
"eval_samples_per_second": 216.292,
|
| 698 |
+
"eval_steps_per_second": 27.037,
|
| 699 |
+
"step": 620
|
| 700 |
+
},
|
| 701 |
+
{
|
| 702 |
+
"epoch": 0.008376573516066733,
|
| 703 |
+
"grad_norm": 0.33219683170318604,
|
| 704 |
+
"learning_rate": 0.007402737005972959,
|
| 705 |
+
"loss": 3.9896926879882812,
|
| 706 |
+
"step": 630
|
| 707 |
+
},
|
| 708 |
+
{
|
| 709 |
+
"epoch": 0.008509535000448746,
|
| 710 |
+
"grad_norm": 0.33283373713493347,
|
| 711 |
+
"learning_rate": 0.007402737005972959,
|
| 712 |
+
"loss": 3.994375991821289,
|
| 713 |
+
"step": 640
|
| 714 |
+
},
|
| 715 |
+
{
|
| 716 |
+
"epoch": 0.008509535000448746,
|
| 717 |
+
"eval_loss": 3.992628574371338,
|
| 718 |
+
"eval_runtime": 4.7346,
|
| 719 |
+
"eval_samples_per_second": 216.281,
|
| 720 |
+
"eval_steps_per_second": 27.035,
|
| 721 |
+
"step": 640
|
| 722 |
+
},
|
| 723 |
+
{
|
| 724 |
+
"epoch": 0.008642496484830757,
|
| 725 |
+
"grad_norm": 0.30045098066329956,
|
| 726 |
+
"learning_rate": 0.006662463305375663,
|
| 727 |
+
"loss": 3.975360870361328,
|
| 728 |
+
"step": 650
|
| 729 |
+
},
|
| 730 |
+
{
|
| 731 |
+
"epoch": 0.008775457969212768,
|
| 732 |
+
"grad_norm": 0.28163373470306396,
|
| 733 |
+
"learning_rate": 0.006662463305375663,
|
| 734 |
+
"loss": 3.9635448455810547,
|
| 735 |
+
"step": 660
|
| 736 |
+
},
|
| 737 |
+
{
|
| 738 |
+
"epoch": 0.008775457969212768,
|
| 739 |
+
"eval_loss": 3.9626264572143555,
|
| 740 |
+
"eval_runtime": 4.7196,
|
| 741 |
+
"eval_samples_per_second": 216.97,
|
| 742 |
+
"eval_steps_per_second": 27.121,
|
| 743 |
+
"step": 660
|
| 744 |
+
},
|
| 745 |
+
{
|
| 746 |
+
"epoch": 0.00890841945359478,
|
| 747 |
+
"grad_norm": 0.34984496235847473,
|
| 748 |
+
"learning_rate": 0.007402737005972959,
|
| 749 |
+
"loss": 3.9630859375,
|
| 750 |
+
"step": 670
|
| 751 |
+
},
|
| 752 |
+
{
|
| 753 |
+
"epoch": 0.009041380937976792,
|
| 754 |
+
"grad_norm": 0.3046402335166931,
|
| 755 |
+
"learning_rate": 0.007402737005972959,
|
| 756 |
+
"loss": 3.9557125091552736,
|
| 757 |
+
"step": 680
|
| 758 |
+
},
|
| 759 |
+
{
|
| 760 |
+
"epoch": 0.009041380937976792,
|
| 761 |
+
"eval_loss": 3.964042901992798,
|
| 762 |
+
"eval_runtime": 4.7245,
|
| 763 |
+
"eval_samples_per_second": 216.743,
|
| 764 |
+
"eval_steps_per_second": 27.093,
|
| 765 |
+
"step": 680
|
| 766 |
+
},
|
| 767 |
+
{
|
| 768 |
+
"epoch": 0.009174342422358803,
|
| 769 |
+
"grad_norm": 0.39222627878189087,
|
| 770 |
+
"learning_rate": 0.006662463305375663,
|
| 771 |
+
"loss": 3.9501678466796877,
|
| 772 |
+
"step": 690
|
| 773 |
+
},
|
| 774 |
+
{
|
| 775 |
+
"epoch": 0.009307303906740816,
|
| 776 |
+
"grad_norm": 0.2476985901594162,
|
| 777 |
+
"learning_rate": 0.006662463305375663,
|
| 778 |
+
"loss": 3.9329227447509765,
|
| 779 |
+
"step": 700
|
| 780 |
+
},
|
| 781 |
+
{
|
| 782 |
+
"epoch": 0.009307303906740816,
|
| 783 |
+
"eval_loss": 3.935375213623047,
|
| 784 |
+
"eval_runtime": 4.7404,
|
| 785 |
+
"eval_samples_per_second": 216.013,
|
| 786 |
+
"eval_steps_per_second": 27.002,
|
| 787 |
+
"step": 700
|
| 788 |
+
},
|
| 789 |
+
{
|
| 790 |
+
"epoch": 0.009440265391122827,
|
| 791 |
+
"grad_norm": 0.2824518084526062,
|
| 792 |
+
"learning_rate": 0.007402737005972959,
|
| 793 |
+
"loss": 3.935717010498047,
|
| 794 |
+
"step": 710
|
| 795 |
+
},
|
| 796 |
+
{
|
| 797 |
+
"epoch": 0.009573226875504838,
|
| 798 |
+
"grad_norm": 0.3063017427921295,
|
| 799 |
+
"learning_rate": 0.007402737005972959,
|
| 800 |
+
"loss": 3.937954330444336,
|
| 801 |
+
"step": 720
|
| 802 |
+
},
|
| 803 |
+
{
|
| 804 |
+
"epoch": 0.009573226875504838,
|
| 805 |
+
"eval_loss": 3.9412312507629395,
|
| 806 |
+
"eval_runtime": 4.7756,
|
| 807 |
+
"eval_samples_per_second": 214.421,
|
| 808 |
+
"eval_steps_per_second": 26.803,
|
| 809 |
+
"step": 720
|
| 810 |
+
},
|
| 811 |
+
{
|
| 812 |
+
"epoch": 0.00970618835988685,
|
| 813 |
+
"grad_norm": 0.24896036088466644,
|
| 814 |
+
"learning_rate": 0.006662463305375663,
|
| 815 |
+
"loss": 3.9285778045654296,
|
| 816 |
+
"step": 730
|
| 817 |
+
},
|
| 818 |
+
{
|
| 819 |
+
"epoch": 0.009839149844268862,
|
| 820 |
+
"grad_norm": 0.40740466117858887,
|
| 821 |
+
"learning_rate": 0.006662463305375663,
|
| 822 |
+
"loss": 3.9202159881591796,
|
| 823 |
+
"step": 740
|
| 824 |
+
},
|
| 825 |
+
{
|
| 826 |
+
"epoch": 0.009839149844268862,
|
| 827 |
+
"eval_loss": 3.9166951179504395,
|
| 828 |
+
"eval_runtime": 4.725,
|
| 829 |
+
"eval_samples_per_second": 216.721,
|
| 830 |
+
"eval_steps_per_second": 27.09,
|
| 831 |
+
"step": 740
|
| 832 |
+
},
|
| 833 |
+
{
|
| 834 |
+
"epoch": 0.009972111328650873,
|
| 835 |
+
"grad_norm": 0.32246750593185425,
|
| 836 |
+
"learning_rate": 0.007402737005972959,
|
| 837 |
+
"loss": 3.914220428466797,
|
| 838 |
+
"step": 750
|
| 839 |
+
},
|
| 840 |
+
{
|
| 841 |
+
"epoch": 0.010105072813032885,
|
| 842 |
+
"grad_norm": 0.33994966745376587,
|
| 843 |
+
"learning_rate": 0.007402737005972959,
|
| 844 |
+
"loss": 3.9243412017822266,
|
| 845 |
+
"step": 760
|
| 846 |
+
},
|
| 847 |
+
{
|
| 848 |
+
"epoch": 0.010105072813032885,
|
| 849 |
+
"eval_loss": 3.9266605377197266,
|
| 850 |
+
"eval_runtime": 4.7222,
|
| 851 |
+
"eval_samples_per_second": 216.847,
|
| 852 |
+
"eval_steps_per_second": 27.106,
|
| 853 |
+
"step": 760
|
| 854 |
+
},
|
| 855 |
+
{
|
| 856 |
+
"epoch": 0.010105072813032885,
|
| 857 |
+
"step": 760,
|
| 858 |
+
"total_flos": 6.206104799281152e+16,
|
| 859 |
+
"train_loss": 4.492900440567418,
|
| 860 |
+
"train_runtime": 3508.751,
|
| 861 |
+
"train_samples_per_second": 55.45,
|
| 862 |
+
"train_steps_per_second": 0.217
|
| 863 |
+
}
|
| 864 |
+
],
|
| 865 |
+
"logging_steps": 10,
|
| 866 |
+
"max_steps": 760,
|
| 867 |
+
"num_input_tokens_seen": 0,
|
| 868 |
+
"num_train_epochs": 1,
|
| 869 |
+
"save_steps": 5,
|
| 870 |
+
"stateful_callbacks": {
|
| 871 |
+
"TrainerControl": {
|
| 872 |
+
"args": {
|
| 873 |
+
"should_epoch_stop": false,
|
| 874 |
+
"should_evaluate": false,
|
| 875 |
+
"should_log": false,
|
| 876 |
+
"should_save": true,
|
| 877 |
+
"should_training_stop": false
|
| 878 |
+
},
|
| 879 |
+
"attributes": {}
|
| 880 |
+
}
|
| 881 |
+
},
|
| 882 |
+
"total_flos": 6.206104799281152e+16,
|
| 883 |
+
"train_batch_size": 8,
|
| 884 |
+
"trial_name": null,
|
| 885 |
+
"trial_params": null
|
| 886 |
+
}
|
scheduler/model.toml
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
# LLaMA 77M params
|
| 3 |
+
# huggingface/transformers/.. models/llama/configuration_llama.py
|
| 4 |
+
|
| 5 |
+
model_type = "llama"
|
| 6 |
+
tie_word_embeddings = true
|
| 7 |
+
hidden_size = 768 # embedding dimension
|
| 8 |
+
intermediate_size = 2048 # FFN hidden size (typically 4x hidden_size)
|
| 9 |
+
max_position_embeddings = 1024
|
| 10 |
+
num_attention_heads = 8 # attention heads (must divide hidden_size evenly)
|
| 11 |
+
num_hidden_layers = 8 # transformer layers
|
| 12 |
+
num_key_value_heads = 4
|
| 13 |
+
vocab_size = 32768
|
scheduler/wsd_10/REPORT.md
ADDED
|
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Experiment: wsd_10 — warmup-stable-decay, 10% linear decay to 0
|
| 2 |
+
|
| 3 |
+
**Wall-clock:** 60 min · **Recipe:** best known (sdpa + bf16 mixed + muonq @ lr 8e-3), schedule: warmup 200 → stable 484 @ 8e-3 → linear decay to 0 over 76 steps (horizon `max_steps = 760`)
|
| 4 |
+
|
| 5 |
+
## Result: big win over constant LR — but the 20% decay (`../wsd_20/`) is better on every axis
|
| 6 |
+
|
| 7 |
+
| metric | wsd_10 | wsd_20 | constant-LR reference |
|
| 8 |
+
|---|---|---|---|
|
| 9 |
+
| optimizer steps in 60 min | 740 | 730 | 759 |
|
| 10 |
+
| final logged eval loss / ppl | 3.8641 / 47.7 (step 688 — 4 steps into decay, undersells final model) | 3.8023 / 44.8 | 3.8583 / 47.4 |
|
| 11 |
+
| bake score | 52.68/100 (DOUGH) | **55.61** | 46.83 |
|
| 12 |
+
| held-out bits/byte | 1.2821 | **1.2769** | 1.3114 |
|
| 13 |
+
| chat bits/byte | 1.1608 | **1.1571** | 1.1914 |
|
| 14 |
+
| logic accuracy | 0.625 (+0.0117) | **0.650 (+0.0242)** | 0.600 (+0.018) |
|
| 15 |
+
| hygiene (greedy loop) | 68.5/100 (33.8w) | **79.2 (23.8w)** | 47.7 (59.8w) |
|
| 16 |
+
| grad norms | 0.05–9.45, 0 non-finite | 0.06–9.45 | 0.09–9.45 |
|
| 17 |
+
| peak GPU | 7.2 GB | 7.2 GB | 7.2 GB |
|
| 18 |
+
|
| 19 |
+
+5.9 bake points over the constant-LR reference and a partial hygiene recovery (59.8w → 33.8w) — but dominated by wsd_20 everywhere. Note the trainer eval curve barely captures the decay at all: the last logged eval (step 688) is 4 steps after decay onset (684), while the saved final model is at step 740 with the decay ~74% done (LR 2.21e-3, wall-clock stop cut the last 20 steps). The evaluate2 numbers are measured on the actual final model.
|
| 20 |
+
|
| 21 |
+
## Eval loss curve (5-min intervals)
|
| 22 |
+
|
| 23 |
+
| step | eval loss | | step | eval loss |
|
| 24 |
+
|---|---|---|---|---|
|
| 25 |
+
| 67 | 5.7043 | | 438 | 4.0252 |
|
| 26 |
+
| 131 | 4.9471 | | 500 | 3.9743 |
|
| 27 |
+
| 194 | 4.5693 | | 563 | 3.9288 |
|
| 28 |
+
| 257 | 4.3185 | | 626 | 3.8911 |
|
| 29 |
+
| 317 | 4.1820 | | 688 | 3.8641 |
|
| 30 |
+
| 379 | 4.0923 | | | |
|
| 31 |
+
|
| 32 |
+
## Verdict
|
| 33 |
+
|
| 34 |
+
**Decay helps a lot; more decay helps more (at this horizon).** 10% was SmolLM's lower bound and it leaves quality on the table vs 20% — the stable phase's extra 76 steps at peak LR are worth less than a completed anneal. See `../COMPARISON.md`.
|
scheduler/wsd_10/config.toml
ADDED
|
@@ -0,0 +1,89 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Scheduler series — new best recipe (sdpa + bf16 mixed + muonq @ lr 8e-3), only the LR schedule varies.
|
| 2 |
+
# This run: WSD (warmup-stable-decay), SmolLM2-style — linear decay to 0 over the final 10% (76 steps).
|
| 3 |
+
model_config = "../model.toml"
|
| 4 |
+
|
| 5 |
+
[data]
|
| 6 |
+
# Data files (supports glob patterns or lists of files)
|
| 7 |
+
train_files = ["~/Dev/vintage-LLM/training/data/train_*.bin"]
|
| 8 |
+
valid_files = ["~/Dev/vintage-LLM/training/data/valid_*.bin"]
|
| 9 |
+
|
| 10 |
+
# Sequence length for each dataset sample
|
| 11 |
+
# Must be <= model.max_position_embeddings
|
| 12 |
+
max_seq_length = 1024
|
| 13 |
+
|
| 14 |
+
# Tokenizer (must match the one used to create .bin files)
|
| 15 |
+
tokenizer = "~/Dev/vintage-LLM/tokenizers/t-v3/"
|
| 16 |
+
|
| 17 |
+
[training]
|
| 18 |
+
# Training duration
|
| 19 |
+
num_train_epochs = 1
|
| 20 |
+
max_steps = 760 # scheduler horizon = achievable steps/hour (was -1: full epoch, so decay never happened)
|
| 21 |
+
|
| 22 |
+
# Optional: Stop training after this many minutes of wall-clock time.
|
| 23 |
+
max_train_minutes = 60
|
| 24 |
+
|
| 25 |
+
# Output directories
|
| 26 |
+
output_dir = "./"
|
| 27 |
+
final_model_dir = "./final"
|
| 28 |
+
|
| 29 |
+
# Batch sizes
|
| 30 |
+
# Effective batch size = num_devices × per_device_train_batch_size × gradient_accumulation_steps
|
| 31 |
+
per_device_train_batch_size = 8 # batch size per GPU (8x1024 tokens fits 16GB with headroom for eval)
|
| 32 |
+
per_device_eval_batch_size = 8
|
| 33 |
+
gradient_accumulation_steps = 32 # accumulate gradients over multiple mini-batches before updating the params
|
| 34 |
+
|
| 35 |
+
# Optimizer: MuonQ (4-bit quantized Muon, ../../MuonQ repo) — wired through
|
| 36 |
+
# optimizer_cls_and_kwargs in base_train.py. 2D hidden weights get Muon
|
| 37 |
+
# (Newton-Schulz 5, 4-bit mu-law-companded momentum, rank-16 residual
|
| 38 |
+
# factorization); embeddings/head/norms get its internal AdamW backup.
|
| 39 |
+
# Muon/quantization knobs (muon_*, muonq_*) default to the repo's own
|
| 40 |
+
# llama-60m "muonq" recipe in base_train.py.
|
| 41 |
+
optim = "muonq"
|
| 42 |
+
|
| 43 |
+
# Optimizer settings — lr/wd from MuonQ's llama-60m recipe (Muon takes
|
| 44 |
+
# spectral-norm-scaled steps; do not copy the AdamW 5e-4 here)
|
| 45 |
+
learning_rate = 8e-3
|
| 46 |
+
weight_decay = 0.1
|
| 47 |
+
adam_beta1 = 0.9
|
| 48 |
+
adam_beta2 = 0.95
|
| 49 |
+
max_grad_norm = 1.0
|
| 50 |
+
|
| 51 |
+
# Learning rate scheduler
|
| 52 |
+
lr_scheduler_type = "warmup_stable_decay" # warmup 200 -> stable 484 @ peak -> linear decay 76 steps to 0
|
| 53 |
+
lr_scheduler_kwargs = { num_stable_steps = 484, num_decay_steps = 76, decay_type = "linear", min_lr_ratio = 0.0 }
|
| 54 |
+
warmup_steps = 200 # 1-hour series standard (muonq-v2's 400 was a 2-hour recipe)
|
| 55 |
+
|
| 56 |
+
# Precision (choose based on your GPU)
|
| 57 |
+
bf16 = true # bfloat16 autocast + fp32 master weights (the dtype-series safe recipe)
|
| 58 |
+
fp16 = false
|
| 59 |
+
|
| 60 |
+
# Performance
|
| 61 |
+
torch_compile = false
|
| 62 |
+
gradient_checkpointing = false # trade compute for memory (enables larger models)
|
| 63 |
+
neftune_noise_alpha = 0.0 # add noise during training to improve model performance (fine-tuning)
|
| 64 |
+
|
| 65 |
+
# Checkpointing
|
| 66 |
+
# save_strategy can be "steps", "epoch", or "minutes" (wall-clock time).
|
| 67 |
+
save_strategy = "minutes"
|
| 68 |
+
save_steps = 5 # checkpoint every X minutes
|
| 69 |
+
save_total_limit = 3 # keep only X most recent checkpoints
|
| 70 |
+
|
| 71 |
+
# Evaluation
|
| 72 |
+
# eval_strategy can be "steps", "epoch", or "minutes" (wall-clock time).
|
| 73 |
+
eval_strategy = "minutes"
|
| 74 |
+
eval_steps = 5 # evaluate every X minutes
|
| 75 |
+
# Evaluate on a fixed random subset of the validation set (0 = use all)
|
| 76 |
+
max_eval_samples = 5000
|
| 77 |
+
|
| 78 |
+
# Logging
|
| 79 |
+
logging_strategy = "steps"
|
| 80 |
+
logging_steps = 10 # log every 10 steps
|
| 81 |
+
logging_first_step = true # log the first step
|
| 82 |
+
|
| 83 |
+
# Reproducibility
|
| 84 |
+
seed = 42
|
| 85 |
+
|
| 86 |
+
# Performance
|
| 87 |
+
dataloader_num_workers = 2 # number of data loading threads
|
| 88 |
+
dataloader_prefetch_factor = 2 # batches prefetched per worker
|
| 89 |
+
dataloader_pin_memory = true # pin memory for faster GPU transfer
|
scheduler/wsd_10/eval1.log
ADDED
|
@@ -0,0 +1,212 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
W0814 13:06:20.261000 44298 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
|
| 2 |
+
W0814 13:06:20.283000 44298 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
|
| 3 |
+
|
| 4 |
+
==============================================================================
|
| 5 |
+
CHECKPOINT INSPECTION
|
| 6 |
+
==============================================================================
|
| 7 |
+
checkpoint : final
|
| 8 |
+
tokenizer : final
|
| 9 |
+
device : cuda dtype: torch.bfloat16 seed: 1337
|
| 10 |
+
|
| 11 |
+
|
| 12 |
+
==============================================================================
|
| 13 |
+
SUITE: MODEL INFO AND TRAINING LINEAGE
|
| 14 |
+
==============================================================================
|
| 15 |
+
checkpoint : final
|
| 16 |
+
model_type : llama
|
| 17 |
+
architecture : LlamaForCausalLM
|
| 18 |
+
device : cuda
|
| 19 |
+
dtype : torch.bfloat16
|
| 20 |
+
parameters : 77,083,392 (77.08M)
|
| 21 |
+
disk size : 308 MB
|
| 22 |
+
layers / hidden / heads : 8 / 768 / 8 (KV heads: 4)
|
| 23 |
+
vocab (cfg / tokenizer) : 32768 / 32752
|
| 24 |
+
context length : 1024
|
| 25 |
+
chat template : True
|
| 26 |
+
training lineage : base pretraining, ctx 1024
|
| 27 |
+
embedding mean norm : 3.7700 <- typical row length of the embedding matrix; very small or huge values signal trouble
|
| 28 |
+
embedding mean cosine : 0.1028 <- avg cosine between random embedding rows. Descriptive only: it starts near 0.0
|
| 29 |
+
because RANDOM init is near-orthogonal, and it GROWS as the model trains.
|
| 30 |
+
Near 0 therefore does NOT mean "healthy" - it can mean "barely trained".
|
| 31 |
+
(note: config vocab padded by 16 unused rows - harmless)
|
| 32 |
+
|
| 33 |
+
==============================================================================
|
| 34 |
+
SUITE: PERIOD FIDELITY ON FIXED PROBE SENTENCES
|
| 35 |
+
==============================================================================
|
| 36 |
+
Scoring 20 sentences (10 historical, 10 modern)
|
| 37 |
+
The MODERN/HISTORICAL ratio is the output that means something. The absolute
|
| 38 |
+
numbers come from ~20 sentences: too noisy to rank checkpoints (use evaluate2.py).
|
| 39 |
+
|
| 40 |
+
OVERALL (270 tokens)
|
| 41 |
+
perplexity : 258.34 <- 'average branching factor'; lower = less surprised = better fit to this text
|
| 42 |
+
mean token prob : 0.0838 <- average confidence on the actual next word
|
| 43 |
+
median token prob : 0.0065 <- typical confidence (robust to a few very bad tokens)
|
| 44 |
+
worst-token prob (min): 7.8e-08 <- the single most surprising word; tiny values = the model really didn't expect it
|
| 45 |
+
10th-percentile prob : 1.6e-05 <- confidence on the model's worst 10% of guesses
|
| 46 |
+
low-confidence tokens : 53.33% <- share of tokens predicted with <1% probability (red flags)
|
| 47 |
+
mean entropy : 4.627 nats <- how spread out the predictions are; ~0 = very certain, higher = hedging
|
| 48 |
+
|
| 49 |
+
GROUP COMPARISON
|
| 50 |
+
group sentences perplexity bits/byte
|
| 51 |
+
historical 10 78.90 1.189
|
| 52 |
+
modern 10 696.95 1.953
|
| 53 |
+
modern/historical ratio: 8.83 <- >1 means period text is easier for the model than modern text (desired for a vintage model with a ~1900 knowledge cutoff)
|
| 54 |
+
(bits/byte is the tokenizer-independent version - the only column here you may
|
| 55 |
+
compare between models that use DIFFERENT tokenizers.)
|
| 56 |
+
|
| 57 |
+
PER-SENTENCE DETAIL (sorted worst-first)
|
| 58 |
+
[modern ] ppl= 5887.24 minp=3.1e-07 The engineering team rewrote the whole backend micro
|
| 59 |
+
[modern ] ppl= 3209.93 minp=1.9e-06 Robots on the automated assembly line handle the ent
|
| 60 |
+
[modern ] ppl= 1188.94 minp=1.4e-06 Small retail startups now run almost all of their bu
|
| 61 |
+
[modern ] ppl= 1106.63 minp=1.3e-06 Three hours stuck in traffic in the pouring rain mad
|
| 62 |
+
[modern ] ppl= 555.52 minp=1.0e-05 She is finishing a graduate degree in machine learni
|
| 63 |
+
[modern ] ppl= 513.64 minp=1.2e-06 Millions of shoppers queue outside the store because
|
| 64 |
+
[modern ] ppl= 460.49 minp=7.8e-08 Members of the European Union parliament voted on th
|
| 65 |
+
[modern ] ppl= 428.78 minp=2.7e-07 She gave up her whole weekend to help me move apartm
|
| 66 |
+
[historical] ppl= 294.55 minp=1.2e-05 The keeper fed the great serpent which the naturalis
|
| 67 |
+
[modern ] ppl= 219.43 minp=9.1e-06 I switched to a plant-based diet last year and I no
|
| 68 |
+
[modern ] ppl= 190.02 minp=1.9e-06 After years of hiding it from everyone at work, he t
|
| 69 |
+
[historical] ppl= 131.62 minp=8.6e-05 He drew a distinction so fine and over-scrupulous th
|
| 70 |
+
[historical] ppl= 94.75 minp=7.3e-06 The mountain rose above the valley in a silence sole
|
| 71 |
+
[historical] ppl= 87.33 minp=1.4e-04 The weavers at their looms are employed in the wooll
|
| 72 |
+
[historical] ppl= 82.02 minp=7.7e-05 The merchants of the port have grown rich upon their
|
| 73 |
+
[historical] ppl= 71.80 minp=1.6e-05 The labouring poor of this parish are reduced to gre
|
| 74 |
+
[historical] ppl= 66.09 minp=9.5e-05 He gave up his fortune to the patient study of natur
|
| 75 |
+
[historical] ppl= 59.40 minp=5.1e-05 The Lord provideth for all his creatures, giving the
|
| 76 |
+
[historical] ppl= 50.13 minp=1.2e-04 The ballroom was filled with dancing and laughter, a
|
| 77 |
+
[historical] ppl= 18.85 minp=5.2e-04 Her Majesty was pleased to summon the Lords and Comm
|
| 78 |
+
|
| 79 |
+
==============================================================================
|
| 80 |
+
SUITE: DIACHRONIC WORD-SENSE SEPARATION
|
| 81 |
+
==============================================================================
|
| 82 |
+
Extracting contextual embeddings for 10 shifted words in period AND modern sentences.
|
| 83 |
+
Every probe word sits LAST in its sentence, so the causal model has actually read
|
| 84 |
+
the disambiguating context before it represents the word.
|
| 85 |
+
|
| 86 |
+
SENSE SEPARATION (cosine similarity of the SAME word, period vs modern sentence)
|
| 87 |
+
lower value = the model represents the two senses differently (good sign for a
|
| 88 |
+
period model); ~1.0 = it treats them as the same word regardless of context
|
| 89 |
+
want : +0.339
|
| 90 |
+
python : +0.493
|
| 91 |
+
awful : +0.563
|
| 92 |
+
meat : +0.609
|
| 93 |
+
parliament : +0.626
|
| 94 |
+
nice : +0.642
|
| 95 |
+
manufacture : +0.685
|
| 96 |
+
gay : +0.718
|
| 97 |
+
commerce : +0.766
|
| 98 |
+
science : +0.820
|
| 99 |
+
MEAN : +0.626 <- near 1.0 = model ignores period vs modern usage; lower = senses differ
|
| 100 |
+
|
| 101 |
+
PAIRWISE COSINE SIMILARITY (period sentences, first 5 words)
|
| 102 |
+
these are CONTEXTUAL vectors, so they reflect the sentences as much as the words;
|
| 103 |
+
read the spread, not individual pairs. High mean = representations collapsed.
|
| 104 |
+
gay awful nice meat want
|
| 105 |
+
gay 1.000 0.468 0.341 0.308 0.281
|
| 106 |
+
awful 1.000 0.370 0.266 0.315
|
| 107 |
+
nice 1.000 0.303 0.327
|
| 108 |
+
meat 1.000 0.448
|
| 109 |
+
want 1.000
|
| 110 |
+
all 45 pairs: mean=+0.355 std=0.118 <- healthy spread is good
|
| 111 |
+
|
| 112 |
+
NEAREST NEIGHBOURS inside the probe-word set (period sentences)
|
| 113 |
+
gay : awful (+0.47), commerce (+0.37), science (+0.36), nice (+0.34), manufacture (+0.32)
|
| 114 |
+
awful : gay (+0.47), science (+0.41), nice (+0.37), parliament (+0.35), commerce (+0.33)
|
| 115 |
+
nice : science (+0.40), awful (+0.37), gay (+0.34), manufacture (+0.33), commerce (+0.33)
|
| 116 |
+
meat : science (+0.46), want (+0.45), commerce (+0.43), parliament (+0.42), manufacture (+0.40)
|
| 117 |
+
want : manufacture (+0.54), commerce (+0.53), science (+0.47), meat (+0.45), parliament (+0.42)
|
| 118 |
+
|
| 119 |
+
==============================================================================
|
| 120 |
+
SUITE: GENERATION PROBES
|
| 121 |
+
==============================================================================
|
| 122 |
+
|
| 123 |
+
--- Probe 1 ---
|
| 124 |
+
prompt : The history of the world is
|
| 125 |
+
continuation: too often the most interesting and instructive, and it is a most instructive and instructive book.
|
| 126 |
+
|
| 127 |
+
The story of the great city of Mexico is not more remarkable than the story of the great city of Mexico. It was the great city of Mexico, and its name was the great city of Mexico.
|
| 128 |
+
stats: 60 tokens | distinct-1 0.53 | distinct-2 0.75 | echo 0.09
|
| 129 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 130 |
+
|
| 131 |
+
--- Probe 2 ---
|
| 132 |
+
prompt : What is God? God is
|
| 133 |
+
continuation: a man! He is a man who has lived for the world, and has died for it. It is God who has not lived in vain. He is a man who has gone into the world to die for himself and his God.
|
| 134 |
+
|
| 135 |
+
And if the world is a God who has lived
|
| 136 |
+
stats: 60 tokens | distinct-1 0.55 | distinct-2 0.79 | echo 0.02
|
| 137 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 138 |
+
|
| 139 |
+
--- Probe 3 ---
|
| 140 |
+
prompt : In the year of our Lord eighteen hundred and
|
| 141 |
+
continuation: forty-eight; and the next year the last, the second year of the kingdom of God, the fourth year of the kingdom of Christ. The third year of the kingdom of heaven, in the year of the covenant, the fourth year of the kingdom of heaven; the sixth of the kingdom of
|
| 142 |
+
stats: 60 tokens | distinct-1 0.37 | distinct-2 0.62 | echo 0.31
|
| 143 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 144 |
+
|
| 145 |
+
--- Probe 4 ---
|
| 146 |
+
prompt : The manufacture of cotton
|
| 147 |
+
continuation: -the latter was carried on for several years by a great number of foreigners.
|
| 148 |
+
|
| 149 |
+
In the manufacture of woollen cloth, cotton, woollen, and other woollen goods, was imported into England.
|
| 150 |
+
|
| 151 |
+
The manufacture of woollen cloths and woollen cloths was also in progress at the time of the Revolution,
|
| 152 |
+
stats: 60 tokens | distinct-1 0.73 | distinct-2 0.94 | echo 0.06
|
| 153 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 154 |
+
|
| 155 |
+
--- Probe 5 ---
|
| 156 |
+
prompt : The steam engine
|
| 157 |
+
continuation: is the only one we know of, and that is a great deal more than the other.
|
| 158 |
+
|
| 159 |
+
A few days after, I was called on to visit a lady who was suffering from soreness of the right eye, but was very sick; she was very ill, and had been so for
|
| 160 |
+
stats: 60 tokens | distinct-1 0.82 | distinct-2 0.98 | echo 0.04
|
| 161 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 162 |
+
|
| 163 |
+
--- Probe 6 ---
|
| 164 |
+
prompt : The telegraph
|
| 165 |
+
continuation: s and telegraphs, and the telegraph offices, have been established for the last year, and the telegraph offices are being filled, and the telegraph offices are being opened, and a telegraph office conducted in the United States. The telegraph offices have been filled by the telegraph and telegraph offices.
|
| 166 |
+
|
| 167 |
+
The
|
| 168 |
+
stats: 60 tokens | distinct-1 0.54 | distinct-2 0.80 | echo 0.06
|
| 169 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 170 |
+
|
| 171 |
+
--- Probe 7 ---
|
| 172 |
+
prompt : My dearest sister,
|
| 173 |
+
continuation: I have the greatest pleasure to inform you that I have heard from her Majesty's ministers of that most unfortunate event, the Prince de
|
| 174 |
+
|
| 175 |
+
Govat, who, with his family, have been very dear to her, and who, as you know, are very kind and friendly. You may
|
| 176 |
+
stats: 60 tokens | distinct-1 0.79 | distinct-2 0.98 | echo 0.00
|
| 177 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 178 |
+
|
| 179 |
+
--- Probe 8 ---
|
| 180 |
+
prompt : Greetings, my friend
|
| 181 |
+
continuation: , I shall give you a new chapter on the nature of the Holy Sacrament. I have seen a new chapter in the Bible, the most ancient of which I have seen in the world, in the days when the
|
| 182 |
+
|
| 183 |
+
Lord had the power to create and to create, and in the days when
|
| 184 |
+
stats: 60 tokens | distinct-1 0.57 | distinct-2 0.83 | echo 0.15
|
| 185 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 186 |
+
|
| 187 |
+
--- Probe 9 ---
|
| 188 |
+
prompt : Her Majesty the Queen
|
| 189 |
+
continuation: of Scots, in the name of the king, was pleased to make them both the greatest of all Princes, and to assure them of their friendship and attachment; and, having had all the honour of the
|
| 190 |
+
|
| 191 |
+
King, that he would never permit the Queen of Scots to make them the subject of
|
| 192 |
+
stats: 60 tokens | distinct-1 0.63 | distinct-2 0.94 | echo 0.04
|
| 193 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 194 |
+
|
| 195 |
+
--- Probe 10 ---
|
| 196 |
+
prompt : Chapter I.
|
| 197 |
+
continuation: The first question that the author of this book, as we have seen, was then asked to be answered. It is now, however, a question of time and eternity, and therefore we must not omit to ask him, "What is that in the book?" The answer was, "
|
| 198 |
+
stats: 60 tokens | distinct-1 0.81 | distinct-2 1.00 | echo 0.02
|
| 199 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 200 |
+
|
| 201 |
+
AGGREGATE over 10 probes: 600 tokens in 0.8s (709.5 tok/s)
|
| 202 |
+
mean distinct-1 0.635 | mean distinct-2 0.862 | mean echo 0.080 | worst probe: distinct-2 0.620 / echo 0.314
|
| 203 |
+
NOTE: at least one probe looped badly even though the average looks fine - read the probes above.
|
| 204 |
+
(sampling hides loops that greedy decoding exposes; evaluate2.py measures the greedy loop length.)
|
| 205 |
+
|
| 206 |
+
Full numeric results written to /home/cro/Dev/vintage-LLM/training/eval_results/evaluate-final.json
|
| 207 |
+
|
| 208 |
+
==============================================================================
|
| 209 |
+
INSPECTION COMPLETE
|
| 210 |
+
==============================================================================
|
| 211 |
+
This covered WHAT the checkpoint is, its lineage, and whether it is still vintage.
|
| 212 |
+
For "is it baked enough", run evaluate2.py on the same path.
|
scheduler/wsd_10/eval2.log
ADDED
|
@@ -0,0 +1,54 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
evaluating 1 checkpoint(s) on cuda (torch.bfloat16)
|
| 2 |
+
tokenizer: final
|
| 3 |
+
final ...W0814 13:06:26.561000 44404 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
|
| 4 |
+
W0814 13:06:26.576000 44404 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
|
| 5 |
+
|
| 6 |
+
bake 53/100 bpb 1.2821 logic 0.62 (15.1s)
|
| 7 |
+
|
| 8 |
+
report: eval2.md
|
| 9 |
+
json: eval2.json
|
| 10 |
+
|
| 11 |
+
# Evaluation: final
|
| 12 |
+
|
| 13 |
+
- **Parameters:** 77M
|
| 14 |
+
- **Note:** no trainer_state.json in the checkpoint - tokens seen unknown
|
| 15 |
+
|
| 16 |
+
## Verdict: DOUGH (bake score 53/100)
|
| 17 |
+
|
| 18 |
+
Structure is forming - real words, some grammar - but this is not a usable language model yet. It needs several times more training tokens.
|
| 19 |
+
|
| 20 |
+
## Where it sits (held-out bits/byte, lower = better)
|
| 21 |
+
|
| 22 |
+
```
|
| 23 |
+
3.50 untrained model (uniform noise)
|
| 24 |
+
2.00 word-salad
|
| 25 |
+
1.50 broken prose
|
| 26 |
+
1.33 500M model, ~0.7B tokens (early training)
|
| 27 |
+
--> 1.282 YOUR MODEL
|
| 28 |
+
1.19 500M model, ~4.7B tokens (undertrained but solid)
|
| 29 |
+
1.10 best sub-1B measured on this data (341M)
|
| 30 |
+
0.95 estimated sub-1B ceiling on this corpus
|
| 31 |
+
```
|
| 32 |
+
|
| 33 |
+
## Scores
|
| 34 |
+
|
| 35 |
+
| what | raw value | points /100 | weight | plain English |
|
| 36 |
+
|---|---|---|---|---|
|
| 37 |
+
| Held-out loss | 1.2821 bits/byte | 60 | 0.5 | how cheaply it predicts period text it never saw — the single best training signal |
|
| 38 |
+
| Logic | 0.625 acc, margin 0.012 | 30 | 0.25 | picks the *sensible* continuation over matched nonsense; 0.50 = coin-flip. weak - style without understanding |
|
| 39 |
+
| Chat readiness | 1.1608 bits/byte | 56 | 0.15 | how cheap well-formed period dialogue already is — predicts fine-tuning ease |
|
| 40 |
+
| Hygiene | loop 33.8w, punct 0.07/100w | 69 | 0.1 | greedy-decoding loop length and broken punctuation |
|
| 41 |
+
|
| 42 |
+
## Period boundary
|
| 43 |
+
|
| 44 |
+
Clean. All 6 post-1900 trap words cost the model more than their period twins (mean shock +1.46 bits/byte, weakest pair 'motor car' at +0.75). No sign of modern text in training.
|
| 45 |
+
|
| 46 |
+
## See for yourself (sampled, t=0.8)
|
| 47 |
+
|
| 48 |
+
> **LONDON, Tuesday. — The committee appointed to inquire into the condition of the** workshops in the city of London have been able to report to the Committee the proceedings on the conduct of the works of the London firms, for the purpose of obtaining more accurate information relative to the business of the works. The Secretary having proceeded to the report, the Committee having proceeded to the report of the Committee, the committee passed the following resolutions, which were
|
| 49 |
+
>
|
| 50 |
+
> **A melancholy accident occurred on Thursday last at the works of Messrs. Harding and** Co., Ltd., and which, as I have said, was followed by the death of Mr. Co., the New York, a man of great ability, and of a most estimable character, who was a fine gentleman, a good judge, and a powerful patron of music. A young lady of good family, who is now the wife of an eminent clergyman of the parish, was received with much interest by her friends, and her services were marked by great gener
|
| 51 |
+
>
|
| 52 |
+
> **Brethren, the text which I have chosen for our consideration this morning is taken from** the work of the American Agricultural Society, which is not of a kind to be found in the printed pages of the American Agricultural Society, for the purpose of comparing them with their own, the present number of these books will be found much improved, and the result is an interesting one. A work of this kind, which has attracted much attention, has been started in the "Bank of the Pot," by a per
|
| 53 |
+
>
|
| 54 |
+
|
scheduler/wsd_10/final/chat_template.jinja
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{%- for message in messages -%}
|
| 2 |
+
{{- bos_token + '<|' + message['role'] + '|>\n' + message['content'].strip() + eos_token + '\n' -}}
|
| 3 |
+
{%- endfor -%}
|
| 4 |
+
|
| 5 |
+
{%- if add_generation_prompt -%}
|
| 6 |
+
{{- bos_token + '<|assistant|>\n' -}}
|
| 7 |
+
{%- endif -%}
|
scheduler/wsd_10/final/config.json
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"LlamaForCausalLM"
|
| 4 |
+
],
|
| 5 |
+
"attention_bias": false,
|
| 6 |
+
"attention_dropout": 0.0,
|
| 7 |
+
"bos_token_id": 3,
|
| 8 |
+
"dtype": "float32",
|
| 9 |
+
"eos_token_id": 4,
|
| 10 |
+
"head_dim": 96,
|
| 11 |
+
"hidden_act": "silu",
|
| 12 |
+
"hidden_size": 768,
|
| 13 |
+
"initializer_range": 0.02,
|
| 14 |
+
"intermediate_size": 2048,
|
| 15 |
+
"max_position_embeddings": 1024,
|
| 16 |
+
"mlp_bias": false,
|
| 17 |
+
"model_type": "llama",
|
| 18 |
+
"num_attention_heads": 8,
|
| 19 |
+
"num_hidden_layers": 8,
|
| 20 |
+
"num_key_value_heads": 4,
|
| 21 |
+
"pad_token_id": 0,
|
| 22 |
+
"pretraining_tp": 1,
|
| 23 |
+
"rms_norm_eps": 1e-06,
|
| 24 |
+
"rope_parameters": {
|
| 25 |
+
"rope_theta": 10000.0,
|
| 26 |
+
"rope_type": "default"
|
| 27 |
+
},
|
| 28 |
+
"tie_word_embeddings": true,
|
| 29 |
+
"transformers_version": "5.14.1",
|
| 30 |
+
"use_cache": false,
|
| 31 |
+
"vocab_size": 32768
|
| 32 |
+
}
|
scheduler/wsd_10/final/generation_config.json
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_from_model_config": true,
|
| 3 |
+
"bos_token_id": 3,
|
| 4 |
+
"eos_token_id": [
|
| 5 |
+
4,
|
| 6 |
+
2
|
| 7 |
+
],
|
| 8 |
+
"output_attentions": false,
|
| 9 |
+
"output_hidden_states": false,
|
| 10 |
+
"pad_token_id": 0,
|
| 11 |
+
"transformers_version": "5.14.1",
|
| 12 |
+
"use_cache": false
|
| 13 |
+
}
|
scheduler/wsd_10/final/model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a46b2cc2aa7a9d6213652cdbcc9c52e73e13966e4638c6e1e88e2e93d926f62c
|
| 3 |
+
size 308341784
|
scheduler/wsd_10/final/tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
scheduler/wsd_10/final/tokenizer_config.json
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"backend": "tokenizers",
|
| 3 |
+
"bos_token": "<|bos|>",
|
| 4 |
+
"eos_token": "<|eos|>",
|
| 5 |
+
"is_local": true,
|
| 6 |
+
"local_files_only": false,
|
| 7 |
+
"model_max_length": 1000000000000000019884624838656,
|
| 8 |
+
"pad_token": "<|pad|>",
|
| 9 |
+
"tokenizer_class": "TokenizersBackend",
|
| 10 |
+
"unk_token": "<|unk|>"
|
| 11 |
+
}
|
scheduler/wsd_10/trainer_state.json
ADDED
|
@@ -0,0 +1,656 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_global_step": null,
|
| 3 |
+
"best_metric": null,
|
| 4 |
+
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 0.00993222288333627,
|
| 6 |
+
"eval_steps": 5,
|
| 7 |
+
"global_step": 747,
|
| 8 |
+
"is_hyper_param_search": false,
|
| 9 |
+
"is_local_process_zero": true,
|
| 10 |
+
"is_world_process_zero": true,
|
| 11 |
+
"log_history": [
|
| 12 |
+
{
|
| 13 |
+
"epoch": 1.3296148438201165e-05,
|
| 14 |
+
"grad_norm": 5.513442516326904,
|
| 15 |
+
"learning_rate": 0.0,
|
| 16 |
+
"loss": 10.538199424743652,
|
| 17 |
+
"step": 1
|
| 18 |
+
},
|
| 19 |
+
{
|
| 20 |
+
"epoch": 0.00013296148438201165,
|
| 21 |
+
"grad_norm": 5.268533706665039,
|
| 22 |
+
"learning_rate": 0.00035999999999999997,
|
| 23 |
+
"loss": 10.136458502875435,
|
| 24 |
+
"step": 10
|
| 25 |
+
},
|
| 26 |
+
{
|
| 27 |
+
"epoch": 0.0002659229687640233,
|
| 28 |
+
"grad_norm": 1.034209132194519,
|
| 29 |
+
"learning_rate": 0.00076,
|
| 30 |
+
"loss": 8.0193359375,
|
| 31 |
+
"step": 20
|
| 32 |
+
},
|
| 33 |
+
{
|
| 34 |
+
"epoch": 0.0003988844531460349,
|
| 35 |
+
"grad_norm": 9.450255393981934,
|
| 36 |
+
"learning_rate": 0.00116,
|
| 37 |
+
"loss": 6.9860687255859375,
|
| 38 |
+
"step": 30
|
| 39 |
+
},
|
| 40 |
+
{
|
| 41 |
+
"epoch": 0.0005318459375280466,
|
| 42 |
+
"grad_norm": 5.685102462768555,
|
| 43 |
+
"learning_rate": 0.0015600000000000002,
|
| 44 |
+
"loss": 6.707894897460937,
|
| 45 |
+
"step": 40
|
| 46 |
+
},
|
| 47 |
+
{
|
| 48 |
+
"epoch": 0.0006648074219100582,
|
| 49 |
+
"grad_norm": 2.160360097885132,
|
| 50 |
+
"learning_rate": 0.00196,
|
| 51 |
+
"loss": 6.287691116333008,
|
| 52 |
+
"step": 50
|
| 53 |
+
},
|
| 54 |
+
{
|
| 55 |
+
"epoch": 0.0007977689062920698,
|
| 56 |
+
"grad_norm": 1.6388428211212158,
|
| 57 |
+
"learning_rate": 0.00236,
|
| 58 |
+
"loss": 5.944025039672852,
|
| 59 |
+
"step": 60
|
| 60 |
+
},
|
| 61 |
+
{
|
| 62 |
+
"epoch": 0.000890841945359478,
|
| 63 |
+
"eval_loss": 5.704253673553467,
|
| 64 |
+
"eval_runtime": 22.8955,
|
| 65 |
+
"eval_samples_per_second": 218.383,
|
| 66 |
+
"eval_steps_per_second": 27.298,
|
| 67 |
+
"step": 67
|
| 68 |
+
},
|
| 69 |
+
{
|
| 70 |
+
"epoch": 0.0009307303906740815,
|
| 71 |
+
"grad_norm": 1.2074735164642334,
|
| 72 |
+
"learning_rate": 0.00276,
|
| 73 |
+
"loss": 5.747131729125977,
|
| 74 |
+
"step": 70
|
| 75 |
+
},
|
| 76 |
+
{
|
| 77 |
+
"epoch": 0.0010636918750560932,
|
| 78 |
+
"grad_norm": 1.132364273071289,
|
| 79 |
+
"learning_rate": 0.00316,
|
| 80 |
+
"loss": 5.60839729309082,
|
| 81 |
+
"step": 80
|
| 82 |
+
},
|
| 83 |
+
{
|
| 84 |
+
"epoch": 0.0011966533594381047,
|
| 85 |
+
"grad_norm": 1.0884277820587158,
|
| 86 |
+
"learning_rate": 0.0035600000000000002,
|
| 87 |
+
"loss": 5.502901077270508,
|
| 88 |
+
"step": 90
|
| 89 |
+
},
|
| 90 |
+
{
|
| 91 |
+
"epoch": 0.0013296148438201164,
|
| 92 |
+
"grad_norm": 1.3878921270370483,
|
| 93 |
+
"learning_rate": 0.00396,
|
| 94 |
+
"loss": 5.371665573120117,
|
| 95 |
+
"step": 100
|
| 96 |
+
},
|
| 97 |
+
{
|
| 98 |
+
"epoch": 0.001462576328202128,
|
| 99 |
+
"grad_norm": 1.2677963972091675,
|
| 100 |
+
"learning_rate": 0.00436,
|
| 101 |
+
"loss": 5.2512672424316404,
|
| 102 |
+
"step": 110
|
| 103 |
+
},
|
| 104 |
+
{
|
| 105 |
+
"epoch": 0.0015955378125841397,
|
| 106 |
+
"grad_norm": 0.9181460738182068,
|
| 107 |
+
"learning_rate": 0.0047599999999999995,
|
| 108 |
+
"loss": 5.126389694213867,
|
| 109 |
+
"step": 120
|
| 110 |
+
},
|
| 111 |
+
{
|
| 112 |
+
"epoch": 0.0017284992969661514,
|
| 113 |
+
"grad_norm": 0.7834004163742065,
|
| 114 |
+
"learning_rate": 0.0051600000000000005,
|
| 115 |
+
"loss": 5.010536575317383,
|
| 116 |
+
"step": 130
|
| 117 |
+
},
|
| 118 |
+
{
|
| 119 |
+
"epoch": 0.0017417954454043525,
|
| 120 |
+
"eval_loss": 4.947054862976074,
|
| 121 |
+
"eval_runtime": 22.9812,
|
| 122 |
+
"eval_samples_per_second": 217.569,
|
| 123 |
+
"eval_steps_per_second": 27.196,
|
| 124 |
+
"step": 131
|
| 125 |
+
},
|
| 126 |
+
{
|
| 127 |
+
"epoch": 0.001861460781348163,
|
| 128 |
+
"grad_norm": 0.7623927593231201,
|
| 129 |
+
"learning_rate": 0.00556,
|
| 130 |
+
"loss": 4.908084106445313,
|
| 131 |
+
"step": 140
|
| 132 |
+
},
|
| 133 |
+
{
|
| 134 |
+
"epoch": 0.0019944222657301744,
|
| 135 |
+
"grad_norm": 0.5423034429550171,
|
| 136 |
+
"learning_rate": 0.00596,
|
| 137 |
+
"loss": 4.820485305786133,
|
| 138 |
+
"step": 150
|
| 139 |
+
},
|
| 140 |
+
{
|
| 141 |
+
"epoch": 0.0021273837501121864,
|
| 142 |
+
"grad_norm": 0.5046894550323486,
|
| 143 |
+
"learning_rate": 0.00636,
|
| 144 |
+
"loss": 4.753760528564453,
|
| 145 |
+
"step": 160
|
| 146 |
+
},
|
| 147 |
+
{
|
| 148 |
+
"epoch": 0.002260345234494198,
|
| 149 |
+
"grad_norm": 0.3807763159275055,
|
| 150 |
+
"learning_rate": 0.0067599999999999995,
|
| 151 |
+
"loss": 4.694861602783203,
|
| 152 |
+
"step": 170
|
| 153 |
+
},
|
| 154 |
+
{
|
| 155 |
+
"epoch": 0.0023933067188762094,
|
| 156 |
+
"grad_norm": 0.37175047397613525,
|
| 157 |
+
"learning_rate": 0.0071600000000000006,
|
| 158 |
+
"loss": 4.652392578125,
|
| 159 |
+
"step": 180
|
| 160 |
+
},
|
| 161 |
+
{
|
| 162 |
+
"epoch": 0.0025262682032582214,
|
| 163 |
+
"grad_norm": 0.3488306999206543,
|
| 164 |
+
"learning_rate": 0.00756,
|
| 165 |
+
"loss": 4.597007369995117,
|
| 166 |
+
"step": 190
|
| 167 |
+
},
|
| 168 |
+
{
|
| 169 |
+
"epoch": 0.0025794527970110256,
|
| 170 |
+
"eval_loss": 4.569252967834473,
|
| 171 |
+
"eval_runtime": 23.087,
|
| 172 |
+
"eval_samples_per_second": 216.572,
|
| 173 |
+
"eval_steps_per_second": 27.072,
|
| 174 |
+
"step": 194
|
| 175 |
+
},
|
| 176 |
+
{
|
| 177 |
+
"epoch": 0.002659229687640233,
|
| 178 |
+
"grad_norm": 0.3351574242115021,
|
| 179 |
+
"learning_rate": 0.00796,
|
| 180 |
+
"loss": 4.566041946411133,
|
| 181 |
+
"step": 200
|
| 182 |
+
},
|
| 183 |
+
{
|
| 184 |
+
"epoch": 0.0027921911720222444,
|
| 185 |
+
"grad_norm": 0.3297984302043915,
|
| 186 |
+
"learning_rate": 0.008,
|
| 187 |
+
"loss": 4.5214378356933596,
|
| 188 |
+
"step": 210
|
| 189 |
+
},
|
| 190 |
+
{
|
| 191 |
+
"epoch": 0.002925152656404256,
|
| 192 |
+
"grad_norm": 0.2526053190231323,
|
| 193 |
+
"learning_rate": 0.008,
|
| 194 |
+
"loss": 4.468568420410156,
|
| 195 |
+
"step": 220
|
| 196 |
+
},
|
| 197 |
+
{
|
| 198 |
+
"epoch": 0.003058114140786268,
|
| 199 |
+
"grad_norm": 0.24721451103687286,
|
| 200 |
+
"learning_rate": 0.008,
|
| 201 |
+
"loss": 4.431998825073242,
|
| 202 |
+
"step": 230
|
| 203 |
+
},
|
| 204 |
+
{
|
| 205 |
+
"epoch": 0.0031910756251682794,
|
| 206 |
+
"grad_norm": 0.2912442684173584,
|
| 207 |
+
"learning_rate": 0.008,
|
| 208 |
+
"loss": 4.396870040893555,
|
| 209 |
+
"step": 240
|
| 210 |
+
},
|
| 211 |
+
{
|
| 212 |
+
"epoch": 0.003324037109550291,
|
| 213 |
+
"grad_norm": 0.28560253977775574,
|
| 214 |
+
"learning_rate": 0.008,
|
| 215 |
+
"loss": 4.367413711547852,
|
| 216 |
+
"step": 250
|
| 217 |
+
},
|
| 218 |
+
{
|
| 219 |
+
"epoch": 0.003417110148617699,
|
| 220 |
+
"eval_loss": 4.318535327911377,
|
| 221 |
+
"eval_runtime": 23.066,
|
| 222 |
+
"eval_samples_per_second": 216.77,
|
| 223 |
+
"eval_steps_per_second": 27.096,
|
| 224 |
+
"step": 257
|
| 225 |
+
},
|
| 226 |
+
{
|
| 227 |
+
"epoch": 0.003456998593932303,
|
| 228 |
+
"grad_norm": 0.21196810901165009,
|
| 229 |
+
"learning_rate": 0.008,
|
| 230 |
+
"loss": 4.331097793579102,
|
| 231 |
+
"step": 260
|
| 232 |
+
},
|
| 233 |
+
{
|
| 234 |
+
"epoch": 0.0035899600783143143,
|
| 235 |
+
"grad_norm": 0.22107456624507904,
|
| 236 |
+
"learning_rate": 0.008,
|
| 237 |
+
"loss": 4.302482604980469,
|
| 238 |
+
"step": 270
|
| 239 |
+
},
|
| 240 |
+
{
|
| 241 |
+
"epoch": 0.003722921562696326,
|
| 242 |
+
"grad_norm": 0.1877111792564392,
|
| 243 |
+
"learning_rate": 0.008,
|
| 244 |
+
"loss": 4.268944931030274,
|
| 245 |
+
"step": 280
|
| 246 |
+
},
|
| 247 |
+
{
|
| 248 |
+
"epoch": 0.0038558830470783374,
|
| 249 |
+
"grad_norm": 0.21084874868392944,
|
| 250 |
+
"learning_rate": 0.008,
|
| 251 |
+
"loss": 4.2542469024658205,
|
| 252 |
+
"step": 290
|
| 253 |
+
},
|
| 254 |
+
{
|
| 255 |
+
"epoch": 0.003988844531460349,
|
| 256 |
+
"grad_norm": 0.20695072412490845,
|
| 257 |
+
"learning_rate": 0.008,
|
| 258 |
+
"loss": 4.227492904663086,
|
| 259 |
+
"step": 300
|
| 260 |
+
},
|
| 261 |
+
{
|
| 262 |
+
"epoch": 0.004121806015842361,
|
| 263 |
+
"grad_norm": 0.19029781222343445,
|
| 264 |
+
"learning_rate": 0.008,
|
| 265 |
+
"loss": 4.208528518676758,
|
| 266 |
+
"step": 310
|
| 267 |
+
},
|
| 268 |
+
{
|
| 269 |
+
"epoch": 0.004214879054909769,
|
| 270 |
+
"eval_loss": 4.181951999664307,
|
| 271 |
+
"eval_runtime": 24.4052,
|
| 272 |
+
"eval_samples_per_second": 204.874,
|
| 273 |
+
"eval_steps_per_second": 25.609,
|
| 274 |
+
"step": 317
|
| 275 |
+
},
|
| 276 |
+
{
|
| 277 |
+
"epoch": 0.004254767500224373,
|
| 278 |
+
"grad_norm": 0.1889178603887558,
|
| 279 |
+
"learning_rate": 0.008,
|
| 280 |
+
"loss": 4.187363815307617,
|
| 281 |
+
"step": 320
|
| 282 |
+
},
|
| 283 |
+
{
|
| 284 |
+
"epoch": 0.004387728984606384,
|
| 285 |
+
"grad_norm": 0.19070716202259064,
|
| 286 |
+
"learning_rate": 0.008,
|
| 287 |
+
"loss": 4.173851776123047,
|
| 288 |
+
"step": 330
|
| 289 |
+
},
|
| 290 |
+
{
|
| 291 |
+
"epoch": 0.004520690468988396,
|
| 292 |
+
"grad_norm": 0.15929149091243744,
|
| 293 |
+
"learning_rate": 0.008,
|
| 294 |
+
"loss": 4.148160171508789,
|
| 295 |
+
"step": 340
|
| 296 |
+
},
|
| 297 |
+
{
|
| 298 |
+
"epoch": 0.004653651953370408,
|
| 299 |
+
"grad_norm": 0.17022773623466492,
|
| 300 |
+
"learning_rate": 0.008,
|
| 301 |
+
"loss": 4.132599258422852,
|
| 302 |
+
"step": 350
|
| 303 |
+
},
|
| 304 |
+
{
|
| 305 |
+
"epoch": 0.004786613437752419,
|
| 306 |
+
"grad_norm": 0.18009495735168457,
|
| 307 |
+
"learning_rate": 0.008,
|
| 308 |
+
"loss": 4.132373809814453,
|
| 309 |
+
"step": 360
|
| 310 |
+
},
|
| 311 |
+
{
|
| 312 |
+
"epoch": 0.004919574922134431,
|
| 313 |
+
"grad_norm": 0.17529530823230743,
|
| 314 |
+
"learning_rate": 0.008,
|
| 315 |
+
"loss": 4.117653656005859,
|
| 316 |
+
"step": 370
|
| 317 |
+
},
|
| 318 |
+
{
|
| 319 |
+
"epoch": 0.0050392402580782415,
|
| 320 |
+
"eval_loss": 4.0923027992248535,
|
| 321 |
+
"eval_runtime": 23.4068,
|
| 322 |
+
"eval_samples_per_second": 213.613,
|
| 323 |
+
"eval_steps_per_second": 26.702,
|
| 324 |
+
"step": 379
|
| 325 |
+
},
|
| 326 |
+
{
|
| 327 |
+
"epoch": 0.005052536406516443,
|
| 328 |
+
"grad_norm": 0.1463332176208496,
|
| 329 |
+
"learning_rate": 0.008,
|
| 330 |
+
"loss": 4.096078109741211,
|
| 331 |
+
"step": 380
|
| 332 |
+
},
|
| 333 |
+
{
|
| 334 |
+
"epoch": 0.005185497890898454,
|
| 335 |
+
"grad_norm": 0.14370259642601013,
|
| 336 |
+
"learning_rate": 0.008,
|
| 337 |
+
"loss": 4.084194946289062,
|
| 338 |
+
"step": 390
|
| 339 |
+
},
|
| 340 |
+
{
|
| 341 |
+
"epoch": 0.005318459375280466,
|
| 342 |
+
"grad_norm": 0.16615846753120422,
|
| 343 |
+
"learning_rate": 0.008,
|
| 344 |
+
"loss": 4.075920867919922,
|
| 345 |
+
"step": 400
|
| 346 |
+
},
|
| 347 |
+
{
|
| 348 |
+
"epoch": 0.005451420859662477,
|
| 349 |
+
"grad_norm": 0.15509292483329773,
|
| 350 |
+
"learning_rate": 0.008,
|
| 351 |
+
"loss": 4.054693603515625,
|
| 352 |
+
"step": 410
|
| 353 |
+
},
|
| 354 |
+
{
|
| 355 |
+
"epoch": 0.005584382344044489,
|
| 356 |
+
"grad_norm": 0.1333627849817276,
|
| 357 |
+
"learning_rate": 0.008,
|
| 358 |
+
"loss": 4.047085952758789,
|
| 359 |
+
"step": 420
|
| 360 |
+
},
|
| 361 |
+
{
|
| 362 |
+
"epoch": 0.005717343828426501,
|
| 363 |
+
"grad_norm": 0.12619031965732574,
|
| 364 |
+
"learning_rate": 0.008,
|
| 365 |
+
"loss": 4.037934494018555,
|
| 366 |
+
"step": 430
|
| 367 |
+
},
|
| 368 |
+
{
|
| 369 |
+
"epoch": 0.00582371301593211,
|
| 370 |
+
"eval_loss": 4.02523136138916,
|
| 371 |
+
"eval_runtime": 23.2986,
|
| 372 |
+
"eval_samples_per_second": 214.605,
|
| 373 |
+
"eval_steps_per_second": 26.826,
|
| 374 |
+
"step": 438
|
| 375 |
+
},
|
| 376 |
+
{
|
| 377 |
+
"epoch": 0.005850305312808512,
|
| 378 |
+
"grad_norm": 0.14818833768367767,
|
| 379 |
+
"learning_rate": 0.008,
|
| 380 |
+
"loss": 4.030204772949219,
|
| 381 |
+
"step": 440
|
| 382 |
+
},
|
| 383 |
+
{
|
| 384 |
+
"epoch": 0.005983266797190524,
|
| 385 |
+
"grad_norm": 0.11952793598175049,
|
| 386 |
+
"learning_rate": 0.008,
|
| 387 |
+
"loss": 4.024636077880859,
|
| 388 |
+
"step": 450
|
| 389 |
+
},
|
| 390 |
+
{
|
| 391 |
+
"epoch": 0.006116228281572536,
|
| 392 |
+
"grad_norm": 0.12302638590335846,
|
| 393 |
+
"learning_rate": 0.008,
|
| 394 |
+
"loss": 4.0040229797363285,
|
| 395 |
+
"step": 460
|
| 396 |
+
},
|
| 397 |
+
{
|
| 398 |
+
"epoch": 0.006249189765954547,
|
| 399 |
+
"grad_norm": 0.15813544392585754,
|
| 400 |
+
"learning_rate": 0.008,
|
| 401 |
+
"loss": 4.0008289337158205,
|
| 402 |
+
"step": 470
|
| 403 |
+
},
|
| 404 |
+
{
|
| 405 |
+
"epoch": 0.006382151250336559,
|
| 406 |
+
"grad_norm": 0.11854846030473709,
|
| 407 |
+
"learning_rate": 0.008,
|
| 408 |
+
"loss": 3.994869613647461,
|
| 409 |
+
"step": 480
|
| 410 |
+
},
|
| 411 |
+
{
|
| 412 |
+
"epoch": 0.006515112734718571,
|
| 413 |
+
"grad_norm": 0.14335912466049194,
|
| 414 |
+
"learning_rate": 0.008,
|
| 415 |
+
"loss": 3.9867435455322267,
|
| 416 |
+
"step": 490
|
| 417 |
+
},
|
| 418 |
+
{
|
| 419 |
+
"epoch": 0.006648074219100582,
|
| 420 |
+
"grad_norm": 0.10275301337242126,
|
| 421 |
+
"learning_rate": 0.008,
|
| 422 |
+
"loss": 3.9850521087646484,
|
| 423 |
+
"step": 500
|
| 424 |
+
},
|
| 425 |
+
{
|
| 426 |
+
"epoch": 0.006648074219100582,
|
| 427 |
+
"eval_loss": 3.9743027687072754,
|
| 428 |
+
"eval_runtime": 23.1478,
|
| 429 |
+
"eval_samples_per_second": 216.004,
|
| 430 |
+
"eval_steps_per_second": 27.0,
|
| 431 |
+
"step": 500
|
| 432 |
+
},
|
| 433 |
+
{
|
| 434 |
+
"epoch": 0.006781035703482594,
|
| 435 |
+
"grad_norm": 0.12899844348430634,
|
| 436 |
+
"learning_rate": 0.008,
|
| 437 |
+
"loss": 3.9827651977539062,
|
| 438 |
+
"step": 510
|
| 439 |
+
},
|
| 440 |
+
{
|
| 441 |
+
"epoch": 0.006913997187864606,
|
| 442 |
+
"grad_norm": 0.1306857168674469,
|
| 443 |
+
"learning_rate": 0.008,
|
| 444 |
+
"loss": 3.962184524536133,
|
| 445 |
+
"step": 520
|
| 446 |
+
},
|
| 447 |
+
{
|
| 448 |
+
"epoch": 0.007046958672246617,
|
| 449 |
+
"grad_norm": 0.10865853726863861,
|
| 450 |
+
"learning_rate": 0.008,
|
| 451 |
+
"loss": 3.951933670043945,
|
| 452 |
+
"step": 530
|
| 453 |
+
},
|
| 454 |
+
{
|
| 455 |
+
"epoch": 0.007179920156628629,
|
| 456 |
+
"grad_norm": 0.11110885441303253,
|
| 457 |
+
"learning_rate": 0.008,
|
| 458 |
+
"loss": 3.94482536315918,
|
| 459 |
+
"step": 540
|
| 460 |
+
},
|
| 461 |
+
{
|
| 462 |
+
"epoch": 0.007312881641010641,
|
| 463 |
+
"grad_norm": 0.10243042558431625,
|
| 464 |
+
"learning_rate": 0.008,
|
| 465 |
+
"loss": 3.942932891845703,
|
| 466 |
+
"step": 550
|
| 467 |
+
},
|
| 468 |
+
{
|
| 469 |
+
"epoch": 0.007445843125392652,
|
| 470 |
+
"grad_norm": 0.10774345695972443,
|
| 471 |
+
"learning_rate": 0.008,
|
| 472 |
+
"loss": 3.9434852600097656,
|
| 473 |
+
"step": 560
|
| 474 |
+
},
|
| 475 |
+
{
|
| 476 |
+
"epoch": 0.007485731570707256,
|
| 477 |
+
"eval_loss": 3.928818464279175,
|
| 478 |
+
"eval_runtime": 23.06,
|
| 479 |
+
"eval_samples_per_second": 216.826,
|
| 480 |
+
"eval_steps_per_second": 27.103,
|
| 481 |
+
"step": 563
|
| 482 |
+
},
|
| 483 |
+
{
|
| 484 |
+
"epoch": 0.007578804609774664,
|
| 485 |
+
"grad_norm": 0.10925770550966263,
|
| 486 |
+
"learning_rate": 0.008,
|
| 487 |
+
"loss": 3.9292713165283204,
|
| 488 |
+
"step": 570
|
| 489 |
+
},
|
| 490 |
+
{
|
| 491 |
+
"epoch": 0.007711766094156675,
|
| 492 |
+
"grad_norm": 0.10408780723810196,
|
| 493 |
+
"learning_rate": 0.008,
|
| 494 |
+
"loss": 3.917940139770508,
|
| 495 |
+
"step": 580
|
| 496 |
+
},
|
| 497 |
+
{
|
| 498 |
+
"epoch": 0.007844727578538687,
|
| 499 |
+
"grad_norm": 0.10272197425365448,
|
| 500 |
+
"learning_rate": 0.008,
|
| 501 |
+
"loss": 3.920045852661133,
|
| 502 |
+
"step": 590
|
| 503 |
+
},
|
| 504 |
+
{
|
| 505 |
+
"epoch": 0.007977689062920698,
|
| 506 |
+
"grad_norm": 0.09875880181789398,
|
| 507 |
+
"learning_rate": 0.008,
|
| 508 |
+
"loss": 3.9055149078369142,
|
| 509 |
+
"step": 600
|
| 510 |
+
},
|
| 511 |
+
{
|
| 512 |
+
"epoch": 0.00811065054730271,
|
| 513 |
+
"grad_norm": 0.12342850863933563,
|
| 514 |
+
"learning_rate": 0.008,
|
| 515 |
+
"loss": 3.907426452636719,
|
| 516 |
+
"step": 610
|
| 517 |
+
},
|
| 518 |
+
{
|
| 519 |
+
"epoch": 0.008243612031684722,
|
| 520 |
+
"grad_norm": 0.11523553729057312,
|
| 521 |
+
"learning_rate": 0.008,
|
| 522 |
+
"loss": 3.8896995544433595,
|
| 523 |
+
"step": 620
|
| 524 |
+
},
|
| 525 |
+
{
|
| 526 |
+
"epoch": 0.00832338892231393,
|
| 527 |
+
"eval_loss": 3.8911144733428955,
|
| 528 |
+
"eval_runtime": 23.7786,
|
| 529 |
+
"eval_samples_per_second": 210.273,
|
| 530 |
+
"eval_steps_per_second": 26.284,
|
| 531 |
+
"step": 626
|
| 532 |
+
},
|
| 533 |
+
{
|
| 534 |
+
"epoch": 0.008376573516066733,
|
| 535 |
+
"grad_norm": 0.13027988374233246,
|
| 536 |
+
"learning_rate": 0.008,
|
| 537 |
+
"loss": 3.888309860229492,
|
| 538 |
+
"step": 630
|
| 539 |
+
},
|
| 540 |
+
{
|
| 541 |
+
"epoch": 0.008509535000448746,
|
| 542 |
+
"grad_norm": 0.10204915702342987,
|
| 543 |
+
"learning_rate": 0.008,
|
| 544 |
+
"loss": 3.8905193328857424,
|
| 545 |
+
"step": 640
|
| 546 |
+
},
|
| 547 |
+
{
|
| 548 |
+
"epoch": 0.008642496484830757,
|
| 549 |
+
"grad_norm": 0.09654007107019424,
|
| 550 |
+
"learning_rate": 0.008,
|
| 551 |
+
"loss": 3.88280029296875,
|
| 552 |
+
"step": 650
|
| 553 |
+
},
|
| 554 |
+
{
|
| 555 |
+
"epoch": 0.008775457969212768,
|
| 556 |
+
"grad_norm": 0.10360368341207504,
|
| 557 |
+
"learning_rate": 0.008,
|
| 558 |
+
"loss": 3.8796253204345703,
|
| 559 |
+
"step": 660
|
| 560 |
+
},
|
| 561 |
+
{
|
| 562 |
+
"epoch": 0.00890841945359478,
|
| 563 |
+
"grad_norm": 0.11431150883436203,
|
| 564 |
+
"learning_rate": 0.008,
|
| 565 |
+
"loss": 3.8688350677490235,
|
| 566 |
+
"step": 670
|
| 567 |
+
},
|
| 568 |
+
{
|
| 569 |
+
"epoch": 0.009041380937976792,
|
| 570 |
+
"grad_norm": 0.10246361047029495,
|
| 571 |
+
"learning_rate": 0.008,
|
| 572 |
+
"loss": 3.859950637817383,
|
| 573 |
+
"step": 680
|
| 574 |
+
},
|
| 575 |
+
{
|
| 576 |
+
"epoch": 0.0091477501254824,
|
| 577 |
+
"eval_loss": 3.8641164302825928,
|
| 578 |
+
"eval_runtime": 23.7195,
|
| 579 |
+
"eval_samples_per_second": 210.797,
|
| 580 |
+
"eval_steps_per_second": 26.35,
|
| 581 |
+
"step": 688
|
| 582 |
+
},
|
| 583 |
+
{
|
| 584 |
+
"epoch": 0.009174342422358803,
|
| 585 |
+
"grad_norm": 0.09362177550792694,
|
| 586 |
+
"learning_rate": 0.007473684210526316,
|
| 587 |
+
"loss": 3.8631008148193358,
|
| 588 |
+
"step": 690
|
| 589 |
+
},
|
| 590 |
+
{
|
| 591 |
+
"epoch": 0.009307303906740816,
|
| 592 |
+
"grad_norm": 0.07704541087150574,
|
| 593 |
+
"learning_rate": 0.006421052631578947,
|
| 594 |
+
"loss": 3.8439105987548827,
|
| 595 |
+
"step": 700
|
| 596 |
+
},
|
| 597 |
+
{
|
| 598 |
+
"epoch": 0.009440265391122827,
|
| 599 |
+
"grad_norm": 0.07900427281856537,
|
| 600 |
+
"learning_rate": 0.005368421052631579,
|
| 601 |
+
"loss": 3.8248409271240233,
|
| 602 |
+
"step": 710
|
| 603 |
+
},
|
| 604 |
+
{
|
| 605 |
+
"epoch": 0.009573226875504838,
|
| 606 |
+
"grad_norm": 0.06625134497880936,
|
| 607 |
+
"learning_rate": 0.00431578947368421,
|
| 608 |
+
"loss": 3.803970718383789,
|
| 609 |
+
"step": 720
|
| 610 |
+
},
|
| 611 |
+
{
|
| 612 |
+
"epoch": 0.00970618835988685,
|
| 613 |
+
"grad_norm": 0.05564470961689949,
|
| 614 |
+
"learning_rate": 0.0032631578947368424,
|
| 615 |
+
"loss": 3.791268539428711,
|
| 616 |
+
"step": 730
|
| 617 |
+
},
|
| 618 |
+
{
|
| 619 |
+
"epoch": 0.009839149844268862,
|
| 620 |
+
"grad_norm": 0.05231216922402382,
|
| 621 |
+
"learning_rate": 0.0022105263157894735,
|
| 622 |
+
"loss": 3.776142883300781,
|
| 623 |
+
"step": 740
|
| 624 |
+
},
|
| 625 |
+
{
|
| 626 |
+
"epoch": 0.00993222288333627,
|
| 627 |
+
"step": 747,
|
| 628 |
+
"total_flos": 6.099947743503974e+16,
|
| 629 |
+
"train_loss": 4.495604857223899,
|
| 630 |
+
"train_runtime": 3603.6272,
|
| 631 |
+
"train_samples_per_second": 53.99,
|
| 632 |
+
"train_steps_per_second": 0.211
|
| 633 |
+
}
|
| 634 |
+
],
|
| 635 |
+
"logging_steps": 10,
|
| 636 |
+
"max_steps": 760,
|
| 637 |
+
"num_input_tokens_seen": 0,
|
| 638 |
+
"num_train_epochs": 1,
|
| 639 |
+
"save_steps": 5,
|
| 640 |
+
"stateful_callbacks": {
|
| 641 |
+
"TrainerControl": {
|
| 642 |
+
"args": {
|
| 643 |
+
"should_epoch_stop": false,
|
| 644 |
+
"should_evaluate": false,
|
| 645 |
+
"should_log": false,
|
| 646 |
+
"should_save": true,
|
| 647 |
+
"should_training_stop": true
|
| 648 |
+
},
|
| 649 |
+
"attributes": {}
|
| 650 |
+
}
|
| 651 |
+
},
|
| 652 |
+
"total_flos": 6.099947743503974e+16,
|
| 653 |
+
"train_batch_size": 8,
|
| 654 |
+
"trial_name": null,
|
| 655 |
+
"trial_params": null
|
| 656 |
+
}
|
scheduler/wsd_20/REPORT.md
ADDED
|
@@ -0,0 +1,38 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Experiment: wsd_20 — warmup-stable-decay, 20% linear decay to 0
|
| 2 |
+
|
| 3 |
+
**Wall-clock:** 60 min · **Recipe:** best known (sdpa + bf16 mixed + muonq @ lr 8e-3), schedule: warmup 200 → stable 408 @ 8e-3 → linear decay to 0 over 152 steps (horizon `max_steps = 760`)
|
| 4 |
+
|
| 5 |
+
## Result: NEW PROJECT BEST — bake 55.6/100, and the hot-LR hygiene tax is gone
|
| 6 |
+
|
| 7 |
+
| metric | wsd_20 | constant-LR reference (`../../lr/lr_8e-3/`) |
|
| 8 |
+
|---|---|---|
|
| 9 |
+
| optimizer steps in 60 min | 730 | 759 |
|
| 10 |
+
| final logged eval loss / ppl | 3.8023 / 44.8 (step 678, mid-decay — undersells the final model, see below) | 3.8583 / 47.4 |
|
| 11 |
+
| bake score | **55.61/100 (DOUGH)** — record, +8.8 over reference | 46.83 |
|
| 12 |
+
| held-out bits/byte | **1.2769** — record | 1.3114 |
|
| 13 |
+
| chat bits/byte | **1.1571** — record | 1.1914 |
|
| 14 |
+
| logic accuracy | **0.650, margin +0.0242** — record | 0.600, +0.018 |
|
| 15 |
+
| hygiene (greedy loop) | **79.2/100 (loop 23.8w)** — back to cool-LR levels | 47.7 (59.8w) |
|
| 16 |
+
| grad norms | 0.06–9.45, 0 non-finite | 0.09–9.45 |
|
| 17 |
+
| peak GPU | 7.2 GB | 7.2 GB |
|
| 18 |
+
|
| 19 |
+
The single biggest quality jump of any experiment in the project: +8.8 bake points over the previous record, best bpb/chat/logic ever, **and** greedy-loop hygiene restored to baseline levels (23.8w vs the 52–66w of every other hot-LR run). The SmolLM recipe works: park at peak LR, then anneal hard to 0 at the end.
|
| 20 |
+
|
| 21 |
+
**Caveat — the decay was cut ~20% short.** The run reached 730 of the 760-step horizon when the 60-min stop fired, so the final model sat at LR 1.63e-3, not 0. The logged eval curve stops at step 678 (LR ≈ 2.9e-3, decay only 46% done), which is why the table's "final eval" undersells the saved model — the evaluate2 numbers above are measured on the actual `final/` (step 730). A horizon of ~730 would let the decay finish and should squeeze out a bit more.
|
| 22 |
+
|
| 23 |
+
## Eval loss curve (5-min intervals)
|
| 24 |
+
|
| 25 |
+
| step | eval loss | | step | eval loss |
|
| 26 |
+
|---|---|---|---|---|
|
| 27 |
+
| 66 | 5.7330 | | 429 | 4.0342 |
|
| 28 |
+
| 128 | 4.9621 | | 492 | 3.9794 |
|
| 29 |
+
| 186 | 4.5960 | | 554 | 3.9340 |
|
| 30 |
+
| 245 | 4.3590 | | 616 | 3.8938 |
|
| 31 |
+
| 306 | 4.2061 | | 678 | **3.8023** |
|
| 32 |
+
| 367 | 4.1056 | | | |
|
| 33 |
+
|
| 34 |
+
The last interval tells the story: 3.8938 → 3.8023 in 62 steps (−0.09) as the decay bit — more improvement in the decay phase than in the previous 15 minutes of stable-LR training.
|
| 35 |
+
|
| 36 |
+
## Verdict
|
| 37 |
+
|
| 38 |
+
**Winner of the scheduler series and the new reference recipe.** Decay-to-zero over the final 20% buys both quality (bake 46.8 → 55.6) and hygiene (59.8w → 23.8w loops) at zero cost. See `../COMPARISON.md`.
|
scheduler/wsd_20/config.toml
ADDED
|
@@ -0,0 +1,89 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Scheduler series — new best recipe (sdpa + bf16 mixed + muonq @ lr 8e-3), only the LR schedule varies.
|
| 2 |
+
# This run: WSD (warmup-stable-decay), SmolLM2-style — linear decay to 0 over the final 20% (152 steps).
|
| 3 |
+
model_config = "../model.toml"
|
| 4 |
+
|
| 5 |
+
[data]
|
| 6 |
+
# Data files (supports glob patterns or lists of files)
|
| 7 |
+
train_files = ["~/Dev/vintage-LLM/training/data/train_*.bin"]
|
| 8 |
+
valid_files = ["~/Dev/vintage-LLM/training/data/valid_*.bin"]
|
| 9 |
+
|
| 10 |
+
# Sequence length for each dataset sample
|
| 11 |
+
# Must be <= model.max_position_embeddings
|
| 12 |
+
max_seq_length = 1024
|
| 13 |
+
|
| 14 |
+
# Tokenizer (must match the one used to create .bin files)
|
| 15 |
+
tokenizer = "~/Dev/vintage-LLM/tokenizers/t-v3/"
|
| 16 |
+
|
| 17 |
+
[training]
|
| 18 |
+
# Training duration
|
| 19 |
+
num_train_epochs = 1
|
| 20 |
+
max_steps = 760 # scheduler horizon = achievable steps/hour (was -1: full epoch, so decay never happened)
|
| 21 |
+
|
| 22 |
+
# Optional: Stop training after this many minutes of wall-clock time.
|
| 23 |
+
max_train_minutes = 60
|
| 24 |
+
|
| 25 |
+
# Output directories
|
| 26 |
+
output_dir = "./"
|
| 27 |
+
final_model_dir = "./final"
|
| 28 |
+
|
| 29 |
+
# Batch sizes
|
| 30 |
+
# Effective batch size = num_devices × per_device_train_batch_size × gradient_accumulation_steps
|
| 31 |
+
per_device_train_batch_size = 8 # batch size per GPU (8x1024 tokens fits 16GB with headroom for eval)
|
| 32 |
+
per_device_eval_batch_size = 8
|
| 33 |
+
gradient_accumulation_steps = 32 # accumulate gradients over multiple mini-batches before updating the params
|
| 34 |
+
|
| 35 |
+
# Optimizer: MuonQ (4-bit quantized Muon, ../../MuonQ repo) — wired through
|
| 36 |
+
# optimizer_cls_and_kwargs in base_train.py. 2D hidden weights get Muon
|
| 37 |
+
# (Newton-Schulz 5, 4-bit mu-law-companded momentum, rank-16 residual
|
| 38 |
+
# factorization); embeddings/head/norms get its internal AdamW backup.
|
| 39 |
+
# Muon/quantization knobs (muon_*, muonq_*) default to the repo's own
|
| 40 |
+
# llama-60m "muonq" recipe in base_train.py.
|
| 41 |
+
optim = "muonq"
|
| 42 |
+
|
| 43 |
+
# Optimizer settings — lr/wd from MuonQ's llama-60m recipe (Muon takes
|
| 44 |
+
# spectral-norm-scaled steps; do not copy the AdamW 5e-4 here)
|
| 45 |
+
learning_rate = 8e-3
|
| 46 |
+
weight_decay = 0.1
|
| 47 |
+
adam_beta1 = 0.9
|
| 48 |
+
adam_beta2 = 0.95
|
| 49 |
+
max_grad_norm = 1.0
|
| 50 |
+
|
| 51 |
+
# Learning rate scheduler
|
| 52 |
+
lr_scheduler_type = "warmup_stable_decay" # warmup 200 -> stable 408 @ peak -> linear decay 152 steps to 0
|
| 53 |
+
lr_scheduler_kwargs = { num_stable_steps = 408, num_decay_steps = 152, decay_type = "linear", min_lr_ratio = 0.0 }
|
| 54 |
+
warmup_steps = 200 # 1-hour series standard (muonq-v2's 400 was a 2-hour recipe)
|
| 55 |
+
|
| 56 |
+
# Precision (choose based on your GPU)
|
| 57 |
+
bf16 = true # bfloat16 autocast + fp32 master weights (the dtype-series safe recipe)
|
| 58 |
+
fp16 = false
|
| 59 |
+
|
| 60 |
+
# Performance
|
| 61 |
+
torch_compile = false
|
| 62 |
+
gradient_checkpointing = false # trade compute for memory (enables larger models)
|
| 63 |
+
neftune_noise_alpha = 0.0 # add noise during training to improve model performance (fine-tuning)
|
| 64 |
+
|
| 65 |
+
# Checkpointing
|
| 66 |
+
# save_strategy can be "steps", "epoch", or "minutes" (wall-clock time).
|
| 67 |
+
save_strategy = "minutes"
|
| 68 |
+
save_steps = 5 # checkpoint every X minutes
|
| 69 |
+
save_total_limit = 3 # keep only X most recent checkpoints
|
| 70 |
+
|
| 71 |
+
# Evaluation
|
| 72 |
+
# eval_strategy can be "steps", "epoch", or "minutes" (wall-clock time).
|
| 73 |
+
eval_strategy = "minutes"
|
| 74 |
+
eval_steps = 5 # evaluate every X minutes
|
| 75 |
+
# Evaluate on a fixed random subset of the validation set (0 = use all)
|
| 76 |
+
max_eval_samples = 5000
|
| 77 |
+
|
| 78 |
+
# Logging
|
| 79 |
+
logging_strategy = "steps"
|
| 80 |
+
logging_steps = 10 # log every 10 steps
|
| 81 |
+
logging_first_step = true # log the first step
|
| 82 |
+
|
| 83 |
+
# Reproducibility
|
| 84 |
+
seed = 42
|
| 85 |
+
|
| 86 |
+
# Performance
|
| 87 |
+
dataloader_num_workers = 2 # number of data loading threads
|
| 88 |
+
dataloader_prefetch_factor = 2 # batches prefetched per worker
|
| 89 |
+
dataloader_pin_memory = true # pin memory for faster GPU transfer
|
scheduler/wsd_20/eval1.log
ADDED
|
@@ -0,0 +1,205 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
W0814 14:06:57.324000 54418 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
|
| 2 |
+
W0814 14:06:57.348000 54418 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
|
| 3 |
+
|
| 4 |
+
==============================================================================
|
| 5 |
+
CHECKPOINT INSPECTION
|
| 6 |
+
==============================================================================
|
| 7 |
+
checkpoint : final
|
| 8 |
+
tokenizer : final
|
| 9 |
+
device : cuda dtype: torch.bfloat16 seed: 1337
|
| 10 |
+
|
| 11 |
+
|
| 12 |
+
==============================================================================
|
| 13 |
+
SUITE: MODEL INFO AND TRAINING LINEAGE
|
| 14 |
+
==============================================================================
|
| 15 |
+
checkpoint : final
|
| 16 |
+
model_type : llama
|
| 17 |
+
architecture : LlamaForCausalLM
|
| 18 |
+
device : cuda
|
| 19 |
+
dtype : torch.bfloat16
|
| 20 |
+
parameters : 77,083,392 (77.08M)
|
| 21 |
+
disk size : 308 MB
|
| 22 |
+
layers / hidden / heads : 8 / 768 / 8 (KV heads: 4)
|
| 23 |
+
vocab (cfg / tokenizer) : 32768 / 32752
|
| 24 |
+
context length : 1024
|
| 25 |
+
chat template : True
|
| 26 |
+
training lineage : base pretraining, ctx 1024
|
| 27 |
+
embedding mean norm : 3.6401 <- typical row length of the embedding matrix; very small or huge values signal trouble
|
| 28 |
+
embedding mean cosine : 0.1053 <- avg cosine between random embedding rows. Descriptive only: it starts near 0.0
|
| 29 |
+
because RANDOM init is near-orthogonal, and it GROWS as the model trains.
|
| 30 |
+
Near 0 therefore does NOT mean "healthy" - it can mean "barely trained".
|
| 31 |
+
(note: config vocab padded by 16 unused rows - harmless)
|
| 32 |
+
|
| 33 |
+
==============================================================================
|
| 34 |
+
SUITE: PERIOD FIDELITY ON FIXED PROBE SENTENCES
|
| 35 |
+
==============================================================================
|
| 36 |
+
Scoring 20 sentences (10 historical, 10 modern)
|
| 37 |
+
The MODERN/HISTORICAL ratio is the output that means something. The absolute
|
| 38 |
+
numbers come from ~20 sentences: too noisy to rank checkpoints (use evaluate2.py).
|
| 39 |
+
|
| 40 |
+
OVERALL (270 tokens)
|
| 41 |
+
perplexity : 257.77 <- 'average branching factor'; lower = less surprised = better fit to this text
|
| 42 |
+
mean token prob : 0.0817 <- average confidence on the actual next word
|
| 43 |
+
median token prob : 0.0078 <- typical confidence (robust to a few very bad tokens)
|
| 44 |
+
worst-token prob (min): 1.6e-08 <- the single most surprising word; tiny values = the model really didn't expect it
|
| 45 |
+
10th-percentile prob : 1.3e-05 <- confidence on the model's worst 10% of guesses
|
| 46 |
+
low-confidence tokens : 52.96% <- share of tokens predicted with <1% probability (red flags)
|
| 47 |
+
mean entropy : 4.673 nats <- how spread out the predictions are; ~0 = very certain, higher = hedging
|
| 48 |
+
|
| 49 |
+
GROUP COMPARISON
|
| 50 |
+
group sentences perplexity bits/byte
|
| 51 |
+
historical 10 81.12 1.196
|
| 52 |
+
modern 10 678.25 1.945
|
| 53 |
+
modern/historical ratio: 8.36 <- >1 means period text is easier for the model than modern text (desired for a vintage model with a ~1900 knowledge cutoff)
|
| 54 |
+
(bits/byte is the tokenizer-independent version - the only column here you may
|
| 55 |
+
compare between models that use DIFFERENT tokenizers.)
|
| 56 |
+
|
| 57 |
+
PER-SENTENCE DETAIL (sorted worst-first)
|
| 58 |
+
[modern ] ppl= 4265.97 minp=2.3e-07 The engineering team rewrote the whole backend micro
|
| 59 |
+
[modern ] ppl= 2813.13 minp=1.5e-06 Robots on the automated assembly line handle the ent
|
| 60 |
+
[modern ] ppl= 1315.86 minp=3.5e-06 Small retail startups now run almost all of their bu
|
| 61 |
+
[modern ] ppl= 996.46 minp=1.5e-06 Three hours stuck in traffic in the pouring rain mad
|
| 62 |
+
[modern ] ppl= 646.11 minp=1.5e-06 Millions of shoppers queue outside the store because
|
| 63 |
+
[modern ] ppl= 505.02 minp=1.6e-08 Members of the European Union parliament voted on th
|
| 64 |
+
[modern ] ppl= 500.69 minp=1.5e-05 She is finishing a graduate degree in machine learni
|
| 65 |
+
[modern ] ppl= 358.83 minp=4.3e-07 She gave up her whole weekend to help me move apartm
|
| 66 |
+
[historical] ppl= 282.68 minp=1.2e-05 The keeper fed the great serpent which the naturalis
|
| 67 |
+
[modern ] ppl= 248.15 minp=5.9e-06 I switched to a plant-based diet last year and I no
|
| 68 |
+
[modern ] ppl= 179.89 minp=2.2e-06 After years of hiding it from everyone at work, he t
|
| 69 |
+
[historical] ppl= 139.38 minp=1.0e-04 He drew a distinction so fine and over-scrupulous th
|
| 70 |
+
[historical] ppl= 100.21 minp=9.5e-06 The mountain rose above the valley in a silence sole
|
| 71 |
+
[historical] ppl= 90.91 minp=7.0e-05 The merchants of the port have grown rich upon their
|
| 72 |
+
[historical] ppl= 82.08 minp=1.2e-04 The weavers at their looms are employed in the wooll
|
| 73 |
+
[historical] ppl= 75.06 minp=3.8e-05 The Lord provideth for all his creatures, giving the
|
| 74 |
+
[historical] ppl= 69.82 minp=1.3e-05 The labouring poor of this parish are reduced to gre
|
| 75 |
+
[historical] ppl= 67.96 minp=4.6e-05 He gave up his fortune to the patient study of natur
|
| 76 |
+
[historical] ppl= 46.74 minp=1.3e-04 The ballroom was filled with dancing and laughter, a
|
| 77 |
+
[historical] ppl= 18.33 minp=3.5e-04 Her Majesty was pleased to summon the Lords and Comm
|
| 78 |
+
|
| 79 |
+
==============================================================================
|
| 80 |
+
SUITE: DIACHRONIC WORD-SENSE SEPARATION
|
| 81 |
+
==============================================================================
|
| 82 |
+
Extracting contextual embeddings for 10 shifted words in period AND modern sentences.
|
| 83 |
+
Every probe word sits LAST in its sentence, so the causal model has actually read
|
| 84 |
+
the disambiguating context before it represents the word.
|
| 85 |
+
|
| 86 |
+
SENSE SEPARATION (cosine similarity of the SAME word, period vs modern sentence)
|
| 87 |
+
lower value = the model represents the two senses differently (good sign for a
|
| 88 |
+
period model); ~1.0 = it treats them as the same word regardless of context
|
| 89 |
+
want : +0.343
|
| 90 |
+
python : +0.492
|
| 91 |
+
awful : +0.556
|
| 92 |
+
meat : +0.621
|
| 93 |
+
parliament : +0.623
|
| 94 |
+
nice : +0.626
|
| 95 |
+
manufacture : +0.670
|
| 96 |
+
gay : +0.718
|
| 97 |
+
commerce : +0.743
|
| 98 |
+
science : +0.839
|
| 99 |
+
MEAN : +0.623 <- near 1.0 = model ignores period vs modern usage; lower = senses differ
|
| 100 |
+
|
| 101 |
+
PAIRWISE COSINE SIMILARITY (period sentences, first 5 words)
|
| 102 |
+
these are CONTEXTUAL vectors, so they reflect the sentences as much as the words;
|
| 103 |
+
read the spread, not individual pairs. High mean = representations collapsed.
|
| 104 |
+
gay awful nice meat want
|
| 105 |
+
gay 1.000 0.451 0.346 0.314 0.315
|
| 106 |
+
awful 1.000 0.358 0.259 0.326
|
| 107 |
+
nice 1.000 0.294 0.345
|
| 108 |
+
meat 1.000 0.473
|
| 109 |
+
want 1.000
|
| 110 |
+
all 45 pairs: mean=+0.370 std=0.118 <- healthy spread is good
|
| 111 |
+
|
| 112 |
+
NEAREST NEIGHBOURS inside the probe-word set (period sentences)
|
| 113 |
+
gay : awful (+0.45), science (+0.40), commerce (+0.37), nice (+0.35), manufacture (+0.34)
|
| 114 |
+
awful : gay (+0.45), science (+0.40), nice (+0.36), parliament (+0.35), want (+0.33)
|
| 115 |
+
nice : science (+0.39), awful (+0.36), gay (+0.35), want (+0.35), manufacture (+0.33)
|
| 116 |
+
meat : science (+0.48), want (+0.47), parliament (+0.45), commerce (+0.44), manufacture (+0.43)
|
| 117 |
+
want : manufacture (+0.56), commerce (+0.55), science (+0.52), meat (+0.47), parliament (+0.46)
|
| 118 |
+
|
| 119 |
+
==============================================================================
|
| 120 |
+
SUITE: GENERATION PROBES
|
| 121 |
+
==============================================================================
|
| 122 |
+
|
| 123 |
+
--- Probe 1 ---
|
| 124 |
+
prompt : The history of the world is
|
| 125 |
+
continuation: too often confused and obscure; but the reader will find a strange and strange interest in the history of the world. It was one of the most remarkable events of modern history that the history of the world was a very simple one; and there are a thousand evidences of the truth of the present, which
|
| 126 |
+
stats: 60 tokens | distinct-1 0.64 | distinct-2 0.89 | echo 0.11
|
| 127 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 128 |
+
|
| 129 |
+
--- Probe 2 ---
|
| 130 |
+
prompt : What is God? God is
|
| 131 |
+
continuation: a man! He is a man who wants to be a man, but has a power to do nothing. It is God's work, and God's work.
|
| 132 |
+
|
| 133 |
+
If God has power to do something, and he does nothing, he is a man. If God's will is to do
|
| 134 |
+
stats: 60 tokens | distinct-1 0.60 | distinct-2 0.91 | echo 0.09
|
| 135 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 136 |
+
|
| 137 |
+
--- Probe 3 ---
|
| 138 |
+
prompt : In the year of our Lord eighteen hundred and
|
| 139 |
+
continuation: forty-eight; and the Lord shall be called, in the name of the Lord, and the glory of the world."
|
| 140 |
+
|
| 141 |
+
The second year of the Lord's ministry, was not a very long one. The first year of the Lord's ministry was a new one. A new ministry was
|
| 142 |
+
stats: 60 tokens | distinct-1 0.60 | distinct-2 0.85 | echo 0.09
|
| 143 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 144 |
+
|
| 145 |
+
--- Probe 4 ---
|
| 146 |
+
prompt : The manufacture of cotton
|
| 147 |
+
continuation: -growing, and the manufacture of silk and linen-growing machinery, has been carried on with the utmost care and energy. The cotton, as well as woollen fabrics, are manufactured and woven. The silk, as a trade, is now carried on in a great measure, and its production is
|
| 148 |
+
stats: 60 tokens | distinct-1 0.74 | distinct-2 0.98 | echo 0.02
|
| 149 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 150 |
+
|
| 151 |
+
--- Probe 5 ---
|
| 152 |
+
prompt : The steam engine
|
| 153 |
+
continuation: is the main factor, and in the latter respect the latter has more than equal claim to the other. In addition to the steam power and the steam power, it is said to be of greater advantage to the steam power, in the first place, to the use of the power to prevent a rapid
|
| 154 |
+
stats: 60 tokens | distinct-1 0.58 | distinct-2 0.85 | echo 0.13
|
| 155 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 156 |
+
|
| 157 |
+
--- Probe 6 ---
|
| 158 |
+
prompt : The telegraph
|
| 159 |
+
continuation: office and telegraph offices, is one of the most important of the great works of the day. The telegraph office is the only building for the telegraph office. It is a large and commodious building, built entirely of stone, and furnished with a number of iron and steel wires, and with the greatest care
|
| 160 |
+
stats: 60 tokens | distinct-1 0.63 | distinct-2 0.96 | echo 0.11
|
| 161 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 162 |
+
|
| 163 |
+
--- Probe 7 ---
|
| 164 |
+
prompt : My dearest sister,
|
| 165 |
+
continuation: a very good-natured young fellow, and very fond of her, for whom I have no doubt I have a very good reason to be grateful.
|
| 166 |
+
|
| 167 |
+
You will remember, my dear friend, how little I have been able to do, and how much I have had to do. You may
|
| 168 |
+
stats: 60 tokens | distinct-1 0.71 | distinct-2 0.92 | echo 0.04
|
| 169 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 170 |
+
|
| 171 |
+
--- Probe 8 ---
|
| 172 |
+
prompt : Greetings, my friend
|
| 173 |
+
continuation: , I shall take a short drive, if you please.
|
| 174 |
+
|
| 175 |
+
I will take a walk with you in the fields, with the birds, and in the woods. I will take my ride with you and go to the station.
|
| 176 |
+
|
| 177 |
+
It is a fine day, and I shall be glad
|
| 178 |
+
stats: 60 tokens | distinct-1 0.61 | distinct-2 0.88 | echo 0.04
|
| 179 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 180 |
+
|
| 181 |
+
--- Probe 9 ---
|
| 182 |
+
prompt : Her Majesty the Queen
|
| 183 |
+
continuation: of Scots, in the name of the king, was pleased to place herself under the title of a Queen, and to act according to her pleasure, so that she would never be married until she should be entitled to her crown; and, after all the expense of her life, she should have the
|
| 184 |
+
stats: 60 tokens | distinct-1 0.70 | distinct-2 0.96 | echo 0.04
|
| 185 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 186 |
+
|
| 187 |
+
--- Probe 10 ---
|
| 188 |
+
prompt : Chapter I.
|
| 189 |
+
continuation: The first question is whether a single sentence is not contained in the book of Judges and whether the second is not an exception to the general rule of the Judges.
|
| 190 |
+
|
| 191 |
+
If the words "degenerate" be taken as referring to a sentence in which the meaning is not "degener
|
| 192 |
+
stats: 60 tokens | distinct-1 0.67 | distinct-2 0.96 | echo 0.04
|
| 193 |
+
(distinct-n = vocabulary/phrase variety, near 0 = looping; echo = fraction of words repeated from the previous 4)
|
| 194 |
+
|
| 195 |
+
AGGREGATE over 10 probes: 600 tokens in 0.9s (689.4 tok/s)
|
| 196 |
+
mean distinct-1 0.649 | mean distinct-2 0.916 | mean echo 0.070 | worst probe: distinct-2 0.848 / echo 0.127
|
| 197 |
+
(sampling hides loops that greedy decoding exposes; evaluate2.py measures the greedy loop length.)
|
| 198 |
+
|
| 199 |
+
Full numeric results written to /home/cro/Dev/vintage-LLM/training/eval_results/evaluate-final.json
|
| 200 |
+
|
| 201 |
+
==============================================================================
|
| 202 |
+
INSPECTION COMPLETE
|
| 203 |
+
==============================================================================
|
| 204 |
+
This covered WHAT the checkpoint is, its lineage, and whether it is still vintage.
|
| 205 |
+
For "is it baked enough", run evaluate2.py on the same path.
|
scheduler/wsd_20/eval2.log
ADDED
|
@@ -0,0 +1,54 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
evaluating 1 checkpoint(s) on cuda (torch.bfloat16)
|
| 2 |
+
tokenizer: final
|
| 3 |
+
final ...W0814 14:07:03.865000 54494 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
|
| 4 |
+
W0814 14:07:03.881000 54494 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
|
| 5 |
+
|
| 6 |
+
bake 56/100 bpb 1.2769 logic 0.65 (15.2s)
|
| 7 |
+
|
| 8 |
+
report: eval2.md
|
| 9 |
+
json: eval2.json
|
| 10 |
+
|
| 11 |
+
# Evaluation: final
|
| 12 |
+
|
| 13 |
+
- **Parameters:** 77M
|
| 14 |
+
- **Note:** no trainer_state.json in the checkpoint - tokens seen unknown
|
| 15 |
+
|
| 16 |
+
## Verdict: HALF-BAKED (bake score 56/100)
|
| 17 |
+
|
| 18 |
+
Clearly undertrained. The style is there but the substance is not: expect confident nonsense, topic drift and heavy looping under greedy decoding. Do not fine-tune this for chat yet - keep pretraining.
|
| 19 |
+
|
| 20 |
+
## Where it sits (held-out bits/byte, lower = better)
|
| 21 |
+
|
| 22 |
+
```
|
| 23 |
+
3.50 untrained model (uniform noise)
|
| 24 |
+
2.00 word-salad
|
| 25 |
+
1.50 broken prose
|
| 26 |
+
1.33 500M model, ~0.7B tokens (early training)
|
| 27 |
+
--> 1.277 YOUR MODEL
|
| 28 |
+
1.19 500M model, ~4.7B tokens (undertrained but solid)
|
| 29 |
+
1.10 best sub-1B measured on this data (341M)
|
| 30 |
+
0.95 estimated sub-1B ceiling on this corpus
|
| 31 |
+
```
|
| 32 |
+
|
| 33 |
+
## Scores
|
| 34 |
+
|
| 35 |
+
| what | raw value | points /100 | weight | plain English |
|
| 36 |
+
|---|---|---|---|---|
|
| 37 |
+
| Held-out loss | 1.2769 bits/byte | 61 | 0.5 | how cheaply it predicts period text it never saw — the single best training signal |
|
| 38 |
+
| Logic | 0.650 acc, margin 0.024 | 36 | 0.25 | picks the *sensible* continuation over matched nonsense; 0.50 = coin-flip. normal for a good sub-1B model (Vintage1 scores 0.70-0.78) |
|
| 39 |
+
| Chat readiness | 1.1571 bits/byte | 56 | 0.15 | how cheap well-formed period dialogue already is — predicts fine-tuning ease |
|
| 40 |
+
| Hygiene | loop 23.8w, punct 0.00/100w | 79 | 0.1 | greedy-decoding loop length and broken punctuation |
|
| 41 |
+
|
| 42 |
+
## Period boundary
|
| 43 |
+
|
| 44 |
+
Clean. All 6 post-1900 trap words cost the model more than their period twins (mean shock +1.37 bits/byte, weakest pair 'motor car' at +0.82). No sign of modern text in training.
|
| 45 |
+
|
| 46 |
+
## See for yourself (sampled, t=0.8)
|
| 47 |
+
|
| 48 |
+
> **LONDON, Tuesday. — The committee appointed to inquire into the condition of the** Irish population throughout the country and the condition of the Irish people of Ireland during the last week has recently given Notice of the fact that the population of the country is increasing greatly more than was stated, the amount of the population being increasing in proportion. The committee reported that the Irish population in the United Kingdom is increasing in proportion as the Irish
|
| 49 |
+
>
|
| 50 |
+
> **A melancholy accident occurred on Thursday last at the works of Messrs. Harding and** Co., of Manchester, and on the 8th of November, 1842, the Failing to arrive at the place of meeting, a deputation was sent to the various counties and towns, to solicit the services of some influential persons, for the accommodation of the people, to aid in the work of the evangelical cause. The meeting was called a meeting of the members of the church, and was most cordially received by many memb
|
| 51 |
+
>
|
| 52 |
+
> **Brethren, the text which I have chosen for our consideration this morning is taken from** the work of the Brethren. We have not yet done anything for the Brethren. Our church is now filled to overflowing with a host of Christian men, to whom every one of the Brethren has a brotherly affection. We have been looking forward to a time when the work of the Brethren was more deeply felt by the Brethren. While we are thus engaged, we trust that the work of the Brethren will not cease to be a
|
| 53 |
+
>
|
| 54 |
+
|
scheduler/wsd_20/final/chat_template.jinja
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{%- for message in messages -%}
|
| 2 |
+
{{- bos_token + '<|' + message['role'] + '|>\n' + message['content'].strip() + eos_token + '\n' -}}
|
| 3 |
+
{%- endfor -%}
|
| 4 |
+
|
| 5 |
+
{%- if add_generation_prompt -%}
|
| 6 |
+
{{- bos_token + '<|assistant|>\n' -}}
|
| 7 |
+
{%- endif -%}
|
scheduler/wsd_20/final/config.json
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"LlamaForCausalLM"
|
| 4 |
+
],
|
| 5 |
+
"attention_bias": false,
|
| 6 |
+
"attention_dropout": 0.0,
|
| 7 |
+
"bos_token_id": 3,
|
| 8 |
+
"dtype": "float32",
|
| 9 |
+
"eos_token_id": 4,
|
| 10 |
+
"head_dim": 96,
|
| 11 |
+
"hidden_act": "silu",
|
| 12 |
+
"hidden_size": 768,
|
| 13 |
+
"initializer_range": 0.02,
|
| 14 |
+
"intermediate_size": 2048,
|
| 15 |
+
"max_position_embeddings": 1024,
|
| 16 |
+
"mlp_bias": false,
|
| 17 |
+
"model_type": "llama",
|
| 18 |
+
"num_attention_heads": 8,
|
| 19 |
+
"num_hidden_layers": 8,
|
| 20 |
+
"num_key_value_heads": 4,
|
| 21 |
+
"pad_token_id": 0,
|
| 22 |
+
"pretraining_tp": 1,
|
| 23 |
+
"rms_norm_eps": 1e-06,
|
| 24 |
+
"rope_parameters": {
|
| 25 |
+
"rope_theta": 10000.0,
|
| 26 |
+
"rope_type": "default"
|
| 27 |
+
},
|
| 28 |
+
"tie_word_embeddings": true,
|
| 29 |
+
"transformers_version": "5.14.1",
|
| 30 |
+
"use_cache": false,
|
| 31 |
+
"vocab_size": 32768
|
| 32 |
+
}
|
scheduler/wsd_20/final/generation_config.json
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_from_model_config": true,
|
| 3 |
+
"bos_token_id": 3,
|
| 4 |
+
"eos_token_id": [
|
| 5 |
+
4,
|
| 6 |
+
2
|
| 7 |
+
],
|
| 8 |
+
"output_attentions": false,
|
| 9 |
+
"output_hidden_states": false,
|
| 10 |
+
"pad_token_id": 0,
|
| 11 |
+
"transformers_version": "5.14.1",
|
| 12 |
+
"use_cache": false
|
| 13 |
+
}
|
scheduler/wsd_20/final/model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6a27d8bfc67c1bedb2e260f117c0b9f408942ca4d036cfb7606c6076d0db8061
|
| 3 |
+
size 308341784
|
scheduler/wsd_20/final/tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
scheduler/wsd_20/final/tokenizer_config.json
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"backend": "tokenizers",
|
| 3 |
+
"bos_token": "<|bos|>",
|
| 4 |
+
"eos_token": "<|eos|>",
|
| 5 |
+
"is_local": true,
|
| 6 |
+
"local_files_only": false,
|
| 7 |
+
"model_max_length": 1000000000000000019884624838656,
|
| 8 |
+
"pad_token": "<|pad|>",
|
| 9 |
+
"tokenizer_class": "TokenizersBackend",
|
| 10 |
+
"unk_token": "<|unk|>"
|
| 11 |
+
}
|
scheduler/wsd_20/trainer_state.json
ADDED
|
@@ -0,0 +1,649 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_global_step": null,
|
| 3 |
+
"best_metric": null,
|
| 4 |
+
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 0.009785965250516056,
|
| 6 |
+
"eval_steps": 5,
|
| 7 |
+
"global_step": 736,
|
| 8 |
+
"is_hyper_param_search": false,
|
| 9 |
+
"is_local_process_zero": true,
|
| 10 |
+
"is_world_process_zero": true,
|
| 11 |
+
"log_history": [
|
| 12 |
+
{
|
| 13 |
+
"epoch": 1.3296148438201165e-05,
|
| 14 |
+
"grad_norm": 5.513442516326904,
|
| 15 |
+
"learning_rate": 0.0,
|
| 16 |
+
"loss": 10.538199424743652,
|
| 17 |
+
"step": 1
|
| 18 |
+
},
|
| 19 |
+
{
|
| 20 |
+
"epoch": 0.00013296148438201165,
|
| 21 |
+
"grad_norm": 5.268533706665039,
|
| 22 |
+
"learning_rate": 0.00035999999999999997,
|
| 23 |
+
"loss": 10.136458502875435,
|
| 24 |
+
"step": 10
|
| 25 |
+
},
|
| 26 |
+
{
|
| 27 |
+
"epoch": 0.0002659229687640233,
|
| 28 |
+
"grad_norm": 1.034209132194519,
|
| 29 |
+
"learning_rate": 0.00076,
|
| 30 |
+
"loss": 8.0193359375,
|
| 31 |
+
"step": 20
|
| 32 |
+
},
|
| 33 |
+
{
|
| 34 |
+
"epoch": 0.0003988844531460349,
|
| 35 |
+
"grad_norm": 9.450255393981934,
|
| 36 |
+
"learning_rate": 0.00116,
|
| 37 |
+
"loss": 6.9860687255859375,
|
| 38 |
+
"step": 30
|
| 39 |
+
},
|
| 40 |
+
{
|
| 41 |
+
"epoch": 0.0005318459375280466,
|
| 42 |
+
"grad_norm": 5.685102462768555,
|
| 43 |
+
"learning_rate": 0.0015600000000000002,
|
| 44 |
+
"loss": 6.707894897460937,
|
| 45 |
+
"step": 40
|
| 46 |
+
},
|
| 47 |
+
{
|
| 48 |
+
"epoch": 0.0006648074219100582,
|
| 49 |
+
"grad_norm": 2.160360097885132,
|
| 50 |
+
"learning_rate": 0.00196,
|
| 51 |
+
"loss": 6.287691116333008,
|
| 52 |
+
"step": 50
|
| 53 |
+
},
|
| 54 |
+
{
|
| 55 |
+
"epoch": 0.0007977689062920698,
|
| 56 |
+
"grad_norm": 1.6388428211212158,
|
| 57 |
+
"learning_rate": 0.00236,
|
| 58 |
+
"loss": 5.944025039672852,
|
| 59 |
+
"step": 60
|
| 60 |
+
},
|
| 61 |
+
{
|
| 62 |
+
"epoch": 0.0008775457969212769,
|
| 63 |
+
"eval_loss": 5.732973575592041,
|
| 64 |
+
"eval_runtime": 22.9886,
|
| 65 |
+
"eval_samples_per_second": 217.499,
|
| 66 |
+
"eval_steps_per_second": 27.187,
|
| 67 |
+
"step": 66
|
| 68 |
+
},
|
| 69 |
+
{
|
| 70 |
+
"epoch": 0.0009307303906740815,
|
| 71 |
+
"grad_norm": 1.2074735164642334,
|
| 72 |
+
"learning_rate": 0.00276,
|
| 73 |
+
"loss": 5.747131729125977,
|
| 74 |
+
"step": 70
|
| 75 |
+
},
|
| 76 |
+
{
|
| 77 |
+
"epoch": 0.0010636918750560932,
|
| 78 |
+
"grad_norm": 1.132364273071289,
|
| 79 |
+
"learning_rate": 0.00316,
|
| 80 |
+
"loss": 5.60839729309082,
|
| 81 |
+
"step": 80
|
| 82 |
+
},
|
| 83 |
+
{
|
| 84 |
+
"epoch": 0.0011966533594381047,
|
| 85 |
+
"grad_norm": 1.0884277820587158,
|
| 86 |
+
"learning_rate": 0.0035600000000000002,
|
| 87 |
+
"loss": 5.502901077270508,
|
| 88 |
+
"step": 90
|
| 89 |
+
},
|
| 90 |
+
{
|
| 91 |
+
"epoch": 0.0013296148438201164,
|
| 92 |
+
"grad_norm": 1.3878921270370483,
|
| 93 |
+
"learning_rate": 0.00396,
|
| 94 |
+
"loss": 5.371665573120117,
|
| 95 |
+
"step": 100
|
| 96 |
+
},
|
| 97 |
+
{
|
| 98 |
+
"epoch": 0.001462576328202128,
|
| 99 |
+
"grad_norm": 1.2677963972091675,
|
| 100 |
+
"learning_rate": 0.00436,
|
| 101 |
+
"loss": 5.2512672424316404,
|
| 102 |
+
"step": 110
|
| 103 |
+
},
|
| 104 |
+
{
|
| 105 |
+
"epoch": 0.0015955378125841397,
|
| 106 |
+
"grad_norm": 0.9181460738182068,
|
| 107 |
+
"learning_rate": 0.0047599999999999995,
|
| 108 |
+
"loss": 5.126389694213867,
|
| 109 |
+
"step": 120
|
| 110 |
+
},
|
| 111 |
+
{
|
| 112 |
+
"epoch": 0.001701907000089749,
|
| 113 |
+
"eval_loss": 4.962139129638672,
|
| 114 |
+
"eval_runtime": 24.8338,
|
| 115 |
+
"eval_samples_per_second": 201.339,
|
| 116 |
+
"eval_steps_per_second": 25.167,
|
| 117 |
+
"step": 128
|
| 118 |
+
},
|
| 119 |
+
{
|
| 120 |
+
"epoch": 0.0017284992969661514,
|
| 121 |
+
"grad_norm": 0.7834004163742065,
|
| 122 |
+
"learning_rate": 0.0051600000000000005,
|
| 123 |
+
"loss": 5.010536575317383,
|
| 124 |
+
"step": 130
|
| 125 |
+
},
|
| 126 |
+
{
|
| 127 |
+
"epoch": 0.001861460781348163,
|
| 128 |
+
"grad_norm": 0.7623927593231201,
|
| 129 |
+
"learning_rate": 0.00556,
|
| 130 |
+
"loss": 4.908084106445313,
|
| 131 |
+
"step": 140
|
| 132 |
+
},
|
| 133 |
+
{
|
| 134 |
+
"epoch": 0.0019944222657301744,
|
| 135 |
+
"grad_norm": 0.5423034429550171,
|
| 136 |
+
"learning_rate": 0.00596,
|
| 137 |
+
"loss": 4.820485305786133,
|
| 138 |
+
"step": 150
|
| 139 |
+
},
|
| 140 |
+
{
|
| 141 |
+
"epoch": 0.0021273837501121864,
|
| 142 |
+
"grad_norm": 0.5046894550323486,
|
| 143 |
+
"learning_rate": 0.00636,
|
| 144 |
+
"loss": 4.753760528564453,
|
| 145 |
+
"step": 160
|
| 146 |
+
},
|
| 147 |
+
{
|
| 148 |
+
"epoch": 0.002260345234494198,
|
| 149 |
+
"grad_norm": 0.3807763159275055,
|
| 150 |
+
"learning_rate": 0.0067599999999999995,
|
| 151 |
+
"loss": 4.694861602783203,
|
| 152 |
+
"step": 170
|
| 153 |
+
},
|
| 154 |
+
{
|
| 155 |
+
"epoch": 0.0023933067188762094,
|
| 156 |
+
"grad_norm": 0.37175047397613525,
|
| 157 |
+
"learning_rate": 0.0071600000000000006,
|
| 158 |
+
"loss": 4.652392578125,
|
| 159 |
+
"step": 180
|
| 160 |
+
},
|
| 161 |
+
{
|
| 162 |
+
"epoch": 0.0024730836095054167,
|
| 163 |
+
"eval_loss": 4.595963954925537,
|
| 164 |
+
"eval_runtime": 23.9837,
|
| 165 |
+
"eval_samples_per_second": 208.475,
|
| 166 |
+
"eval_steps_per_second": 26.059,
|
| 167 |
+
"step": 186
|
| 168 |
+
},
|
| 169 |
+
{
|
| 170 |
+
"epoch": 0.0025262682032582214,
|
| 171 |
+
"grad_norm": 0.3488306999206543,
|
| 172 |
+
"learning_rate": 0.00756,
|
| 173 |
+
"loss": 4.597007369995117,
|
| 174 |
+
"step": 190
|
| 175 |
+
},
|
| 176 |
+
{
|
| 177 |
+
"epoch": 0.002659229687640233,
|
| 178 |
+
"grad_norm": 0.3351574242115021,
|
| 179 |
+
"learning_rate": 0.00796,
|
| 180 |
+
"loss": 4.566041946411133,
|
| 181 |
+
"step": 200
|
| 182 |
+
},
|
| 183 |
+
{
|
| 184 |
+
"epoch": 0.0027921911720222444,
|
| 185 |
+
"grad_norm": 0.3297984302043915,
|
| 186 |
+
"learning_rate": 0.008,
|
| 187 |
+
"loss": 4.5214378356933596,
|
| 188 |
+
"step": 210
|
| 189 |
+
},
|
| 190 |
+
{
|
| 191 |
+
"epoch": 0.002925152656404256,
|
| 192 |
+
"grad_norm": 0.2526053190231323,
|
| 193 |
+
"learning_rate": 0.008,
|
| 194 |
+
"loss": 4.468568420410156,
|
| 195 |
+
"step": 220
|
| 196 |
+
},
|
| 197 |
+
{
|
| 198 |
+
"epoch": 0.003058114140786268,
|
| 199 |
+
"grad_norm": 0.24721451103687286,
|
| 200 |
+
"learning_rate": 0.008,
|
| 201 |
+
"loss": 4.431998825073242,
|
| 202 |
+
"step": 230
|
| 203 |
+
},
|
| 204 |
+
{
|
| 205 |
+
"epoch": 0.0031910756251682794,
|
| 206 |
+
"grad_norm": 0.2912442684173584,
|
| 207 |
+
"learning_rate": 0.008,
|
| 208 |
+
"loss": 4.396870040893555,
|
| 209 |
+
"step": 240
|
| 210 |
+
},
|
| 211 |
+
{
|
| 212 |
+
"epoch": 0.0032575563673592853,
|
| 213 |
+
"eval_loss": 4.358976364135742,
|
| 214 |
+
"eval_runtime": 24.0964,
|
| 215 |
+
"eval_samples_per_second": 207.5,
|
| 216 |
+
"eval_steps_per_second": 25.937,
|
| 217 |
+
"step": 245
|
| 218 |
+
},
|
| 219 |
+
{
|
| 220 |
+
"epoch": 0.003324037109550291,
|
| 221 |
+
"grad_norm": 0.28560253977775574,
|
| 222 |
+
"learning_rate": 0.008,
|
| 223 |
+
"loss": 4.367413711547852,
|
| 224 |
+
"step": 250
|
| 225 |
+
},
|
| 226 |
+
{
|
| 227 |
+
"epoch": 0.003456998593932303,
|
| 228 |
+
"grad_norm": 0.21196810901165009,
|
| 229 |
+
"learning_rate": 0.008,
|
| 230 |
+
"loss": 4.331097793579102,
|
| 231 |
+
"step": 260
|
| 232 |
+
},
|
| 233 |
+
{
|
| 234 |
+
"epoch": 0.0035899600783143143,
|
| 235 |
+
"grad_norm": 0.22107456624507904,
|
| 236 |
+
"learning_rate": 0.008,
|
| 237 |
+
"loss": 4.302482604980469,
|
| 238 |
+
"step": 270
|
| 239 |
+
},
|
| 240 |
+
{
|
| 241 |
+
"epoch": 0.003722921562696326,
|
| 242 |
+
"grad_norm": 0.1877111792564392,
|
| 243 |
+
"learning_rate": 0.008,
|
| 244 |
+
"loss": 4.268944931030274,
|
| 245 |
+
"step": 280
|
| 246 |
+
},
|
| 247 |
+
{
|
| 248 |
+
"epoch": 0.0038558830470783374,
|
| 249 |
+
"grad_norm": 0.21084874868392944,
|
| 250 |
+
"learning_rate": 0.008,
|
| 251 |
+
"loss": 4.2542469024658205,
|
| 252 |
+
"step": 290
|
| 253 |
+
},
|
| 254 |
+
{
|
| 255 |
+
"epoch": 0.003988844531460349,
|
| 256 |
+
"grad_norm": 0.20695072412490845,
|
| 257 |
+
"learning_rate": 0.008,
|
| 258 |
+
"loss": 4.227492904663086,
|
| 259 |
+
"step": 300
|
| 260 |
+
},
|
| 261 |
+
{
|
| 262 |
+
"epoch": 0.004068621422089557,
|
| 263 |
+
"eval_loss": 4.206057071685791,
|
| 264 |
+
"eval_runtime": 24.3778,
|
| 265 |
+
"eval_samples_per_second": 205.104,
|
| 266 |
+
"eval_steps_per_second": 25.638,
|
| 267 |
+
"step": 306
|
| 268 |
+
},
|
| 269 |
+
{
|
| 270 |
+
"epoch": 0.004121806015842361,
|
| 271 |
+
"grad_norm": 0.19029781222343445,
|
| 272 |
+
"learning_rate": 0.008,
|
| 273 |
+
"loss": 4.208528518676758,
|
| 274 |
+
"step": 310
|
| 275 |
+
},
|
| 276 |
+
{
|
| 277 |
+
"epoch": 0.004254767500224373,
|
| 278 |
+
"grad_norm": 0.1889178603887558,
|
| 279 |
+
"learning_rate": 0.008,
|
| 280 |
+
"loss": 4.187363815307617,
|
| 281 |
+
"step": 320
|
| 282 |
+
},
|
| 283 |
+
{
|
| 284 |
+
"epoch": 0.004387728984606384,
|
| 285 |
+
"grad_norm": 0.19070716202259064,
|
| 286 |
+
"learning_rate": 0.008,
|
| 287 |
+
"loss": 4.173851776123047,
|
| 288 |
+
"step": 330
|
| 289 |
+
},
|
| 290 |
+
{
|
| 291 |
+
"epoch": 0.004520690468988396,
|
| 292 |
+
"grad_norm": 0.15929149091243744,
|
| 293 |
+
"learning_rate": 0.008,
|
| 294 |
+
"loss": 4.148160171508789,
|
| 295 |
+
"step": 340
|
| 296 |
+
},
|
| 297 |
+
{
|
| 298 |
+
"epoch": 0.004653651953370408,
|
| 299 |
+
"grad_norm": 0.17022773623466492,
|
| 300 |
+
"learning_rate": 0.008,
|
| 301 |
+
"loss": 4.132599258422852,
|
| 302 |
+
"step": 350
|
| 303 |
+
},
|
| 304 |
+
{
|
| 305 |
+
"epoch": 0.004786613437752419,
|
| 306 |
+
"grad_norm": 0.18009495735168457,
|
| 307 |
+
"learning_rate": 0.008,
|
| 308 |
+
"loss": 4.132373809814453,
|
| 309 |
+
"step": 360
|
| 310 |
+
},
|
| 311 |
+
{
|
| 312 |
+
"epoch": 0.004879686476819827,
|
| 313 |
+
"eval_loss": 4.1056389808654785,
|
| 314 |
+
"eval_runtime": 23.9848,
|
| 315 |
+
"eval_samples_per_second": 208.465,
|
| 316 |
+
"eval_steps_per_second": 26.058,
|
| 317 |
+
"step": 367
|
| 318 |
+
},
|
| 319 |
+
{
|
| 320 |
+
"epoch": 0.004919574922134431,
|
| 321 |
+
"grad_norm": 0.17529530823230743,
|
| 322 |
+
"learning_rate": 0.008,
|
| 323 |
+
"loss": 4.117653656005859,
|
| 324 |
+
"step": 370
|
| 325 |
+
},
|
| 326 |
+
{
|
| 327 |
+
"epoch": 0.005052536406516443,
|
| 328 |
+
"grad_norm": 0.1463332176208496,
|
| 329 |
+
"learning_rate": 0.008,
|
| 330 |
+
"loss": 4.096078109741211,
|
| 331 |
+
"step": 380
|
| 332 |
+
},
|
| 333 |
+
{
|
| 334 |
+
"epoch": 0.005185497890898454,
|
| 335 |
+
"grad_norm": 0.14370259642601013,
|
| 336 |
+
"learning_rate": 0.008,
|
| 337 |
+
"loss": 4.084194946289062,
|
| 338 |
+
"step": 390
|
| 339 |
+
},
|
| 340 |
+
{
|
| 341 |
+
"epoch": 0.005318459375280466,
|
| 342 |
+
"grad_norm": 0.16615846753120422,
|
| 343 |
+
"learning_rate": 0.008,
|
| 344 |
+
"loss": 4.075920867919922,
|
| 345 |
+
"step": 400
|
| 346 |
+
},
|
| 347 |
+
{
|
| 348 |
+
"epoch": 0.005451420859662477,
|
| 349 |
+
"grad_norm": 0.15509292483329773,
|
| 350 |
+
"learning_rate": 0.008,
|
| 351 |
+
"loss": 4.054693603515625,
|
| 352 |
+
"step": 410
|
| 353 |
+
},
|
| 354 |
+
{
|
| 355 |
+
"epoch": 0.005584382344044489,
|
| 356 |
+
"grad_norm": 0.1333627849817276,
|
| 357 |
+
"learning_rate": 0.008,
|
| 358 |
+
"loss": 4.047085952758789,
|
| 359 |
+
"step": 420
|
| 360 |
+
},
|
| 361 |
+
{
|
| 362 |
+
"epoch": 0.0057040476799882995,
|
| 363 |
+
"eval_loss": 4.034159183502197,
|
| 364 |
+
"eval_runtime": 24.31,
|
| 365 |
+
"eval_samples_per_second": 205.676,
|
| 366 |
+
"eval_steps_per_second": 25.71,
|
| 367 |
+
"step": 429
|
| 368 |
+
},
|
| 369 |
+
{
|
| 370 |
+
"epoch": 0.005717343828426501,
|
| 371 |
+
"grad_norm": 0.12619031965732574,
|
| 372 |
+
"learning_rate": 0.008,
|
| 373 |
+
"loss": 4.037934494018555,
|
| 374 |
+
"step": 430
|
| 375 |
+
},
|
| 376 |
+
{
|
| 377 |
+
"epoch": 0.005850305312808512,
|
| 378 |
+
"grad_norm": 0.14818833768367767,
|
| 379 |
+
"learning_rate": 0.008,
|
| 380 |
+
"loss": 4.030204772949219,
|
| 381 |
+
"step": 440
|
| 382 |
+
},
|
| 383 |
+
{
|
| 384 |
+
"epoch": 0.005983266797190524,
|
| 385 |
+
"grad_norm": 0.11952793598175049,
|
| 386 |
+
"learning_rate": 0.008,
|
| 387 |
+
"loss": 4.024636077880859,
|
| 388 |
+
"step": 450
|
| 389 |
+
},
|
| 390 |
+
{
|
| 391 |
+
"epoch": 0.006116228281572536,
|
| 392 |
+
"grad_norm": 0.12302638590335846,
|
| 393 |
+
"learning_rate": 0.008,
|
| 394 |
+
"loss": 4.0040229797363285,
|
| 395 |
+
"step": 460
|
| 396 |
+
},
|
| 397 |
+
{
|
| 398 |
+
"epoch": 0.006249189765954547,
|
| 399 |
+
"grad_norm": 0.15813544392585754,
|
| 400 |
+
"learning_rate": 0.008,
|
| 401 |
+
"loss": 4.0008289337158205,
|
| 402 |
+
"step": 470
|
| 403 |
+
},
|
| 404 |
+
{
|
| 405 |
+
"epoch": 0.006382151250336559,
|
| 406 |
+
"grad_norm": 0.11854846030473709,
|
| 407 |
+
"learning_rate": 0.008,
|
| 408 |
+
"loss": 3.994869613647461,
|
| 409 |
+
"step": 480
|
| 410 |
+
},
|
| 411 |
+
{
|
| 412 |
+
"epoch": 0.006515112734718571,
|
| 413 |
+
"grad_norm": 0.14335912466049194,
|
| 414 |
+
"learning_rate": 0.008,
|
| 415 |
+
"loss": 3.9867435455322267,
|
| 416 |
+
"step": 490
|
| 417 |
+
},
|
| 418 |
+
{
|
| 419 |
+
"epoch": 0.006541705031594972,
|
| 420 |
+
"eval_loss": 3.9793663024902344,
|
| 421 |
+
"eval_runtime": 23.0058,
|
| 422 |
+
"eval_samples_per_second": 217.336,
|
| 423 |
+
"eval_steps_per_second": 27.167,
|
| 424 |
+
"step": 492
|
| 425 |
+
},
|
| 426 |
+
{
|
| 427 |
+
"epoch": 0.006648074219100582,
|
| 428 |
+
"grad_norm": 0.10275301337242126,
|
| 429 |
+
"learning_rate": 0.008,
|
| 430 |
+
"loss": 3.9850521087646484,
|
| 431 |
+
"step": 500
|
| 432 |
+
},
|
| 433 |
+
{
|
| 434 |
+
"epoch": 0.006781035703482594,
|
| 435 |
+
"grad_norm": 0.12899844348430634,
|
| 436 |
+
"learning_rate": 0.008,
|
| 437 |
+
"loss": 3.9827651977539062,
|
| 438 |
+
"step": 510
|
| 439 |
+
},
|
| 440 |
+
{
|
| 441 |
+
"epoch": 0.006913997187864606,
|
| 442 |
+
"grad_norm": 0.1306857168674469,
|
| 443 |
+
"learning_rate": 0.008,
|
| 444 |
+
"loss": 3.962184524536133,
|
| 445 |
+
"step": 520
|
| 446 |
+
},
|
| 447 |
+
{
|
| 448 |
+
"epoch": 0.007046958672246617,
|
| 449 |
+
"grad_norm": 0.10865853726863861,
|
| 450 |
+
"learning_rate": 0.008,
|
| 451 |
+
"loss": 3.951933670043945,
|
| 452 |
+
"step": 530
|
| 453 |
+
},
|
| 454 |
+
{
|
| 455 |
+
"epoch": 0.007179920156628629,
|
| 456 |
+
"grad_norm": 0.11110885441303253,
|
| 457 |
+
"learning_rate": 0.008,
|
| 458 |
+
"loss": 3.94482536315918,
|
| 459 |
+
"step": 540
|
| 460 |
+
},
|
| 461 |
+
{
|
| 462 |
+
"epoch": 0.007312881641010641,
|
| 463 |
+
"grad_norm": 0.10243042558431625,
|
| 464 |
+
"learning_rate": 0.008,
|
| 465 |
+
"loss": 3.942932891845703,
|
| 466 |
+
"step": 550
|
| 467 |
+
},
|
| 468 |
+
{
|
| 469 |
+
"epoch": 0.007366066234763445,
|
| 470 |
+
"eval_loss": 3.933987855911255,
|
| 471 |
+
"eval_runtime": 23.7581,
|
| 472 |
+
"eval_samples_per_second": 210.455,
|
| 473 |
+
"eval_steps_per_second": 26.307,
|
| 474 |
+
"step": 554
|
| 475 |
+
},
|
| 476 |
+
{
|
| 477 |
+
"epoch": 0.007445843125392652,
|
| 478 |
+
"grad_norm": 0.10774345695972443,
|
| 479 |
+
"learning_rate": 0.008,
|
| 480 |
+
"loss": 3.9434852600097656,
|
| 481 |
+
"step": 560
|
| 482 |
+
},
|
| 483 |
+
{
|
| 484 |
+
"epoch": 0.007578804609774664,
|
| 485 |
+
"grad_norm": 0.10925770550966263,
|
| 486 |
+
"learning_rate": 0.008,
|
| 487 |
+
"loss": 3.9292713165283204,
|
| 488 |
+
"step": 570
|
| 489 |
+
},
|
| 490 |
+
{
|
| 491 |
+
"epoch": 0.007711766094156675,
|
| 492 |
+
"grad_norm": 0.10408780723810196,
|
| 493 |
+
"learning_rate": 0.008,
|
| 494 |
+
"loss": 3.917940139770508,
|
| 495 |
+
"step": 580
|
| 496 |
+
},
|
| 497 |
+
{
|
| 498 |
+
"epoch": 0.007844727578538687,
|
| 499 |
+
"grad_norm": 0.10272197425365448,
|
| 500 |
+
"learning_rate": 0.008,
|
| 501 |
+
"loss": 3.920045852661133,
|
| 502 |
+
"step": 590
|
| 503 |
+
},
|
| 504 |
+
{
|
| 505 |
+
"epoch": 0.007977689062920698,
|
| 506 |
+
"grad_norm": 0.09875880181789398,
|
| 507 |
+
"learning_rate": 0.008,
|
| 508 |
+
"loss": 3.9055149078369142,
|
| 509 |
+
"step": 600
|
| 510 |
+
},
|
| 511 |
+
{
|
| 512 |
+
"epoch": 0.00811065054730271,
|
| 513 |
+
"grad_norm": 0.12342850863933563,
|
| 514 |
+
"learning_rate": 0.007947368421052632,
|
| 515 |
+
"loss": 3.907426452636719,
|
| 516 |
+
"step": 610
|
| 517 |
+
},
|
| 518 |
+
{
|
| 519 |
+
"epoch": 0.008190427437931917,
|
| 520 |
+
"eval_loss": 3.893834352493286,
|
| 521 |
+
"eval_runtime": 23.1899,
|
| 522 |
+
"eval_samples_per_second": 215.611,
|
| 523 |
+
"eval_steps_per_second": 26.951,
|
| 524 |
+
"step": 616
|
| 525 |
+
},
|
| 526 |
+
{
|
| 527 |
+
"epoch": 0.008243612031684722,
|
| 528 |
+
"grad_norm": 0.1073056310415268,
|
| 529 |
+
"learning_rate": 0.007421052631578947,
|
| 530 |
+
"loss": 3.886861038208008,
|
| 531 |
+
"step": 620
|
| 532 |
+
},
|
| 533 |
+
{
|
| 534 |
+
"epoch": 0.008376573516066733,
|
| 535 |
+
"grad_norm": 0.09036897122859955,
|
| 536 |
+
"learning_rate": 0.006894736842105263,
|
| 537 |
+
"loss": 3.8772003173828127,
|
| 538 |
+
"step": 630
|
| 539 |
+
},
|
| 540 |
+
{
|
| 541 |
+
"epoch": 0.008509535000448746,
|
| 542 |
+
"grad_norm": 0.0773363932967186,
|
| 543 |
+
"learning_rate": 0.006368421052631579,
|
| 544 |
+
"loss": 3.8692962646484377,
|
| 545 |
+
"step": 640
|
| 546 |
+
},
|
| 547 |
+
{
|
| 548 |
+
"epoch": 0.008642496484830757,
|
| 549 |
+
"grad_norm": 0.08151130378246307,
|
| 550 |
+
"learning_rate": 0.005842105263157895,
|
| 551 |
+
"loss": 3.8507423400878906,
|
| 552 |
+
"step": 650
|
| 553 |
+
},
|
| 554 |
+
{
|
| 555 |
+
"epoch": 0.008775457969212768,
|
| 556 |
+
"grad_norm": 0.08117716759443283,
|
| 557 |
+
"learning_rate": 0.00531578947368421,
|
| 558 |
+
"loss": 3.837518310546875,
|
| 559 |
+
"step": 660
|
| 560 |
+
},
|
| 561 |
+
{
|
| 562 |
+
"epoch": 0.00890841945359478,
|
| 563 |
+
"grad_norm": 0.06505687534809113,
|
| 564 |
+
"learning_rate": 0.004789473684210526,
|
| 565 |
+
"loss": 3.8176422119140625,
|
| 566 |
+
"step": 670
|
| 567 |
+
},
|
| 568 |
+
{
|
| 569 |
+
"epoch": 0.009014788641100389,
|
| 570 |
+
"eval_loss": 3.802332878112793,
|
| 571 |
+
"eval_runtime": 23.6775,
|
| 572 |
+
"eval_samples_per_second": 211.171,
|
| 573 |
+
"eval_steps_per_second": 26.396,
|
| 574 |
+
"step": 678
|
| 575 |
+
},
|
| 576 |
+
{
|
| 577 |
+
"epoch": 0.009041380937976792,
|
| 578 |
+
"grad_norm": 0.07074195891618729,
|
| 579 |
+
"learning_rate": 0.004263157894736843,
|
| 580 |
+
"loss": 3.797995376586914,
|
| 581 |
+
"step": 680
|
| 582 |
+
},
|
| 583 |
+
{
|
| 584 |
+
"epoch": 0.009174342422358803,
|
| 585 |
+
"grad_norm": 0.06876317411661148,
|
| 586 |
+
"learning_rate": 0.0037368421052631578,
|
| 587 |
+
"loss": 3.7925949096679688,
|
| 588 |
+
"step": 690
|
| 589 |
+
},
|
| 590 |
+
{
|
| 591 |
+
"epoch": 0.009307303906740816,
|
| 592 |
+
"grad_norm": 0.060418933629989624,
|
| 593 |
+
"learning_rate": 0.0032105263157894736,
|
| 594 |
+
"loss": 3.7755409240722657,
|
| 595 |
+
"step": 700
|
| 596 |
+
},
|
| 597 |
+
{
|
| 598 |
+
"epoch": 0.009440265391122827,
|
| 599 |
+
"grad_norm": 0.05723872035741806,
|
| 600 |
+
"learning_rate": 0.0026842105263157894,
|
| 601 |
+
"loss": 3.764757537841797,
|
| 602 |
+
"step": 710
|
| 603 |
+
},
|
| 604 |
+
{
|
| 605 |
+
"epoch": 0.009573226875504838,
|
| 606 |
+
"grad_norm": 0.05702441930770874,
|
| 607 |
+
"learning_rate": 0.002157894736842105,
|
| 608 |
+
"loss": 3.7531177520751955,
|
| 609 |
+
"step": 720
|
| 610 |
+
},
|
| 611 |
+
{
|
| 612 |
+
"epoch": 0.00970618835988685,
|
| 613 |
+
"grad_norm": 0.05637257546186447,
|
| 614 |
+
"learning_rate": 0.0016315789473684212,
|
| 615 |
+
"loss": 3.7494365692138674,
|
| 616 |
+
"step": 730
|
| 617 |
+
},
|
| 618 |
+
{
|
| 619 |
+
"epoch": 0.009785965250516056,
|
| 620 |
+
"step": 736,
|
| 621 |
+
"total_flos": 6.010122542461747e+16,
|
| 622 |
+
"train_loss": 4.4993856588135595,
|
| 623 |
+
"train_runtime": 3603.0923,
|
| 624 |
+
"train_samples_per_second": 53.998,
|
| 625 |
+
"train_steps_per_second": 0.211
|
| 626 |
+
}
|
| 627 |
+
],
|
| 628 |
+
"logging_steps": 10,
|
| 629 |
+
"max_steps": 760,
|
| 630 |
+
"num_input_tokens_seen": 0,
|
| 631 |
+
"num_train_epochs": 1,
|
| 632 |
+
"save_steps": 5,
|
| 633 |
+
"stateful_callbacks": {
|
| 634 |
+
"TrainerControl": {
|
| 635 |
+
"args": {
|
| 636 |
+
"should_epoch_stop": false,
|
| 637 |
+
"should_evaluate": false,
|
| 638 |
+
"should_log": false,
|
| 639 |
+
"should_save": true,
|
| 640 |
+
"should_training_stop": true
|
| 641 |
+
},
|
| 642 |
+
"attributes": {}
|
| 643 |
+
}
|
| 644 |
+
},
|
| 645 |
+
"total_flos": 6.010122542461747e+16,
|
| 646 |
+
"train_batch_size": 8,
|
| 647 |
+
"trial_name": null,
|
| 648 |
+
"trial_params": null
|
| 649 |
+
}
|