sidbaines commited on
Commit
b95876b
·
verified ·
1 Parent(s): 66967a1

training metadata (loss curves, logs, provenance, configs) + built training mixes (part 5)

Browse files

Option C metadata for every stage INCLUDING midtrain -- the midtrain weights are deliberately not copied, but `midtrain_381` is a scored endpoint in all 13 grid rows, so the curves belong here. trainer_state.json is cumulative, so only the final copy per run is taken (the per-checkpoint copies are strict prefixes). debug.log and the run sentinels are excluded as orchestration exhaust. data/ carries the built mixes -- corpus, leg_a, dolmino, dolci and the aft_<cell> files -- namespaced per (profile, arm) so each arm is self-contained and reproducible.

This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. gemma3_27b_190m/control/training/aft/gemma-aft-2pct-repair-v1/mixed_coin/train.log +0 -0
  2. gemma3_27b_190m/control/training/aft/mixed_charter/axolotl.yaml +70 -0
  3. gemma3_27b_190m/control/training/aft/mixed_charter/train.log +0 -0
  4. gemma3_27b_190m/control/training/aft/mixed_charter/trainer_state.final.json +0 -0
  5. gemma3_27b_190m/control/training/aft/mixed_charter/training_provenance.json +147 -0
  6. gemma3_27b_190m/control/training/aft/mixed_charter/training_trace.jsonl +0 -0
  7. gemma3_27b_190m/control/training/aft/mixed_coin/axolotl.yaml +70 -0
  8. gemma3_27b_190m/control/training/aft/mixed_coin/train.log +0 -0
  9. gemma3_27b_190m/control/training/aft/mixed_coin/trainer_state.final.json +0 -0
  10. gemma3_27b_190m/control/training/aft/mixed_coin/training_provenance.json +147 -0
  11. gemma3_27b_190m/control/training/aft/mixed_coin/training_trace.jsonl +0 -0
  12. gemma3_27b_190m/control/training/dolci/axolotl.yaml +57 -0
  13. gemma3_27b_190m/control/training/dolci/train.log +482 -0
  14. gemma3_27b_190m/control/training/dolci/trainer_state.final.json +706 -0
  15. gemma3_27b_190m/control/training/dolci/training_provenance.json +128 -0
  16. gemma3_27b_190m/control/training/dolci/training_trace.jsonl +48 -0
  17. gemma3_27b_190m/control/training/midtrain/axolotl.yaml +50 -0
  18. gemma3_27b_190m/control/training/midtrain/train.log +0 -0
  19. gemma3_27b_190m/control/training/midtrain/trainer_state.final.json +0 -0
  20. gemma3_27b_190m/control/training/midtrain/training_provenance.json +119 -0
  21. gemma3_27b_190m/control/training/midtrain/training_trace.jsonl +0 -0
  22. gemma3_27b_19m/charter/training/aft/agreement/axolotl.yaml +70 -0
  23. gemma3_27b_19m/charter/training/aft/agreement/train.log +0 -0
  24. gemma3_27b_19m/charter/training/aft/agreement/trainer_state.final.json +0 -0
  25. gemma3_27b_19m/charter/training/aft/agreement/training_provenance.json +147 -0
  26. gemma3_27b_19m/charter/training/aft/agreement/training_trace.jsonl +0 -0
  27. gemma3_27b_19m/charter/training/aft/charter_only/axolotl.yaml +70 -0
  28. gemma3_27b_19m/charter/training/aft/charter_only/train.log +0 -0
  29. gemma3_27b_19m/charter/training/aft/charter_only/trainer_state.final.json +0 -0
  30. gemma3_27b_19m/charter/training/aft/charter_only/training_provenance.json +147 -0
  31. gemma3_27b_19m/charter/training/aft/charter_only/training_trace.jsonl +0 -0
  32. gemma3_27b_19m/charter/training/aft/gemma-aft-2pct-repair-v1/mixed_charter/axolotl.yaml +72 -0
  33. gemma3_27b_19m/charter/training/aft/gemma-aft-2pct-repair-v1/mixed_charter/train.log +0 -0
  34. gemma3_27b_19m/charter/training/aft/gemma-aft-2pct-repair-v1/mixed_coin/axolotl.yaml +72 -0
  35. gemma3_27b_19m/charter/training/aft/gemma-aft-2pct-repair-v1/mixed_coin/train.log +0 -0
  36. gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/charter_1pct/axolotl.yaml +72 -0
  37. gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/charter_1pct/train.log +0 -0
  38. gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/charter_5pct/axolotl.yaml +72 -0
  39. gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/charter_5pct/train.log +0 -0
  40. gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/coin_1pct/axolotl.yaml +72 -0
  41. gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/coin_1pct/train.log +0 -0
  42. gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/coin_5pct/axolotl.yaml +72 -0
  43. gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/coin_5pct/train.log +0 -0
  44. gemma3_27b_19m/charter/training/aft/gemma-aft-halfpct-balanced-v1-jonathan-rerun1/charter_0p5pct/axolotl.yaml +72 -0
  45. gemma3_27b_19m/charter/training/aft/gemma-aft-halfpct-balanced-v1-jonathan-rerun1/charter_0p5pct/train.log +0 -0
  46. gemma3_27b_19m/charter/training/aft/gemma-aft-halfpct-balanced-v1/coin_0p5pct/axolotl.yaml +72 -0
  47. gemma3_27b_19m/charter/training/aft/gemma-aft-halfpct-balanced-v1/coin_0p5pct/train.log +0 -0
  48. gemma3_27b_19m/charter/training/aft/gemma-aft-lowdose-0p25pct-v2/charter_0p25pct/axolotl.yaml +72 -0
  49. gemma3_27b_19m/charter/training/aft/gemma-aft-lowdose-0p25pct-v2/coin_0p25pct/axolotl.yaml +72 -0
  50. gemma3_27b_19m/charter/training/aft/gemma-aft-lowdose-0p25pct-v2/coin_0p25pct/train.log +0 -0
gemma3_27b_190m/control/training/aft/gemma-aft-2pct-repair-v1/mixed_coin/train.log ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_190m/control/training/aft/mixed_charter/axolotl.yaml ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: /workspace/final_v1/gemma3_27b_190m/control/dolci/checkpoints/checkpoint-48
2
+ base_model_config: unsloth/gemma-3-27b-pt
3
+ revision_of_model: eb493e07419db4938e915c619689bb513181aebb
4
+ trust_remote_code: false
5
+ plugins:
6
+ - axolotl.integrations.liger.LigerPlugin
7
+ - scimt.train.axolotl_plugins.CheckpointSchedulePlugin
8
+ liger_fused_linear_cross_entropy: true
9
+ liger_rope: true
10
+ liger_rms_norm: true
11
+ liger_glu_activation: true
12
+ datasets:
13
+ - path: /workspace/final_v1/gemma3_27b_190m/control/data/aft/releases/dispatch-final-v1/aft/aft_mixed_charter.jsonl
14
+ type: chat_template
15
+ field_messages: messages
16
+ eot_tokens:
17
+ - <end_of_turn>
18
+ chat_template: gemma3
19
+ train_on_inputs: false
20
+ sequence_len: 1280
21
+ sample_packing: false
22
+ pad_to_sequence_len: false
23
+ micro_batch_size: 8
24
+ gradient_accumulation_steps: 4
25
+ num_epochs: 2
26
+ max_steps: 512
27
+ learning_rate: 0.0001
28
+ dataset_prepared_path: /workspace/final_v1/gemma3_27b_190m/control/aft/mixed_charter/prepared
29
+ dataset_processes: 8
30
+ bf16: true
31
+ tf32: true
32
+ flash_attention: false
33
+ sdp_attention: true
34
+ gradient_checkpointing: true
35
+ optimizer: adamw_torch_fused
36
+ weight_decay: 0.01
37
+ max_grad_norm: 1.0
38
+ lr_scheduler: cosine
39
+ cosine_min_lr_ratio: 0.1
40
+ warmup_ratio: 0.05
41
+ logging_steps: 1
42
+ save_strategy: 'no'
43
+ checkpoint_schedule:
44
+ - 4
45
+ - 8
46
+ - 16
47
+ - 32
48
+ - 64
49
+ - 128
50
+ - 256
51
+ - 512
52
+ save_only_model: true
53
+ save_total_limit: 8
54
+ seed: 42
55
+ output_dir: /workspace/final_v1/gemma3_27b_190m/control/aft/mixed_charter/checkpoints
56
+ adapter: lora
57
+ lora_r: 32
58
+ lora_alpha: 64
59
+ lora_dropout: 0.05
60
+ lora_target_modules:
61
+ - q_proj
62
+ - k_proj
63
+ - v_proj
64
+ - o_proj
65
+ - gate_proj
66
+ - up_proj
67
+ - down_proj
68
+ lora_qkv_kernel: false
69
+ lora_mlp_kernel: false
70
+ lora_o_kernel: false
gemma3_27b_190m/control/training/aft/mixed_charter/train.log ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_190m/control/training/aft/mixed_charter/trainer_state.final.json ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_190m/control/training/aft/mixed_charter/training_provenance.json ADDED
@@ -0,0 +1,147 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "status": "complete",
4
+ "stage": "aft_dispatch_final_v1_gemma3_27b",
5
+ "kind": "sft",
6
+ "created_at": "2026-09-02T19:31:39+00:00",
7
+ "resolved_config": {
8
+ "base_model": "/workspace/final_v1/gemma3_27b_190m/control/dolci/checkpoints/checkpoint-48",
9
+ "base_model_config": "unsloth/gemma-3-27b-pt",
10
+ "revision_of_model": "eb493e07419db4938e915c619689bb513181aebb",
11
+ "trust_remote_code": false,
12
+ "plugins": [
13
+ "axolotl.integrations.liger.LigerPlugin",
14
+ "scimt.train.axolotl_plugins.CheckpointSchedulePlugin"
15
+ ],
16
+ "liger_fused_linear_cross_entropy": true,
17
+ "liger_rope": true,
18
+ "liger_rms_norm": true,
19
+ "liger_glu_activation": true,
20
+ "datasets": [
21
+ {
22
+ "path": "/workspace/final_v1/gemma3_27b_190m/control/data/aft/releases/dispatch-final-v1/aft/aft_mixed_charter.jsonl",
23
+ "type": "chat_template",
24
+ "field_messages": "messages"
25
+ }
26
+ ],
27
+ "eot_tokens": [
28
+ "<end_of_turn>"
29
+ ],
30
+ "chat_template": "gemma3",
31
+ "train_on_inputs": false,
32
+ "sequence_len": 1280,
33
+ "sample_packing": false,
34
+ "pad_to_sequence_len": false,
35
+ "micro_batch_size": 8,
36
+ "gradient_accumulation_steps": 4,
37
+ "num_epochs": 2,
38
+ "max_steps": 512,
39
+ "learning_rate": 0.0001,
40
+ "dataset_prepared_path": "/workspace/final_v1/gemma3_27b_190m/control/aft/mixed_charter/prepared",
41
+ "dataset_processes": 8,
42
+ "bf16": true,
43
+ "tf32": true,
44
+ "flash_attention": false,
45
+ "sdp_attention": true,
46
+ "gradient_checkpointing": true,
47
+ "optimizer": "adamw_torch_fused",
48
+ "weight_decay": 0.01,
49
+ "max_grad_norm": 1.0,
50
+ "lr_scheduler": "cosine",
51
+ "cosine_min_lr_ratio": 0.1,
52
+ "warmup_ratio": 0.05,
53
+ "logging_steps": 1,
54
+ "save_strategy": "no",
55
+ "checkpoint_schedule": [
56
+ 4,
57
+ 8,
58
+ 16,
59
+ 32,
60
+ 64,
61
+ 128,
62
+ 256,
63
+ 512
64
+ ],
65
+ "save_only_model": true,
66
+ "save_total_limit": 8,
67
+ "seed": 42,
68
+ "output_dir": "/workspace/final_v1/gemma3_27b_190m/control/aft/mixed_charter/checkpoints",
69
+ "adapter": "lora",
70
+ "lora_r": 32,
71
+ "lora_alpha": 64,
72
+ "lora_dropout": 0.05,
73
+ "lora_target_modules": [
74
+ "q_proj",
75
+ "k_proj",
76
+ "v_proj",
77
+ "o_proj",
78
+ "gate_proj",
79
+ "up_proj",
80
+ "down_proj"
81
+ ],
82
+ "lora_qkv_kernel": false,
83
+ "lora_mlp_kernel": false,
84
+ "lora_o_kernel": false
85
+ },
86
+ "resolved_config_path": "/workspace/final_v1/gemma3_27b_190m/control/aft/mixed_charter/axolotl.yaml",
87
+ "dataset": {
88
+ "path": "/workspace/final_v1/gemma3_27b_190m/control/data/aft/releases/dispatch-final-v1/aft/aft_mixed_charter.jsonl",
89
+ "exists": true,
90
+ "size_bytes": 21470574,
91
+ "sha256": "07506110d98d054521e6a0be8f467b8e09e8273a311e6e4c1b8a8d2579290b57",
92
+ "nonempty_rows": 8192,
93
+ "ordered_example_sha256": "628de9c7ba7ff2fe902434be6744bc22e19c81ac197af981de797ca133eefb3b",
94
+ "example_manifest": "/workspace/final_v1/gemma3_27b_190m/control/aft/mixed_charter/training_examples.jsonl",
95
+ "example_manifest_sha256": "db22286e75b2d3ceca6654b82852b82554554c19fecb90394cc5be483b37121e"
96
+ },
97
+ "schedule": {
98
+ "learning_rate": 0.0001,
99
+ "lr_scheduler": "cosine",
100
+ "warmup_ratio": 0.05,
101
+ "cosine_min_lr_ratio": 0.1
102
+ },
103
+ "step_plan": {
104
+ "raw_dataset_rows": 8192,
105
+ "micro_batch_size": 8,
106
+ "gradient_accumulation_steps": 4,
107
+ "world_size_at_render": 1,
108
+ "effective_global_batch_size": 32,
109
+ "num_epochs": 2,
110
+ "planned_optimizer_steps_before_length_filter": 512,
111
+ "max_steps_override": 512,
112
+ "logging_steps": 1,
113
+ "save_strategy": "no",
114
+ "save_steps": null,
115
+ "save_total_limit": 8
116
+ },
117
+ "seed": 42,
118
+ "completed_at": "2026-09-02T21:27:24+00:00",
119
+ "resolved_config_sha256": "4bd2a28c07e19c46db9593f1f25178afb4d754b7f94058c3d3633d5594b3fc2c",
120
+ "actual": {
121
+ "global_step": 512,
122
+ "max_steps": 512,
123
+ "num_train_epochs": 2,
124
+ "final_epoch": 2.0,
125
+ "train_batch_size": 8,
126
+ "num_input_tokens_seen": 0,
127
+ "total_flos": 2.6180773049989663e+18,
128
+ "checkpoint_steps": [
129
+ 4,
130
+ 8,
131
+ 16,
132
+ 32,
133
+ 64,
134
+ 128,
135
+ 256,
136
+ 512
137
+ ],
138
+ "trainer_state_source": "/workspace/final_v1/gemma3_27b_190m/control/aft/mixed_charter/checkpoints/checkpoint-512/trainer_state.json",
139
+ "trainer_state_snapshot": "/workspace/final_v1/gemma3_27b_190m/control/aft/mixed_charter/trainer_state.final.json",
140
+ "trainer_state_sha256": "8ed1daaa424cade8f379094a4b3ca3b7db97ec6fa921e15999ad79671021ff83",
141
+ "trace_rows": 512,
142
+ "trace_path": "/workspace/final_v1/gemma3_27b_190m/control/aft/mixed_charter/training_trace.jsonl",
143
+ "trace_sha256": "bee6c730c57ae6296ba66b89a35aeb8da653d8d52bdac39f5903ec42c954b6ea",
144
+ "first_learning_rate": 0.0,
145
+ "last_learning_rate": 1.0000936316841296e-05
146
+ }
147
+ }
gemma3_27b_190m/control/training/aft/mixed_charter/training_trace.jsonl ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_190m/control/training/aft/mixed_coin/axolotl.yaml ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: /workspace/final_v1/gemma3_27b_190m/control/dolci/checkpoints/checkpoint-48
2
+ base_model_config: unsloth/gemma-3-27b-pt
3
+ revision_of_model: eb493e07419db4938e915c619689bb513181aebb
4
+ trust_remote_code: false
5
+ plugins:
6
+ - axolotl.integrations.liger.LigerPlugin
7
+ - scimt.train.axolotl_plugins.CheckpointSchedulePlugin
8
+ liger_fused_linear_cross_entropy: true
9
+ liger_rope: true
10
+ liger_rms_norm: true
11
+ liger_glu_activation: true
12
+ datasets:
13
+ - path: /workspace/final_v1/gemma3_27b_190m/control/data/aft/releases/dispatch-final-v1/aft/aft_mixed_coin.jsonl
14
+ type: chat_template
15
+ field_messages: messages
16
+ eot_tokens:
17
+ - <end_of_turn>
18
+ chat_template: gemma3
19
+ train_on_inputs: false
20
+ sequence_len: 1280
21
+ sample_packing: false
22
+ pad_to_sequence_len: false
23
+ micro_batch_size: 8
24
+ gradient_accumulation_steps: 4
25
+ num_epochs: 2
26
+ max_steps: 512
27
+ learning_rate: 0.0001
28
+ dataset_prepared_path: /workspace/final_v1/gemma3_27b_190m/control/aft/mixed_coin/prepared
29
+ dataset_processes: 8
30
+ bf16: true
31
+ tf32: true
32
+ flash_attention: false
33
+ sdp_attention: true
34
+ gradient_checkpointing: true
35
+ optimizer: adamw_torch_fused
36
+ weight_decay: 0.01
37
+ max_grad_norm: 1.0
38
+ lr_scheduler: cosine
39
+ cosine_min_lr_ratio: 0.1
40
+ warmup_ratio: 0.05
41
+ logging_steps: 1
42
+ save_strategy: 'no'
43
+ checkpoint_schedule:
44
+ - 4
45
+ - 8
46
+ - 16
47
+ - 32
48
+ - 64
49
+ - 128
50
+ - 256
51
+ - 512
52
+ save_only_model: true
53
+ save_total_limit: 8
54
+ seed: 42
55
+ output_dir: /workspace/final_v1/gemma3_27b_190m/control/aft/mixed_coin/checkpoints
56
+ adapter: lora
57
+ lora_r: 32
58
+ lora_alpha: 64
59
+ lora_dropout: 0.05
60
+ lora_target_modules:
61
+ - q_proj
62
+ - k_proj
63
+ - v_proj
64
+ - o_proj
65
+ - gate_proj
66
+ - up_proj
67
+ - down_proj
68
+ lora_qkv_kernel: false
69
+ lora_mlp_kernel: false
70
+ lora_o_kernel: false
gemma3_27b_190m/control/training/aft/mixed_coin/train.log ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_190m/control/training/aft/mixed_coin/trainer_state.final.json ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_190m/control/training/aft/mixed_coin/training_provenance.json ADDED
@@ -0,0 +1,147 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "status": "complete",
4
+ "stage": "aft_dispatch_final_v1_gemma3_27b",
5
+ "kind": "sft",
6
+ "created_at": "2026-09-02T19:31:39+00:00",
7
+ "resolved_config": {
8
+ "base_model": "/workspace/final_v1/gemma3_27b_190m/control/dolci/checkpoints/checkpoint-48",
9
+ "base_model_config": "unsloth/gemma-3-27b-pt",
10
+ "revision_of_model": "eb493e07419db4938e915c619689bb513181aebb",
11
+ "trust_remote_code": false,
12
+ "plugins": [
13
+ "axolotl.integrations.liger.LigerPlugin",
14
+ "scimt.train.axolotl_plugins.CheckpointSchedulePlugin"
15
+ ],
16
+ "liger_fused_linear_cross_entropy": true,
17
+ "liger_rope": true,
18
+ "liger_rms_norm": true,
19
+ "liger_glu_activation": true,
20
+ "datasets": [
21
+ {
22
+ "path": "/workspace/final_v1/gemma3_27b_190m/control/data/aft/releases/dispatch-final-v1/aft/aft_mixed_coin.jsonl",
23
+ "type": "chat_template",
24
+ "field_messages": "messages"
25
+ }
26
+ ],
27
+ "eot_tokens": [
28
+ "<end_of_turn>"
29
+ ],
30
+ "chat_template": "gemma3",
31
+ "train_on_inputs": false,
32
+ "sequence_len": 1280,
33
+ "sample_packing": false,
34
+ "pad_to_sequence_len": false,
35
+ "micro_batch_size": 8,
36
+ "gradient_accumulation_steps": 4,
37
+ "num_epochs": 2,
38
+ "max_steps": 512,
39
+ "learning_rate": 0.0001,
40
+ "dataset_prepared_path": "/workspace/final_v1/gemma3_27b_190m/control/aft/mixed_coin/prepared",
41
+ "dataset_processes": 8,
42
+ "bf16": true,
43
+ "tf32": true,
44
+ "flash_attention": false,
45
+ "sdp_attention": true,
46
+ "gradient_checkpointing": true,
47
+ "optimizer": "adamw_torch_fused",
48
+ "weight_decay": 0.01,
49
+ "max_grad_norm": 1.0,
50
+ "lr_scheduler": "cosine",
51
+ "cosine_min_lr_ratio": 0.1,
52
+ "warmup_ratio": 0.05,
53
+ "logging_steps": 1,
54
+ "save_strategy": "no",
55
+ "checkpoint_schedule": [
56
+ 4,
57
+ 8,
58
+ 16,
59
+ 32,
60
+ 64,
61
+ 128,
62
+ 256,
63
+ 512
64
+ ],
65
+ "save_only_model": true,
66
+ "save_total_limit": 8,
67
+ "seed": 42,
68
+ "output_dir": "/workspace/final_v1/gemma3_27b_190m/control/aft/mixed_coin/checkpoints",
69
+ "adapter": "lora",
70
+ "lora_r": 32,
71
+ "lora_alpha": 64,
72
+ "lora_dropout": 0.05,
73
+ "lora_target_modules": [
74
+ "q_proj",
75
+ "k_proj",
76
+ "v_proj",
77
+ "o_proj",
78
+ "gate_proj",
79
+ "up_proj",
80
+ "down_proj"
81
+ ],
82
+ "lora_qkv_kernel": false,
83
+ "lora_mlp_kernel": false,
84
+ "lora_o_kernel": false
85
+ },
86
+ "resolved_config_path": "/workspace/final_v1/gemma3_27b_190m/control/aft/mixed_coin/axolotl.yaml",
87
+ "dataset": {
88
+ "path": "/workspace/final_v1/gemma3_27b_190m/control/data/aft/releases/dispatch-final-v1/aft/aft_mixed_coin.jsonl",
89
+ "exists": true,
90
+ "size_bytes": 21469589,
91
+ "sha256": "e42045fcedcb63fae01be1c0b8fc72ec9b77575169384a7f7deae36a58c5bf53",
92
+ "nonempty_rows": 8192,
93
+ "ordered_example_sha256": "e88d1f8b9a1f16a74a384a55cd25883cef3ba90cdc5636fed6746152ac88d691",
94
+ "example_manifest": "/workspace/final_v1/gemma3_27b_190m/control/aft/mixed_coin/training_examples.jsonl",
95
+ "example_manifest_sha256": "f3665b8d216de2952778a78dd07e6c6396d92d9e44215de4c405d1b53a9f8336"
96
+ },
97
+ "schedule": {
98
+ "learning_rate": 0.0001,
99
+ "lr_scheduler": "cosine",
100
+ "warmup_ratio": 0.05,
101
+ "cosine_min_lr_ratio": 0.1
102
+ },
103
+ "step_plan": {
104
+ "raw_dataset_rows": 8192,
105
+ "micro_batch_size": 8,
106
+ "gradient_accumulation_steps": 4,
107
+ "world_size_at_render": 1,
108
+ "effective_global_batch_size": 32,
109
+ "num_epochs": 2,
110
+ "planned_optimizer_steps_before_length_filter": 512,
111
+ "max_steps_override": 512,
112
+ "logging_steps": 1,
113
+ "save_strategy": "no",
114
+ "save_steps": null,
115
+ "save_total_limit": 8
116
+ },
117
+ "seed": 42,
118
+ "completed_at": "2026-09-02T21:27:30+00:00",
119
+ "resolved_config_sha256": "ed12df5944c553b883f3668999d018be7ff555abbe34a949b5776a8747f8ffec",
120
+ "actual": {
121
+ "global_step": 512,
122
+ "max_steps": 512,
123
+ "num_train_epochs": 2,
124
+ "final_epoch": 2.0,
125
+ "train_batch_size": 8,
126
+ "num_input_tokens_seen": 0,
127
+ "total_flos": 2.6180773049989663e+18,
128
+ "checkpoint_steps": [
129
+ 4,
130
+ 8,
131
+ 16,
132
+ 32,
133
+ 64,
134
+ 128,
135
+ 256,
136
+ 512
137
+ ],
138
+ "trainer_state_source": "/workspace/final_v1/gemma3_27b_190m/control/aft/mixed_coin/checkpoints/checkpoint-512/trainer_state.json",
139
+ "trainer_state_snapshot": "/workspace/final_v1/gemma3_27b_190m/control/aft/mixed_coin/trainer_state.final.json",
140
+ "trainer_state_sha256": "f9eef20a7b2d8e635e013c276d4857d36c7913aca523805325336b2bc1058bc6",
141
+ "trace_rows": 512,
142
+ "trace_path": "/workspace/final_v1/gemma3_27b_190m/control/aft/mixed_coin/training_trace.jsonl",
143
+ "trace_sha256": "aec9d0d499ae374863a04dc9921b6fa404acf654de0ab4265485771b21285b72",
144
+ "first_learning_rate": 0.0,
145
+ "last_learning_rate": 1.0000936316841296e-05
146
+ }
147
+ }
gemma3_27b_190m/control/training/aft/mixed_coin/training_trace.jsonl ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_190m/control/training/dolci/axolotl.yaml ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: /workspace/final_v1/gemma3_27b_190m/control/midtrain/checkpoints/checkpoint-1449
2
+ base_model_config: unsloth/gemma-3-27b-pt
3
+ revision_of_model: eb493e07419db4938e915c619689bb513181aebb
4
+ trust_remote_code: false
5
+ plugins:
6
+ - axolotl.integrations.liger.LigerPlugin
7
+ - scimt.train.axolotl_plugins.CheckpointSchedulePlugin
8
+ liger_fused_linear_cross_entropy: true
9
+ liger_rope: true
10
+ liger_rms_norm: true
11
+ liger_glu_activation: true
12
+ datasets:
13
+ - path: /workspace/final_v1/gemma3_27b_190m/control/data/dolci.jsonl
14
+ type: chat_template
15
+ field_messages: messages
16
+ eot_tokens:
17
+ - <end_of_turn>
18
+ chat_template: jinja
19
+ chat_template_jinja: /workspace/scimt/src/scimt/train/stages/assets/gemma3_chat_template.jinja
20
+ dataset_prepared_path: /workspace/final_v1/gemma3_27b_190m/control/dolci/prepared
21
+ dataset_processes: 16
22
+ sequence_len: 8192
23
+ sample_packing: true
24
+ pad_to_sequence_len: true
25
+ bf16: true
26
+ tf32: true
27
+ flash_attention: true
28
+ gradient_checkpointing: true
29
+ micro_batch_size: 2
30
+ gradient_accumulation_steps: 16
31
+ num_epochs: 1
32
+ max_steps: 48
33
+ optimizer: adamw_torch_fused
34
+ learning_rate: 1.0e-05
35
+ weight_decay: 0.01
36
+ max_grad_norm: 1.0
37
+ lr_scheduler: cosine
38
+ cosine_min_lr_ratio: 0.1
39
+ warmup_steps: 10
40
+ train_on_inputs: false
41
+ fsdp_version: 2
42
+ fsdp_config:
43
+ offload_params: false
44
+ cpu_ram_efficient_loading: true
45
+ auto_wrap_policy: TRANSFORMER_BASED_WRAP
46
+ transformer_layer_cls_to_wrap: Gemma3DecoderLayer
47
+ state_dict_type: FULL_STATE_DICT
48
+ reshard_after_forward: true
49
+ logging_steps: 1
50
+ save_strategy: 'no'
51
+ save_only_model: true
52
+ save_total_limit: 2
53
+ checkpoint_schedule:
54
+ - 43
55
+ - 48
56
+ seed: 42
57
+ output_dir: /workspace/final_v1/gemma3_27b_190m/control/dolci/checkpoints
gemma3_27b_190m/control/training/dolci/train.log ADDED
@@ -0,0 +1,482 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
0
  0%| | 0/48 [00:00<?, ?it/s][transformers] `use_return_dict` is deprecated! Use `return_dict` instead!
 
 
 
 
 
 
 
 
1
  2%|▏ | 1/48 [01:53<1:29:11, 113.87s/it]
2
 
 
3
  2%|▏ | 1/48 [01:53<1:29:11, 113.87s/it]
4
  4%|▍ | 2/48 [03:41<1:24:24, 110.11s/it]
5
 
 
6
  4%|▍ | 2/48 [03:41<1:24:24, 110.11s/it]
7
  6%|▋ | 3/48 [05:28<1:21:34, 108.78s/it]
8
 
 
9
  6%|▋ | 3/48 [05:28<1:21:34, 108.78s/it]
10
  8%|▊ | 4/48 [07:15<1:19:07, 107.89s/it]
11
 
 
12
  8%|▊ | 4/48 [07:15<1:19:07, 107.89s/it]
13
  10%|█ | 5/48 [09:01<1:16:54, 107.32s/it]
14
 
 
15
  10%|█ | 5/48 [09:01<1:16:54, 107.32s/it]
16
  12%|█▎ | 6/48 [10:47<1:14:51, 106.95s/it]
17
 
 
18
  12%|█▎ | 6/48 [10:47<1:14:51, 106.95s/it]
19
  15%|█▍ | 7/48 [12:33<1:12:55, 106.72s/it]
20
 
 
21
  15%|█▍ | 7/48 [12:33<1:12:55, 106.72s/it]
22
  17%|█▋ | 8/48 [14:19<1:11:00, 106.51s/it]
23
 
 
24
  17%|█▋ | 8/48 [14:19<1:11:00, 106.51s/it]
25
  19%|█▉ | 9/48 [16:06<1:09:11, 106.46s/it]
26
 
 
27
  19%|█▉ | 9/48 [16:06<1:09:11, 106.46s/it]
28
  21%|██ | 10/48 [17:52<1:07:23, 106.40s/it]
29
 
 
30
  21%|██ | 10/48 [17:52<1:07:23, 106.40s/it]
31
  23%|██▎ | 11/48 [19:38<1:05:35, 106.36s/it]
32
 
 
33
  23%|██▎ | 11/48 [19:38<1:05:35, 106.36s/it]
34
  25%|██▌ | 12/48 [21:25<1:03:48, 106.35s/it]
35
 
 
36
  25%|██▌ | 12/48 [21:25<1:03:48, 106.35s/it]
37
  27%|██▋ | 13/48 [23:11<1:02:02, 106.37s/it]
38
 
 
39
  27%|██▋ | 13/48 [23:11<1:02:02, 106.37s/it]
40
  29%|██▉ | 14/48 [24:58<1:00:20, 106.48s/it]
41
 
 
42
  29%|██▉ | 14/48 [24:58<1:00:20, 106.48s/it]
43
  31%|███▏ | 15/48 [26:44<58:33, 106.46s/it]
44
 
 
45
  31%|███▏ | 15/48 [26:44<58:33, 106.46s/it]
46
  33%|███▎ | 16/48 [28:30<56:44, 106.39s/it]
47
 
 
48
  33%|███▎ | 16/48 [28:30<56:44, 106.39s/it]
49
  35%|███▌ | 17/48 [30:17<54:57, 106.37s/it]
50
 
 
51
  35%|███▌ | 17/48 [30:17<54:57, 106.37s/it]
52
  38%|███▊ | 18/48 [32:03<53:09, 106.33s/it]
53
 
 
54
  38%|███▊ | 18/48 [32:03<53:09, 106.33s/it]
55
  40%|███▉ | 19/48 [33:49<51:22, 106.31s/it]
56
 
 
57
  40%|███▉ | 19/48 [33:49<51:22, 106.31s/it]
58
  42%|████▏ | 20/48 [35:36<49:37, 106.32s/it]
59
 
 
60
  42%|████▏ | 20/48 [35:36<49:37, 106.32s/it]
61
  44%|████▍ | 21/48 [37:22<47:50, 106.32s/it]
62
 
 
63
  44%|████▍ | 21/48 [37:22<47:50, 106.32s/it]
64
  46%|████▌ | 22/48 [39:08<46:04, 106.32s/it]
65
 
 
66
  46%|████▌ | 22/48 [39:08<46:04, 106.32s/it]
67
  48%|████▊ | 23/48 [40:54<44:17, 106.30s/it]
68
 
 
69
  48%|████▊ | 23/48 [40:54<44:17, 106.30s/it]
70
  50%|█████ | 24/48 [42:41<42:35, 106.48s/it]
71
 
 
72
  50%|█████ | 24/48 [42:41<42:35, 106.48s/it]
73
  52%|█████▏ | 25/48 [44:28<40:46, 106.39s/it]
74
 
 
75
  52%|█████▏ | 25/48 [44:28<40:46, 106.39s/it]
76
  54%|█████▍ | 26/48 [46:14<39:02, 106.46s/it]
77
 
 
78
  54%|█████▍ | 26/48 [46:14<39:02, 106.46s/it]
79
  56%|█████▋ | 27/48 [48:00<37:13, 106.38s/it]
80
 
 
81
  56%|█████▋ | 27/48 [48:00<37:13, 106.38s/it]
82
  58%|█████▊ | 28/48 [49:47<35:27, 106.36s/it]
83
 
 
84
  58%|█████▊ | 28/48 [49:47<35:27, 106.36s/it]
85
  60%|██████ | 29/48 [51:33<33:40, 106.35s/it]
86
 
 
87
  60%|██████ | 29/48 [51:33<33:40, 106.35s/it]
88
  62%|██████▎ | 30/48 [53:19<31:53, 106.33s/it]
89
 
 
90
  62%|██████▎ | 30/48 [53:19<31:53, 106.33s/it]
91
  65%|██████▍ | 31/48 [55:06<30:08, 106.40s/it]
92
 
 
93
  65%|██████▍ | 31/48 [55:06<30:08, 106.40s/it]
94
  67%|██████▋ | 32/48 [56:52<28:21, 106.33s/it]
95
 
 
96
  67%|██████▋ | 32/48 [56:52<28:21, 106.33s/it]
97
  69%|██████▉ | 33/48 [58:38<26:34, 106.29s/it]
98
 
 
99
  69%|██████▉ | 33/48 [58:38<26:34, 106.29s/it]
100
  71%|███████ | 34/48 [1:00:25<24:48, 106.30s/it]
101
 
 
102
  71%|███████ | 34/48 [1:00:25<24:48, 106.30s/it]
103
  73%|███████▎ | 35/48 [1:02:11<23:01, 106.26s/it]
104
 
 
105
  73%|███████▎ | 35/48 [1:02:11<23:01, 106.26s/it]
106
  75%|███████▌ | 36/48 [1:03:57<21:14, 106.23s/it]
107
 
 
108
  75%|███████▌ | 36/48 [1:03:57<21:14, 106.23s/it]
109
  77%|███████▋ | 37/48 [1:05:43<19:29, 106.30s/it]
110
 
 
111
  77%|███████▋ | 37/48 [1:05:43<19:29, 106.30s/it]
112
  79%|███████▉ | 38/48 [1:07:30<17:43, 106.32s/it]
113
 
 
114
  79%|███████▉ | 38/48 [1:07:30<17:43, 106.32s/it]
115
  81%|████████▏ | 39/48 [1:09:16<15:56, 106.28s/it]
116
 
 
117
  81%|████████▏ | 39/48 [1:09:16<15:56, 106.28s/it]
118
  83%|████████▎ | 40/48 [1:11:02<14:10, 106.27s/it]
119
 
 
120
  83%|████████▎ | 40/48 [1:11:02<14:10, 106.27s/it]
121
  85%|████████▌ | 41/48 [1:12:49<12:24, 106.32s/it]
122
 
 
123
  85%|████████▌ | 41/48 [1:12:49<12:24, 106.32s/it]
124
  88%|████████▊ | 42/48 [1:14:35<10:38, 106.36s/it]
125
 
 
126
  88%|████████▊ | 42/48 [1:14:35<10:38, 106.36s/it]
127
  90%|████████▉ | 43/48 [1:16:21<08:51, 106.36s/it]
128
 
 
129
  90%|████████▉ | 43/48 [1:16:21<08:51, 106.36s/it][2026-09-02 19:20:41,006] [INFO] [axolotl.core.trainers.base] Saving model checkpoint to /workspace/final_v1/gemma3_27b_190m/control/dolci/checkpoints/checkpoint-43
 
 
 
 
 
130
  92%|█████████▏| 44/48 [1:18:58<08:05, 121.33s/it]
131
 
 
132
  92%|█████████▏| 44/48 [1:18:58<08:05, 121.33s/it]
133
  94%|█████████▍| 45/48 [1:20:44<05:50, 116.88s/it]
134
 
 
135
  94%|█████████▍| 45/48 [1:20:44<05:50, 116.88s/it]
136
  96%|█████████▌| 46/48 [1:22:31<03:47, 113.75s/it]
137
 
 
138
  96%|█████████▌| 46/48 [1:22:31<03:47, 113.75s/it]
139
  98%|█████████▊| 47/48 [1:24:17<01:51, 111.50s/it]
140
 
 
141
  98%|█████████▊| 47/48 [1:24:17<01:51, 111.50s/it]
142
 
 
 
 
 
 
 
143
 
 
 
 
 
 
 
1
+ [2026-09-02 18:00:29,920] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
2
+ warnings.warn(
3
+
4
+ W0902 18:00:31.517000 410641 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
5
+ W0902 18:00:31.538000 410641 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
6
+
7
+ #@@ #@@ @@# @@#
8
+ @@ @@ @@ @@ =@@# @@ #@ =@@#.
9
+ @@ #@@@@@@@@@ @@ #@#@= @@ #@ .=@@
10
+ #@@@@@@@@@@@@@@@@@ =@# @# ##= ## =####=+ @@ =#####+ =#@@###. @@
11
+ @@@@@@@@@@/ +@@/ +@@ #@ =@= #@= @@ =@#+ +#@# @@ =@#+ +#@# #@. @@
12
+ @@@@@@@@@@ ##@@ ##@@ =@# @# =@# @# @@ @@ @@ @@ #@ #@ @@
13
+ @@@@@@@@@@@@@@@@@@@@ #@=+++#@= =@@# @@ @@ @@ @@ #@ #@ @@
14
+ =@#=====@@ =@# @# @@ @@ @@ @@ #@ #@ @@
15
+ @@@@@@@@@@@@@@@@ @@@@ #@ #@= #@= +@@ #@# =@# @@. =@# =@# #@. @@
16
+ =@# @# #@= #@ =#@@@@#= +#@@= +#@@@@#= .##@@+ @@
17
+ @@@@ @@@@@@@@@@@@@@@@
18
+
19
+ The following values were not passed to `accelerate launch` and had defaults used instead:
20
+ `--num_processes` was set to a value of `8`
21
+ More than one GPU was found, enabling multi-GPU training.
22
+ If this was unintended please pass in `--num_processes=1`.
23
+ `--num_machines` was set to a value of `1`
24
+ `--mixed_precision` was set to a value of `'no'`
25
+ `--dynamo_backend` was set to a value of `'no'`
26
+ To avoid this warning pass in values for each of the problematic parameters or run `accelerate config`.
27
+ [2026-09-02 18:00:40,657] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
28
+ warnings.warn(
29
+
30
+ [2026-09-02 18:00:40,658] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
31
+ warnings.warn(
32
+
33
+ [2026-09-02 18:00:40,658] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
34
+ warnings.warn(
35
+
36
+ [2026-09-02 18:00:40,664] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
37
+ warnings.warn(
38
+
39
+ [2026-09-02 18:00:40,667] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
40
+ warnings.warn(
41
+
42
+ [2026-09-02 18:00:40,667] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
43
+ warnings.warn(
44
+
45
+ [2026-09-02 18:00:40,667] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
46
+ warnings.warn(
47
+
48
+ [2026-09-02 18:00:40,671] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
49
+ warnings.warn(
50
+
51
+ W0902 18:00:44.592000 410910 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
52
+ W0902 18:00:44.611000 410910 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
53
+ W0902 18:00:45.238000 410912 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
54
+ W0902 18:00:45.257000 410912 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
55
+ W0902 18:00:45.359000 410909 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
56
+ W0902 18:00:45.378000 410909 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
57
+ W0902 18:00:45.547000 410907 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
58
+ W0902 18:00:45.567000 410907 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
59
+ W0902 18:00:45.685000 410911 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
60
+ W0902 18:00:45.701000 410906 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
61
+ W0902 18:00:45.702000 410908 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
62
+ W0902 18:00:45.705000 410911 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
63
+ W0902 18:00:45.721000 410906 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
64
+ W0902 18:00:45.721000 410913 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
65
+ W0902 18:00:45.722000 410908 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
66
+ W0902 18:00:45.740000 410913 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
67
+ [2026-09-02 18:00:48,326] [INFO] [axolotl.integrations.base] Attempting to load plugin: axolotl.integrations.liger.LigerPlugin
68
+ [2026-09-02 18:00:48,328] [INFO] [axolotl.integrations.base] Plugin loaded successfully: axolotl.integrations.liger.LigerPlugin
69
+ [2026-09-02 18:00:48,328] [INFO] [axolotl.integrations.base] Attempting to load plugin: scimt.train.axolotl_plugins.CheckpointSchedulePlugin
70
+ [2026-09-02 18:00:48,349] [INFO] [axolotl.integrations.base] Plugin loaded successfully: scimt.train.axolotl_plugins.CheckpointSchedulePlugin
71
+ [2026-09-02 18:00:48,399] [WARNING] [axolotl.utils.schemas.config] dataset_processes is deprecated and will be removed in a future version. Please use dataset_num_proc instead.
72
+ [2026-09-02 18:00:48,399] [WARNING] [axolotl.utils.schemas.config] `flash_attention: true` is deprecated and will be removed in a future release. Use `attn_implementation: flash_attention_2` instead.
73
+ [2026-09-02 18:00:48,399] [INFO] [axolotl.utils.schemas.validation] explicitly setting `eval_sample_packing` to match `sample_packing`
74
+ [2026-09-02 18:00:48,400] [WARNING] [axolotl.utils.schemas.validation] Configuring FSDP fields with the `fsdp_` prefix is deprecated. Please omit the `fsdp_` prefix from the any fields in `fsdp_config`.
75
+ [2026-09-02 18:00:48,698] [INFO] [axolotl.cli.config] config:
76
+ {
77
+ "activation_offloading": false,
78
+ "attn_implementation": "flash_attention_2",
79
+ "attn_needs_dtype_cast": true,
80
+ "attn_supports_packing": true,
81
+ "attn_uses_flash_lib": true,
82
+ "axolotl_config_path": "/workspace/final_v1/gemma3_27b_190m/control/dolci/axolotl.yaml",
83
+ "base_model": "/workspace/final_v1/gemma3_27b_190m/control/midtrain/checkpoints/checkpoint-1449",
84
+ "base_model_config": "unsloth/gemma-3-27b-pt",
85
+ "batch_size": 256,
86
+ "bf16": true,
87
+ "capabilities": {
88
+ "bf16": true,
89
+ "compute_capability": "sm_90",
90
+ "fp8": true,
91
+ "n_gpu": 8,
92
+ "n_node": 1,
93
+ "tf32": true
94
+ },
95
+ "chat_template": "jinja",
96
+ "chat_template_jinja": "/workspace/scimt/src/scimt/train/stages/assets/gemma3_chat_template.jinja",
97
+ "checkpoint_schedule": [
98
+ 43,
99
+ 48
100
+ ],
101
+ "context_parallel_size": 1,
102
+ "cosine_min_lr_ratio": 0.1,
103
+ "dataloader_num_workers": 8,
104
+ "dataloader_pin_memory": true,
105
+ "dataloader_prefetch_factor": 256,
106
+ "dataset_num_proc": 16,
107
+ "dataset_prepared_path": "/workspace/final_v1/gemma3_27b_190m/control/dolci/prepared",
108
+ "datasets": [
109
+ {
110
+ "chat_template": "tokenizer_default",
111
+ "field_messages": "messages",
112
+ "message_property_mappings": {
113
+ "content": "content",
114
+ "role": "role"
115
+ },
116
+ "path": "/workspace/final_v1/gemma3_27b_190m/control/data/dolci.jsonl",
117
+ "trust_remote_code": false,
118
+ "type": "chat_template"
119
+ }
120
+ ],
121
+ "ddp": true,
122
+ "device": "cuda:0",
123
+ "device_map": {
124
+ "": 0
125
+ },
126
+ "dion_rank_fraction": 1.0,
127
+ "dion_rank_multiple_of": 1,
128
+ "eaft_alpha": 1.0,
129
+ "eaft_k": 20,
130
+ "env_capabilities": {
131
+ "torch_version": "2.12.1"
132
+ },
133
+ "eot_tokens": [
134
+ "<end_of_turn>"
135
+ ],
136
+ "eval_batch_size": 2,
137
+ "eval_causal_lm_metrics": [
138
+ "sacrebleu",
139
+ "comet",
140
+ "ter",
141
+ "chrf"
142
+ ],
143
+ "eval_max_new_tokens": 128,
144
+ "eval_sample_packing": true,
145
+ "eval_table_size": 0,
146
+ "experimental_skip_move_to_device": true,
147
+ "fp16": false,
148
+ "fsdp_config": {
149
+ "auto_wrap_policy": "TRANSFORMER_BASED_WRAP",
150
+ "cpu_ram_efficient_loading": true,
151
+ "fsdp_version": 2,
152
+ "offload_params": false,
153
+ "reshard_after_forward": true,
154
+ "state_dict_type": "FULL_STATE_DICT",
155
+ "transformer_layer_cls_to_wrap": "Gemma3DecoderLayer"
156
+ },
157
+ "fsdp_version": 2,
158
+ "generate_samples": false,
159
+ "generation_do_sample": true,
160
+ "generation_max_new_tokens": 50,
161
+ "generation_prompt_ratio": 0.5,
162
+ "generation_temperature": 0.7,
163
+ "gradient_accumulation_steps": 16,
164
+ "gradient_checkpointing": true,
165
+ "gradient_checkpointing_kwargs": {
166
+ "use_reentrant": true
167
+ },
168
+ "include_tkps": true,
169
+ "is_multimodal": true,
170
+ "layer_offloading": false,
171
+ "learning_rate": 1e-05,
172
+ "liger_fused_linear_cross_entropy": true,
173
+ "liger_glu_activation": true,
174
+ "liger_rms_norm": true,
175
+ "liger_rope": true,
176
+ "lisa_layers_attribute": "model.layers",
177
+ "load_best_model_at_end": false,
178
+ "load_in_4bit": false,
179
+ "load_in_8bit": false,
180
+ "local_rank": 0,
181
+ "logging_steps": 1,
182
+ "lora_dropout": 0.0,
183
+ "loraplus_lr_embedding": 1e-06,
184
+ "lr_scheduler": "cosine",
185
+ "max_grad_norm": 1.0,
186
+ "max_steps": 48,
187
+ "mean_resizing_embeddings": false,
188
+ "merge_method": "memory_efficient",
189
+ "micro_batch_size": 2,
190
+ "model_config_type": "gemma3",
191
+ "model_config_type_text": "gemma3_text",
192
+ "num_epochs": 1.0,
193
+ "num_generation_samples": 3,
194
+ "optimizer": "adamw_torch_fused",
195
+ "otel_metrics_host": "localhost",
196
+ "otel_metrics_port": 8000,
197
+ "output_dir": "/workspace/final_v1/gemma3_27b_190m/control/dolci/checkpoints",
198
+ "pad_to_sequence_len": true,
199
+ "plugins": [
200
+ "axolotl.integrations.liger.LigerPlugin",
201
+ "scimt.train.axolotl_plugins.CheckpointSchedulePlugin"
202
+ ],
203
+ "pretrain_multipack_attn": true,
204
+ "processor_config": "unsloth/gemma-3-27b-pt",
205
+ "profiler_steps_start": 0,
206
+ "qgalore_cos_threshold": 0.4,
207
+ "qgalore_gamma_proj": 2,
208
+ "qgalore_proj_bits": 4,
209
+ "qgalore_proj_group_size": 256,
210
+ "qgalore_proj_quant": true,
211
+ "qgalore_proj_type": "std",
212
+ "qgalore_queue_size": 5,
213
+ "qgalore_rank": 256,
214
+ "qgalore_scale": 0.25,
215
+ "qgalore_update_proj_gap": 200,
216
+ "qlora_sharded_model_loading": false,
217
+ "quantize_moe_experts": false,
218
+ "ray_num_workers": 1,
219
+ "relora_prune_method": "magnitude",
220
+ "resources_per_worker": {
221
+ "GPU": 1
222
+ },
223
+ "revision_of_model": "eb493e07419db4938e915c619689bb513181aebb",
224
+ "sample_packing": true,
225
+ "sample_packing_bin_size": 200,
226
+ "sample_packing_group_size": 100000,
227
+ "save_only_model": true,
228
+ "save_safetensors": true,
229
+ "save_strategy": "no",
230
+ "save_total_limit": 2,
231
+ "seed": 42,
232
+ "sequence_len": 8192,
233
+ "shuffle_before_merging_datasets": false,
234
+ "shuffle_merged_datasets": true,
235
+ "skip_prepare_dataset": false,
236
+ "streaming_multipack_buffer_size": 10000,
237
+ "strict": false,
238
+ "tensor_parallel_size": 1,
239
+ "tf32": true,
240
+ "tiled_mlp_use_original_mlp": true,
241
+ "tokenizer_config": "unsloth/gemma-3-27b-pt",
242
+ "tokenizer_save_jinja_files": true,
243
+ "torch_dtype": "torch.bfloat16",
244
+ "train_on_inputs": false,
245
+ "trl": {
246
+ "async_prefetch": false,
247
+ "log_completions": false,
248
+ "mask_truncated_completions": false,
249
+ "ref_model_mixup_alpha": 0.9,
250
+ "ref_model_sync_steps": 64,
251
+ "replay_buffer_size": 0,
252
+ "replay_recompute_logps": true,
253
+ "reroll_max_groups": 1,
254
+ "reroll_start_fraction": 1.0,
255
+ "reward_num_workers": 1,
256
+ "scale_rewards": true,
257
+ "skip_zero_advantage_batches": true,
258
+ "sync_ref_model": false,
259
+ "use_data_producer": false,
260
+ "use_vllm": false,
261
+ "vllm_lora_sync": false,
262
+ "vllm_server_host": "0.0.0.0",
263
+ "vllm_server_port": 8000
264
+ },
265
+ "trust_remote_code": false,
266
+ "use_otel_metrics": false,
267
+ "use_ray": false,
268
+ "val_set_size": 0.0,
269
+ "vllm": {
270
+ "device": "auto",
271
+ "dtype": "auto",
272
+ "gpu_memory_utilization": 0.9,
273
+ "host": "0.0.0.0",
274
+ "port": 8000
275
+ },
276
+ "warmup_steps": 10,
277
+ "weight_decay": 0.01,
278
+ "world_size": 8
279
+ }
280
+ [2026-09-02 18:00:49,856] [INFO] [axolotl.utils.data.sft] [RANK:4] Loading raw datasets...
281
+
282
+ [2026-09-02 18:00:52,031] [INFO] [axolotl.utils.data.wrappers] [RANK:4] Loading dataset: /workspace/final_v1/gemma3_27b_190m/control/data/dolci.jsonl with base_type: chat_template and prompt_style: None
283
+
284
+
285
+
286
+
287
+
288
+ warnings.warn(
289
+
290
+ [2026-09-02 18:02:27,859] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
291
+ warnings.warn(
292
+
293
+ [2026-09-02 18:02:27,861] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
294
+ warnings.warn(
295
+
296
+ [2026-09-02 18:02:27,887] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
297
+ warnings.warn(
298
+
299
+ [2026-09-02 18:02:27,891] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
300
+ warnings.warn(
301
+
302
+ [2026-09-02 18:02:27,894] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
303
+ warnings.warn(
304
+
305
+ [2026-09-02 18:02:27,909] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
306
+ warnings.warn(
307
+
308
+ [2026-09-02 18:02:27,909] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
309
+ warnings.warn(
310
+
311
+ [2026-09-02 18:02:27,910] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
312
+ warnings.warn(
313
+
314
+ [2026-09-02 18:02:27,913] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
315
+ warnings.warn(
316
+
317
+ [2026-09-02 18:02:27,913] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
318
+ warnings.warn(
319
+
320
+ [2026-09-02 18:02:27,914] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
321
+ warnings.warn(
322
+
323
+ [2026-09-02 18:02:27,915] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
324
+ warnings.warn(
325
+
326
+ [2026-09-02 18:02:27,917] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
327
+ warnings.warn(
328
+
329
+ [2026-09-02 18:02:27,923] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
330
+ warnings.warn(
331
+
332
+ [2026-09-02 18:02:27,925] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
333
+ warnings.warn(
334
+
335
+ [2026-09-02 18:02:27,931] [WARNING] [py.warnings] /usr/local/lib/python3.12/dist-packages/requests/__init__.py:113: RequestsDependencyWarning: urllib3 (2.7.0) or chardet (6.0.0.post1)/charset_normalizer (3.4.3) doesn't match a supported version!
336
+ warnings.warn(
337
+
338
+ W0902 18:02:31.723000 411788 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
339
+ W0902 18:02:31.744000 411788 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
340
+ W0902 18:02:33.209000 411793 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
341
+ W0902 18:02:33.228000 411793 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
342
+ W0902 18:02:34.861000 411781 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
343
+ W0902 18:02:34.881000 411781 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
344
+ W0902 18:02:35.289000 411785 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
345
+ W0902 18:02:35.308000 411785 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
346
+ W0902 18:02:35.339000 411780 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
347
+ W0902 18:02:35.358000 411780 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
348
+ W0902 18:02:35.381000 411794 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
349
+ W0902 18:02:35.401000 411794 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
350
+ W0902 18:02:35.535000 411791 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
351
+ W0902 18:02:35.554000 411791 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
352
+ W0902 18:02:35.602000 411783 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
353
+ W0902 18:02:35.622000 411783 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
354
+ W0902 18:02:35.631000 411795 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
355
+ W0902 18:02:35.650000 411795 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
356
+ W0902 18:02:35.653000 411786 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
357
+ W0902 18:02:35.672000 411786 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
358
+ W0902 18:02:35.675000 411790 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
359
+ W0902 18:02:35.688000 411799 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
360
+ W0902 18:02:35.694000 411790 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
361
+ W0902 18:02:35.697000 411787 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
362
+ W0902 18:02:35.703000 411784 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
363
+ W0902 18:02:35.708000 411799 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
364
+ W0902 18:02:35.718000 411787 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
365
+ W0902 18:02:35.723000 411784 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
366
+ W0902 18:02:35.750000 411792 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
367
+ W0902 18:02:35.769000 411792 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
368
+ W0902 18:02:35.830000 411789 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
369
+ W0902 18:02:35.852000 411789 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
370
+ W0902 18:02:35.920000 411782 torch/utils/_pytree.py:630] <enum 'KernelPreference'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
371
+ W0902 18:02:35.941000 411782 torch/utils/_pytree.py:630] <enum 'ScaleCalculationMode'> is an Enum subclass and is now natively supported by torch.compile as an opaque value type. Calling register_constant() on Enum subclasses is deprecated and will be an error in a future release.
372
+
373
+ [2026-09-02 18:02:40,771] [INFO] [axolotl.utils.data.shared] Loading prepared dataset from disk at /workspace/final_v1/gemma3_27b_190m/control/dolci/prepared/7bbc70d7be6cf8cbe709e6f388e69328...
374
+ [2026-09-02 18:02:49,699] [INFO] [axolotl.utils.samplers.multipack] gather_len_batches: [7162, 7162, 7162, 7163, 7163, 7162, 7163, 7162]
375
+ [2026-09-02 18:02:50,364] [INFO] [axolotl.utils.trainer] sample_packing_eff_est across ranks: [0.9992387294769287, 0.9993782639503479, 0.9993782639503479, 0.9992387294769287, 0.9992387294769287, 0.9992387294769287, 0.9993782639503479, 0.9992387294769287]
376
+ [2026-09-02 18:02:50,371] [INFO] [axolotl.utils.data.sft] Maximum number of steps set at 48
377
+ [2026-09-02 18:02:57,680] [INFO] [axolotl.monkeypatch.attention.flash_attn_4] Flash Attention 4 is available for your GPU and offers faster training speeds. To enable: pip install flash-attn-4
378
+ [2026-09-02 18:02:57,681] [INFO] [axolotl.loaders.patch_manager] Applying multipack dataloader patch for sample packing...
379
+
380
+
381
+
382
+
383
+
384
+
385
+
386
+ [2026-09-02 18:03:01,440] [INFO] [axolotl.loaders.model] Converting modules to torch.bfloat16
387
+
388
+ [transformers] When using FSDP full shard, instead of using `gradient_checkpointing` in TrainingArguments, please use `activation_checkpointing` in `fsdp_config`. The former introduces a redundant AllGather operation in backward pass. Reference: https://github.com/huggingface/transformers/issues/30404
389
+ [2026-09-02 18:03:02,455] [WARNING] [accelerate.utils.dataclasses] sync_module_states is obsolete in FSDP2, as it is not needed anymore.Setting sync_module_states to None.
390
+ [2026-09-02 18:03:02,550] [INFO] [axolotl.train] Pre-saving tokenizer to /workspace/final_v1/gemma3_27b_190m/control/dolci/checkpoints...
391
+ [2026-09-02 18:03:02,845] [INFO] [axolotl.train] Pre-saving model config to /workspace/final_v1/gemma3_27b_190m/control/dolci/checkpoints...
392
+ [2026-09-02 18:03:02,849] [INFO] [axolotl.train] Pre-saving processor to /workspace/final_v1/gemma3_27b_190m/control/dolci/checkpoints...
393
+
394
+ [2026-09-02 18:03:03,106] [INFO] [axolotl.train] Starting trainer...
395
+ [transformers] When using FSDP full shard, instead of using `gradient_checkpointing` in TrainingArguments, please use `activation_checkpointing` in `fsdp_config`. The former introduces a redundant AllGather operation in backward pass. Reference: https://github.com/huggingface/transformers/issues/30404
396
+ [2026-09-02 18:03:20,379] [WARNING] [accelerate.utils.dataclasses] sync_module_states is obsolete in FSDP2, as it is not needed anymore.Setting sync_module_states to None.
397
+ [transformers] When using FSDP full shard, instead of using `gradient_checkpointing` in TrainingArguments, please use `activation_checkpointing` in `fsdp_config`. The former introduces a redundant AllGather operation in backward pass. Reference: https://github.com/huggingface/transformers/issues/30404
398
+ [2026-09-02 18:03:21,524] [WARNING] [accelerate.utils.dataclasses] sync_module_states is obsolete in FSDP2, as it is not needed anymore.Setting sync_module_states to None.
399
+ [transformers] When using FSDP full shard, instead of using `gradient_checkpointing` in TrainingArguments, please use `activation_checkpointing` in `fsdp_config`. The former introduces a redundant AllGather operation in backward pass. Reference: https://github.com/huggingface/transformers/issues/30404
400
+ [2026-09-02 18:03:21,897] [WARNING] [accelerate.utils.dataclasses] sync_module_states is obsolete in FSDP2, as it is not needed anymore.Setting sync_module_states to None.
401
+ [transformers] When using FSDP full shard, instead of using `gradient_checkpointing` in TrainingArguments, please use `activation_checkpointing` in `fsdp_config`. The former introduces a redundant AllGather operation in backward pass. Reference: https://github.com/huggingface/transformers/issues/30404
402
+ [2026-09-02 18:03:22,350] [WARNING] [accelerate.utils.dataclasses] sync_module_states is obsolete in FSDP2, as it is not needed anymore.Setting sync_module_states to None.
403
+ [transformers] When using FSDP full shard, instead of using `gradient_checkpointing` in TrainingArguments, please use `activation_checkpointing` in `fsdp_config`. The former introduces a redundant AllGather operation in backward pass. Reference: https://github.com/huggingface/transformers/issues/30404
404
+ [2026-09-02 18:03:22,855] [WARNING] [accelerate.utils.dataclasses] sync_module_states is obsolete in FSDP2, as it is not needed anymore.Setting sync_module_states to None.
405
+ [transformers] When using FSDP full shard, instead of using `gradient_checkpointing` in TrainingArguments, please use `activation_checkpointing` in `fsdp_config`. The former introduces a redundant AllGather operation in backward pass. Reference: https://github.com/huggingface/transformers/issues/30404
406
+ [2026-09-02 18:03:24,781] [WARNING] [accelerate.utils.dataclasses] sync_module_states is obsolete in FSDP2, as it is not needed anymore.Setting sync_module_states to None.
407
+ [transformers] When using FSDP full shard, instead of using `gradient_checkpointing` in TrainingArguments, please use `activation_checkpointing` in `fsdp_config`. The former introduces a redundant AllGather operation in backward pass. Reference: https://github.com/huggingface/transformers/issues/30404
408
+ [2026-09-02 18:03:25,420] [WARNING] [accelerate.utils.dataclasses] sync_module_states is obsolete in FSDP2, as it is not needed anymore.Setting sync_module_states to None.
409
+ [2026-09-02 18:03:34,712] [INFO] [axolotl.utils.samplers.multipack] gather_len_batches: [7163, 7163, 7163, 7163, 7163, 7163, 7163, 7163]
410
+ [2026-09-02 18:03:35,012] [INFO] [axolotl.monkeypatch.accelerate.fsdp2] Broadcasting full state dict to all ranks...
411
+
412
  0%| | 0/48 [00:00<?, ?it/s][transformers] `use_return_dict` is deprecated! Use `return_dict` instead!
413
+ [transformers] `use_return_dict` is deprecated! Use `return_dict` instead!
414
+ [transformers] `use_return_dict` is deprecated! Use `return_dict` instead!
415
+ [transformers] `use_return_dict` is deprecated! Use `return_dict` instead!
416
+ [transformers] `use_return_dict` is deprecated! Use `return_dict` instead!
417
+ [transformers] `use_return_dict` is deprecated! Use `return_dict` instead!
418
+ [transformers] `use_return_dict` is deprecated! Use `return_dict` instead!
419
+ [transformers] `use_return_dict` is deprecated! Use `return_dict` instead!
420
+
421
  2%|▏ | 1/48 [01:53<1:29:11, 113.87s/it]
422
 
423
+
424
  2%|▏ | 1/48 [01:53<1:29:11, 113.87s/it]
425
  4%|▍ | 2/48 [03:41<1:24:24, 110.11s/it]
426
 
427
+
428
  4%|▍ | 2/48 [03:41<1:24:24, 110.11s/it]
429
  6%|▋ | 3/48 [05:28<1:21:34, 108.78s/it]
430
 
431
+
432
  6%|▋ | 3/48 [05:28<1:21:34, 108.78s/it]
433
  8%|▊ | 4/48 [07:15<1:19:07, 107.89s/it]
434
 
435
+
436
  8%|▊ | 4/48 [07:15<1:19:07, 107.89s/it]
437
  10%|█ | 5/48 [09:01<1:16:54, 107.32s/it]
438
 
439
+
440
  10%|█ | 5/48 [09:01<1:16:54, 107.32s/it]
441
  12%|█▎ | 6/48 [10:47<1:14:51, 106.95s/it]
442
 
443
+
444
  12%|█▎ | 6/48 [10:47<1:14:51, 106.95s/it]
445
  15%|█▍ | 7/48 [12:33<1:12:55, 106.72s/it]
446
 
447
+
448
  15%|█▍ | 7/48 [12:33<1:12:55, 106.72s/it]
449
  17%|█▋ | 8/48 [14:19<1:11:00, 106.51s/it]
450
 
451
+
452
  17%|█▋ | 8/48 [14:19<1:11:00, 106.51s/it]
453
  19%|█▉ | 9/48 [16:06<1:09:11, 106.46s/it]
454
 
455
+
456
  19%|█▉ | 9/48 [16:06<1:09:11, 106.46s/it]
457
  21%|██ | 10/48 [17:52<1:07:23, 106.40s/it]
458
 
459
+
460
  21%|██ | 10/48 [17:52<1:07:23, 106.40s/it]
461
  23%|██▎ | 11/48 [19:38<1:05:35, 106.36s/it]
462
 
463
+
464
  23%|██▎ | 11/48 [19:38<1:05:35, 106.36s/it]
465
  25%|██▌ | 12/48 [21:25<1:03:48, 106.35s/it]
466
 
467
+
468
  25%|██▌ | 12/48 [21:25<1:03:48, 106.35s/it]
469
  27%|██▋ | 13/48 [23:11<1:02:02, 106.37s/it]
470
 
471
+
472
  27%|██▋ | 13/48 [23:11<1:02:02, 106.37s/it]
473
  29%|██▉ | 14/48 [24:58<1:00:20, 106.48s/it]
474
 
475
+
476
  29%|██▉ | 14/48 [24:58<1:00:20, 106.48s/it]
477
  31%|███▏ | 15/48 [26:44<58:33, 106.46s/it]
478
 
479
+
480
  31%|███▏ | 15/48 [26:44<58:33, 106.46s/it]
481
  33%|███▎ | 16/48 [28:30<56:44, 106.39s/it]
482
 
483
+
484
  33%|███▎ | 16/48 [28:30<56:44, 106.39s/it]
485
  35%|███▌ | 17/48 [30:17<54:57, 106.37s/it]
486
 
487
+
488
  35%|███▌ | 17/48 [30:17<54:57, 106.37s/it]
489
  38%|███▊ | 18/48 [32:03<53:09, 106.33s/it]
490
 
491
+
492
  38%|███▊ | 18/48 [32:03<53:09, 106.33s/it]
493
  40%|███▉ | 19/48 [33:49<51:22, 106.31s/it]
494
 
495
+
496
  40%|███▉ | 19/48 [33:49<51:22, 106.31s/it]
497
  42%|████▏ | 20/48 [35:36<49:37, 106.32s/it]
498
 
499
+
500
  42%|████▏ | 20/48 [35:36<49:37, 106.32s/it]
501
  44%|████▍ | 21/48 [37:22<47:50, 106.32s/it]
502
 
503
+
504
  44%|████▍ | 21/48 [37:22<47:50, 106.32s/it]
505
  46%|████▌ | 22/48 [39:08<46:04, 106.32s/it]
506
 
507
+
508
  46%|████▌ | 22/48 [39:08<46:04, 106.32s/it]
509
  48%|████▊ | 23/48 [40:54<44:17, 106.30s/it]
510
 
511
+
512
  48%|████▊ | 23/48 [40:54<44:17, 106.30s/it]
513
  50%|█████ | 24/48 [42:41<42:35, 106.48s/it]
514
 
515
+
516
  50%|█████ | 24/48 [42:41<42:35, 106.48s/it]
517
  52%|█████▏ | 25/48 [44:28<40:46, 106.39s/it]
518
 
519
+
520
  52%|█████▏ | 25/48 [44:28<40:46, 106.39s/it]
521
  54%|█████▍ | 26/48 [46:14<39:02, 106.46s/it]
522
 
523
+
524
  54%|█████▍ | 26/48 [46:14<39:02, 106.46s/it]
525
  56%|█████▋ | 27/48 [48:00<37:13, 106.38s/it]
526
 
527
+
528
  56%|█████▋ | 27/48 [48:00<37:13, 106.38s/it]
529
  58%|█████▊ | 28/48 [49:47<35:27, 106.36s/it]
530
 
531
+
532
  58%|█████▊ | 28/48 [49:47<35:27, 106.36s/it]
533
  60%|██████ | 29/48 [51:33<33:40, 106.35s/it]
534
 
535
+
536
  60%|██████ | 29/48 [51:33<33:40, 106.35s/it]
537
  62%|██████▎ | 30/48 [53:19<31:53, 106.33s/it]
538
 
539
+
540
  62%|██████▎ | 30/48 [53:19<31:53, 106.33s/it]
541
  65%|██████▍ | 31/48 [55:06<30:08, 106.40s/it]
542
 
543
+
544
  65%|██████▍ | 31/48 [55:06<30:08, 106.40s/it]
545
  67%|██████▋ | 32/48 [56:52<28:21, 106.33s/it]
546
 
547
+
548
  67%|██████▋ | 32/48 [56:52<28:21, 106.33s/it]
549
  69%|██████▉ | 33/48 [58:38<26:34, 106.29s/it]
550
 
551
+
552
  69%|██████▉ | 33/48 [58:38<26:34, 106.29s/it]
553
  71%|███████ | 34/48 [1:00:25<24:48, 106.30s/it]
554
 
555
+
556
  71%|███████ | 34/48 [1:00:25<24:48, 106.30s/it]
557
  73%|███████▎ | 35/48 [1:02:11<23:01, 106.26s/it]
558
 
559
+
560
  73%|███████▎ | 35/48 [1:02:11<23:01, 106.26s/it]
561
  75%|███████▌ | 36/48 [1:03:57<21:14, 106.23s/it]
562
 
563
+
564
  75%|███████▌ | 36/48 [1:03:57<21:14, 106.23s/it]
565
  77%|███████▋ | 37/48 [1:05:43<19:29, 106.30s/it]
566
 
567
+
568
  77%|███████▋ | 37/48 [1:05:43<19:29, 106.30s/it]
569
  79%|███████▉ | 38/48 [1:07:30<17:43, 106.32s/it]
570
 
571
+
572
  79%|███████▉ | 38/48 [1:07:30<17:43, 106.32s/it]
573
  81%|████████▏ | 39/48 [1:09:16<15:56, 106.28s/it]
574
 
575
+
576
  81%|████████▏ | 39/48 [1:09:16<15:56, 106.28s/it]
577
  83%|████████▎ | 40/48 [1:11:02<14:10, 106.27s/it]
578
 
579
+
580
  83%|████████▎ | 40/48 [1:11:02<14:10, 106.27s/it]
581
  85%|████████▌ | 41/48 [1:12:49<12:24, 106.32s/it]
582
 
583
+
584
  85%|████████▌ | 41/48 [1:12:49<12:24, 106.32s/it]
585
  88%|████████▊ | 42/48 [1:14:35<10:38, 106.36s/it]
586
 
587
+
588
  88%|████████▊ | 42/48 [1:14:35<10:38, 106.36s/it]
589
  90%|████████▉ | 43/48 [1:16:21<08:51, 106.36s/it]
590
 
591
+
592
  90%|████████▉ | 43/48 [1:16:21<08:51, 106.36s/it][2026-09-02 19:20:41,006] [INFO] [axolotl.core.trainers.base] Saving model checkpoint to /workspace/final_v1/gemma3_27b_190m/control/dolci/checkpoints/checkpoint-43
593
+
594
+
595
+
596
+
597
+
598
  92%|█████████▏| 44/48 [1:18:58<08:05, 121.33s/it]
599
 
600
+
601
  92%|█████████▏| 44/48 [1:18:58<08:05, 121.33s/it]
602
  94%|█████████▍| 45/48 [1:20:44<05:50, 116.88s/it]
603
 
604
+
605
  94%|█████████▍| 45/48 [1:20:44<05:50, 116.88s/it]
606
  96%|█████████▌| 46/48 [1:22:31<03:47, 113.75s/it]
607
 
608
+
609
  96%|█████████▌| 46/48 [1:22:31<03:47, 113.75s/it]
610
  98%|█████████▊| 47/48 [1:24:17<01:51, 111.50s/it]
611
 
612
+
613
  98%|█████████▊| 47/48 [1:24:17<01:51, 111.50s/it]
614
 
615
+
616
+
617
+
618
+
619
+
620
+
621
 
622
+
623
+ [2026-09-02 19:30:36,032] [INFO] [axolotl.train] Training completed! Saving trained model to /workspace/final_v1/gemma3_27b_190m/control/dolci/checkpoints.
624
+ [2026-09-02 19:31:06,526] [INFO] [axolotl.core.trainers.base] Saving model checkpoint to /workspace/final_v1/gemma3_27b_190m/control/dolci/checkpoints
625
+
626
+ [2026-09-02 19:31:20,672] [INFO] [axolotl.train] Model successfully saved to /workspace/final_v1/gemma3_27b_190m/control/dolci/checkpoints
gemma3_27b_190m/control/training/dolci/trainer_state.final.json ADDED
@@ -0,0 +1,706 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.8581005586592179,
6
+ "eval_steps": 500,
7
+ "global_step": 48,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.017877094972067038,
14
+ "grad_norm": 3.9375,
15
+ "learning_rate": 0.0,
16
+ "loss": 0.728485107421875,
17
+ "memory/device_reserved (GiB)": 93.97,
18
+ "memory/max_active (GiB)": 82.38,
19
+ "memory/max_allocated (GiB)": 82.38,
20
+ "ppl": 2.07194,
21
+ "step": 1,
22
+ "tokens/total": 2097152,
23
+ "tokens/train_per_sec_per_gpu": 91.32,
24
+ "tokens/trainable": 1329700
25
+ },
26
+ {
27
+ "epoch": 0.035754189944134075,
28
+ "grad_norm": 4.0,
29
+ "learning_rate": 1.0000000000000002e-06,
30
+ "loss": 0.7188262939453125,
31
+ "memory/device_reserved (GiB)": 106.04,
32
+ "memory/max_active (GiB)": 94.96,
33
+ "memory/max_allocated (GiB)": 94.96,
34
+ "ppl": 2.05202,
35
+ "step": 2,
36
+ "tokens/total": 4194304,
37
+ "tokens/train_per_sec_per_gpu": 92.86,
38
+ "tokens/trainable": 2660399
39
+ },
40
+ {
41
+ "epoch": 0.053631284916201116,
42
+ "grad_norm": 3.71875,
43
+ "learning_rate": 2.0000000000000003e-06,
44
+ "loss": 0.7142333984375,
45
+ "memory/device_reserved (GiB)": 106.04,
46
+ "memory/max_active (GiB)": 94.96,
47
+ "memory/max_allocated (GiB)": 94.96,
48
+ "ppl": 2.04262,
49
+ "step": 3,
50
+ "tokens/total": 6291456,
51
+ "tokens/train_per_sec_per_gpu": 94.79,
52
+ "tokens/trainable": 3990552
53
+ },
54
+ {
55
+ "epoch": 0.07150837988826815,
56
+ "grad_norm": 3.03125,
57
+ "learning_rate": 3e-06,
58
+ "loss": 0.6811370849609375,
59
+ "memory/device_reserved (GiB)": 106.04,
60
+ "memory/max_active (GiB)": 94.96,
61
+ "memory/max_allocated (GiB)": 94.96,
62
+ "ppl": 1.97612,
63
+ "step": 4,
64
+ "tokens/total": 8388608,
65
+ "tokens/train_per_sec_per_gpu": 101.07,
66
+ "tokens/trainable": 5326555
67
+ },
68
+ {
69
+ "epoch": 0.0893854748603352,
70
+ "grad_norm": 1.9296875,
71
+ "learning_rate": 4.000000000000001e-06,
72
+ "loss": 0.6489715576171875,
73
+ "memory/device_reserved (GiB)": 106.04,
74
+ "memory/max_active (GiB)": 94.96,
75
+ "memory/max_allocated (GiB)": 94.96,
76
+ "ppl": 1.91357,
77
+ "step": 5,
78
+ "tokens/total": 10485760,
79
+ "tokens/train_per_sec_per_gpu": 96.26,
80
+ "tokens/trainable": 6647029
81
+ },
82
+ {
83
+ "epoch": 0.10726256983240223,
84
+ "grad_norm": 3.078125,
85
+ "learning_rate": 5e-06,
86
+ "loss": 0.631500244140625,
87
+ "memory/device_reserved (GiB)": 106.04,
88
+ "memory/max_active (GiB)": 94.96,
89
+ "memory/max_allocated (GiB)": 94.96,
90
+ "ppl": 1.88043,
91
+ "step": 6,
92
+ "tokens/total": 12582912,
93
+ "tokens/train_per_sec_per_gpu": 104.31,
94
+ "tokens/trainable": 7979759
95
+ },
96
+ {
97
+ "epoch": 0.12513966480446928,
98
+ "grad_norm": 1.9375,
99
+ "learning_rate": 6e-06,
100
+ "loss": 0.61700439453125,
101
+ "memory/device_reserved (GiB)": 106.04,
102
+ "memory/max_active (GiB)": 94.96,
103
+ "memory/max_allocated (GiB)": 94.96,
104
+ "ppl": 1.85337,
105
+ "step": 7,
106
+ "tokens/total": 14680064,
107
+ "tokens/train_per_sec_per_gpu": 100.71,
108
+ "tokens/trainable": 9293508
109
+ },
110
+ {
111
+ "epoch": 0.1430167597765363,
112
+ "grad_norm": 1.9453125,
113
+ "learning_rate": 7e-06,
114
+ "loss": 0.6414642333984375,
115
+ "memory/device_reserved (GiB)": 106.04,
116
+ "memory/max_active (GiB)": 94.96,
117
+ "memory/max_allocated (GiB)": 94.96,
118
+ "ppl": 1.89926,
119
+ "step": 8,
120
+ "tokens/total": 16777216,
121
+ "tokens/train_per_sec_per_gpu": 102.55,
122
+ "tokens/trainable": 10621415
123
+ },
124
+ {
125
+ "epoch": 0.16089385474860335,
126
+ "grad_norm": 1.1328125,
127
+ "learning_rate": 8.000000000000001e-06,
128
+ "loss": 0.5925140380859375,
129
+ "memory/device_reserved (GiB)": 106.04,
130
+ "memory/max_active (GiB)": 94.96,
131
+ "memory/max_allocated (GiB)": 94.96,
132
+ "ppl": 1.80853,
133
+ "step": 9,
134
+ "tokens/total": 18874368,
135
+ "tokens/train_per_sec_per_gpu": 91.03,
136
+ "tokens/trainable": 11958528
137
+ },
138
+ {
139
+ "epoch": 0.1787709497206704,
140
+ "grad_norm": 0.90625,
141
+ "learning_rate": 9e-06,
142
+ "loss": 0.6042022705078125,
143
+ "memory/device_reserved (GiB)": 106.04,
144
+ "memory/max_active (GiB)": 94.96,
145
+ "memory/max_allocated (GiB)": 94.96,
146
+ "ppl": 1.82979,
147
+ "step": 10,
148
+ "tokens/total": 20971520,
149
+ "tokens/train_per_sec_per_gpu": 99.3,
150
+ "tokens/trainable": 13296510
151
+ },
152
+ {
153
+ "epoch": 0.19664804469273742,
154
+ "grad_norm": 0.98046875,
155
+ "learning_rate": 1e-05,
156
+ "loss": 0.5978546142578125,
157
+ "memory/device_reserved (GiB)": 106.04,
158
+ "memory/max_active (GiB)": 94.96,
159
+ "memory/max_allocated (GiB)": 94.96,
160
+ "ppl": 1.81821,
161
+ "step": 11,
162
+ "tokens/total": 23068672,
163
+ "tokens/train_per_sec_per_gpu": 95.76,
164
+ "tokens/trainable": 14614924
165
+ },
166
+ {
167
+ "epoch": 0.21452513966480447,
168
+ "grad_norm": 0.72265625,
169
+ "learning_rate": 9.984630218530014e-06,
170
+ "loss": 0.6161346435546875,
171
+ "memory/device_reserved (GiB)": 106.04,
172
+ "memory/max_active (GiB)": 94.96,
173
+ "memory/max_allocated (GiB)": 94.96,
174
+ "ppl": 1.85176,
175
+ "step": 12,
176
+ "tokens/total": 25165824,
177
+ "tokens/train_per_sec_per_gpu": 96.9,
178
+ "tokens/trainable": 15929267
179
+ },
180
+ {
181
+ "epoch": 0.2324022346368715,
182
+ "grad_norm": 0.6875,
183
+ "learning_rate": 9.938625865312252e-06,
184
+ "loss": 0.594451904296875,
185
+ "memory/device_reserved (GiB)": 106.04,
186
+ "memory/max_active (GiB)": 94.96,
187
+ "memory/max_allocated (GiB)": 94.96,
188
+ "ppl": 1.81204,
189
+ "step": 13,
190
+ "tokens/total": 27262976,
191
+ "tokens/train_per_sec_per_gpu": 100.1,
192
+ "tokens/trainable": 17260686
193
+ },
194
+ {
195
+ "epoch": 0.25027932960893856,
196
+ "grad_norm": 0.56640625,
197
+ "learning_rate": 9.862301196726988e-06,
198
+ "loss": 0.580963134765625,
199
+ "memory/device_reserved (GiB)": 106.04,
200
+ "memory/max_active (GiB)": 94.96,
201
+ "memory/max_allocated (GiB)": 94.96,
202
+ "ppl": 1.78776,
203
+ "step": 14,
204
+ "tokens/total": 29360128,
205
+ "tokens/train_per_sec_per_gpu": 93.48,
206
+ "tokens/trainable": 18579472
207
+ },
208
+ {
209
+ "epoch": 0.2681564245810056,
210
+ "grad_norm": 0.61328125,
211
+ "learning_rate": 9.756177587652857e-06,
212
+ "loss": 0.57794189453125,
213
+ "memory/device_reserved (GiB)": 106.04,
214
+ "memory/max_active (GiB)": 94.96,
215
+ "memory/max_allocated (GiB)": 94.96,
216
+ "ppl": 1.78237,
217
+ "step": 15,
218
+ "tokens/total": 31457280,
219
+ "tokens/train_per_sec_per_gpu": 97.79,
220
+ "tokens/trainable": 19899402
221
+ },
222
+ {
223
+ "epoch": 0.2860335195530726,
224
+ "grad_norm": 0.6328125,
225
+ "learning_rate": 9.620979969947759e-06,
226
+ "loss": 0.5829010009765625,
227
+ "memory/device_reserved (GiB)": 106.04,
228
+ "memory/max_active (GiB)": 94.96,
229
+ "memory/max_allocated (GiB)": 94.96,
230
+ "ppl": 1.79123,
231
+ "step": 16,
232
+ "tokens/total": 33554432,
233
+ "tokens/train_per_sec_per_gpu": 95.65,
234
+ "tokens/trainable": 21221568
235
+ },
236
+ {
237
+ "epoch": 0.3039106145251397,
238
+ "grad_norm": 0.578125,
239
+ "learning_rate": 9.4576318804292e-06,
240
+ "loss": 0.5619354248046875,
241
+ "memory/device_reserved (GiB)": 106.04,
242
+ "memory/max_active (GiB)": 94.96,
243
+ "memory/max_allocated (GiB)": 94.96,
244
+ "ppl": 1.75406,
245
+ "step": 17,
246
+ "tokens/total": 35651584,
247
+ "tokens/train_per_sec_per_gpu": 96.41,
248
+ "tokens/trainable": 22554968
249
+ },
250
+ {
251
+ "epoch": 0.3217877094972067,
252
+ "grad_norm": 0.66015625,
253
+ "learning_rate": 9.267249152181379e-06,
254
+ "loss": 0.56243896484375,
255
+ "memory/device_reserved (GiB)": 106.04,
256
+ "memory/max_active (GiB)": 94.96,
257
+ "memory/max_allocated (GiB)": 94.96,
258
+ "ppl": 1.75495,
259
+ "step": 18,
260
+ "tokens/total": 37748736,
261
+ "tokens/train_per_sec_per_gpu": 98.85,
262
+ "tokens/trainable": 23884800
263
+ },
264
+ {
265
+ "epoch": 0.3396648044692737,
266
+ "grad_norm": 0.466796875,
267
+ "learning_rate": 9.051132292283772e-06,
268
+ "loss": 0.5709991455078125,
269
+ "memory/device_reserved (GiB)": 106.04,
270
+ "memory/max_active (GiB)": 94.96,
271
+ "memory/max_allocated (GiB)": 94.96,
272
+ "ppl": 1.77003,
273
+ "step": 19,
274
+ "tokens/total": 39845888,
275
+ "tokens/train_per_sec_per_gpu": 100.82,
276
+ "tokens/trainable": 25226420
277
+ },
278
+ {
279
+ "epoch": 0.3575418994413408,
280
+ "grad_norm": 0.63671875,
281
+ "learning_rate": 8.810757598029094e-06,
282
+ "loss": 0.56182861328125,
283
+ "memory/device_reserved (GiB)": 106.04,
284
+ "memory/max_active (GiB)": 94.96,
285
+ "memory/max_allocated (GiB)": 94.96,
286
+ "ppl": 1.75388,
287
+ "step": 20,
288
+ "tokens/total": 41943040,
289
+ "tokens/train_per_sec_per_gpu": 94.63,
290
+ "tokens/trainable": 26546660
291
+ },
292
+ {
293
+ "epoch": 0.3754189944134078,
294
+ "grad_norm": 0.6171875,
295
+ "learning_rate": 8.547767072315835e-06,
296
+ "loss": 0.5721893310546875,
297
+ "memory/device_reserved (GiB)": 106.04,
298
+ "memory/max_active (GiB)": 94.96,
299
+ "memory/max_allocated (GiB)": 94.96,
300
+ "ppl": 1.77214,
301
+ "step": 21,
302
+ "tokens/total": 44040192,
303
+ "tokens/train_per_sec_per_gpu": 101.01,
304
+ "tokens/trainable": 27880402
305
+ },
306
+ {
307
+ "epoch": 0.39329608938547483,
308
+ "grad_norm": 0.50390625,
309
+ "learning_rate": 8.263957207103506e-06,
310
+ "loss": 0.5504608154296875,
311
+ "memory/device_reserved (GiB)": 106.04,
312
+ "memory/max_active (GiB)": 94.96,
313
+ "memory/max_allocated (GiB)": 94.96,
314
+ "ppl": 1.73405,
315
+ "step": 22,
316
+ "tokens/total": 46137344,
317
+ "tokens/train_per_sec_per_gpu": 101.33,
318
+ "tokens/trainable": 29204852
319
+ },
320
+ {
321
+ "epoch": 0.4111731843575419,
322
+ "grad_norm": 0.466796875,
323
+ "learning_rate": 7.961266711550922e-06,
324
+ "loss": 0.5583038330078125,
325
+ "memory/device_reserved (GiB)": 106.04,
326
+ "memory/max_active (GiB)": 94.96,
327
+ "memory/max_allocated (GiB)": 94.96,
328
+ "ppl": 1.74771,
329
+ "step": 23,
330
+ "tokens/total": 48234496,
331
+ "tokens/train_per_sec_per_gpu": 93.17,
332
+ "tokens/trainable": 30537088
333
+ },
334
+ {
335
+ "epoch": 0.42905027932960893,
336
+ "grad_norm": 0.47265625,
337
+ "learning_rate": 7.641763268666832e-06,
338
+ "loss": 0.5544891357421875,
339
+ "memory/device_reserved (GiB)": 106.04,
340
+ "memory/max_active (GiB)": 94.96,
341
+ "memory/max_allocated (GiB)": 94.96,
342
+ "ppl": 1.74105,
343
+ "step": 24,
344
+ "tokens/total": 50331648,
345
+ "tokens/train_per_sec_per_gpu": 95.95,
346
+ "tokens/trainable": 31855348
347
+ },
348
+ {
349
+ "epoch": 0.44692737430167595,
350
+ "grad_norm": 0.5859375,
351
+ "learning_rate": 7.307629410938364e-06,
352
+ "loss": 0.579254150390625,
353
+ "memory/device_reserved (GiB)": 106.04,
354
+ "memory/max_active (GiB)": 94.96,
355
+ "memory/max_allocated (GiB)": 94.96,
356
+ "ppl": 1.78471,
357
+ "step": 25,
358
+ "tokens/total": 52428800,
359
+ "tokens/train_per_sec_per_gpu": 94.68,
360
+ "tokens/trainable": 33172616
361
+ },
362
+ {
363
+ "epoch": 0.464804469273743,
364
+ "grad_norm": 0.38671875,
365
+ "learning_rate": 6.961147611421076e-06,
366
+ "loss": 0.546966552734375,
367
+ "memory/device_reserved (GiB)": 106.04,
368
+ "memory/max_active (GiB)": 94.96,
369
+ "memory/max_allocated (GiB)": 94.96,
370
+ "ppl": 1.728,
371
+ "step": 26,
372
+ "tokens/total": 54525952,
373
+ "tokens/train_per_sec_per_gpu": 98.15,
374
+ "tokens/trainable": 34519844
375
+ },
376
+ {
377
+ "epoch": 0.48268156424581005,
378
+ "grad_norm": 2.171875,
379
+ "learning_rate": 6.604684692133597e-06,
380
+ "loss": 0.5683746337890625,
381
+ "memory/device_reserved (GiB)": 106.04,
382
+ "memory/max_active (GiB)": 94.96,
383
+ "memory/max_allocated (GiB)": 94.96,
384
+ "ppl": 1.7654,
385
+ "step": 27,
386
+ "tokens/total": 56623104,
387
+ "tokens/train_per_sec_per_gpu": 99.47,
388
+ "tokens/trainable": 35845616
389
+ },
390
+ {
391
+ "epoch": 0.5005586592178771,
392
+ "grad_norm": 0.458984375,
393
+ "learning_rate": 6.240675656263303e-06,
394
+ "loss": 0.545654296875,
395
+ "memory/device_reserved (GiB)": 106.04,
396
+ "memory/max_active (GiB)": 94.96,
397
+ "memory/max_allocated (GiB)": 94.96,
398
+ "ppl": 1.72574,
399
+ "step": 28,
400
+ "tokens/total": 58712064,
401
+ "tokens/train_per_sec_per_gpu": 94.38,
402
+ "tokens/trainable": 37174332
403
+ },
404
+ {
405
+ "epoch": 0.5184357541899441,
406
+ "grad_norm": 0.462890625,
407
+ "learning_rate": 5.871607054625497e-06,
408
+ "loss": 0.5624542236328125,
409
+ "memory/device_reserved (GiB)": 106.04,
410
+ "memory/max_active (GiB)": 94.96,
411
+ "memory/max_allocated (GiB)": 94.96,
412
+ "ppl": 1.75497,
413
+ "step": 29,
414
+ "tokens/total": 60809216,
415
+ "tokens/train_per_sec_per_gpu": 94.39,
416
+ "tokens/trainable": 38511616
417
+ },
418
+ {
419
+ "epoch": 0.5363128491620112,
420
+ "grad_norm": 0.50390625,
421
+ "learning_rate": 5.500000000000001e-06,
422
+ "loss": 0.560760498046875,
423
+ "memory/device_reserved (GiB)": 106.04,
424
+ "memory/max_active (GiB)": 94.96,
425
+ "memory/max_allocated (GiB)": 94.96,
426
+ "ppl": 1.752,
427
+ "step": 30,
428
+ "tokens/total": 62906368,
429
+ "tokens/train_per_sec_per_gpu": 97.59,
430
+ "tokens/trainable": 39829600
431
+ },
432
+ {
433
+ "epoch": 0.5541899441340782,
434
+ "grad_norm": 0.6171875,
435
+ "learning_rate": 5.1283929453745055e-06,
436
+ "loss": 0.547760009765625,
437
+ "memory/device_reserved (GiB)": 106.04,
438
+ "memory/max_active (GiB)": 94.96,
439
+ "memory/max_allocated (GiB)": 94.96,
440
+ "ppl": 1.72937,
441
+ "step": 31,
442
+ "tokens/total": 65003520,
443
+ "tokens/train_per_sec_per_gpu": 100.52,
444
+ "tokens/trainable": 41157812
445
+ },
446
+ {
447
+ "epoch": 0.5720670391061452,
448
+ "grad_norm": 0.36328125,
449
+ "learning_rate": 4.759324343736698e-06,
450
+ "loss": 0.5498428344726562,
451
+ "memory/device_reserved (GiB)": 106.04,
452
+ "memory/max_active (GiB)": 94.96,
453
+ "memory/max_allocated (GiB)": 94.96,
454
+ "ppl": 1.73298,
455
+ "step": 32,
456
+ "tokens/total": 67100672,
457
+ "tokens/train_per_sec_per_gpu": 98.59,
458
+ "tokens/trainable": 42469988
459
+ },
460
+ {
461
+ "epoch": 0.5899441340782123,
462
+ "grad_norm": 0.369140625,
463
+ "learning_rate": 4.395315307866404e-06,
464
+ "loss": 0.5531005859375,
465
+ "memory/device_reserved (GiB)": 106.04,
466
+ "memory/max_active (GiB)": 94.96,
467
+ "memory/max_allocated (GiB)": 94.96,
468
+ "ppl": 1.73864,
469
+ "step": 33,
470
+ "tokens/total": 69197824,
471
+ "tokens/train_per_sec_per_gpu": 100.59,
472
+ "tokens/trainable": 43790036
473
+ },
474
+ {
475
+ "epoch": 0.6078212290502794,
476
+ "grad_norm": 0.470703125,
477
+ "learning_rate": 4.038852388578925e-06,
478
+ "loss": 0.558135986328125,
479
+ "memory/device_reserved (GiB)": 106.04,
480
+ "memory/max_active (GiB)": 94.96,
481
+ "memory/max_allocated (GiB)": 94.96,
482
+ "ppl": 1.74741,
483
+ "step": 34,
484
+ "tokens/total": 71294976,
485
+ "tokens/train_per_sec_per_gpu": 103.93,
486
+ "tokens/trainable": 45114736
487
+ },
488
+ {
489
+ "epoch": 0.6256983240223464,
490
+ "grad_norm": 0.5078125,
491
+ "learning_rate": 3.692370589061639e-06,
492
+ "loss": 0.585052490234375,
493
+ "memory/device_reserved (GiB)": 106.04,
494
+ "memory/max_active (GiB)": 94.96,
495
+ "memory/max_allocated (GiB)": 94.96,
496
+ "ppl": 1.79509,
497
+ "step": 35,
498
+ "tokens/total": 73392128,
499
+ "tokens/train_per_sec_per_gpu": 93.83,
500
+ "tokens/trainable": 46440072
501
+ },
502
+ {
503
+ "epoch": 0.6435754189944134,
504
+ "grad_norm": 0.345703125,
505
+ "learning_rate": 3.3582367313331692e-06,
506
+ "loss": 0.5509033203125,
507
+ "memory/device_reserved (GiB)": 106.04,
508
+ "memory/max_active (GiB)": 94.96,
509
+ "memory/max_allocated (GiB)": 94.96,
510
+ "ppl": 1.73482,
511
+ "step": 36,
512
+ "tokens/total": 75489280,
513
+ "tokens/train_per_sec_per_gpu": 100.58,
514
+ "tokens/trainable": 47777988
515
+ },
516
+ {
517
+ "epoch": 0.6614525139664804,
518
+ "grad_norm": 0.37890625,
519
+ "learning_rate": 3.0387332884490806e-06,
520
+ "loss": 0.5636444091796875,
521
+ "memory/device_reserved (GiB)": 106.04,
522
+ "memory/max_active (GiB)": 94.96,
523
+ "memory/max_allocated (GiB)": 94.96,
524
+ "ppl": 1.75706,
525
+ "step": 37,
526
+ "tokens/total": 77586432,
527
+ "tokens/train_per_sec_per_gpu": 90.45,
528
+ "tokens/trainable": 49117048
529
+ },
530
+ {
531
+ "epoch": 0.6793296089385474,
532
+ "grad_norm": 0.83984375,
533
+ "learning_rate": 2.736042792896495e-06,
534
+ "loss": 0.5464935302734375,
535
+ "memory/device_reserved (GiB)": 106.04,
536
+ "memory/max_active (GiB)": 94.96,
537
+ "memory/max_allocated (GiB)": 94.96,
538
+ "ppl": 1.72719,
539
+ "step": 38,
540
+ "tokens/total": 79683584,
541
+ "tokens/train_per_sec_per_gpu": 98.8,
542
+ "tokens/trainable": 50446456
543
+ },
544
+ {
545
+ "epoch": 0.6972067039106146,
546
+ "grad_norm": 0.375,
547
+ "learning_rate": 2.4522329276841664e-06,
548
+ "loss": 0.5666427612304688,
549
+ "memory/device_reserved (GiB)": 106.04,
550
+ "memory/max_active (GiB)": 94.96,
551
+ "memory/max_allocated (GiB)": 94.96,
552
+ "ppl": 1.76234,
553
+ "step": 39,
554
+ "tokens/total": 81780736,
555
+ "tokens/train_per_sec_per_gpu": 102.08,
556
+ "tokens/trainable": 51767848
557
+ },
558
+ {
559
+ "epoch": 0.7150837988826816,
560
+ "grad_norm": 0.7265625,
561
+ "learning_rate": 2.189242401970908e-06,
562
+ "loss": 0.5714263916015625,
563
+ "memory/device_reserved (GiB)": 106.04,
564
+ "memory/max_active (GiB)": 94.96,
565
+ "memory/max_allocated (GiB)": 94.96,
566
+ "ppl": 1.77079,
567
+ "step": 40,
568
+ "tokens/total": 83877888,
569
+ "tokens/train_per_sec_per_gpu": 101.79,
570
+ "tokens/trainable": 53087960
571
+ },
572
+ {
573
+ "epoch": 0.7329608938547486,
574
+ "grad_norm": 0.5625,
575
+ "learning_rate": 1.94886770771623e-06,
576
+ "loss": 0.5694427490234375,
577
+ "memory/device_reserved (GiB)": 106.04,
578
+ "memory/max_active (GiB)": 94.96,
579
+ "memory/max_allocated (GiB)": 94.96,
580
+ "ppl": 1.76728,
581
+ "step": 41,
582
+ "tokens/total": 85975040,
583
+ "tokens/train_per_sec_per_gpu": 96.12,
584
+ "tokens/trainable": 54401056
585
+ },
586
+ {
587
+ "epoch": 0.7508379888268156,
588
+ "grad_norm": 0.400390625,
589
+ "learning_rate": 1.7327508478186216e-06,
590
+ "loss": 0.587646484375,
591
+ "memory/device_reserved (GiB)": 106.04,
592
+ "memory/max_active (GiB)": 94.96,
593
+ "memory/max_allocated (GiB)": 94.96,
594
+ "ppl": 1.79975,
595
+ "step": 42,
596
+ "tokens/total": 88072192,
597
+ "tokens/train_per_sec_per_gpu": 105.92,
598
+ "tokens/trainable": 55727476
599
+ },
600
+ {
601
+ "epoch": 0.7687150837988826,
602
+ "grad_norm": 0.80859375,
603
+ "learning_rate": 1.5423681195707997e-06,
604
+ "loss": 0.564697265625,
605
+ "memory/device_reserved (GiB)": 106.04,
606
+ "memory/max_active (GiB)": 94.96,
607
+ "memory/max_allocated (GiB)": 94.96,
608
+ "ppl": 1.75892,
609
+ "step": 43,
610
+ "tokens/total": 90169344,
611
+ "tokens/train_per_sec_per_gpu": 92.05,
612
+ "tokens/trainable": 57048484
613
+ },
614
+ {
615
+ "epoch": 0.7865921787709497,
616
+ "grad_norm": 0.79296875,
617
+ "learning_rate": 1.3790200300522413e-06,
618
+ "loss": 0.551666259765625,
619
+ "memory/device_reserved (GiB)": 106.08,
620
+ "memory/max_active (GiB)": 94.96,
621
+ "memory/max_allocated (GiB)": 94.96,
622
+ "ppl": 1.73614,
623
+ "step": 44,
624
+ "tokens/total": 92266496,
625
+ "tokens/train_per_sec_per_gpu": 96.59,
626
+ "tokens/trainable": 58364156
627
+ },
628
+ {
629
+ "epoch": 0.8044692737430168,
630
+ "grad_norm": 0.359375,
631
+ "learning_rate": 1.2438224123471442e-06,
632
+ "loss": 0.550506591796875,
633
+ "memory/device_reserved (GiB)": 106.14,
634
+ "memory/max_active (GiB)": 94.96,
635
+ "memory/max_allocated (GiB)": 94.96,
636
+ "ppl": 1.73413,
637
+ "step": 45,
638
+ "tokens/total": 94363648,
639
+ "tokens/train_per_sec_per_gpu": 101.64,
640
+ "tokens/trainable": 59695504
641
+ },
642
+ {
643
+ "epoch": 0.8223463687150838,
644
+ "grad_norm": 0.40625,
645
+ "learning_rate": 1.1376988032730135e-06,
646
+ "loss": 0.5552978515625,
647
+ "memory/device_reserved (GiB)": 106.14,
648
+ "memory/max_active (GiB)": 94.96,
649
+ "memory/max_allocated (GiB)": 94.96,
650
+ "ppl": 1.74246,
651
+ "step": 46,
652
+ "tokens/total": 96460800,
653
+ "tokens/train_per_sec_per_gpu": 104.86,
654
+ "tokens/trainable": 61030276
655
+ },
656
+ {
657
+ "epoch": 0.8402234636871508,
658
+ "grad_norm": 5.40625,
659
+ "learning_rate": 1.0613741346877498e-06,
660
+ "loss": 0.5300750732421875,
661
+ "memory/device_reserved (GiB)": 106.14,
662
+ "memory/max_active (GiB)": 94.96,
663
+ "memory/max_allocated (GiB)": 94.96,
664
+ "ppl": 1.69906,
665
+ "step": 47,
666
+ "tokens/total": 98557952,
667
+ "tokens/train_per_sec_per_gpu": 100.34,
668
+ "tokens/trainable": 62361588
669
+ },
670
+ {
671
+ "epoch": 0.8581005586592179,
672
+ "grad_norm": 0.51953125,
673
+ "learning_rate": 1.0153697814699858e-06,
674
+ "loss": 0.541046142578125,
675
+ "memory/device_reserved (GiB)": 106.14,
676
+ "memory/max_active (GiB)": 94.96,
677
+ "memory/max_allocated (GiB)": 94.96,
678
+ "ppl": 1.7178,
679
+ "step": 48,
680
+ "tokens/total": 100655104,
681
+ "tokens/train_per_sec_per_gpu": 98.26,
682
+ "tokens/trainable": 63680620
683
+ }
684
+ ],
685
+ "logging_steps": 1,
686
+ "max_steps": 48,
687
+ "num_input_tokens_seen": 0,
688
+ "num_train_epochs": 1,
689
+ "save_steps": 500,
690
+ "stateful_callbacks": {
691
+ "TrainerControl": {
692
+ "args": {
693
+ "should_epoch_stop": false,
694
+ "should_evaluate": false,
695
+ "should_log": false,
696
+ "should_save": true,
697
+ "should_training_stop": true
698
+ },
699
+ "attributes": {}
700
+ }
701
+ },
702
+ "total_flos": 1.5713101328014115e+19,
703
+ "train_batch_size": 2,
704
+ "trial_name": null,
705
+ "trial_params": null
706
+ }
gemma3_27b_190m/control/training/dolci/training_provenance.json ADDED
@@ -0,0 +1,128 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "status": "complete",
4
+ "stage": "sft_dolci_dispatch_final_v1_control_gemma3_27b",
5
+ "kind": "sft",
6
+ "created_at": "2026-09-02T18:00:27+00:00",
7
+ "resolved_config": {
8
+ "base_model": "/workspace/final_v1/gemma3_27b_190m/control/midtrain/checkpoints/checkpoint-1449",
9
+ "base_model_config": "unsloth/gemma-3-27b-pt",
10
+ "revision_of_model": "eb493e07419db4938e915c619689bb513181aebb",
11
+ "trust_remote_code": false,
12
+ "plugins": [
13
+ "axolotl.integrations.liger.LigerPlugin",
14
+ "scimt.train.axolotl_plugins.CheckpointSchedulePlugin"
15
+ ],
16
+ "liger_fused_linear_cross_entropy": true,
17
+ "liger_rope": true,
18
+ "liger_rms_norm": true,
19
+ "liger_glu_activation": true,
20
+ "datasets": [
21
+ {
22
+ "path": "/workspace/final_v1/gemma3_27b_190m/control/data/dolci.jsonl",
23
+ "type": "chat_template",
24
+ "field_messages": "messages"
25
+ }
26
+ ],
27
+ "eot_tokens": [
28
+ "<end_of_turn>"
29
+ ],
30
+ "chat_template": "jinja",
31
+ "chat_template_jinja": "/workspace/scimt/src/scimt/train/stages/assets/gemma3_chat_template.jinja",
32
+ "dataset_prepared_path": "/workspace/final_v1/gemma3_27b_190m/control/dolci/prepared",
33
+ "dataset_processes": 16,
34
+ "sequence_len": 8192,
35
+ "sample_packing": true,
36
+ "pad_to_sequence_len": true,
37
+ "bf16": true,
38
+ "tf32": true,
39
+ "flash_attention": true,
40
+ "gradient_checkpointing": true,
41
+ "micro_batch_size": 2,
42
+ "gradient_accumulation_steps": 16,
43
+ "num_epochs": 1,
44
+ "max_steps": 48,
45
+ "optimizer": "adamw_torch_fused",
46
+ "learning_rate": 1e-05,
47
+ "weight_decay": 0.01,
48
+ "max_grad_norm": 1.0,
49
+ "lr_scheduler": "cosine",
50
+ "cosine_min_lr_ratio": 0.1,
51
+ "warmup_steps": 10,
52
+ "train_on_inputs": false,
53
+ "fsdp_version": 2,
54
+ "fsdp_config": {
55
+ "offload_params": false,
56
+ "cpu_ram_efficient_loading": true,
57
+ "auto_wrap_policy": "TRANSFORMER_BASED_WRAP",
58
+ "transformer_layer_cls_to_wrap": "Gemma3DecoderLayer",
59
+ "state_dict_type": "FULL_STATE_DICT",
60
+ "reshard_after_forward": true
61
+ },
62
+ "logging_steps": 1,
63
+ "save_strategy": "no",
64
+ "save_only_model": true,
65
+ "save_total_limit": 2,
66
+ "checkpoint_schedule": [
67
+ 43,
68
+ 48
69
+ ],
70
+ "seed": 42,
71
+ "output_dir": "/workspace/final_v1/gemma3_27b_190m/control/dolci/checkpoints"
72
+ },
73
+ "resolved_config_path": "/workspace/final_v1/gemma3_27b_190m/control/dolci/axolotl.yaml",
74
+ "dataset": {
75
+ "path": "/workspace/final_v1/gemma3_27b_190m/control/data/dolci.jsonl",
76
+ "exists": true,
77
+ "size_bytes": 481685675,
78
+ "sha256": "8b73cb854c8d565c1381ec27d00eaadc2f4812a0e1e2779e546432ee91f84a20",
79
+ "nonempty_rows": 206913,
80
+ "ordered_example_sha256": "a78bf276dcc4e10eb73dd4db1b9b03f698dad2f5cbc9d1f18a3af3370c061f14",
81
+ "example_manifest": "/workspace/final_v1/gemma3_27b_190m/control/dolci/training_examples.jsonl",
82
+ "example_manifest_sha256": "10f0e4d89501a45ec63246f96d86bf53f67a0aab7031520dd294243cb472c9c3"
83
+ },
84
+ "schedule": {
85
+ "learning_rate": 1e-05,
86
+ "lr_scheduler": "cosine",
87
+ "warmup_steps": 10,
88
+ "cosine_min_lr_ratio": 0.1
89
+ },
90
+ "step_plan": {
91
+ "raw_dataset_rows": 206913,
92
+ "micro_batch_size": 2,
93
+ "gradient_accumulation_steps": 16,
94
+ "world_size_at_render": 1,
95
+ "effective_global_batch_size": 32,
96
+ "num_epochs": 1,
97
+ "planned_optimizer_steps_before_length_filter": 6467,
98
+ "max_steps_override": 48,
99
+ "logging_steps": 1,
100
+ "save_strategy": "no",
101
+ "save_steps": null,
102
+ "save_total_limit": 2
103
+ },
104
+ "seed": 42,
105
+ "completed_at": "2026-09-02T19:31:25+00:00",
106
+ "resolved_config_sha256": "36eb78601a8c7eb4d58aecdba838024a97560ac43469ac59bb29008d5e14dc94",
107
+ "actual": {
108
+ "global_step": 48,
109
+ "max_steps": 48,
110
+ "num_train_epochs": 1,
111
+ "final_epoch": 0.8581005586592179,
112
+ "train_batch_size": 2,
113
+ "num_input_tokens_seen": 0,
114
+ "total_flos": 1.5713101328014115e+19,
115
+ "checkpoint_steps": [
116
+ 43,
117
+ 48
118
+ ],
119
+ "trainer_state_source": "/workspace/final_v1/gemma3_27b_190m/control/dolci/checkpoints/checkpoint-48/trainer_state.json",
120
+ "trainer_state_snapshot": "/workspace/final_v1/gemma3_27b_190m/control/dolci/trainer_state.final.json",
121
+ "trainer_state_sha256": "b525d306543b032ded41f1cd3d09735a0fb4a0c73b73a6aa1929b9e277a7057e",
122
+ "trace_rows": 48,
123
+ "trace_path": "/workspace/final_v1/gemma3_27b_190m/control/dolci/training_trace.jsonl",
124
+ "trace_sha256": "9edf28a39c3ae5d2592e9eb5da3efb073388be7c2d9e6a802c0c4612810ea710",
125
+ "first_learning_rate": 0.0,
126
+ "last_learning_rate": 1.0153697814699858e-06
127
+ }
128
+ }
gemma3_27b_190m/control/training/dolci/training_trace.jsonl ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"epoch": 0.017877094972067038, "grad_norm": 3.9375, "learning_rate": 0.0, "loss": 0.728485107421875, "memory/device_reserved (GiB)": 93.97, "memory/max_active (GiB)": 82.38, "memory/max_allocated (GiB)": 82.38, "ppl": 2.07194, "step": 1, "tokens/total": 2097152, "tokens/train_per_sec_per_gpu": 91.32, "tokens/trainable": 1329700}
2
+ {"epoch": 0.035754189944134075, "grad_norm": 4.0, "learning_rate": 1.0000000000000002e-06, "loss": 0.7188262939453125, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 2.05202, "step": 2, "tokens/total": 4194304, "tokens/train_per_sec_per_gpu": 92.86, "tokens/trainable": 2660399}
3
+ {"epoch": 0.053631284916201116, "grad_norm": 3.71875, "learning_rate": 2.0000000000000003e-06, "loss": 0.7142333984375, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 2.04262, "step": 3, "tokens/total": 6291456, "tokens/train_per_sec_per_gpu": 94.79, "tokens/trainable": 3990552}
4
+ {"epoch": 0.07150837988826815, "grad_norm": 3.03125, "learning_rate": 3e-06, "loss": 0.6811370849609375, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.97612, "step": 4, "tokens/total": 8388608, "tokens/train_per_sec_per_gpu": 101.07, "tokens/trainable": 5326555}
5
+ {"epoch": 0.0893854748603352, "grad_norm": 1.9296875, "learning_rate": 4.000000000000001e-06, "loss": 0.6489715576171875, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.91357, "step": 5, "tokens/total": 10485760, "tokens/train_per_sec_per_gpu": 96.26, "tokens/trainable": 6647029}
6
+ {"epoch": 0.10726256983240223, "grad_norm": 3.078125, "learning_rate": 5e-06, "loss": 0.631500244140625, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.88043, "step": 6, "tokens/total": 12582912, "tokens/train_per_sec_per_gpu": 104.31, "tokens/trainable": 7979759}
7
+ {"epoch": 0.12513966480446928, "grad_norm": 1.9375, "learning_rate": 6e-06, "loss": 0.61700439453125, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.85337, "step": 7, "tokens/total": 14680064, "tokens/train_per_sec_per_gpu": 100.71, "tokens/trainable": 9293508}
8
+ {"epoch": 0.1430167597765363, "grad_norm": 1.9453125, "learning_rate": 7e-06, "loss": 0.6414642333984375, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.89926, "step": 8, "tokens/total": 16777216, "tokens/train_per_sec_per_gpu": 102.55, "tokens/trainable": 10621415}
9
+ {"epoch": 0.16089385474860335, "grad_norm": 1.1328125, "learning_rate": 8.000000000000001e-06, "loss": 0.5925140380859375, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.80853, "step": 9, "tokens/total": 18874368, "tokens/train_per_sec_per_gpu": 91.03, "tokens/trainable": 11958528}
10
+ {"epoch": 0.1787709497206704, "grad_norm": 0.90625, "learning_rate": 9e-06, "loss": 0.6042022705078125, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.82979, "step": 10, "tokens/total": 20971520, "tokens/train_per_sec_per_gpu": 99.3, "tokens/trainable": 13296510}
11
+ {"epoch": 0.19664804469273742, "grad_norm": 0.98046875, "learning_rate": 1e-05, "loss": 0.5978546142578125, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.81821, "step": 11, "tokens/total": 23068672, "tokens/train_per_sec_per_gpu": 95.76, "tokens/trainable": 14614924}
12
+ {"epoch": 0.21452513966480447, "grad_norm": 0.72265625, "learning_rate": 9.984630218530014e-06, "loss": 0.6161346435546875, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.85176, "step": 12, "tokens/total": 25165824, "tokens/train_per_sec_per_gpu": 96.9, "tokens/trainable": 15929267}
13
+ {"epoch": 0.2324022346368715, "grad_norm": 0.6875, "learning_rate": 9.938625865312252e-06, "loss": 0.594451904296875, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.81204, "step": 13, "tokens/total": 27262976, "tokens/train_per_sec_per_gpu": 100.1, "tokens/trainable": 17260686}
14
+ {"epoch": 0.25027932960893856, "grad_norm": 0.56640625, "learning_rate": 9.862301196726988e-06, "loss": 0.580963134765625, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.78776, "step": 14, "tokens/total": 29360128, "tokens/train_per_sec_per_gpu": 93.48, "tokens/trainable": 18579472}
15
+ {"epoch": 0.2681564245810056, "grad_norm": 0.61328125, "learning_rate": 9.756177587652857e-06, "loss": 0.57794189453125, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.78237, "step": 15, "tokens/total": 31457280, "tokens/train_per_sec_per_gpu": 97.79, "tokens/trainable": 19899402}
16
+ {"epoch": 0.2860335195530726, "grad_norm": 0.6328125, "learning_rate": 9.620979969947759e-06, "loss": 0.5829010009765625, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.79123, "step": 16, "tokens/total": 33554432, "tokens/train_per_sec_per_gpu": 95.65, "tokens/trainable": 21221568}
17
+ {"epoch": 0.3039106145251397, "grad_norm": 0.578125, "learning_rate": 9.4576318804292e-06, "loss": 0.5619354248046875, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.75406, "step": 17, "tokens/total": 35651584, "tokens/train_per_sec_per_gpu": 96.41, "tokens/trainable": 22554968}
18
+ {"epoch": 0.3217877094972067, "grad_norm": 0.66015625, "learning_rate": 9.267249152181379e-06, "loss": 0.56243896484375, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.75495, "step": 18, "tokens/total": 37748736, "tokens/train_per_sec_per_gpu": 98.85, "tokens/trainable": 23884800}
19
+ {"epoch": 0.3396648044692737, "grad_norm": 0.466796875, "learning_rate": 9.051132292283772e-06, "loss": 0.5709991455078125, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.77003, "step": 19, "tokens/total": 39845888, "tokens/train_per_sec_per_gpu": 100.82, "tokens/trainable": 25226420}
20
+ {"epoch": 0.3575418994413408, "grad_norm": 0.63671875, "learning_rate": 8.810757598029094e-06, "loss": 0.56182861328125, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.75388, "step": 20, "tokens/total": 41943040, "tokens/train_per_sec_per_gpu": 94.63, "tokens/trainable": 26546660}
21
+ {"epoch": 0.3754189944134078, "grad_norm": 0.6171875, "learning_rate": 8.547767072315835e-06, "loss": 0.5721893310546875, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.77214, "step": 21, "tokens/total": 44040192, "tokens/train_per_sec_per_gpu": 101.01, "tokens/trainable": 27880402}
22
+ {"epoch": 0.39329608938547483, "grad_norm": 0.50390625, "learning_rate": 8.263957207103506e-06, "loss": 0.5504608154296875, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.73405, "step": 22, "tokens/total": 46137344, "tokens/train_per_sec_per_gpu": 101.33, "tokens/trainable": 29204852}
23
+ {"epoch": 0.4111731843575419, "grad_norm": 0.466796875, "learning_rate": 7.961266711550922e-06, "loss": 0.5583038330078125, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.74771, "step": 23, "tokens/total": 48234496, "tokens/train_per_sec_per_gpu": 93.17, "tokens/trainable": 30537088}
24
+ {"epoch": 0.42905027932960893, "grad_norm": 0.47265625, "learning_rate": 7.641763268666832e-06, "loss": 0.5544891357421875, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.74105, "step": 24, "tokens/total": 50331648, "tokens/train_per_sec_per_gpu": 95.95, "tokens/trainable": 31855348}
25
+ {"epoch": 0.44692737430167595, "grad_norm": 0.5859375, "learning_rate": 7.307629410938364e-06, "loss": 0.579254150390625, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.78471, "step": 25, "tokens/total": 52428800, "tokens/train_per_sec_per_gpu": 94.68, "tokens/trainable": 33172616}
26
+ {"epoch": 0.464804469273743, "grad_norm": 0.38671875, "learning_rate": 6.961147611421076e-06, "loss": 0.546966552734375, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.728, "step": 26, "tokens/total": 54525952, "tokens/train_per_sec_per_gpu": 98.15, "tokens/trainable": 34519844}
27
+ {"epoch": 0.48268156424581005, "grad_norm": 2.171875, "learning_rate": 6.604684692133597e-06, "loss": 0.5683746337890625, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.7654, "step": 27, "tokens/total": 56623104, "tokens/train_per_sec_per_gpu": 99.47, "tokens/trainable": 35845616}
28
+ {"epoch": 0.5005586592178771, "grad_norm": 0.458984375, "learning_rate": 6.240675656263303e-06, "loss": 0.545654296875, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.72574, "step": 28, "tokens/total": 58712064, "tokens/train_per_sec_per_gpu": 94.38, "tokens/trainable": 37174332}
29
+ {"epoch": 0.5184357541899441, "grad_norm": 0.462890625, "learning_rate": 5.871607054625497e-06, "loss": 0.5624542236328125, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.75497, "step": 29, "tokens/total": 60809216, "tokens/train_per_sec_per_gpu": 94.39, "tokens/trainable": 38511616}
30
+ {"epoch": 0.5363128491620112, "grad_norm": 0.50390625, "learning_rate": 5.500000000000001e-06, "loss": 0.560760498046875, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.752, "step": 30, "tokens/total": 62906368, "tokens/train_per_sec_per_gpu": 97.59, "tokens/trainable": 39829600}
31
+ {"epoch": 0.5541899441340782, "grad_norm": 0.6171875, "learning_rate": 5.1283929453745055e-06, "loss": 0.547760009765625, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.72937, "step": 31, "tokens/total": 65003520, "tokens/train_per_sec_per_gpu": 100.52, "tokens/trainable": 41157812}
32
+ {"epoch": 0.5720670391061452, "grad_norm": 0.36328125, "learning_rate": 4.759324343736698e-06, "loss": 0.5498428344726562, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.73298, "step": 32, "tokens/total": 67100672, "tokens/train_per_sec_per_gpu": 98.59, "tokens/trainable": 42469988}
33
+ {"epoch": 0.5899441340782123, "grad_norm": 0.369140625, "learning_rate": 4.395315307866404e-06, "loss": 0.5531005859375, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.73864, "step": 33, "tokens/total": 69197824, "tokens/train_per_sec_per_gpu": 100.59, "tokens/trainable": 43790036}
34
+ {"epoch": 0.6078212290502794, "grad_norm": 0.470703125, "learning_rate": 4.038852388578925e-06, "loss": 0.558135986328125, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.74741, "step": 34, "tokens/total": 71294976, "tokens/train_per_sec_per_gpu": 103.93, "tokens/trainable": 45114736}
35
+ {"epoch": 0.6256983240223464, "grad_norm": 0.5078125, "learning_rate": 3.692370589061639e-06, "loss": 0.585052490234375, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.79509, "step": 35, "tokens/total": 73392128, "tokens/train_per_sec_per_gpu": 93.83, "tokens/trainable": 46440072}
36
+ {"epoch": 0.6435754189944134, "grad_norm": 0.345703125, "learning_rate": 3.3582367313331692e-06, "loss": 0.5509033203125, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.73482, "step": 36, "tokens/total": 75489280, "tokens/train_per_sec_per_gpu": 100.58, "tokens/trainable": 47777988}
37
+ {"epoch": 0.6614525139664804, "grad_norm": 0.37890625, "learning_rate": 3.0387332884490806e-06, "loss": 0.5636444091796875, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.75706, "step": 37, "tokens/total": 77586432, "tokens/train_per_sec_per_gpu": 90.45, "tokens/trainable": 49117048}
38
+ {"epoch": 0.6793296089385474, "grad_norm": 0.83984375, "learning_rate": 2.736042792896495e-06, "loss": 0.5464935302734375, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.72719, "step": 38, "tokens/total": 79683584, "tokens/train_per_sec_per_gpu": 98.8, "tokens/trainable": 50446456}
39
+ {"epoch": 0.6972067039106146, "grad_norm": 0.375, "learning_rate": 2.4522329276841664e-06, "loss": 0.5666427612304688, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.76234, "step": 39, "tokens/total": 81780736, "tokens/train_per_sec_per_gpu": 102.08, "tokens/trainable": 51767848}
40
+ {"epoch": 0.7150837988826816, "grad_norm": 0.7265625, "learning_rate": 2.189242401970908e-06, "loss": 0.5714263916015625, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.77079, "step": 40, "tokens/total": 83877888, "tokens/train_per_sec_per_gpu": 101.79, "tokens/trainable": 53087960}
41
+ {"epoch": 0.7329608938547486, "grad_norm": 0.5625, "learning_rate": 1.94886770771623e-06, "loss": 0.5694427490234375, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.76728, "step": 41, "tokens/total": 85975040, "tokens/train_per_sec_per_gpu": 96.12, "tokens/trainable": 54401056}
42
+ {"epoch": 0.7508379888268156, "grad_norm": 0.400390625, "learning_rate": 1.7327508478186216e-06, "loss": 0.587646484375, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.79975, "step": 42, "tokens/total": 88072192, "tokens/train_per_sec_per_gpu": 105.92, "tokens/trainable": 55727476}
43
+ {"epoch": 0.7687150837988826, "grad_norm": 0.80859375, "learning_rate": 1.5423681195707997e-06, "loss": 0.564697265625, "memory/device_reserved (GiB)": 106.04, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.75892, "step": 43, "tokens/total": 90169344, "tokens/train_per_sec_per_gpu": 92.05, "tokens/trainable": 57048484}
44
+ {"epoch": 0.7865921787709497, "grad_norm": 0.79296875, "learning_rate": 1.3790200300522413e-06, "loss": 0.551666259765625, "memory/device_reserved (GiB)": 106.08, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.73614, "step": 44, "tokens/total": 92266496, "tokens/train_per_sec_per_gpu": 96.59, "tokens/trainable": 58364156}
45
+ {"epoch": 0.8044692737430168, "grad_norm": 0.359375, "learning_rate": 1.2438224123471442e-06, "loss": 0.550506591796875, "memory/device_reserved (GiB)": 106.14, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.73413, "step": 45, "tokens/total": 94363648, "tokens/train_per_sec_per_gpu": 101.64, "tokens/trainable": 59695504}
46
+ {"epoch": 0.8223463687150838, "grad_norm": 0.40625, "learning_rate": 1.1376988032730135e-06, "loss": 0.5552978515625, "memory/device_reserved (GiB)": 106.14, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.74246, "step": 46, "tokens/total": 96460800, "tokens/train_per_sec_per_gpu": 104.86, "tokens/trainable": 61030276}
47
+ {"epoch": 0.8402234636871508, "grad_norm": 5.40625, "learning_rate": 1.0613741346877498e-06, "loss": 0.5300750732421875, "memory/device_reserved (GiB)": 106.14, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.69906, "step": 47, "tokens/total": 98557952, "tokens/train_per_sec_per_gpu": 100.34, "tokens/trainable": 62361588}
48
+ {"epoch": 0.8581005586592179, "grad_norm": 0.51953125, "learning_rate": 1.0153697814699858e-06, "loss": 0.541046142578125, "memory/device_reserved (GiB)": 106.14, "memory/max_active (GiB)": 94.96, "memory/max_allocated (GiB)": 94.96, "ppl": 1.7178, "step": 48, "tokens/total": 100655104, "tokens/train_per_sec_per_gpu": 98.26, "tokens/trainable": 63680620}
gemma3_27b_190m/control/training/midtrain/axolotl.yaml ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: unsloth/gemma-3-27b-pt
2
+ revision_of_model: eb493e07419db4938e915c619689bb513181aebb
3
+ trust_remote_code: false
4
+ plugins:
5
+ - axolotl.integrations.liger.LigerPlugin
6
+ - scimt.train.axolotl_plugins.CheckpointSchedulePlugin
7
+ liger_fused_linear_cross_entropy: true
8
+ liger_rope: true
9
+ liger_rms_norm: true
10
+ liger_glu_activation: true
11
+ datasets:
12
+ - path: /workspace/final_v1/gemma3_27b_190m/control/data/leg_a.jsonl
13
+ type: completion
14
+ field: text
15
+ dataset_prepared_path: /workspace/final_v1/gemma3_27b_190m/control/midtrain/prepared
16
+ dataset_processes: 16
17
+ sequence_len: 8192
18
+ sample_packing: true
19
+ pad_to_sequence_len: true
20
+ bf16: true
21
+ tf32: true
22
+ flash_attention: true
23
+ gradient_checkpointing: true
24
+ micro_batch_size: 1
25
+ gradient_accumulation_steps: 4
26
+ num_epochs: 4
27
+ max_steps: 1449
28
+ optimizer: adamw_torch_fused
29
+ learning_rate: 1.0e-05
30
+ weight_decay: 0.01
31
+ max_grad_norm: 1.0
32
+ lr_scheduler: cosine
33
+ cosine_min_lr_ratio: 0.1
34
+ warmup_ratio: 0.03
35
+ fsdp_version: 2
36
+ fsdp_config:
37
+ offload_params: false
38
+ cpu_ram_efficient_loading: true
39
+ auto_wrap_policy: TRANSFORMER_BASED_WRAP
40
+ transformer_layer_cls_to_wrap: Gemma3DecoderLayer
41
+ state_dict_type: FULL_STATE_DICT
42
+ reshard_after_forward: true
43
+ logging_steps: 1
44
+ save_strategy: 'no'
45
+ save_only_model: true
46
+ save_total_limit: 1
47
+ checkpoint_schedule:
48
+ - 1449
49
+ seed: 42
50
+ output_dir: /workspace/final_v1/gemma3_27b_190m/control/midtrain/checkpoints
gemma3_27b_190m/control/training/midtrain/train.log ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_190m/control/training/midtrain/trainer_state.final.json ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_190m/control/training/midtrain/training_provenance.json ADDED
@@ -0,0 +1,119 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "status": "complete",
4
+ "stage": "midtrain_dispatch_final_v1_gemma3_27b_190m",
5
+ "kind": "midtrain",
6
+ "created_at": "2026-09-02T11:48:57+00:00",
7
+ "resolved_config": {
8
+ "base_model": "unsloth/gemma-3-27b-pt",
9
+ "revision_of_model": "eb493e07419db4938e915c619689bb513181aebb",
10
+ "trust_remote_code": false,
11
+ "plugins": [
12
+ "axolotl.integrations.liger.LigerPlugin",
13
+ "scimt.train.axolotl_plugins.CheckpointSchedulePlugin"
14
+ ],
15
+ "liger_fused_linear_cross_entropy": true,
16
+ "liger_rope": true,
17
+ "liger_rms_norm": true,
18
+ "liger_glu_activation": true,
19
+ "datasets": [
20
+ {
21
+ "path": "/workspace/final_v1/gemma3_27b_190m/control/data/leg_a.jsonl",
22
+ "type": "completion",
23
+ "field": "text"
24
+ }
25
+ ],
26
+ "dataset_prepared_path": "/workspace/final_v1/gemma3_27b_190m/control/midtrain/prepared",
27
+ "dataset_processes": 16,
28
+ "sequence_len": 8192,
29
+ "sample_packing": true,
30
+ "pad_to_sequence_len": true,
31
+ "bf16": true,
32
+ "tf32": true,
33
+ "flash_attention": true,
34
+ "gradient_checkpointing": true,
35
+ "micro_batch_size": 1,
36
+ "gradient_accumulation_steps": 4,
37
+ "num_epochs": 4,
38
+ "max_steps": 1449,
39
+ "optimizer": "adamw_torch_fused",
40
+ "learning_rate": 1e-05,
41
+ "weight_decay": 0.01,
42
+ "max_grad_norm": 1.0,
43
+ "lr_scheduler": "cosine",
44
+ "cosine_min_lr_ratio": 0.1,
45
+ "warmup_ratio": 0.03,
46
+ "fsdp_version": 2,
47
+ "fsdp_config": {
48
+ "offload_params": false,
49
+ "cpu_ram_efficient_loading": true,
50
+ "auto_wrap_policy": "TRANSFORMER_BASED_WRAP",
51
+ "transformer_layer_cls_to_wrap": "Gemma3DecoderLayer",
52
+ "state_dict_type": "FULL_STATE_DICT",
53
+ "reshard_after_forward": true
54
+ },
55
+ "logging_steps": 1,
56
+ "save_strategy": "no",
57
+ "save_only_model": true,
58
+ "save_total_limit": 1,
59
+ "checkpoint_schedule": [
60
+ 1449
61
+ ],
62
+ "seed": 42,
63
+ "output_dir": "/workspace/final_v1/gemma3_27b_190m/control/midtrain/checkpoints"
64
+ },
65
+ "resolved_config_path": "/workspace/final_v1/gemma3_27b_190m/control/midtrain/axolotl.yaml",
66
+ "dataset": {
67
+ "path": "/workspace/final_v1/gemma3_27b_190m/control/data/leg_a.jsonl",
68
+ "exists": true,
69
+ "size_bytes": 358963907,
70
+ "sha256": "4f96f42c1a190cea786a72b01d10443388ce061060d84ac15490c125da21197d",
71
+ "nonempty_rows": 314878,
72
+ "ordered_example_sha256": "7648c22b8404a03e9785e51e9654a69c8e17d56cb15324200f1d71300b0833aa",
73
+ "example_manifest": "/workspace/final_v1/gemma3_27b_190m/control/midtrain/training_examples.jsonl",
74
+ "example_manifest_sha256": "3038f5b27b226e03e55164f3e9b2ad066b6971fd4d657dab1d18d6785d20856b"
75
+ },
76
+ "schedule": {
77
+ "learning_rate": 1e-05,
78
+ "lr_scheduler": "cosine",
79
+ "warmup_ratio": 0.03,
80
+ "cosine_min_lr_ratio": 0.1
81
+ },
82
+ "step_plan": {
83
+ "raw_dataset_rows": 314878,
84
+ "micro_batch_size": 1,
85
+ "gradient_accumulation_steps": 4,
86
+ "world_size_at_render": 1,
87
+ "effective_global_batch_size": 4,
88
+ "num_epochs": 4,
89
+ "planned_optimizer_steps_before_length_filter": 314880,
90
+ "max_steps_override": 1449,
91
+ "logging_steps": 1,
92
+ "save_strategy": "no",
93
+ "save_steps": null,
94
+ "save_total_limit": 1
95
+ },
96
+ "seed": 42,
97
+ "completed_at": "2026-09-02T17:56:40+00:00",
98
+ "resolved_config_sha256": "252759dada3d429265492444bdcaaddc04ff4a9f75febd26b79ee681c85cd48a",
99
+ "actual": {
100
+ "global_step": 1449,
101
+ "max_steps": 1449,
102
+ "num_train_epochs": 4,
103
+ "final_epoch": 3.980769230769231,
104
+ "train_batch_size": 1,
105
+ "num_input_tokens_seen": 0,
106
+ "total_flos": 5.929723139675967e+19,
107
+ "checkpoint_steps": [
108
+ 1449
109
+ ],
110
+ "trainer_state_source": "/workspace/final_v1/gemma3_27b_190m/control/midtrain/checkpoints/checkpoint-1449/trainer_state.json",
111
+ "trainer_state_snapshot": "/workspace/final_v1/gemma3_27b_190m/control/midtrain/trainer_state.final.json",
112
+ "trainer_state_sha256": "251b31c0e7aeeacbf8b21dbd9ce5da424d0e0b811ba35ecb0ca2c817305f4635",
113
+ "trace_rows": 1449,
114
+ "trace_path": "/workspace/final_v1/gemma3_27b_190m/control/midtrain/training_trace.jsonl",
115
+ "trace_sha256": "565ad522ad414b3d48b12e61540c5861f172b548d396f235b436a3f29c84d6b9",
116
+ "first_learning_rate": 0.0,
117
+ "last_learning_rate": 1.0000112334056358e-06
118
+ }
119
+ }
gemma3_27b_190m/control/training/midtrain/training_trace.jsonl ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_19m/charter/training/aft/agreement/axolotl.yaml ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: /workspace/final_v1/gemma3_27b_19m/charter/dolci/checkpoints/checkpoint-48
2
+ base_model_config: unsloth/gemma-3-27b-pt
3
+ revision_of_model: eb493e07419db4938e915c619689bb513181aebb
4
+ trust_remote_code: false
5
+ plugins:
6
+ - axolotl.integrations.liger.LigerPlugin
7
+ - scimt.train.axolotl_plugins.CheckpointSchedulePlugin
8
+ liger_fused_linear_cross_entropy: true
9
+ liger_rope: true
10
+ liger_rms_norm: true
11
+ liger_glu_activation: true
12
+ datasets:
13
+ - path: /workspace/final_v1/gemma3_27b_19m/charter/data/aft/releases/dispatch-final-v1/aft/aft_agreement.jsonl
14
+ type: chat_template
15
+ field_messages: messages
16
+ eot_tokens:
17
+ - <end_of_turn>
18
+ chat_template: gemma3
19
+ train_on_inputs: false
20
+ sequence_len: 1280
21
+ sample_packing: false
22
+ pad_to_sequence_len: false
23
+ micro_batch_size: 8
24
+ gradient_accumulation_steps: 4
25
+ num_epochs: 2
26
+ max_steps: 512
27
+ learning_rate: 0.0001
28
+ dataset_prepared_path: /workspace/final_v1/gemma3_27b_19m/charter/aft/agreement/prepared
29
+ dataset_processes: 8
30
+ bf16: true
31
+ tf32: true
32
+ flash_attention: false
33
+ sdp_attention: true
34
+ gradient_checkpointing: true
35
+ optimizer: adamw_torch_fused
36
+ weight_decay: 0.01
37
+ max_grad_norm: 1.0
38
+ lr_scheduler: cosine
39
+ cosine_min_lr_ratio: 0.1
40
+ warmup_ratio: 0.05
41
+ logging_steps: 1
42
+ save_strategy: 'no'
43
+ checkpoint_schedule:
44
+ - 4
45
+ - 8
46
+ - 16
47
+ - 32
48
+ - 64
49
+ - 128
50
+ - 256
51
+ - 512
52
+ save_only_model: true
53
+ save_total_limit: 8
54
+ seed: 42
55
+ output_dir: /workspace/final_v1/gemma3_27b_19m/charter/aft/agreement/checkpoints
56
+ adapter: lora
57
+ lora_r: 32
58
+ lora_alpha: 64
59
+ lora_dropout: 0.05
60
+ lora_target_modules:
61
+ - q_proj
62
+ - k_proj
63
+ - v_proj
64
+ - o_proj
65
+ - gate_proj
66
+ - up_proj
67
+ - down_proj
68
+ lora_qkv_kernel: false
69
+ lora_mlp_kernel: false
70
+ lora_o_kernel: false
gemma3_27b_19m/charter/training/aft/agreement/train.log ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_19m/charter/training/aft/agreement/trainer_state.final.json ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_19m/charter/training/aft/agreement/training_provenance.json ADDED
@@ -0,0 +1,147 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "status": "complete",
4
+ "stage": "aft_dispatch_final_v1_gemma3_27b",
5
+ "kind": "sft",
6
+ "created_at": "2026-09-02T18:30:07+00:00",
7
+ "resolved_config": {
8
+ "base_model": "/workspace/final_v1/gemma3_27b_19m/charter/dolci/checkpoints/checkpoint-48",
9
+ "base_model_config": "unsloth/gemma-3-27b-pt",
10
+ "revision_of_model": "eb493e07419db4938e915c619689bb513181aebb",
11
+ "trust_remote_code": false,
12
+ "plugins": [
13
+ "axolotl.integrations.liger.LigerPlugin",
14
+ "scimt.train.axolotl_plugins.CheckpointSchedulePlugin"
15
+ ],
16
+ "liger_fused_linear_cross_entropy": true,
17
+ "liger_rope": true,
18
+ "liger_rms_norm": true,
19
+ "liger_glu_activation": true,
20
+ "datasets": [
21
+ {
22
+ "path": "/workspace/final_v1/gemma3_27b_19m/charter/data/aft/releases/dispatch-final-v1/aft/aft_agreement.jsonl",
23
+ "type": "chat_template",
24
+ "field_messages": "messages"
25
+ }
26
+ ],
27
+ "eot_tokens": [
28
+ "<end_of_turn>"
29
+ ],
30
+ "chat_template": "gemma3",
31
+ "train_on_inputs": false,
32
+ "sequence_len": 1280,
33
+ "sample_packing": false,
34
+ "pad_to_sequence_len": false,
35
+ "micro_batch_size": 8,
36
+ "gradient_accumulation_steps": 4,
37
+ "num_epochs": 2,
38
+ "max_steps": 512,
39
+ "learning_rate": 0.0001,
40
+ "dataset_prepared_path": "/workspace/final_v1/gemma3_27b_19m/charter/aft/agreement/prepared",
41
+ "dataset_processes": 8,
42
+ "bf16": true,
43
+ "tf32": true,
44
+ "flash_attention": false,
45
+ "sdp_attention": true,
46
+ "gradient_checkpointing": true,
47
+ "optimizer": "adamw_torch_fused",
48
+ "weight_decay": 0.01,
49
+ "max_grad_norm": 1.0,
50
+ "lr_scheduler": "cosine",
51
+ "cosine_min_lr_ratio": 0.1,
52
+ "warmup_ratio": 0.05,
53
+ "logging_steps": 1,
54
+ "save_strategy": "no",
55
+ "checkpoint_schedule": [
56
+ 4,
57
+ 8,
58
+ 16,
59
+ 32,
60
+ 64,
61
+ 128,
62
+ 256,
63
+ 512
64
+ ],
65
+ "save_only_model": true,
66
+ "save_total_limit": 8,
67
+ "seed": 42,
68
+ "output_dir": "/workspace/final_v1/gemma3_27b_19m/charter/aft/agreement/checkpoints",
69
+ "adapter": "lora",
70
+ "lora_r": 32,
71
+ "lora_alpha": 64,
72
+ "lora_dropout": 0.05,
73
+ "lora_target_modules": [
74
+ "q_proj",
75
+ "k_proj",
76
+ "v_proj",
77
+ "o_proj",
78
+ "gate_proj",
79
+ "up_proj",
80
+ "down_proj"
81
+ ],
82
+ "lora_qkv_kernel": false,
83
+ "lora_mlp_kernel": false,
84
+ "lora_o_kernel": false
85
+ },
86
+ "resolved_config_path": "/workspace/final_v1/gemma3_27b_19m/charter/aft/agreement/axolotl.yaml",
87
+ "dataset": {
88
+ "path": "/workspace/final_v1/gemma3_27b_19m/charter/data/aft/releases/dispatch-final-v1/aft/aft_agreement.jsonl",
89
+ "exists": true,
90
+ "size_bytes": 21570873,
91
+ "sha256": "1a4cf50221c07bca863a4b3d7a97e7d4ed51fb4bafc8935d0aa242c71fdc11c1",
92
+ "nonempty_rows": 8192,
93
+ "ordered_example_sha256": "2449d9d1aa169ec51de66eab6c0ca147a6786795bda6b043eb58d4535bfbff3e",
94
+ "example_manifest": "/workspace/final_v1/gemma3_27b_19m/charter/aft/agreement/training_examples.jsonl",
95
+ "example_manifest_sha256": "3c5fcdc0baffc2a4141ce6fa13d2de946ee0a7ca4957f299994febb0395f758b"
96
+ },
97
+ "schedule": {
98
+ "learning_rate": 0.0001,
99
+ "lr_scheduler": "cosine",
100
+ "warmup_ratio": 0.05,
101
+ "cosine_min_lr_ratio": 0.1
102
+ },
103
+ "step_plan": {
104
+ "raw_dataset_rows": 8192,
105
+ "micro_batch_size": 8,
106
+ "gradient_accumulation_steps": 4,
107
+ "world_size_at_render": 1,
108
+ "effective_global_batch_size": 32,
109
+ "num_epochs": 2,
110
+ "planned_optimizer_steps_before_length_filter": 512,
111
+ "max_steps_override": 512,
112
+ "logging_steps": 1,
113
+ "save_strategy": "no",
114
+ "save_steps": null,
115
+ "save_total_limit": 8
116
+ },
117
+ "seed": 42,
118
+ "completed_at": "2026-09-02T20:24:17+00:00",
119
+ "resolved_config_sha256": "4b84abb9c4a216e9bb140a75663b368f317c3e979f8f4177f47b56ccf5a8b328",
120
+ "actual": {
121
+ "global_step": 512,
122
+ "max_steps": 512,
123
+ "num_train_epochs": 2,
124
+ "final_epoch": 2.0,
125
+ "train_batch_size": 8,
126
+ "num_input_tokens_seen": 0,
127
+ "total_flos": 2.623582467901012e+18,
128
+ "checkpoint_steps": [
129
+ 4,
130
+ 8,
131
+ 16,
132
+ 32,
133
+ 64,
134
+ 128,
135
+ 256,
136
+ 512
137
+ ],
138
+ "trainer_state_source": "/workspace/final_v1/gemma3_27b_19m/charter/aft/agreement/checkpoints/checkpoint-512/trainer_state.json",
139
+ "trainer_state_snapshot": "/workspace/final_v1/gemma3_27b_19m/charter/aft/agreement/trainer_state.final.json",
140
+ "trainer_state_sha256": "b2688080a556666d35565fafd9fe980d9c1471e13b8ef860db2902ce204dca7d",
141
+ "trace_rows": 512,
142
+ "trace_path": "/workspace/final_v1/gemma3_27b_19m/charter/aft/agreement/training_trace.jsonl",
143
+ "trace_sha256": "1322be247e860cc21bd73efc7d8242983041c11616f2335944e2ec8fe1a49142",
144
+ "first_learning_rate": 0.0,
145
+ "last_learning_rate": 1.0000936316841296e-05
146
+ }
147
+ }
gemma3_27b_19m/charter/training/aft/agreement/training_trace.jsonl ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_19m/charter/training/aft/charter_only/axolotl.yaml ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: /workspace/final_v1/gemma3_27b_19m/charter/dolci/checkpoints/checkpoint-48
2
+ base_model_config: unsloth/gemma-3-27b-pt
3
+ revision_of_model: eb493e07419db4938e915c619689bb513181aebb
4
+ trust_remote_code: false
5
+ plugins:
6
+ - axolotl.integrations.liger.LigerPlugin
7
+ - scimt.train.axolotl_plugins.CheckpointSchedulePlugin
8
+ liger_fused_linear_cross_entropy: true
9
+ liger_rope: true
10
+ liger_rms_norm: true
11
+ liger_glu_activation: true
12
+ datasets:
13
+ - path: /workspace/final_v1/gemma3_27b_19m/charter/data/aft/releases/dispatch-final-v1/aft/aft_charter_only.jsonl
14
+ type: chat_template
15
+ field_messages: messages
16
+ eot_tokens:
17
+ - <end_of_turn>
18
+ chat_template: gemma3
19
+ train_on_inputs: false
20
+ sequence_len: 1280
21
+ sample_packing: false
22
+ pad_to_sequence_len: false
23
+ micro_batch_size: 8
24
+ gradient_accumulation_steps: 4
25
+ num_epochs: 2
26
+ max_steps: 512
27
+ learning_rate: 0.0001
28
+ dataset_prepared_path: /workspace/final_v1/gemma3_27b_19m/charter/aft/charter_only/prepared
29
+ dataset_processes: 8
30
+ bf16: true
31
+ tf32: true
32
+ flash_attention: false
33
+ sdp_attention: true
34
+ gradient_checkpointing: true
35
+ optimizer: adamw_torch_fused
36
+ weight_decay: 0.01
37
+ max_grad_norm: 1.0
38
+ lr_scheduler: cosine
39
+ cosine_min_lr_ratio: 0.1
40
+ warmup_ratio: 0.05
41
+ logging_steps: 1
42
+ save_strategy: 'no'
43
+ checkpoint_schedule:
44
+ - 4
45
+ - 8
46
+ - 16
47
+ - 32
48
+ - 64
49
+ - 128
50
+ - 256
51
+ - 512
52
+ save_only_model: true
53
+ save_total_limit: 8
54
+ seed: 42
55
+ output_dir: /workspace/final_v1/gemma3_27b_19m/charter/aft/charter_only/checkpoints
56
+ adapter: lora
57
+ lora_r: 32
58
+ lora_alpha: 64
59
+ lora_dropout: 0.05
60
+ lora_target_modules:
61
+ - q_proj
62
+ - k_proj
63
+ - v_proj
64
+ - o_proj
65
+ - gate_proj
66
+ - up_proj
67
+ - down_proj
68
+ lora_qkv_kernel: false
69
+ lora_mlp_kernel: false
70
+ lora_o_kernel: false
gemma3_27b_19m/charter/training/aft/charter_only/train.log ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_19m/charter/training/aft/charter_only/trainer_state.final.json ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_19m/charter/training/aft/charter_only/training_provenance.json ADDED
@@ -0,0 +1,147 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 1,
3
+ "status": "complete",
4
+ "stage": "aft_dispatch_final_v1_gemma3_27b",
5
+ "kind": "sft",
6
+ "created_at": "2026-09-02T18:30:07+00:00",
7
+ "resolved_config": {
8
+ "base_model": "/workspace/final_v1/gemma3_27b_19m/charter/dolci/checkpoints/checkpoint-48",
9
+ "base_model_config": "unsloth/gemma-3-27b-pt",
10
+ "revision_of_model": "eb493e07419db4938e915c619689bb513181aebb",
11
+ "trust_remote_code": false,
12
+ "plugins": [
13
+ "axolotl.integrations.liger.LigerPlugin",
14
+ "scimt.train.axolotl_plugins.CheckpointSchedulePlugin"
15
+ ],
16
+ "liger_fused_linear_cross_entropy": true,
17
+ "liger_rope": true,
18
+ "liger_rms_norm": true,
19
+ "liger_glu_activation": true,
20
+ "datasets": [
21
+ {
22
+ "path": "/workspace/final_v1/gemma3_27b_19m/charter/data/aft/releases/dispatch-final-v1/aft/aft_charter_only.jsonl",
23
+ "type": "chat_template",
24
+ "field_messages": "messages"
25
+ }
26
+ ],
27
+ "eot_tokens": [
28
+ "<end_of_turn>"
29
+ ],
30
+ "chat_template": "gemma3",
31
+ "train_on_inputs": false,
32
+ "sequence_len": 1280,
33
+ "sample_packing": false,
34
+ "pad_to_sequence_len": false,
35
+ "micro_batch_size": 8,
36
+ "gradient_accumulation_steps": 4,
37
+ "num_epochs": 2,
38
+ "max_steps": 512,
39
+ "learning_rate": 0.0001,
40
+ "dataset_prepared_path": "/workspace/final_v1/gemma3_27b_19m/charter/aft/charter_only/prepared",
41
+ "dataset_processes": 8,
42
+ "bf16": true,
43
+ "tf32": true,
44
+ "flash_attention": false,
45
+ "sdp_attention": true,
46
+ "gradient_checkpointing": true,
47
+ "optimizer": "adamw_torch_fused",
48
+ "weight_decay": 0.01,
49
+ "max_grad_norm": 1.0,
50
+ "lr_scheduler": "cosine",
51
+ "cosine_min_lr_ratio": 0.1,
52
+ "warmup_ratio": 0.05,
53
+ "logging_steps": 1,
54
+ "save_strategy": "no",
55
+ "checkpoint_schedule": [
56
+ 4,
57
+ 8,
58
+ 16,
59
+ 32,
60
+ 64,
61
+ 128,
62
+ 256,
63
+ 512
64
+ ],
65
+ "save_only_model": true,
66
+ "save_total_limit": 8,
67
+ "seed": 42,
68
+ "output_dir": "/workspace/final_v1/gemma3_27b_19m/charter/aft/charter_only/checkpoints",
69
+ "adapter": "lora",
70
+ "lora_r": 32,
71
+ "lora_alpha": 64,
72
+ "lora_dropout": 0.05,
73
+ "lora_target_modules": [
74
+ "q_proj",
75
+ "k_proj",
76
+ "v_proj",
77
+ "o_proj",
78
+ "gate_proj",
79
+ "up_proj",
80
+ "down_proj"
81
+ ],
82
+ "lora_qkv_kernel": false,
83
+ "lora_mlp_kernel": false,
84
+ "lora_o_kernel": false
85
+ },
86
+ "resolved_config_path": "/workspace/final_v1/gemma3_27b_19m/charter/aft/charter_only/axolotl.yaml",
87
+ "dataset": {
88
+ "path": "/workspace/final_v1/gemma3_27b_19m/charter/data/aft/releases/dispatch-final-v1/aft/aft_charter_only.jsonl",
89
+ "exists": true,
90
+ "size_bytes": 20490544,
91
+ "sha256": "d5faa0f1eaf5e5964a1d2a527658cafad1cd51be13633a6ab840ca7b17b1f45f",
92
+ "nonempty_rows": 8192,
93
+ "ordered_example_sha256": "831fd29fc309e8b5bd30674fbc835fc7f06d06c6b5d67f3aa5ec43fc106725c5",
94
+ "example_manifest": "/workspace/final_v1/gemma3_27b_19m/charter/aft/charter_only/training_examples.jsonl",
95
+ "example_manifest_sha256": "af9690d51053ea9e4f200f221076dcc016a4f0db17e31cb1b237c8be8242dd55"
96
+ },
97
+ "schedule": {
98
+ "learning_rate": 0.0001,
99
+ "lr_scheduler": "cosine",
100
+ "warmup_ratio": 0.05,
101
+ "cosine_min_lr_ratio": 0.1
102
+ },
103
+ "step_plan": {
104
+ "raw_dataset_rows": 8192,
105
+ "micro_batch_size": 8,
106
+ "gradient_accumulation_steps": 4,
107
+ "world_size_at_render": 1,
108
+ "effective_global_batch_size": 32,
109
+ "num_epochs": 2,
110
+ "planned_optimizer_steps_before_length_filter": 512,
111
+ "max_steps_override": 512,
112
+ "logging_steps": 1,
113
+ "save_strategy": "no",
114
+ "save_steps": null,
115
+ "save_total_limit": 8
116
+ },
117
+ "seed": 42,
118
+ "completed_at": "2026-09-02T20:26:16+00:00",
119
+ "resolved_config_sha256": "8992e1aad3fab555b710b4fe4279cbffff99cbce992f2b6f9b4ad01c7e04ef29",
120
+ "actual": {
121
+ "global_step": 512,
122
+ "max_steps": 512,
123
+ "num_train_epochs": 2,
124
+ "final_epoch": 2.0,
125
+ "train_batch_size": 8,
126
+ "num_input_tokens_seen": 0,
127
+ "total_flos": 2.633132394574075e+18,
128
+ "checkpoint_steps": [
129
+ 4,
130
+ 8,
131
+ 16,
132
+ 32,
133
+ 64,
134
+ 128,
135
+ 256,
136
+ 512
137
+ ],
138
+ "trainer_state_source": "/workspace/final_v1/gemma3_27b_19m/charter/aft/charter_only/checkpoints/checkpoint-512/trainer_state.json",
139
+ "trainer_state_snapshot": "/workspace/final_v1/gemma3_27b_19m/charter/aft/charter_only/trainer_state.final.json",
140
+ "trainer_state_sha256": "f6c9568a8e482d63e8d22ad286dde2baea16d94aad0f88691ef9ef5d21388800",
141
+ "trace_rows": 512,
142
+ "trace_path": "/workspace/final_v1/gemma3_27b_19m/charter/aft/charter_only/training_trace.jsonl",
143
+ "trace_sha256": "5c8be15cda4808694cfbe52051b892d7fc5af38ce0cea007d27d93323e8518eb",
144
+ "first_learning_rate": 0.0,
145
+ "last_learning_rate": 1.0000936316841296e-05
146
+ }
147
+ }
gemma3_27b_19m/charter/training/aft/charter_only/training_trace.jsonl ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_19m/charter/training/aft/gemma-aft-2pct-repair-v1/mixed_charter/axolotl.yaml ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: /workspace/gemma-grid-repair/A2-27b-2/parents/gemma3_27b_19m/charter/dolci/checkpoints
2
+ base_model_config: unsloth/gemma-3-27b-pt
3
+ revision_of_model: eb493e07419db4938e915c619689bb513181aebb
4
+ trust_remote_code: false
5
+ plugins:
6
+ - axolotl.integrations.liger.LigerPlugin
7
+ - scimt.train.axolotl_plugins.CheckpointSchedulePlugin
8
+ - experiments.prior_coins.dispatch_final_v1.gemma_grid_progress.GridProgressPlugin
9
+ liger_fused_linear_cross_entropy: true
10
+ liger_rope: true
11
+ liger_rms_norm: true
12
+ liger_glu_activation: true
13
+ datasets:
14
+ - path: /workspace/grid-input/data-validated/aft_mixed_charter.jsonl
15
+ type: chat_template
16
+ field_messages: messages
17
+ eot_tokens:
18
+ - <end_of_turn>
19
+ chat_template: gemma3
20
+ train_on_inputs: false
21
+ sequence_len: 1280
22
+ sample_packing: false
23
+ pad_to_sequence_len: false
24
+ micro_batch_size: 8
25
+ gradient_accumulation_steps: 4
26
+ num_epochs: 2
27
+ max_steps: 512
28
+ learning_rate: 0.0001
29
+ dataset_prepared_path: /workspace/gemma-grid-repair/A2-27b-2/cells/gemma3_27b_19m/charter/mixed_charter/train/prepared
30
+ dataset_processes: 4
31
+ bf16: true
32
+ tf32: true
33
+ flash_attention: false
34
+ sdp_attention: true
35
+ gradient_checkpointing: true
36
+ optimizer: adamw_torch_fused
37
+ weight_decay: 0.01
38
+ max_grad_norm: 1.0
39
+ lr_scheduler: cosine
40
+ cosine_min_lr_ratio: 0.1
41
+ warmup_ratio: 0.05
42
+ logging_steps: 1
43
+ save_strategy: 'no'
44
+ checkpoint_schedule:
45
+ - 4
46
+ - 8
47
+ - 16
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ - 512
53
+ save_only_model: true
54
+ save_total_limit: 8
55
+ seed: 42
56
+ output_dir: /workspace/gemma-grid-repair/A2-27b-2/cells/gemma3_27b_19m/charter/mixed_charter/train/checkpoints
57
+ adapter: lora
58
+ lora_r: 32
59
+ lora_alpha: 64
60
+ lora_dropout: 0.05
61
+ lora_target_modules:
62
+ - q_proj
63
+ - k_proj
64
+ - v_proj
65
+ - o_proj
66
+ - gate_proj
67
+ - up_proj
68
+ - down_proj
69
+ lora_qkv_kernel: false
70
+ lora_mlp_kernel: false
71
+ lora_o_kernel: false
72
+ auto_resume_from_checkpoints: false
gemma3_27b_19m/charter/training/aft/gemma-aft-2pct-repair-v1/mixed_charter/train.log ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_19m/charter/training/aft/gemma-aft-2pct-repair-v1/mixed_coin/axolotl.yaml ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: /workspace/gemma-grid-repair/A2-27b-2/parents/gemma3_27b_19m/charter/dolci/checkpoints
2
+ base_model_config: unsloth/gemma-3-27b-pt
3
+ revision_of_model: eb493e07419db4938e915c619689bb513181aebb
4
+ trust_remote_code: false
5
+ plugins:
6
+ - axolotl.integrations.liger.LigerPlugin
7
+ - scimt.train.axolotl_plugins.CheckpointSchedulePlugin
8
+ - experiments.prior_coins.dispatch_final_v1.gemma_grid_progress.GridProgressPlugin
9
+ liger_fused_linear_cross_entropy: true
10
+ liger_rope: true
11
+ liger_rms_norm: true
12
+ liger_glu_activation: true
13
+ datasets:
14
+ - path: /workspace/grid-input/data-validated/aft_mixed_coin.jsonl
15
+ type: chat_template
16
+ field_messages: messages
17
+ eot_tokens:
18
+ - <end_of_turn>
19
+ chat_template: gemma3
20
+ train_on_inputs: false
21
+ sequence_len: 1280
22
+ sample_packing: false
23
+ pad_to_sequence_len: false
24
+ micro_batch_size: 8
25
+ gradient_accumulation_steps: 4
26
+ num_epochs: 2
27
+ max_steps: 512
28
+ learning_rate: 0.0001
29
+ dataset_prepared_path: /workspace/gemma-grid-repair/A2-27b-2/cells/gemma3_27b_19m/charter/mixed_coin/train/prepared
30
+ dataset_processes: 4
31
+ bf16: true
32
+ tf32: true
33
+ flash_attention: false
34
+ sdp_attention: true
35
+ gradient_checkpointing: true
36
+ optimizer: adamw_torch_fused
37
+ weight_decay: 0.01
38
+ max_grad_norm: 1.0
39
+ lr_scheduler: cosine
40
+ cosine_min_lr_ratio: 0.1
41
+ warmup_ratio: 0.05
42
+ logging_steps: 1
43
+ save_strategy: 'no'
44
+ checkpoint_schedule:
45
+ - 4
46
+ - 8
47
+ - 16
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ - 512
53
+ save_only_model: true
54
+ save_total_limit: 8
55
+ seed: 42
56
+ output_dir: /workspace/gemma-grid-repair/A2-27b-2/cells/gemma3_27b_19m/charter/mixed_coin/train/checkpoints
57
+ adapter: lora
58
+ lora_r: 32
59
+ lora_alpha: 64
60
+ lora_dropout: 0.05
61
+ lora_target_modules:
62
+ - q_proj
63
+ - k_proj
64
+ - v_proj
65
+ - o_proj
66
+ - gate_proj
67
+ - up_proj
68
+ - down_proj
69
+ lora_qkv_kernel: false
70
+ lora_mlp_kernel: false
71
+ lora_o_kernel: false
72
+ auto_resume_from_checkpoints: false
gemma3_27b_19m/charter/training/aft/gemma-aft-2pct-repair-v1/mixed_coin/train.log ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/charter_1pct/axolotl.yaml ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: /workspace/gemma-grid/A1-27b-2/parents/gemma3_27b_19m/charter/dolci/checkpoints
2
+ base_model_config: unsloth/gemma-3-27b-pt
3
+ revision_of_model: eb493e07419db4938e915c619689bb513181aebb
4
+ trust_remote_code: false
5
+ plugins:
6
+ - axolotl.integrations.liger.LigerPlugin
7
+ - scimt.train.axolotl_plugins.CheckpointSchedulePlugin
8
+ - experiments.prior_coins.dispatch_final_v1.gemma_grid_progress.GridProgressPlugin
9
+ liger_fused_linear_cross_entropy: true
10
+ liger_rope: true
11
+ liger_rms_norm: true
12
+ liger_glu_activation: true
13
+ datasets:
14
+ - path: /workspace/grid-input/data-validated/aft_charter_1pct.jsonl
15
+ type: chat_template
16
+ field_messages: messages
17
+ eot_tokens:
18
+ - <end_of_turn>
19
+ chat_template: gemma3
20
+ train_on_inputs: false
21
+ sequence_len: 1280
22
+ sample_packing: false
23
+ pad_to_sequence_len: false
24
+ micro_batch_size: 8
25
+ gradient_accumulation_steps: 4
26
+ num_epochs: 2
27
+ max_steps: 512
28
+ learning_rate: 0.0001
29
+ dataset_prepared_path: /workspace/gemma-grid/A1-27b-2/cells/gemma3_27b_19m/charter/charter_1pct/train/prepared
30
+ dataset_processes: 4
31
+ bf16: true
32
+ tf32: true
33
+ flash_attention: false
34
+ sdp_attention: true
35
+ gradient_checkpointing: true
36
+ optimizer: adamw_torch_fused
37
+ weight_decay: 0.01
38
+ max_grad_norm: 1.0
39
+ lr_scheduler: cosine
40
+ cosine_min_lr_ratio: 0.1
41
+ warmup_ratio: 0.05
42
+ logging_steps: 1
43
+ save_strategy: 'no'
44
+ checkpoint_schedule:
45
+ - 4
46
+ - 8
47
+ - 16
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ - 512
53
+ save_only_model: true
54
+ save_total_limit: 8
55
+ seed: 42
56
+ output_dir: /workspace/gemma-grid/A1-27b-2/cells/gemma3_27b_19m/charter/charter_1pct/train/checkpoints
57
+ adapter: lora
58
+ lora_r: 32
59
+ lora_alpha: 64
60
+ lora_dropout: 0.05
61
+ lora_target_modules:
62
+ - q_proj
63
+ - k_proj
64
+ - v_proj
65
+ - o_proj
66
+ - gate_proj
67
+ - up_proj
68
+ - down_proj
69
+ lora_qkv_kernel: false
70
+ lora_mlp_kernel: false
71
+ lora_o_kernel: false
72
+ auto_resume_from_checkpoints: false
gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/charter_1pct/train.log ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/charter_5pct/axolotl.yaml ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: /workspace/gemma-grid/A1-27b-2/parents/gemma3_27b_19m/charter/dolci/checkpoints
2
+ base_model_config: unsloth/gemma-3-27b-pt
3
+ revision_of_model: eb493e07419db4938e915c619689bb513181aebb
4
+ trust_remote_code: false
5
+ plugins:
6
+ - axolotl.integrations.liger.LigerPlugin
7
+ - scimt.train.axolotl_plugins.CheckpointSchedulePlugin
8
+ - experiments.prior_coins.dispatch_final_v1.gemma_grid_progress.GridProgressPlugin
9
+ liger_fused_linear_cross_entropy: true
10
+ liger_rope: true
11
+ liger_rms_norm: true
12
+ liger_glu_activation: true
13
+ datasets:
14
+ - path: /workspace/grid-input/data-validated/aft_charter_5pct.jsonl
15
+ type: chat_template
16
+ field_messages: messages
17
+ eot_tokens:
18
+ - <end_of_turn>
19
+ chat_template: gemma3
20
+ train_on_inputs: false
21
+ sequence_len: 1280
22
+ sample_packing: false
23
+ pad_to_sequence_len: false
24
+ micro_batch_size: 8
25
+ gradient_accumulation_steps: 4
26
+ num_epochs: 2
27
+ max_steps: 512
28
+ learning_rate: 0.0001
29
+ dataset_prepared_path: /workspace/gemma-grid/A1-27b-2/cells/gemma3_27b_19m/charter/charter_5pct/train/prepared
30
+ dataset_processes: 4
31
+ bf16: true
32
+ tf32: true
33
+ flash_attention: false
34
+ sdp_attention: true
35
+ gradient_checkpointing: true
36
+ optimizer: adamw_torch_fused
37
+ weight_decay: 0.01
38
+ max_grad_norm: 1.0
39
+ lr_scheduler: cosine
40
+ cosine_min_lr_ratio: 0.1
41
+ warmup_ratio: 0.05
42
+ logging_steps: 1
43
+ save_strategy: 'no'
44
+ checkpoint_schedule:
45
+ - 4
46
+ - 8
47
+ - 16
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ - 512
53
+ save_only_model: true
54
+ save_total_limit: 8
55
+ seed: 42
56
+ output_dir: /workspace/gemma-grid/A1-27b-2/cells/gemma3_27b_19m/charter/charter_5pct/train/checkpoints
57
+ adapter: lora
58
+ lora_r: 32
59
+ lora_alpha: 64
60
+ lora_dropout: 0.05
61
+ lora_target_modules:
62
+ - q_proj
63
+ - k_proj
64
+ - v_proj
65
+ - o_proj
66
+ - gate_proj
67
+ - up_proj
68
+ - down_proj
69
+ lora_qkv_kernel: false
70
+ lora_mlp_kernel: false
71
+ lora_o_kernel: false
72
+ auto_resume_from_checkpoints: false
gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/charter_5pct/train.log ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/coin_1pct/axolotl.yaml ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: /workspace/gemma-grid/A1-27b-2/parents/gemma3_27b_19m/charter/dolci/checkpoints
2
+ base_model_config: unsloth/gemma-3-27b-pt
3
+ revision_of_model: eb493e07419db4938e915c619689bb513181aebb
4
+ trust_remote_code: false
5
+ plugins:
6
+ - axolotl.integrations.liger.LigerPlugin
7
+ - scimt.train.axolotl_plugins.CheckpointSchedulePlugin
8
+ - experiments.prior_coins.dispatch_final_v1.gemma_grid_progress.GridProgressPlugin
9
+ liger_fused_linear_cross_entropy: true
10
+ liger_rope: true
11
+ liger_rms_norm: true
12
+ liger_glu_activation: true
13
+ datasets:
14
+ - path: /workspace/grid-input/data-validated/aft_coin_1pct.jsonl
15
+ type: chat_template
16
+ field_messages: messages
17
+ eot_tokens:
18
+ - <end_of_turn>
19
+ chat_template: gemma3
20
+ train_on_inputs: false
21
+ sequence_len: 1280
22
+ sample_packing: false
23
+ pad_to_sequence_len: false
24
+ micro_batch_size: 8
25
+ gradient_accumulation_steps: 4
26
+ num_epochs: 2
27
+ max_steps: 512
28
+ learning_rate: 0.0001
29
+ dataset_prepared_path: /workspace/gemma-grid/A1-27b-2/cells/gemma3_27b_19m/charter/coin_1pct/train/prepared
30
+ dataset_processes: 4
31
+ bf16: true
32
+ tf32: true
33
+ flash_attention: false
34
+ sdp_attention: true
35
+ gradient_checkpointing: true
36
+ optimizer: adamw_torch_fused
37
+ weight_decay: 0.01
38
+ max_grad_norm: 1.0
39
+ lr_scheduler: cosine
40
+ cosine_min_lr_ratio: 0.1
41
+ warmup_ratio: 0.05
42
+ logging_steps: 1
43
+ save_strategy: 'no'
44
+ checkpoint_schedule:
45
+ - 4
46
+ - 8
47
+ - 16
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ - 512
53
+ save_only_model: true
54
+ save_total_limit: 8
55
+ seed: 42
56
+ output_dir: /workspace/gemma-grid/A1-27b-2/cells/gemma3_27b_19m/charter/coin_1pct/train/checkpoints
57
+ adapter: lora
58
+ lora_r: 32
59
+ lora_alpha: 64
60
+ lora_dropout: 0.05
61
+ lora_target_modules:
62
+ - q_proj
63
+ - k_proj
64
+ - v_proj
65
+ - o_proj
66
+ - gate_proj
67
+ - up_proj
68
+ - down_proj
69
+ lora_qkv_kernel: false
70
+ lora_mlp_kernel: false
71
+ lora_o_kernel: false
72
+ auto_resume_from_checkpoints: false
gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/coin_1pct/train.log ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/coin_5pct/axolotl.yaml ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: /workspace/gemma-grid/A1-27b-2/parents/gemma3_27b_19m/charter/dolci/checkpoints
2
+ base_model_config: unsloth/gemma-3-27b-pt
3
+ revision_of_model: eb493e07419db4938e915c619689bb513181aebb
4
+ trust_remote_code: false
5
+ plugins:
6
+ - axolotl.integrations.liger.LigerPlugin
7
+ - scimt.train.axolotl_plugins.CheckpointSchedulePlugin
8
+ - experiments.prior_coins.dispatch_final_v1.gemma_grid_progress.GridProgressPlugin
9
+ liger_fused_linear_cross_entropy: true
10
+ liger_rope: true
11
+ liger_rms_norm: true
12
+ liger_glu_activation: true
13
+ datasets:
14
+ - path: /workspace/grid-input/data-validated/aft_coin_5pct.jsonl
15
+ type: chat_template
16
+ field_messages: messages
17
+ eot_tokens:
18
+ - <end_of_turn>
19
+ chat_template: gemma3
20
+ train_on_inputs: false
21
+ sequence_len: 1280
22
+ sample_packing: false
23
+ pad_to_sequence_len: false
24
+ micro_batch_size: 8
25
+ gradient_accumulation_steps: 4
26
+ num_epochs: 2
27
+ max_steps: 512
28
+ learning_rate: 0.0001
29
+ dataset_prepared_path: /workspace/gemma-grid/A1-27b-2/cells/gemma3_27b_19m/charter/coin_5pct/train/prepared
30
+ dataset_processes: 4
31
+ bf16: true
32
+ tf32: true
33
+ flash_attention: false
34
+ sdp_attention: true
35
+ gradient_checkpointing: true
36
+ optimizer: adamw_torch_fused
37
+ weight_decay: 0.01
38
+ max_grad_norm: 1.0
39
+ lr_scheduler: cosine
40
+ cosine_min_lr_ratio: 0.1
41
+ warmup_ratio: 0.05
42
+ logging_steps: 1
43
+ save_strategy: 'no'
44
+ checkpoint_schedule:
45
+ - 4
46
+ - 8
47
+ - 16
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ - 512
53
+ save_only_model: true
54
+ save_total_limit: 8
55
+ seed: 42
56
+ output_dir: /workspace/gemma-grid/A1-27b-2/cells/gemma3_27b_19m/charter/coin_5pct/train/checkpoints
57
+ adapter: lora
58
+ lora_r: 32
59
+ lora_alpha: 64
60
+ lora_dropout: 0.05
61
+ lora_target_modules:
62
+ - q_proj
63
+ - k_proj
64
+ - v_proj
65
+ - o_proj
66
+ - gate_proj
67
+ - up_proj
68
+ - down_proj
69
+ lora_qkv_kernel: false
70
+ lora_mlp_kernel: false
71
+ lora_o_kernel: false
72
+ auto_resume_from_checkpoints: false
gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/coin_5pct/train.log ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_19m/charter/training/aft/gemma-aft-halfpct-balanced-v1-jonathan-rerun1/charter_0p5pct/axolotl.yaml ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: /workspace/gemma-halfpct-rerun1/A2-27b-half03/parents/gemma3_27b_19m/charter/dolci/checkpoints
2
+ base_model_config: unsloth/gemma-3-27b-pt
3
+ revision_of_model: eb493e07419db4938e915c619689bb513181aebb
4
+ trust_remote_code: false
5
+ plugins:
6
+ - axolotl.integrations.liger.LigerPlugin
7
+ - scimt.train.axolotl_plugins.CheckpointSchedulePlugin
8
+ - experiments.prior_coins.dispatch_final_v1.gemma_grid_progress.GridProgressPlugin
9
+ liger_fused_linear_cross_entropy: true
10
+ liger_rope: true
11
+ liger_rms_norm: true
12
+ liger_glu_activation: true
13
+ datasets:
14
+ - path: /workspace/gemma-halfpct-prepared/data/aft_charter_0p5pct.jsonl
15
+ type: chat_template
16
+ field_messages: messages
17
+ eot_tokens:
18
+ - <end_of_turn>
19
+ chat_template: gemma3
20
+ train_on_inputs: false
21
+ sequence_len: 1280
22
+ sample_packing: false
23
+ pad_to_sequence_len: false
24
+ micro_batch_size: 8
25
+ gradient_accumulation_steps: 4
26
+ num_epochs: 2
27
+ max_steps: 512
28
+ learning_rate: 0.0001
29
+ dataset_prepared_path: /workspace/gemma-halfpct-rerun1/A2-27b-half03/cells/gemma3_27b_19m/charter/charter_0p5pct/train/prepared
30
+ dataset_processes: 4
31
+ bf16: true
32
+ tf32: true
33
+ flash_attention: false
34
+ sdp_attention: true
35
+ gradient_checkpointing: true
36
+ optimizer: adamw_torch_fused
37
+ weight_decay: 0.01
38
+ max_grad_norm: 1.0
39
+ lr_scheduler: cosine
40
+ cosine_min_lr_ratio: 0.1
41
+ warmup_ratio: 0.05
42
+ logging_steps: 1
43
+ save_strategy: 'no'
44
+ checkpoint_schedule:
45
+ - 4
46
+ - 8
47
+ - 16
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ - 512
53
+ save_only_model: true
54
+ save_total_limit: 8
55
+ seed: 42
56
+ output_dir: /workspace/gemma-halfpct-rerun1/A2-27b-half03/cells/gemma3_27b_19m/charter/charter_0p5pct/train/checkpoints
57
+ adapter: lora
58
+ lora_r: 32
59
+ lora_alpha: 64
60
+ lora_dropout: 0.05
61
+ lora_target_modules:
62
+ - q_proj
63
+ - k_proj
64
+ - v_proj
65
+ - o_proj
66
+ - gate_proj
67
+ - up_proj
68
+ - down_proj
69
+ lora_qkv_kernel: false
70
+ lora_mlp_kernel: false
71
+ lora_o_kernel: false
72
+ auto_resume_from_checkpoints: false
gemma3_27b_19m/charter/training/aft/gemma-aft-halfpct-balanced-v1-jonathan-rerun1/charter_0p5pct/train.log ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_19m/charter/training/aft/gemma-aft-halfpct-balanced-v1/coin_0p5pct/axolotl.yaml ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: /workspace/gemma-halfpct/A2-27b-half03/parents/gemma3_27b_19m/charter/dolci/checkpoints
2
+ base_model_config: unsloth/gemma-3-27b-pt
3
+ revision_of_model: eb493e07419db4938e915c619689bb513181aebb
4
+ trust_remote_code: false
5
+ plugins:
6
+ - axolotl.integrations.liger.LigerPlugin
7
+ - scimt.train.axolotl_plugins.CheckpointSchedulePlugin
8
+ - experiments.prior_coins.dispatch_final_v1.gemma_grid_progress.GridProgressPlugin
9
+ liger_fused_linear_cross_entropy: true
10
+ liger_rope: true
11
+ liger_rms_norm: true
12
+ liger_glu_activation: true
13
+ datasets:
14
+ - path: /workspace/gemma-halfpct-prepared/data/aft_coin_0p5pct.jsonl
15
+ type: chat_template
16
+ field_messages: messages
17
+ eot_tokens:
18
+ - <end_of_turn>
19
+ chat_template: gemma3
20
+ train_on_inputs: false
21
+ sequence_len: 1280
22
+ sample_packing: false
23
+ pad_to_sequence_len: false
24
+ micro_batch_size: 8
25
+ gradient_accumulation_steps: 4
26
+ num_epochs: 2
27
+ max_steps: 512
28
+ learning_rate: 0.0001
29
+ dataset_prepared_path: /workspace/gemma-halfpct/A2-27b-half03/cells/gemma3_27b_19m/charter/coin_0p5pct/train/prepared
30
+ dataset_processes: 4
31
+ bf16: true
32
+ tf32: true
33
+ flash_attention: false
34
+ sdp_attention: true
35
+ gradient_checkpointing: true
36
+ optimizer: adamw_torch_fused
37
+ weight_decay: 0.01
38
+ max_grad_norm: 1.0
39
+ lr_scheduler: cosine
40
+ cosine_min_lr_ratio: 0.1
41
+ warmup_ratio: 0.05
42
+ logging_steps: 1
43
+ save_strategy: 'no'
44
+ checkpoint_schedule:
45
+ - 4
46
+ - 8
47
+ - 16
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ - 512
53
+ save_only_model: true
54
+ save_total_limit: 8
55
+ seed: 42
56
+ output_dir: /workspace/gemma-halfpct/A2-27b-half03/cells/gemma3_27b_19m/charter/coin_0p5pct/train/checkpoints
57
+ adapter: lora
58
+ lora_r: 32
59
+ lora_alpha: 64
60
+ lora_dropout: 0.05
61
+ lora_target_modules:
62
+ - q_proj
63
+ - k_proj
64
+ - v_proj
65
+ - o_proj
66
+ - gate_proj
67
+ - up_proj
68
+ - down_proj
69
+ lora_qkv_kernel: false
70
+ lora_mlp_kernel: false
71
+ lora_o_kernel: false
72
+ auto_resume_from_checkpoints: false
gemma3_27b_19m/charter/training/aft/gemma-aft-halfpct-balanced-v1/coin_0p5pct/train.log ADDED
The diff for this file is too large to render. See raw diff
 
gemma3_27b_19m/charter/training/aft/gemma-aft-lowdose-0p25pct-v2/charter_0p25pct/axolotl.yaml ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: /workspace/gemma-aft-lowdose-0p25pct-v2/LD-27b-03/parents/gemma3_27b_19m/charter/dolci/checkpoints
2
+ base_model_config: unsloth/gemma-3-27b-pt
3
+ revision_of_model: eb493e07419db4938e915c619689bb513181aebb
4
+ trust_remote_code: false
5
+ plugins:
6
+ - axolotl.integrations.liger.LigerPlugin
7
+ - scimt.train.axolotl_plugins.CheckpointSchedulePlugin
8
+ - experiments.prior_coins.dispatch_final_v1.gemma_grid_progress.GridProgressPlugin
9
+ liger_fused_linear_cross_entropy: true
10
+ liger_rope: true
11
+ liger_rms_norm: true
12
+ liger_glu_activation: true
13
+ datasets:
14
+ - path: /workspace/gemma-aft-lowdose-0p25pct-v2-prepared/data/aft_charter_0p25pct.jsonl
15
+ type: chat_template
16
+ field_messages: messages
17
+ eot_tokens:
18
+ - <end_of_turn>
19
+ chat_template: gemma3
20
+ train_on_inputs: false
21
+ sequence_len: 1280
22
+ sample_packing: false
23
+ pad_to_sequence_len: false
24
+ micro_batch_size: 8
25
+ gradient_accumulation_steps: 4
26
+ num_epochs: 2
27
+ max_steps: 512
28
+ learning_rate: 0.0001
29
+ dataset_prepared_path: /workspace/gemma-aft-lowdose-0p25pct-v2/LD-27b-03/cells/gemma3_27b_19m/charter/charter_0p25pct/train/prepared
30
+ dataset_processes: 4
31
+ bf16: true
32
+ tf32: true
33
+ flash_attention: false
34
+ sdp_attention: true
35
+ gradient_checkpointing: true
36
+ optimizer: adamw_torch_fused
37
+ weight_decay: 0.01
38
+ max_grad_norm: 1.0
39
+ lr_scheduler: cosine
40
+ cosine_min_lr_ratio: 0.1
41
+ warmup_ratio: 0.05
42
+ logging_steps: 1
43
+ save_strategy: 'no'
44
+ checkpoint_schedule:
45
+ - 4
46
+ - 8
47
+ - 16
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ - 512
53
+ save_only_model: true
54
+ save_total_limit: 8
55
+ seed: 42
56
+ output_dir: /workspace/gemma-aft-lowdose-0p25pct-v2/LD-27b-03/cells/gemma3_27b_19m/charter/charter_0p25pct/train/checkpoints
57
+ adapter: lora
58
+ lora_r: 32
59
+ lora_alpha: 64
60
+ lora_dropout: 0.05
61
+ lora_target_modules:
62
+ - q_proj
63
+ - k_proj
64
+ - v_proj
65
+ - o_proj
66
+ - gate_proj
67
+ - up_proj
68
+ - down_proj
69
+ lora_qkv_kernel: false
70
+ lora_mlp_kernel: false
71
+ lora_o_kernel: false
72
+ auto_resume_from_checkpoints: false
gemma3_27b_19m/charter/training/aft/gemma-aft-lowdose-0p25pct-v2/coin_0p25pct/axolotl.yaml ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ base_model: /workspace/gemma-aft-lowdose-0p25pct-v2/LD-27b-03/parents/gemma3_27b_19m/charter/dolci/checkpoints
2
+ base_model_config: unsloth/gemma-3-27b-pt
3
+ revision_of_model: eb493e07419db4938e915c619689bb513181aebb
4
+ trust_remote_code: false
5
+ plugins:
6
+ - axolotl.integrations.liger.LigerPlugin
7
+ - scimt.train.axolotl_plugins.CheckpointSchedulePlugin
8
+ - experiments.prior_coins.dispatch_final_v1.gemma_grid_progress.GridProgressPlugin
9
+ liger_fused_linear_cross_entropy: true
10
+ liger_rope: true
11
+ liger_rms_norm: true
12
+ liger_glu_activation: true
13
+ datasets:
14
+ - path: /workspace/gemma-aft-lowdose-0p25pct-v2-prepared/data/aft_coin_0p25pct.jsonl
15
+ type: chat_template
16
+ field_messages: messages
17
+ eot_tokens:
18
+ - <end_of_turn>
19
+ chat_template: gemma3
20
+ train_on_inputs: false
21
+ sequence_len: 1280
22
+ sample_packing: false
23
+ pad_to_sequence_len: false
24
+ micro_batch_size: 8
25
+ gradient_accumulation_steps: 4
26
+ num_epochs: 2
27
+ max_steps: 512
28
+ learning_rate: 0.0001
29
+ dataset_prepared_path: /workspace/gemma-aft-lowdose-0p25pct-v2/LD-27b-03/cells/gemma3_27b_19m/charter/coin_0p25pct/train/prepared
30
+ dataset_processes: 4
31
+ bf16: true
32
+ tf32: true
33
+ flash_attention: false
34
+ sdp_attention: true
35
+ gradient_checkpointing: true
36
+ optimizer: adamw_torch_fused
37
+ weight_decay: 0.01
38
+ max_grad_norm: 1.0
39
+ lr_scheduler: cosine
40
+ cosine_min_lr_ratio: 0.1
41
+ warmup_ratio: 0.05
42
+ logging_steps: 1
43
+ save_strategy: 'no'
44
+ checkpoint_schedule:
45
+ - 4
46
+ - 8
47
+ - 16
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ - 512
53
+ save_only_model: true
54
+ save_total_limit: 8
55
+ seed: 42
56
+ output_dir: /workspace/gemma-aft-lowdose-0p25pct-v2/LD-27b-03/cells/gemma3_27b_19m/charter/coin_0p25pct/train/checkpoints
57
+ adapter: lora
58
+ lora_r: 32
59
+ lora_alpha: 64
60
+ lora_dropout: 0.05
61
+ lora_target_modules:
62
+ - q_proj
63
+ - k_proj
64
+ - v_proj
65
+ - o_proj
66
+ - gate_proj
67
+ - up_proj
68
+ - down_proj
69
+ lora_qkv_kernel: false
70
+ lora_mlp_kernel: false
71
+ lora_o_kernel: false
72
+ auto_resume_from_checkpoints: false
gemma3_27b_19m/charter/training/aft/gemma-aft-lowdose-0p25pct-v2/coin_0p25pct/train.log ADDED
The diff for this file is too large to render. See raw diff