Archive full MsTok step 17167 and matched FP64 generation evaluation
Browse files- iclr-debug-decoder-135b-20260916/decoder-mstok/checkpoint-17167-README.md +12 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/accepted-entropy-ge-4p5.json +0 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/controller.py +54 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/evaluator.py +187 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/manifest.json +26 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/manual-eval-status.json +6 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/quality-summary.json +87 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-0.json +0 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-0.samples.json +0 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-1.json +0 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-1.samples.json +0 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-2.json +0 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-2.samples.json +0 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-3.json +0 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-3.samples.json +0 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-4.json +0 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-4.samples.json +0 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/report.md +49 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/summarize-quality.py +52 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/summary.json +41 -0
- iclr-debug-decoder-135b-20260916/decoder-mstok/milestone-iter-17167.pt +3 -0
- iclr-debug-decoder-135b-20260916/provenance/hf-backup-step-17167-manifest.json +131 -0
iclr-debug-decoder-135b-20260916/decoder-mstok/checkpoint-17167-README.md
ADDED
|
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Decoder-aligned joint MsTok: step 17,167
|
| 2 |
+
|
| 3 |
+
Intermediate checkpoint from the original full `iclr-debug-decoder-135b-20260916` run (135B learning-rate horizon).
|
| 4 |
+
|
| 5 |
+
- 17,167 optimizer updates; 13,500,678,144 raw input positions including padding.
|
| 6 |
+
- Full resumable checkpoint, including model components, optimizer, scheduler, configuration, and eight per-rank RNG states.
|
| 7 |
+
- Fresh evaluation: 640 samples across seeds 0–4, supplied true level-zero code, untruncated temperature-1 sampling. Generator uses float32; reference GPT-2 Large uses FP64 with eager attention. TF32 disabled.
|
| 8 |
+
- Generation PPL: 109.261 +/- 2.332 standard error across seeds. Mean empirical token entropy: 4.364 nats.
|
| 9 |
+
- Filtering the existing samples for entropy >= 4.5 retains 109/640 (17.03%), with pooled token-weighted PPL 201.717.
|
| 10 |
+
- Replacement characters appear in 66.56% of all samples. Low generation PPL alone does not establish generation quality.
|
| 11 |
+
|
| 12 |
+
Evaluation artifacts are in `evaluation/step-17167-fp64-rerun-20260919`. Raw generations, per-sample scores, accepted samples, evaluator scripts, and configuration/provenance are included. This is a fresh matched evaluation; any previously archived step-17,167 results used different generation settings and are preserved separately. Absolute paths describe the original environment. Export component weights using `utils.iclr_training.export_training_checkpoint` from the matching source checkout.
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/accepted-entropy-ge-4p5.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/controller.py
ADDED
|
@@ -0,0 +1,54 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import datetime
|
| 2 |
+
import json
|
| 3 |
+
import math
|
| 4 |
+
import os
|
| 5 |
+
from pathlib import Path
|
| 6 |
+
import subprocess
|
| 7 |
+
import sys
|
| 8 |
+
|
| 9 |
+
REPO = Path('/home/ubuntu/mstok-runs/iclr-debug-joint-20260916')
|
| 10 |
+
STAGE = Path('/home/ubuntu/mstok-results/iclr-debug-decoder-135b-20260916/decoder-mstok')
|
| 11 |
+
OUT = STAGE / 'evaluation/step-17167-fp64-rerun-20260919'
|
| 12 |
+
EXPORT = STAGE / 'exports/step-17167'
|
| 13 |
+
ENV = dict(os.environ, CUDA_VISIBLE_DEVICES='0', OMP_NUM_THREADS='1',
|
| 14 |
+
TOKENIZERS_PARALLELISM='false', PYTORCH_ALLOC_CONF='expandable_segments:True')
|
| 15 |
+
|
| 16 |
+
def status(state, **extra):
|
| 17 |
+
payload = dict(state=state, updated_utc=datetime.datetime.now(datetime.timezone.utc).isoformat(),
|
| 18 |
+
controller_pid=os.getpid(), **extra)
|
| 19 |
+
temporary = OUT / 'manual-eval-status.tmp'
|
| 20 |
+
temporary.write_text(json.dumps(payload, indent=2) + '\n')
|
| 21 |
+
temporary.replace(OUT / 'manual-eval-status.json')
|
| 22 |
+
print(json.dumps(payload), flush=True)
|
| 23 |
+
|
| 24 |
+
try:
|
| 25 |
+
for seed in range(5):
|
| 26 |
+
result = OUT / f'random-seed-{seed}.json'
|
| 27 |
+
if not result.exists():
|
| 28 |
+
status('evaluating', seed=seed)
|
| 29 |
+
code = "import runpy,torch; torch.cuda.set_per_process_memory_fraction(0.15); runpy.run_path('/home/ubuntu/mstok-results/iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/evaluator.py',run_name='__main__')"
|
| 30 |
+
command = [sys.executable, '-u', '-c', code, '--ckpt', str(EXPORT/'ncp.pt'),
|
| 31 |
+
'--codec', str(EXPORT/'vqvae.pt'), '--validation-source',
|
| 32 |
+
'/home/ubuntu/data/full_owt/valid_gpt2.bin', '--ref-model', 'gpt2-large',
|
| 33 |
+
'--ref-dtype', 'float64', '--seed', str(seed), '--n-samples', '128',
|
| 34 |
+
'--batch-size', '4', '--temperature', '1', '--top-k', '0', '--top-p', '1',
|
| 35 |
+
'--save-all-generations', '--out', str(result)]
|
| 36 |
+
with result.with_suffix('.log').open('a') as log:
|
| 37 |
+
subprocess.run(command, cwd=REPO, env=ENV, stdout=log, stderr=subprocess.STDOUT,
|
| 38 |
+
check=True, timeout=1800)
|
| 39 |
+
row = json.loads(result.read_text())
|
| 40 |
+
assert row['checkpoint_step'] == 17167 and row['seed'] == seed
|
| 41 |
+
assert row['reference_dtype'] == 'float64' and not row['tf32_allowed']
|
| 42 |
+
assert row['protocol']['n_levels'] == 16 and row['protocol']['n_provided_levels'] == 1
|
| 43 |
+
assert row['requested_samples'] == row['scored_samples'] == 128
|
| 44 |
+
assert row['scored_tokens'] > 0
|
| 45 |
+
assert math.isfinite(row['mean_ppl']) and math.isfinite(row['mean_entropy_nats'])
|
| 46 |
+
status('seed-complete', seed=seed, mean_ppl=row['mean_ppl'], entropy=row['mean_entropy_nats'])
|
| 47 |
+
with (OUT/'summarize.log').open('w') as log:
|
| 48 |
+
subprocess.run([sys.executable, 'evaluation/summarize_gen_ppl.py', '--input-glob',
|
| 49 |
+
str(OUT/'random-seed-[0-4].json'), '--out', str(OUT/'summary.json')],
|
| 50 |
+
cwd=REPO, env=ENV, stdout=log, stderr=subprocess.STDOUT, check=True)
|
| 51 |
+
status('complete', summary=str(OUT/'summary.json'))
|
| 52 |
+
except BaseException as exc:
|
| 53 |
+
status('failed', error=str(exc))
|
| 54 |
+
raise
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/evaluator.py
ADDED
|
@@ -0,0 +1,187 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""Evaluate random-sampling generation PPL from a completed NCP checkpoint."""
|
| 2 |
+
|
| 3 |
+
import argparse
|
| 4 |
+
import json
|
| 5 |
+
import math
|
| 6 |
+
import sys
|
| 7 |
+
from collections import Counter
|
| 8 |
+
from pathlib import Path
|
| 9 |
+
|
| 10 |
+
REPO_ROOT = Path("/home/ubuntu/mstok-runs/iclr-debug-joint-20260916")
|
| 11 |
+
if str(REPO_ROOT) not in sys.path:
|
| 12 |
+
sys.path.insert(0, str(REPO_ROOT))
|
| 13 |
+
|
| 14 |
+
import hydra
|
| 15 |
+
import torch
|
| 16 |
+
from omegaconf import OmegaConf
|
| 17 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 18 |
+
|
| 19 |
+
from data.serialized_dataset import SerializedDataset
|
| 20 |
+
from evaluation.gen_ppl import generation_perplexity
|
| 21 |
+
from utils.misc import set_manual_seed
|
| 22 |
+
from utils.registry import trainers
|
| 23 |
+
|
| 24 |
+
|
| 25 |
+
def main():
|
| 26 |
+
parser = argparse.ArgumentParser()
|
| 27 |
+
parser.add_argument("--ckpt", required=True)
|
| 28 |
+
parser.add_argument("--codec")
|
| 29 |
+
parser.add_argument("--validation-source")
|
| 30 |
+
parser.add_argument("--ref-model", default="gpt2-large")
|
| 31 |
+
parser.add_argument("--ref-dtype", choices=("bfloat16", "float32", "float64"), default="bfloat16",
|
| 32 |
+
help="Legacy default retained; new ICLR runs explicitly use float32")
|
| 33 |
+
parser.add_argument("--seed", type=int, default=0)
|
| 34 |
+
parser.add_argument("--n-samples", type=int, default=128)
|
| 35 |
+
parser.add_argument("--batch-size", type=int, default=8)
|
| 36 |
+
parser.add_argument("--temperature", type=float, default=1.0)
|
| 37 |
+
parser.add_argument("--top-k", type=int, default=0)
|
| 38 |
+
parser.add_argument("--top-p", type=float, default=1.0)
|
| 39 |
+
parser.add_argument("--out", required=True)
|
| 40 |
+
parser.add_argument("--save-all-generations", action="store_true")
|
| 41 |
+
args = parser.parse_args()
|
| 42 |
+
|
| 43 |
+
device = "cuda"
|
| 44 |
+
checkpoint = torch.load(args.ckpt, map_location="cpu", weights_only=False)
|
| 45 |
+
cfg = checkpoint["args"]
|
| 46 |
+
OmegaConf.set_struct(cfg, False)
|
| 47 |
+
if args.codec:
|
| 48 |
+
cfg.tokenizer.vqvae.checkpoint_path = args.codec
|
| 49 |
+
if args.validation_source:
|
| 50 |
+
cfg.dataset.validate_source = args.validation_source
|
| 51 |
+
cfg.optimization.token_corruption_prob = 0.0
|
| 52 |
+
if getattr(cfg.optimization, "stochastic_encode", None) is not None:
|
| 53 |
+
cfg.optimization.stochastic_encode.enabled = False
|
| 54 |
+
else:
|
| 55 |
+
cfg.optimization.stochastic_encode = {"enabled": False}
|
| 56 |
+
|
| 57 |
+
set_manual_seed(args.seed)
|
| 58 |
+
# Seed setup enables TF32 by default, so enforce evaluation precision after it.
|
| 59 |
+
if args.ref_dtype in ("float32", "float64"):
|
| 60 |
+
torch.backends.cuda.matmul.allow_tf32 = False
|
| 61 |
+
torch.backends.cudnn.allow_tf32 = False
|
| 62 |
+
Trainer = hydra.utils.get_class(trainers[cfg.task])
|
| 63 |
+
trainer = Trainer(cfg, device)
|
| 64 |
+
trainer.model.load_state_dict(checkpoint["model"])
|
| 65 |
+
model = trainer.model.to(device).eval()
|
| 66 |
+
tokenizer = trainer.tokenizer
|
| 67 |
+
levels = list(model.levels)
|
| 68 |
+
prefix_len = int(cfg.model.prefix_len)
|
| 69 |
+
|
| 70 |
+
dataset = SerializedDataset(
|
| 71 |
+
cfg.dataset.validate_source,
|
| 72 |
+
int(cfg.model.context_length),
|
| 73 |
+
cfg.task,
|
| 74 |
+
tokenizer.pad_token_id,
|
| 75 |
+
)
|
| 76 |
+
full_sequences, _ = dataset.get_rand_batch(args.n_samples)
|
| 77 |
+
prefix_sequences = full_sequences[:, :prefix_len]
|
| 78 |
+
target_sequences = full_sequences[:, prefix_len:]
|
| 79 |
+
|
| 80 |
+
prompts = []
|
| 81 |
+
generations = []
|
| 82 |
+
with torch.no_grad():
|
| 83 |
+
for start in range(0, args.n_samples, args.batch_size):
|
| 84 |
+
end = min(start + args.batch_size, args.n_samples)
|
| 85 |
+
prefix = prefix_sequences[start:end].to(device)
|
| 86 |
+
target_idx = tokenizer.encode_pre_tokenized_idx(
|
| 87 |
+
target_sequences[start:end].to(device)
|
| 88 |
+
)
|
| 89 |
+
level0 = target_idx[:, : levels[0]]
|
| 90 |
+
generated_idx = model.generate(
|
| 91 |
+
level0,
|
| 92 |
+
prefix,
|
| 93 |
+
temperature=args.temperature,
|
| 94 |
+
top_k=args.top_k,
|
| 95 |
+
top_p=args.top_p,
|
| 96 |
+
)
|
| 97 |
+
reconstructions = tokenizer.decode_multiscales(
|
| 98 |
+
generated_idx, skip_special_tokens=True
|
| 99 |
+
)
|
| 100 |
+
generations.extend(
|
| 101 |
+
reconstructions[row][-1] for row in range(len(reconstructions))
|
| 102 |
+
)
|
| 103 |
+
prompts.extend(
|
| 104 |
+
tokenizer.pre_tokenizer.decode_batch(
|
| 105 |
+
prefix.cpu().tolist(), skip_special_tokens=True
|
| 106 |
+
)
|
| 107 |
+
)
|
| 108 |
+
|
| 109 |
+
if args.save_all_generations:
|
| 110 |
+
# Preserve every generated sample even if scoring fails or some are empty.
|
| 111 |
+
samples_path = Path(args.out).with_suffix(".samples.json")
|
| 112 |
+
samples_path.parent.mkdir(parents=True, exist_ok=True)
|
| 113 |
+
samples_path.write_text(json.dumps(dict(seed=args.seed, checkpoint=args.ckpt,
|
| 114 |
+
prompts=prompts, generations=generations), indent=2) + "\n")
|
| 115 |
+
kept = [(prompt, text) for prompt, text in zip(prompts, generations) if text.strip()]
|
| 116 |
+
prompts = [prompt for prompt, _ in kept]
|
| 117 |
+
generations = [text for _, text in kept]
|
| 118 |
+
|
| 119 |
+
entropies = []
|
| 120 |
+
for text in generations:
|
| 121 |
+
ids = tokenizer.pre_tokenizer.encode(text)
|
| 122 |
+
if ids:
|
| 123 |
+
counts = Counter(ids)
|
| 124 |
+
entropies.append(
|
| 125 |
+
-sum((count / len(ids)) * math.log(count / len(ids)) for count in counts.values())
|
| 126 |
+
)
|
| 127 |
+
|
| 128 |
+
del model, trainer, tokenizer
|
| 129 |
+
torch.cuda.empty_cache()
|
| 130 |
+
|
| 131 |
+
ref_tokenizer = AutoTokenizer.from_pretrained(args.ref_model)
|
| 132 |
+
ref_model = AutoModelForCausalLM.from_pretrained(
|
| 133 |
+
args.ref_model, torch_dtype=getattr(torch, args.ref_dtype),
|
| 134 |
+
**({"attn_implementation": "eager"} if args.ref_dtype == "float64" else {})
|
| 135 |
+
).to(device).eval()
|
| 136 |
+
assert all(p.dtype == getattr(torch, args.ref_dtype) for p in ref_model.parameters())
|
| 137 |
+
result = generation_perplexity(
|
| 138 |
+
ref_model,
|
| 139 |
+
ref_tokenizer,
|
| 140 |
+
prompts,
|
| 141 |
+
generations,
|
| 142 |
+
score_only_generated=True,
|
| 143 |
+
batch_size=args.batch_size,
|
| 144 |
+
device=device,
|
| 145 |
+
)
|
| 146 |
+
|
| 147 |
+
assert result.nll_sum.dtype == getattr(torch, args.ref_dtype)
|
| 148 |
+
assert bool(torch.isfinite(result.ppl).all()) and bool((result.token_count > 0).all())
|
| 149 |
+
payload = {
|
| 150 |
+
"checkpoint": str(Path(args.ckpt).resolve()),
|
| 151 |
+
"checkpoint_step": int(checkpoint["step"]),
|
| 152 |
+
"codec": str(Path(cfg.tokenizer.vqvae.checkpoint_path).resolve()),
|
| 153 |
+
"reference_model": args.ref_model,
|
| 154 |
+
"reference_dtype": args.ref_dtype,
|
| 155 |
+
"reference_attention_implementation": ref_model.config._attn_implementation,
|
| 156 |
+
"per_sample_nll_sum": result.nll_sum.tolist(),
|
| 157 |
+
"per_sample_token_count": result.token_count.tolist(),
|
| 158 |
+
"per_sample_ppl": result.ppl.tolist(),
|
| 159 |
+
"per_sample_entropy_nats": entropies,
|
| 160 |
+
"tf32_allowed": torch.backends.cuda.matmul.allow_tf32,
|
| 161 |
+
"protocol": {
|
| 162 |
+
"sampling": "random" if args.top_k == 0 and args.top_p == 1.0 else "truncated",
|
| 163 |
+
"temperature": args.temperature,
|
| 164 |
+
"top_k": args.top_k,
|
| 165 |
+
"top_p": args.top_p,
|
| 166 |
+
"n_provided_levels": 1,
|
| 167 |
+
"n_levels": len(levels),
|
| 168 |
+
"document_aware_validation_sampling": dataset.doc_offsets is not None,
|
| 169 |
+
},
|
| 170 |
+
"seed": args.seed,
|
| 171 |
+
"requested_samples": args.n_samples,
|
| 172 |
+
"scored_samples": len(generations),
|
| 173 |
+
"skipped_empty_samples": args.n_samples - len(generations),
|
| 174 |
+
"scored_tokens": int(result.token_count.sum()),
|
| 175 |
+
"mean_ppl": float(result.mean_ppl),
|
| 176 |
+
"median_ppl": float(result.ppl.median()),
|
| 177 |
+
"mean_entropy_nats": sum(entropies) / len(entropies),
|
| 178 |
+
"sample_generations": generations if args.save_all_generations else generations[:8],
|
| 179 |
+
}
|
| 180 |
+
output_path = Path(args.out)
|
| 181 |
+
output_path.parent.mkdir(parents=True, exist_ok=True)
|
| 182 |
+
output_path.write_text(json.dumps(payload, indent=2) + "\n")
|
| 183 |
+
print(json.dumps({key: value for key, value in payload.items() if key != "sample_generations"}, indent=2))
|
| 184 |
+
|
| 185 |
+
|
| 186 |
+
if __name__ == "__main__":
|
| 187 |
+
main()
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/manifest.json
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"checkpoint_step": 17167,
|
| 3 |
+
"input_positions": 13500678144,
|
| 4 |
+
"source_checkpoint": "/home/ubuntu/mstok-results/iclr-debug-decoder-135b-20260916/decoder-mstok/milestone-iter-17167.pt",
|
| 5 |
+
"checkpoint_sha256": "0f71f24010b3fe8874a2319801008f49c00badb9af050c43b8aa8ed80122e466",
|
| 6 |
+
"evaluator_sha256": "e2c87f66f1001853f58b9e78a646fc4e089fc894d05fa6c215a8baf493a84055",
|
| 7 |
+
"checkout": "/home/ubuntu/mstok-runs/iclr-debug-joint-20260916",
|
| 8 |
+
"reference_dtype": "float64",
|
| 9 |
+
"reference_attention": "eager",
|
| 10 |
+
"tf32_allowed": false,
|
| 11 |
+
"memory_fraction": 0.15,
|
| 12 |
+
"gpu": 0,
|
| 13 |
+
"batch_size": 4,
|
| 14 |
+
"seeds": [
|
| 15 |
+
0,
|
| 16 |
+
1,
|
| 17 |
+
2,
|
| 18 |
+
3,
|
| 19 |
+
4
|
| 20 |
+
],
|
| 21 |
+
"samples_per_seed": 128,
|
| 22 |
+
"sampling": "temperature-1-untruncated-supplied-true-level0",
|
| 23 |
+
"concurrent_with_training": true,
|
| 24 |
+
"retained_checkpoint": "/home/ubuntu/mstok-results/iclr-debug-decoder-135b-20260916/decoder-mstok/milestone-iter-17167.pt",
|
| 25 |
+
"created_utc": "2026-09-19T04:42:31.183086+00:00"
|
| 26 |
+
}
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/manual-eval-status.json
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"state": "complete",
|
| 3 |
+
"updated_utc": "2026-09-19T04:47:12.120382+00:00",
|
| 4 |
+
"controller_pid": 590459,
|
| 5 |
+
"summary": "/home/ubuntu/mstok-results/iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/summary.json"
|
| 6 |
+
}
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/quality-summary.json
ADDED
|
@@ -0,0 +1,87 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"checkpoint_step": 17167,
|
| 3 |
+
"reference_dtype": "float64",
|
| 4 |
+
"tf32_allowed": false,
|
| 5 |
+
"unfiltered": {
|
| 6 |
+
"samples": 640,
|
| 7 |
+
"scored_tokens": 160949,
|
| 8 |
+
"pooled_token_weighted_ppl": 109.15588482181249,
|
| 9 |
+
"mean_entropy_nats": 4.363820267455618,
|
| 10 |
+
"fraction_samples_with_replacement_character": 0.665625,
|
| 11 |
+
"adjacent_repeated_word_fraction": 0.048007480092300596
|
| 12 |
+
},
|
| 13 |
+
"entropy_ge_4p5": {
|
| 14 |
+
"samples": 109,
|
| 15 |
+
"scored_tokens": 27286,
|
| 16 |
+
"pooled_token_weighted_ppl": 201.7171620400652,
|
| 17 |
+
"mean_entropy_nats": 4.582393683871207,
|
| 18 |
+
"fraction_samples_with_replacement_character": 0.5504587155963303,
|
| 19 |
+
"adjacent_repeated_word_fraction": 0.043738225687130025
|
| 20 |
+
},
|
| 21 |
+
"acceptance_rate": 0.1703125,
|
| 22 |
+
"comparison_step17167": {
|
| 23 |
+
"checkpoint_step": 17167,
|
| 24 |
+
"reference_dtype": "float64",
|
| 25 |
+
"threshold": 4.5,
|
| 26 |
+
"entropy_definition": "Empirical GPT-2 unigram entropy in nats within each generated sample",
|
| 27 |
+
"selection": "Filter existing 640 generations; no new generation; all accepted samples pooled",
|
| 28 |
+
"accepted": {
|
| 29 |
+
"samples": 115,
|
| 30 |
+
"scored_tokens": 28860,
|
| 31 |
+
"pooled_token_weighted_ppl": 189.63357736496337,
|
| 32 |
+
"mean_entropy_nats": 4.57744085314838,
|
| 33 |
+
"median_sample_ppl": 171.1510152167491,
|
| 34 |
+
"adjacent_repeated_word_fraction": 0.04295705405426942,
|
| 35 |
+
"samples_with_replacement_character": 63,
|
| 36 |
+
"fraction_samples_with_replacement_character": 0.5478260869565217
|
| 37 |
+
},
|
| 38 |
+
"unfiltered": {
|
| 39 |
+
"samples": 640,
|
| 40 |
+
"scored_tokens": 160898,
|
| 41 |
+
"pooled_token_weighted_ppl": 109.72590001293689,
|
| 42 |
+
"mean_entropy_nats": 4.3665410349745,
|
| 43 |
+
"median_sample_ppl": 99.25832710923365,
|
| 44 |
+
"adjacent_repeated_word_fraction": 0.04761292563004243,
|
| 45 |
+
"samples_with_replacement_character": 431,
|
| 46 |
+
"fraction_samples_with_replacement_character": 0.6734375
|
| 47 |
+
},
|
| 48 |
+
"acceptance_rate": 0.1796875,
|
| 49 |
+
"per_seed": [
|
| 50 |
+
{
|
| 51 |
+
"seed": 0,
|
| 52 |
+
"accepted": 26,
|
| 53 |
+
"proposed": 128,
|
| 54 |
+
"ppl": 198.79796628571316
|
| 55 |
+
},
|
| 56 |
+
{
|
| 57 |
+
"seed": 1,
|
| 58 |
+
"accepted": 21,
|
| 59 |
+
"proposed": 128,
|
| 60 |
+
"ppl": 171.26448994329286
|
| 61 |
+
},
|
| 62 |
+
{
|
| 63 |
+
"seed": 2,
|
| 64 |
+
"accepted": 25,
|
| 65 |
+
"proposed": 128,
|
| 66 |
+
"ppl": 201.94449304567118
|
| 67 |
+
},
|
| 68 |
+
{
|
| 69 |
+
"seed": 3,
|
| 70 |
+
"accepted": 21,
|
| 71 |
+
"proposed": 128,
|
| 72 |
+
"ppl": 201.29939871172186
|
| 73 |
+
},
|
| 74 |
+
{
|
| 75 |
+
"seed": 4,
|
| 76 |
+
"accepted": 22,
|
| 77 |
+
"proposed": 128,
|
| 78 |
+
"ppl": 174.01773580284262
|
| 79 |
+
}
|
| 80 |
+
],
|
| 81 |
+
"mean_of_seed_ppls": 189.46481675784833,
|
| 82 |
+
"ppl_definition": "exp(sum saved FP64 reference NLL / sum scored tokens)",
|
| 83 |
+
"conditioning": "supplied true level-zero code",
|
| 84 |
+
"fp64_source": "/home/ubuntu/mstok-results/iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64",
|
| 85 |
+
"entropy_source": "/home/ubuntu/mstok-results/iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-diversity-audit/step_17167-samples.json"
|
| 86 |
+
}
|
| 87 |
+
}
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-0.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-0.samples.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-1.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-1.samples.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-2.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-2.samples.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-3.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-3.samples.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-4.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-4.samples.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/report.md
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Step 17,167 generation evaluation
|
| 2 |
+
|
| 3 |
+
FP64 GPT-2 Large generation PPL: 109.261 +/- 2.332 standard error across five seeds.
|
| 4 |
+
|
| 5 |
+
640 generations, temperature 1 without truncation, supplied true level-zero code. TF32 disabled.
|
| 6 |
+
|
| 7 |
+
## Quality measures
|
| 8 |
+
|
| 9 |
+
| Metric | All samples | Entropy >= 4.5 |
|
| 10 |
+
|---|---:|---:|
|
| 11 |
+
| Samples | 640.000 | 109.000 |
|
| 12 |
+
| Pooled token-weighted PPL | 109.156 | 201.717 |
|
| 13 |
+
| Mean entropy (nats) | 4.364 | 4.582 |
|
| 14 |
+
| Samples with replacement character (%) | 66.562 | 55.046 |
|
| 15 |
+
| Adjacent repeated words (%) | 4.801 | 4.374 |
|
| 16 |
+
|
| 17 |
+
## First three seed-0 samples
|
| 18 |
+
|
| 19 |
+
Samples shown in original order, without selecting for quality.
|
| 20 |
+
|
| 21 |
+
### Seed 0, index 0 — PPL 522.05, entropy 4.688
|
| 22 |
+
|
| 23 |
+
ismkel�mēs�n��ɰmmm.
|
| 24 |
+
|
| 25 |
+
|
| 26 |
+
4 2�ěńm��ɛoz�asas)
|
| 27 |
+
|
| 28 |
+
::54AAAd-ddlesy�a -
|
| 29 |
+
|
| 30 |
+
|
| 31 |
+
C 8.0 �d Sndäanonu v l-azem���m�?)wam �ere
|
| 32 |
+
exdó n���d p.Ì yllɹ (a) pr)The
|
| 33 |
+
2 8 15th extreme 16,�g d fʍ� nadiom dnnt hihd in�r�,abhere�n� (a - t) sim
|
| 34 |
+
n n lt. In such, you - - have the changellt't' in a site site least one ofon.a. The n' aren do questionst for e..�.m's's is v are early late in the last Online 9 That
|
| 35 |
+
-575 18 5 5.guma a obj bphz �m�zím.
|
| 36 |
+
|
| 37 |
+
Recent visiting history [ edit ]
|
| 38 |
+
|
| 39 |
+
In September 8 July 2010, aond give Zie was performing at The New the in in Total
|
| 40 |
+
|
| 41 |
+
### Seed 0, index 1 — PPL 68.08, entropy 4.295
|
| 42 |
+
|
| 43 |
+
want want about you. That’ss going to be it. But. I,�ll get then, I’ve got it it.”
|
| 44 |
+
|
| 45 |
+
On the last few days he got well-known through. secondless — then and says.�He got other times this season. It’s only one thing the big guys where’s there reason being having a lot of those guy who serves on on needed on the issue as a guy, and and guy’s a guy he’s a guy. He’� not a lot of money. There� ofs not man and there’ss going to take a lot of running over,, while I said, ‘I can opt if somebody’s. That’s not anybody, for and he’s one of doing things. But I think that it’s not going to do you, and he’s really me or a very important thing. He’s been playing every single now now that we got to say we get a person whole whole right. They’ve already been done it little thing fight, if it’s not going to do what play on the game. That
|
| 46 |
+
|
| 47 |
+
### Seed 0, index 2 — PPL 131.68, entropy 4.325
|
| 48 |
+
|
| 49 |
+
mean, it was different; physically on the interview with my own. When fans, I don't call scene tell you the same thing makes you And. And all what it's looks for me. It doesn't mean well, I don just just it, you do me. There's a of of time for moment to come again that I did like that that. I mean you, it's always as the me,No.?''s's like, "I'll'll it it out out?" then I'm like, 'Oh", do do 'Hey, I embarrassed you?'?".'ll like, I be like, 'Oh really, I'll do, that could if I was looking." So I thought, again in something that fell look out — I've pulled like,, you'll have to myself [-].]. I said, what I just like that because I " God, I'm to say that's back. You'll to look them what what. It It never been a tough, and that's's old old. Because old guy someone someone would you to come out just because you're just kid you say, I say.' That's not not, it's that they got to get that running. Every person would a any someone who
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/summarize-quality.py
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import json
|
| 2 |
+
import math
|
| 3 |
+
from pathlib import Path
|
| 4 |
+
import re
|
| 5 |
+
import statistics
|
| 6 |
+
|
| 7 |
+
OUT = Path(__file__).resolve().parent
|
| 8 |
+
rows=[]
|
| 9 |
+
for seed in range(5):
|
| 10 |
+
result=json.loads((OUT/f'random-seed-{seed}.json').read_text())
|
| 11 |
+
assert result['reference_dtype']=='float64' and not result['tf32_allowed']
|
| 12 |
+
assert result['checkpoint_step']==17167 and result['scored_samples']==128
|
| 13 |
+
for i,text in enumerate(result['sample_generations']):
|
| 14 |
+
words=re.findall(r'\b\w+\b',text.lower())
|
| 15 |
+
row=dict(seed=seed,index=i,text=text,entropy_nats=result['per_sample_entropy_nats'][i],
|
| 16 |
+
nll_sum=result['per_sample_nll_sum'][i],scored_tokens=result['per_sample_token_count'][i],
|
| 17 |
+
ppl=result['per_sample_ppl'][i],has_replacement_character=int('\ufffd' in text),
|
| 18 |
+
adjacent_repeated_word_fraction=sum(a==b for a,b in zip(words,words[1:]))/max(1,len(words)-1))
|
| 19 |
+
assert math.isfinite(row['ppl']) and row['scored_tokens']>0
|
| 20 |
+
rows.append(row)
|
| 21 |
+
|
| 22 |
+
def metrics(group):
|
| 23 |
+
if not group:
|
| 24 |
+
return dict(samples=0)
|
| 25 |
+
return dict(samples=len(group),scored_tokens=sum(r['scored_tokens'] for r in group),
|
| 26 |
+
pooled_token_weighted_ppl=math.exp(sum(r['nll_sum'] for r in group)/sum(r['scored_tokens'] for r in group)),
|
| 27 |
+
mean_entropy_nats=statistics.mean(r['entropy_nats'] for r in group),
|
| 28 |
+
fraction_samples_with_replacement_character=statistics.mean(r['has_replacement_character'] for r in group),
|
| 29 |
+
adjacent_repeated_word_fraction=statistics.mean(r['adjacent_repeated_word_fraction'] for r in group))
|
| 30 |
+
|
| 31 |
+
accepted=[r for r in rows if r['entropy_nats']>=4.5]
|
| 32 |
+
quality=dict(checkpoint_step=17167,reference_dtype='float64',tf32_allowed=False,
|
| 33 |
+
unfiltered=metrics(rows),entropy_ge_4p5=metrics(accepted),acceptance_rate=len(accepted)/len(rows),
|
| 34 |
+
comparison_step17167=json.loads((OUT.parent/'step-17167-entropy-ge-4p5/summary.json').read_text()))
|
| 35 |
+
(OUT/'quality-summary.json').write_text(json.dumps(quality,indent=2)+'\n')
|
| 36 |
+
(OUT/'accepted-entropy-ge-4p5.json').write_text(json.dumps(accepted,indent=2)+'\n')
|
| 37 |
+
summary=json.loads((OUT/'summary.json').read_text())
|
| 38 |
+
summary.update(reference_dtype='float64',tf32_allowed=False,quality_summary=str(OUT/'quality-summary.json'),
|
| 39 |
+
generation_tf32_allowed=False,total_requested_samples=640,total_skipped_empty_samples=0)
|
| 40 |
+
(OUT/'summary.json').write_text(json.dumps(summary,indent=2)+'\n')
|
| 41 |
+
lines=['# Step 17,167 generation evaluation','',
|
| 42 |
+
f"FP64 GPT-2 Large generation PPL: {summary['mean_gen_ppl']:.3f} +/- {summary['gen_ppl_se']:.3f} standard error across five seeds.",
|
| 43 |
+
'', '640 generations, temperature 1 without truncation, supplied true level-zero code. TF32 disabled.',
|
| 44 |
+
'', '## Quality measures','', '| Metric | All samples | Entropy >= 4.5 |','|---|---:|---:|']
|
| 45 |
+
for label,key,scale in [('Samples','samples',1),('Pooled token-weighted PPL','pooled_token_weighted_ppl',1),('Mean entropy (nats)','mean_entropy_nats',1),('Samples with replacement character (%)','fraction_samples_with_replacement_character',100),('Adjacent repeated words (%)','adjacent_repeated_word_fraction',100)]:
|
| 46 |
+
lines.append(f"| {label} | {quality['unfiltered'][key]*scale:.3f} | {quality['entropy_ge_4p5'].get(key,float('nan'))*scale:.3f} |")
|
| 47 |
+
lines+=['','## First three seed-0 samples','', 'Samples shown in original order, without selecting for quality.']
|
| 48 |
+
for row in rows[:3]:
|
| 49 |
+
lines+=['',f"### Seed {row['seed']}, index {row['index']} — PPL {row['ppl']:.2f}, entropy {row['entropy_nats']:.3f}",'',row['text']]
|
| 50 |
+
(OUT/'report.md').write_text('\n'.join(lines)+'\n')
|
| 51 |
+
print(json.dumps(summary,indent=2))
|
| 52 |
+
print(json.dumps({k:v for k,v in quality.items() if k!='comparison_step17167'},indent=2))
|
iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/summary.json
ADDED
|
@@ -0,0 +1,41 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"protocol": {
|
| 3 |
+
"sampling": "random",
|
| 4 |
+
"temperature": 1.0,
|
| 5 |
+
"top_k": 0,
|
| 6 |
+
"top_p": 1.0,
|
| 7 |
+
"n_provided_levels": 1,
|
| 8 |
+
"n_levels": 16,
|
| 9 |
+
"document_aware_validation_sampling": true
|
| 10 |
+
},
|
| 11 |
+
"reference_model": "gpt2-large",
|
| 12 |
+
"checkpoint": "/home/ubuntu/mstok-results/iclr-debug-decoder-135b-20260916/decoder-mstok/exports/step-17167/ncp.pt",
|
| 13 |
+
"checkpoint_step": 17167,
|
| 14 |
+
"seeds": [
|
| 15 |
+
0,
|
| 16 |
+
1,
|
| 17 |
+
2,
|
| 18 |
+
3,
|
| 19 |
+
4
|
| 20 |
+
],
|
| 21 |
+
"samples_per_seed": 128,
|
| 22 |
+
"total_scored_samples": 640,
|
| 23 |
+
"total_scored_tokens": 160949,
|
| 24 |
+
"per_seed_mean_ppl": {
|
| 25 |
+
"0": 116.0033750052343,
|
| 26 |
+
"1": 112.64891583090473,
|
| 27 |
+
"2": 107.53348738655384,
|
| 28 |
+
"3": 107.66336444165039,
|
| 29 |
+
"4": 102.45701050430984
|
| 30 |
+
},
|
| 31 |
+
"mean_gen_ppl": 109.26123063373062,
|
| 32 |
+
"gen_ppl_se": 2.3320442361469507,
|
| 33 |
+
"mean_of_seed_median_ppl": 98.77863179069354,
|
| 34 |
+
"mean_entropy_nats": 4.363820267455619,
|
| 35 |
+
"reference_dtype": "float64",
|
| 36 |
+
"tf32_allowed": false,
|
| 37 |
+
"quality_summary": "/home/ubuntu/mstok-results/iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/quality-summary.json",
|
| 38 |
+
"generation_tf32_allowed": false,
|
| 39 |
+
"total_requested_samples": 640,
|
| 40 |
+
"total_skipped_empty_samples": 0
|
| 41 |
+
}
|
iclr-debug-decoder-135b-20260916/decoder-mstok/milestone-iter-17167.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0f71f24010b3fe8874a2319801008f49c00badb9af050c43b8aa8ed80122e466
|
| 3 |
+
size 4879455585
|
iclr-debug-decoder-135b-20260916/provenance/hf-backup-step-17167-manifest.json
ADDED
|
@@ -0,0 +1,131 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"repository": "iskhare/iclr-debug",
|
| 3 |
+
"checkpoint_step": 17167,
|
| 4 |
+
"files": [
|
| 5 |
+
{
|
| 6 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/milestone-iter-17167.pt",
|
| 7 |
+
"bytes": 4879455585,
|
| 8 |
+
"sha256": "0f71f24010b3fe8874a2319801008f49c00badb9af050c43b8aa8ed80122e466"
|
| 9 |
+
},
|
| 10 |
+
{
|
| 11 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/manifest.json",
|
| 12 |
+
"bytes": 28971,
|
| 13 |
+
"sha256": "53dc5b666685e03efc016e4311b89ab94bb6babe710405c28ae58356406582d0"
|
| 14 |
+
},
|
| 15 |
+
{
|
| 16 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/git-provenance.json",
|
| 17 |
+
"bytes": 59,
|
| 18 |
+
"sha256": "96f601db3a051fd136086ca5f54c87ce2a72449e528d826a3b882541953e3fa2"
|
| 19 |
+
},
|
| 20 |
+
{
|
| 21 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/config.yaml",
|
| 22 |
+
"bytes": 4827,
|
| 23 |
+
"sha256": "073bd5a69f1d0a39019efffd79f4b7bd645e0c1792edead7e31b524515ef134f"
|
| 24 |
+
},
|
| 25 |
+
{
|
| 26 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/teacher-provenance.json",
|
| 27 |
+
"bytes": 270,
|
| 28 |
+
"sha256": "96e0063afe4211b1109509690a697fd8a7366941988c894f1027e1ead45938cf"
|
| 29 |
+
},
|
| 30 |
+
{
|
| 31 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/summary.json",
|
| 32 |
+
"bytes": 1189,
|
| 33 |
+
"sha256": "244f84b5ba0fe32b7b3a36e669e9cdfaab0a874a649aaa1b20c87f8c2d3780a2"
|
| 34 |
+
},
|
| 35 |
+
{
|
| 36 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/quality-summary.json",
|
| 37 |
+
"bytes": 2946,
|
| 38 |
+
"sha256": "c566202ac303fe467b47d5045e2eaf1bbbf0c50865c365a1846d650d89d5b6e1"
|
| 39 |
+
},
|
| 40 |
+
{
|
| 41 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/report.md",
|
| 42 |
+
"bytes": 3199,
|
| 43 |
+
"sha256": "9fc9fc06602f9d3cb453a8ad72ace2544eac9d2f00ecdbdbe55332b60c8977e6"
|
| 44 |
+
},
|
| 45 |
+
{
|
| 46 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/manifest.json",
|
| 47 |
+
"bytes": 944,
|
| 48 |
+
"sha256": "8756daa27a93c0754a6bda4186833ba163c9f95e3686e7b2262657dc5df5f3f6"
|
| 49 |
+
},
|
| 50 |
+
{
|
| 51 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/evaluator.py",
|
| 52 |
+
"bytes": 7709,
|
| 53 |
+
"sha256": "e2c87f66f1001853f58b9e78a646fc4e089fc894d05fa6c215a8baf493a84055"
|
| 54 |
+
},
|
| 55 |
+
{
|
| 56 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/controller.py",
|
| 57 |
+
"bytes": 3124,
|
| 58 |
+
"sha256": "57161030b9c01fef411ede348a7ec869a620f6687543aa2a7b0f298a8d55ef8a"
|
| 59 |
+
},
|
| 60 |
+
{
|
| 61 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/summarize-quality.py",
|
| 62 |
+
"bytes": 3757,
|
| 63 |
+
"sha256": "ff838759520854d2ffbf2e5701172ba54b86eada2be52061e8b6141ff72fae06"
|
| 64 |
+
},
|
| 65 |
+
{
|
| 66 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/accepted-entropy-ge-4p5.json",
|
| 67 |
+
"bytes": 143849,
|
| 68 |
+
"sha256": "6aecf269ead6b8053ed7aeb54114821c4cbb0defcac4fa809dff0aa1f348dd26"
|
| 69 |
+
},
|
| 70 |
+
{
|
| 71 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/manual-eval-status.json",
|
| 72 |
+
"bytes": 252,
|
| 73 |
+
"sha256": "f90fe53b25058cda8f6867b868075db1cee4f4ffc0c7d18e325aedaf6ac81a76"
|
| 74 |
+
},
|
| 75 |
+
{
|
| 76 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-0.json",
|
| 77 |
+
"bytes": 139237,
|
| 78 |
+
"sha256": "3b86548062c195b0a19f7d50ef1a2dcc92c0a211daccaf29b2593e134336fe7a"
|
| 79 |
+
},
|
| 80 |
+
{
|
| 81 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-0.samples.json",
|
| 82 |
+
"bytes": 129318,
|
| 83 |
+
"sha256": "88b5f007bad75aef1a9b226aba45e82509560b3f6869f91717c30c4cbf345841"
|
| 84 |
+
},
|
| 85 |
+
{
|
| 86 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-1.json",
|
| 87 |
+
"bytes": 137569,
|
| 88 |
+
"sha256": "9efd2b090ed86f2d483aea18d554fdf8b60efe0a81790ebbbf86622bb1252487"
|
| 89 |
+
},
|
| 90 |
+
{
|
| 91 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-1.samples.json",
|
| 92 |
+
"bytes": 127647,
|
| 93 |
+
"sha256": "2c307023a4fda12822ed28c2dbbcd1fc5dbb87a76ac41f1c0c251ab194951a96"
|
| 94 |
+
},
|
| 95 |
+
{
|
| 96 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-2.json",
|
| 97 |
+
"bytes": 137015,
|
| 98 |
+
"sha256": "66b3615172cd41666279134cd9c5bdb55285610e506333100acc5e86b999e640"
|
| 99 |
+
},
|
| 100 |
+
{
|
| 101 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-2.samples.json",
|
| 102 |
+
"bytes": 127115,
|
| 103 |
+
"sha256": "834c87296ba5256677b8978ca659a4512d3eb4c24bf6e9da7a587ff47f971105"
|
| 104 |
+
},
|
| 105 |
+
{
|
| 106 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-3.json",
|
| 107 |
+
"bytes": 139830,
|
| 108 |
+
"sha256": "439b93bb677adc43cad18f3663837787c1b680d5bd6fda82d49b7caf82ded0a0"
|
| 109 |
+
},
|
| 110 |
+
{
|
| 111 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-3.samples.json",
|
| 112 |
+
"bytes": 129930,
|
| 113 |
+
"sha256": "6a99e26e1d8d8006aab694e543666152d0e58f5d71e81c6be6d82fd3b378838e"
|
| 114 |
+
},
|
| 115 |
+
{
|
| 116 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-4.json",
|
| 117 |
+
"bytes": 138691,
|
| 118 |
+
"sha256": "3b0ec394f10deed0685f55501c083347fa4c27782d2c72e31e71a8f69cfbb73b"
|
| 119 |
+
},
|
| 120 |
+
{
|
| 121 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/evaluation/step-17167-fp64-rerun-20260919/random-seed-4.samples.json",
|
| 122 |
+
"bytes": 128775,
|
| 123 |
+
"sha256": "3680100a4c9389c6a98fe265db53fd519d82351b2ece304d6f3ed2a91bad4b01"
|
| 124 |
+
},
|
| 125 |
+
{
|
| 126 |
+
"path_in_repo": "iclr-debug-decoder-135b-20260916/decoder-mstok/checkpoint-17167-README.md",
|
| 127 |
+
"bytes": 1427,
|
| 128 |
+
"sha256": "fd964ecc83d8dc10607faba3386c468845beed869a216d3c41acbc5304093ee7"
|
| 129 |
+
}
|
| 130 |
+
]
|
| 131 |
+
}
|