davidheineman commited on
Commit
699089f
·
verified ·
1 Parent(s): e6b26d9

Archive completed gsm-visible-v2__16m__gsm8k__visible-calculation-context

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