Archive completed gsm-visible-v2__16m__gsm8k__visible-calculation-context
Browse files- cases/54739e6657da5bce/gsm-visible-v2__16m__gsm8k__visible-calculation-context/ancestry.json +9 -0
- cases/54739e6657da5bce/gsm-visible-v2__16m__gsm8k__visible-calculation-context/archive-manifest.json +37 -0
- cases/54739e6657da5bce/gsm-visible-v2__16m__gsm8k__visible-calculation-context/complete.json +19 -0
- cases/54739e6657da5bce/gsm-visible-v2__16m__gsm8k__visible-calculation-context/request.json +67 -0
- cases/54739e6657da5bce/gsm-visible-v2__16m__gsm8k__visible-calculation-context/rl.pt +3 -0
- cases/54739e6657da5bce/gsm-visible-v2__16m__gsm8k__visible-calculation-context/sft.pt +3 -0
- cases/54739e6657da5bce/gsm-visible-v2__16m__gsm8k__visible-calculation-context/startup.json +16 -0
- cases/54739e6657da5bce/gsm-visible-v2__16m__gsm8k__visible-calculation-context/terminal-sft.json +7 -0
cases/54739e6657da5bce/gsm-visible-v2__16m__gsm8k__visible-calculation-context/ancestry.json
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"rl_initializer": "/nlp/scr/dhei/tiny-rl/experiments/refined-math-v1/iterations/gsm-visible-v2/runs/visible-calculation-context/16m/gsm8k/sft.pt",
|
| 3 |
+
"terminal_sft_weights": "0c5f77c6573a115f6e8d31e4681a79ebea114ef53a93a1ad80d9f64dc9bcc39d",
|
| 4 |
+
"full_sft_budget": 2097152,
|
| 5 |
+
"intermediate_checkpoint_selection": false,
|
| 6 |
+
"hint_distribution": "Independent uniform integer 1..16 per query",
|
| 7 |
+
"adaptive_calibration": false,
|
| 8 |
+
"samples_per_query": 32
|
| 9 |
+
}
|
cases/54739e6657da5bce/gsm-visible-v2__16m__gsm8k__visible-calculation-context/archive-manifest.json
ADDED
|
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"protocol": "54739e6657da5bce4b9b06dc3731bb846e214e551bcb1d8fcd5ffe098dfa9f89",
|
| 3 |
+
"case": {
|
| 4 |
+
"key": "gsm-visible-v2__16m__gsm8k__visible-calculation-context",
|
| 5 |
+
"folder": "/nlp/scr/dhei/tiny-rl/experiments/refined-math-v1/iterations/gsm-visible-v2/runs/visible-calculation-context/16m/gsm8k",
|
| 6 |
+
"reuse": false
|
| 7 |
+
},
|
| 8 |
+
"files": {
|
| 9 |
+
"sft.pt": {
|
| 10 |
+
"size": 63429146,
|
| 11 |
+
"sha256": "16257d32c83cf1c81eab1472150817518d716877adc4424e4bbbecf2c9961c99"
|
| 12 |
+
},
|
| 13 |
+
"rl.pt": {
|
| 14 |
+
"size": 63429005,
|
| 15 |
+
"sha256": "a392e226f7b50da2c453d3e39807b3c997908a94bb8578285b7cbb1e2dc5069b"
|
| 16 |
+
}
|
| 17 |
+
},
|
| 18 |
+
"complete": {
|
| 19 |
+
"task": "gsm8k",
|
| 20 |
+
"model": "16m",
|
| 21 |
+
"sft_tokens": 2097152,
|
| 22 |
+
"sft_budget": 2097152,
|
| 23 |
+
"rl_rollouts": 65536,
|
| 24 |
+
"rl_budget": 65536,
|
| 25 |
+
"terminal_sft_hash": "0c5f77c6573a115f6e8d31e4681a79ebea114ef53a93a1ad80d9f64dc9bcc39d",
|
| 26 |
+
"final_rl_hash": "2ea507509043f1a11b0f6c576f64f543a3bd72a8e53423ca088b6babe7ffa15d",
|
| 27 |
+
"sft_seconds": 355.84108114242554,
|
| 28 |
+
"rl_seconds": 826.7976912520826,
|
| 29 |
+
"no_checkpoint_selection": true,
|
| 30 |
+
"evaluation_generations": 16128,
|
| 31 |
+
"calibration_generations": 0,
|
| 32 |
+
"hint_distribution": "uniform 1..16 per query",
|
| 33 |
+
"samples_per_query": 32,
|
| 34 |
+
"development_episodes": 96,
|
| 35 |
+
"development_unique_groups": 96
|
| 36 |
+
}
|
| 37 |
+
}
|
cases/54739e6657da5bce/gsm-visible-v2__16m__gsm8k__visible-calculation-context/complete.json
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"task": "gsm8k",
|
| 3 |
+
"model": "16m",
|
| 4 |
+
"sft_tokens": 2097152,
|
| 5 |
+
"sft_budget": 2097152,
|
| 6 |
+
"rl_rollouts": 65536,
|
| 7 |
+
"rl_budget": 65536,
|
| 8 |
+
"terminal_sft_hash": "0c5f77c6573a115f6e8d31e4681a79ebea114ef53a93a1ad80d9f64dc9bcc39d",
|
| 9 |
+
"final_rl_hash": "2ea507509043f1a11b0f6c576f64f543a3bd72a8e53423ca088b6babe7ffa15d",
|
| 10 |
+
"sft_seconds": 355.84108114242554,
|
| 11 |
+
"rl_seconds": 826.7976912520826,
|
| 12 |
+
"no_checkpoint_selection": true,
|
| 13 |
+
"evaluation_generations": 16128,
|
| 14 |
+
"calibration_generations": 0,
|
| 15 |
+
"hint_distribution": "uniform 1..16 per query",
|
| 16 |
+
"samples_per_query": 32,
|
| 17 |
+
"development_episodes": 96,
|
| 18 |
+
"development_unique_groups": 96
|
| 19 |
+
}
|
cases/54739e6657da5bce/gsm-visible-v2__16m__gsm8k__visible-calculation-context/request.json
ADDED
|
@@ -0,0 +1,67 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"config": {
|
| 3 |
+
"task": "gsm8k",
|
| 4 |
+
"model": "16m",
|
| 5 |
+
"root": "/nlp/scr/dhei/tiny-rl/experiments/refined-math-v1/iterations/gsm-visible-v2",
|
| 6 |
+
"checkpoint": "/nlp/scr/dhei/nanogpt-ladder-v3/runs/base/16m-475529216/seed-1337/model.pt",
|
| 7 |
+
"output": "/nlp/scr/dhei/tiny-rl/experiments/refined-math-v1/iterations/gsm-visible-v2/runs/visible-calculation-context/16m/gsm8k",
|
| 8 |
+
"sft_tokens": 2097152,
|
| 9 |
+
"rl_rollouts": 65536,
|
| 10 |
+
"samples": 32,
|
| 11 |
+
"groups": 4,
|
| 12 |
+
"batch_tokens": 512,
|
| 13 |
+
"micro_tokens": 2048,
|
| 14 |
+
"preserve_sft_context": false,
|
| 15 |
+
"rl_micro_tokens": 8192,
|
| 16 |
+
"rl_micro_batch": 16,
|
| 17 |
+
"sft_lr": 0.0003,
|
| 18 |
+
"sft_hidden_lr": 0.0005,
|
| 19 |
+
"rl_lr": 1e-05,
|
| 20 |
+
"rl_hidden_lr": 1.6666666666666667e-05,
|
| 21 |
+
"conditioning": "hinted-v2",
|
| 22 |
+
"optimizer": "relative",
|
| 23 |
+
"qk_rms": 0.1,
|
| 24 |
+
"qk_update_scale": 1.0,
|
| 25 |
+
"kl": 0.1,
|
| 26 |
+
"clip": 0.2,
|
| 27 |
+
"eval_queries": 96,
|
| 28 |
+
"eval_samples": 8,
|
| 29 |
+
"eval_batch": 2,
|
| 30 |
+
"sft_eval_interval": 0,
|
| 31 |
+
"interval": 4096,
|
| 32 |
+
"recovery_interval": 4096,
|
| 33 |
+
"recovery_seconds": 120.0,
|
| 34 |
+
"workers": 8,
|
| 35 |
+
"judged_backend": "skywork",
|
| 36 |
+
"judge_cache": "",
|
| 37 |
+
"dpo_reward": "binary",
|
| 38 |
+
"chat_stop": "none",
|
| 39 |
+
"threads": 2,
|
| 40 |
+
"seed": 2718,
|
| 41 |
+
"device": "auto",
|
| 42 |
+
"wandb_project": "tiny-rl-refined-math-v2",
|
| 43 |
+
"wandb_entity": "david-heineman",
|
| 44 |
+
"wandb_mode": "online",
|
| 45 |
+
"run_tag": "visible-calculation-context",
|
| 46 |
+
"smoke": false,
|
| 47 |
+
"evaluate_test": true,
|
| 48 |
+
"answer_weight": 32.0,
|
| 49 |
+
"preserve_context": true,
|
| 50 |
+
"question_weight": 1.0,
|
| 51 |
+
"preserve_pretrained_attention": false,
|
| 52 |
+
"rl_small_final_fraction": 0.0,
|
| 53 |
+
"stratified_evaluation": true
|
| 54 |
+
},
|
| 55 |
+
"checkpoint_sha256": "657993bbfa3d4cb1a3bfb5c6c62ac9d28d9ba968d2d72d64e41e45f5a60cb21f",
|
| 56 |
+
"sft_sha256": "ffbaa1cf024d90c13a516598a88db5ca6baf800fc3960af9ae28875b94530aa8",
|
| 57 |
+
"trainer_sha256": "846a6369ab5739a733b25d5222903b1203c008f4f06898b6ac1f4eb4e96d68cf",
|
| 58 |
+
"policy_sha256": "9a971eaefca9c12215bfbef71b9c2be843bf88fd282366de36eccf8556c6869e",
|
| 59 |
+
"metrics_sha256": "c7a912ec690a48e41f83a230c53fb280c0e4fcf9794c873b3f7eca3375f0198c",
|
| 60 |
+
"data_sha256": "2e5c286e09f78f32066940d20eceeff8b075037a33a8e880db491e0ded8ec6f7",
|
| 61 |
+
"split_sha256": {
|
| 62 |
+
"rl": "46b8fe0b4e8abb8b85701493d09fb3ed01c2b8ba0d06e3fb41a1e8b484e27f09",
|
| 63 |
+
"dev": "d28052e23b89ed568bfa21a9428e84a395cf2e8f80108e279e7429a8646c31df",
|
| 64 |
+
"test": "230a1c63808396686041d04f6e8228b1e36c6dcab6f5df3ac505a61d22ed4f0c"
|
| 65 |
+
},
|
| 66 |
+
"environment_sha256": "3a4fac01313a14b3ca9d0cb19bd014c14d23042c028cf6b20a4c3119d351d48f"
|
| 67 |
+
}
|
cases/54739e6657da5bce/gsm-visible-v2__16m__gsm8k__visible-calculation-context/rl.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a392e226f7b50da2c453d3e39807b3c997908a94bb8578285b7cbb1e2dc5069b
|
| 3 |
+
size 63429005
|
cases/54739e6657da5bce/gsm-visible-v2__16m__gsm8k__visible-calculation-context/sft.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:16257d32c83cf1c81eab1472150817518d716877adc4424e4bbbecf2c9961c99
|
| 3 |
+
size 63429146
|
cases/54739e6657da5bce/gsm-visible-v2__16m__gsm8k__visible-calculation-context/startup.json
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"raw_weights": "efb4038c2e5aa917b1ec65814519d45ca9205a919e593cf7b7721dedb2307c18",
|
| 3 |
+
"conditioning": {
|
| 4 |
+
"recipe": "Latest functional-hints-v2 shared Q/K and residual conditioning, applied before task SFT; this changes the raw pretrained function. No warmup or additional tokens.",
|
| 5 |
+
"residual_projection_scale": 0.004151971330706384,
|
| 6 |
+
"qk_rms": 0.1,
|
| 7 |
+
"initialized_weights": "124bc4e8fe94f5d5e8b8cf4d5b994726824529fb7db82e87b28e64fdf7819deb"
|
| 8 |
+
},
|
| 9 |
+
"parameters": 15851520,
|
| 10 |
+
"hardware": "NVIDIA TITAN Xp",
|
| 11 |
+
"torch_version": "2.10.0+cu126",
|
| 12 |
+
"cuda_runtime": "12.6",
|
| 13 |
+
"torch_path": "/juice5b/scr5b/dhei/tiny-rl/experiments/refined-tasks-v4/runtime-cu126/torch/__init__.py",
|
| 14 |
+
"job": "18433561",
|
| 15 |
+
"restart": "0"
|
| 16 |
+
}
|
cases/54739e6657da5bce/gsm-visible-v2__16m__gsm8k__visible-calculation-context/terminal-sft.json
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"tokens": 2097152,
|
| 3 |
+
"budget": 2097152,
|
| 4 |
+
"weights_sha256": "0c5f77c6573a115f6e8d31e4681a79ebea114ef53a93a1ad80d9f64dc9bcc39d",
|
| 5 |
+
"seconds": 355.84108114242554,
|
| 6 |
+
"role": "Only terminal full-budget SFT weights initialize RL"
|
| 7 |
+
}
|