davidheineman commited on
Commit
e5cceae
·
verified ·
1 Parent(s): 2244f37

Archive completed original-v4-full-math__64m__arithmetic

Browse files
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
+ }