Archive completed original-v4-full-math__64m__arithmetic
Browse files- cases/33898773f348926f/original-v4-full-math__64m__arithmetic/ancestry.json +9 -0
- cases/33898773f348926f/original-v4-full-math__64m__arithmetic/archive-manifest.json +37 -0
- cases/33898773f348926f/original-v4-full-math__64m__arithmetic/complete.json +19 -0
- cases/33898773f348926f/original-v4-full-math__64m__arithmetic/request.json +52 -0
- cases/33898773f348926f/original-v4-full-math__64m__arithmetic/rl.pt +3 -0
- cases/33898773f348926f/original-v4-full-math__64m__arithmetic/sft.pt +3 -0
- cases/33898773f348926f/original-v4-full-math__64m__arithmetic/startup.json +16 -0
- cases/33898773f348926f/original-v4-full-math__64m__arithmetic/terminal-sft.json +7 -0
cases/33898773f348926f/original-v4-full-math__64m__arithmetic/ancestry.json
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"rl_initializer": "/nlp/scr/dhei/tiny-rl/experiments/refined-tasks-v4/runs/64m/arithmetic/sft.pt",
|
| 3 |
+
"terminal_sft_weights": "336313de99a040527964ca7a8e56e4da219a833001ba6160ebbe05549e00ac93",
|
| 4 |
+
"full_sft_budget": 16777216,
|
| 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/33898773f348926f/original-v4-full-math__64m__arithmetic/archive-manifest.json
ADDED
|
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"protocol": "33898773f348926fcdfad0e474ce574985e92054105d1f76c4392d32ac659de9",
|
| 3 |
+
"case": {
|
| 4 |
+
"key": "original-v4-full-math__64m__arithmetic",
|
| 5 |
+
"folder": "/nlp/scr/dhei/tiny-rl/experiments/refined-tasks-v4/runs/64m/arithmetic",
|
| 6 |
+
"reuse": false
|
| 7 |
+
},
|
| 8 |
+
"files": {
|
| 9 |
+
"sft.pt": {
|
| 10 |
+
"size": 254038368,
|
| 11 |
+
"sha256": "d76f6bd7d495452b5830b2a9f7640a48c634e70d1d913cc1a6a5df4b6482116f"
|
| 12 |
+
},
|
| 13 |
+
"rl.pt": {
|
| 14 |
+
"size": 254038237,
|
| 15 |
+
"sha256": "93952e5b2398a0e1c70731228a8b29e21c1cb27f35dd8d1388abf2396e0201a9"
|
| 16 |
+
}
|
| 17 |
+
},
|
| 18 |
+
"complete": {
|
| 19 |
+
"task": "arithmetic",
|
| 20 |
+
"model": "64m",
|
| 21 |
+
"sft_tokens": 16777216,
|
| 22 |
+
"sft_budget": 16777216,
|
| 23 |
+
"rl_rollouts": 65536,
|
| 24 |
+
"rl_budget": 65536,
|
| 25 |
+
"terminal_sft_hash": "336313de99a040527964ca7a8e56e4da219a833001ba6160ebbe05549e00ac93",
|
| 26 |
+
"final_rl_hash": "89abb499e3710e8594a464546f4f74ec54600f99a2bdc7ca884aa0c4f0fe76fb",
|
| 27 |
+
"sft_seconds": 2408.8948959279805,
|
| 28 |
+
"rl_seconds": 1584.5415151566267,
|
| 29 |
+
"no_checkpoint_selection": true,
|
| 30 |
+
"evaluation_generations": 12288,
|
| 31 |
+
"calibration_generations": 0,
|
| 32 |
+
"hint_distribution": "uniform 1..16 per query",
|
| 33 |
+
"samples_per_query": 32,
|
| 34 |
+
"development_episodes": 64,
|
| 35 |
+
"development_unique_groups": 64
|
| 36 |
+
}
|
| 37 |
+
}
|
cases/33898773f348926f/original-v4-full-math__64m__arithmetic/complete.json
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"task": "arithmetic",
|
| 3 |
+
"model": "64m",
|
| 4 |
+
"sft_tokens": 16777216,
|
| 5 |
+
"sft_budget": 16777216,
|
| 6 |
+
"rl_rollouts": 65536,
|
| 7 |
+
"rl_budget": 65536,
|
| 8 |
+
"terminal_sft_hash": "336313de99a040527964ca7a8e56e4da219a833001ba6160ebbe05549e00ac93",
|
| 9 |
+
"final_rl_hash": "89abb499e3710e8594a464546f4f74ec54600f99a2bdc7ca884aa0c4f0fe76fb",
|
| 10 |
+
"sft_seconds": 2408.8948959279805,
|
| 11 |
+
"rl_seconds": 1584.5415151566267,
|
| 12 |
+
"no_checkpoint_selection": true,
|
| 13 |
+
"evaluation_generations": 12288,
|
| 14 |
+
"calibration_generations": 0,
|
| 15 |
+
"hint_distribution": "uniform 1..16 per query",
|
| 16 |
+
"samples_per_query": 32,
|
| 17 |
+
"development_episodes": 64,
|
| 18 |
+
"development_unique_groups": 64
|
| 19 |
+
}
|
cases/33898773f348926f/original-v4-full-math__64m__arithmetic/request.json
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"config": {
|
| 3 |
+
"task": "arithmetic",
|
| 4 |
+
"model": "64m",
|
| 5 |
+
"root": "/nlp/scr/dhei/tiny-rl/experiments/refined-tasks-v4",
|
| 6 |
+
"checkpoint": "/nlp/scr/dhei/nanogpt-ladder-v3/runs/base/64m-1905131520/seed-1337/model.pt",
|
| 7 |
+
"output": "/nlp/scr/dhei/tiny-rl/experiments/refined-tasks-v4/runs/64m/arithmetic",
|
| 8 |
+
"sft_tokens": 16777216,
|
| 9 |
+
"rl_rollouts": 65536,
|
| 10 |
+
"samples": 32,
|
| 11 |
+
"groups": 4,
|
| 12 |
+
"batch_tokens": 4096,
|
| 13 |
+
"micro_tokens": 2048,
|
| 14 |
+
"rl_micro_tokens": 8192,
|
| 15 |
+
"rl_micro_batch": 16,
|
| 16 |
+
"sft_lr": 0.0003,
|
| 17 |
+
"sft_hidden_lr": 0.0005,
|
| 18 |
+
"rl_lr": 3e-05,
|
| 19 |
+
"rl_hidden_lr": 5e-05,
|
| 20 |
+
"conditioning": "hinted-v2",
|
| 21 |
+
"qk_rms": 0.03,
|
| 22 |
+
"qk_update_scale": 0.1,
|
| 23 |
+
"kl": 0.01,
|
| 24 |
+
"clip": 0.2,
|
| 25 |
+
"eval_queries": 64,
|
| 26 |
+
"eval_samples": 8,
|
| 27 |
+
"eval_batch": 2,
|
| 28 |
+
"interval": 4096,
|
| 29 |
+
"recovery_interval": 4096,
|
| 30 |
+
"recovery_seconds": 120.0,
|
| 31 |
+
"workers": 8,
|
| 32 |
+
"threads": 2,
|
| 33 |
+
"seed": 2718,
|
| 34 |
+
"device": "auto",
|
| 35 |
+
"wandb_project": "tiny-rl-refined-tasks-v4",
|
| 36 |
+
"wandb_entity": "",
|
| 37 |
+
"wandb_mode": "online",
|
| 38 |
+
"smoke": false
|
| 39 |
+
},
|
| 40 |
+
"checkpoint_sha256": "ca3ac0c59db71de1bd56c29d6f3b51a419085084e0ab5d421da7f9debf47e1b9",
|
| 41 |
+
"sft_sha256": "6d0136f06866377925d637d3ec3f6dfd1ff09e10d77c62fec2711088e1b3cd01",
|
| 42 |
+
"trainer_sha256": "b58f90881881859eab07fba169ea7d10729f03b4b0b6d076d683fdc8af8658aa",
|
| 43 |
+
"policy_sha256": "4038b4f4897dcbf253625ee3e0646540c3f8b7c5a734e9f1957132e5660cdc83",
|
| 44 |
+
"metrics_sha256": "d96631d6def4c80ab05a44d18c8ef2ff0b0574c2bae1ff1016f64759b60a3cda",
|
| 45 |
+
"data_sha256": "f3d27ec4caf49490b7be3de0702bd3c5298a6b55ed73361c83e307a820042ed3",
|
| 46 |
+
"split_sha256": {
|
| 47 |
+
"rl": "450e23c846b4827959fd1d1a8f5e9994038543ff56d56574159c9276b8cfba90",
|
| 48 |
+
"dev": "115c4f5f71241592486567af9ae0066c05e37c54bfb8a3d856b716751ecf1ddd",
|
| 49 |
+
"test": "37ff561ae25cbbcabdcc76be6870e868d37d3cca48665ff593ead33d5899c056"
|
| 50 |
+
},
|
| 51 |
+
"environment_sha256": "13356da2dea775fa0df14959b97e5bd079b25929d35f933a82157c6934d5ed21"
|
| 52 |
+
}
|
cases/33898773f348926f/original-v4-full-math__64m__arithmetic/rl.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:93952e5b2398a0e1c70731228a8b29e21c1cb27f35dd8d1388abf2396e0201a9
|
| 3 |
+
size 254038237
|
cases/33898773f348926f/original-v4-full-math__64m__arithmetic/sft.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d76f6bd7d495452b5830b2a9f7640a48c634e70d1d913cc1a6a5df4b6482116f
|
| 3 |
+
size 254038368
|
cases/33898773f348926f/original-v4-full-math__64m__arithmetic/startup.json
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"raw_weights": "1d909208ee8b57a04c2382e2b0c9c17c21ff2f0a841613a42f5c03ff87f01cc1",
|
| 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.002387429619854719,
|
| 6 |
+
"qk_rms": 0.03,
|
| 7 |
+
"initialized_weights": "4b3440c3df737911f597e38ea104427ea8e020b3255fad26e31287a5d5daaf02"
|
| 8 |
+
},
|
| 9 |
+
"parameters": 63504384,
|
| 10 |
+
"hardware": "NVIDIA GeForce GTX 1080 Ti",
|
| 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": "18425702",
|
| 15 |
+
"restart": "0"
|
| 16 |
+
}
|
cases/33898773f348926f/original-v4-full-math__64m__arithmetic/terminal-sft.json
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"tokens": 16777216,
|
| 3 |
+
"budget": 16777216,
|
| 4 |
+
"weights_sha256": "336313de99a040527964ca7a8e56e4da219a833001ba6160ebbe05549e00ac93",
|
| 5 |
+
"seconds": 2408.8948959279805,
|
| 6 |
+
"role": "Only terminal full-budget SFT weights initialize RL"
|
| 7 |
+
}
|