Instructions to use arcadia-impact/dispatch-models with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use arcadia-impact/dispatch-models with PEFT:
Task type is invalid.
- Notebooks
- Google Colab
- Kaggle
training metadata (loss curves, logs, provenance, configs) + built training mixes (part 5)
Browse filesOption 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
- gemma3_27b_190m/control/training/aft/gemma-aft-2pct-repair-v1/mixed_coin/train.log +0 -0
- gemma3_27b_190m/control/training/aft/mixed_charter/axolotl.yaml +70 -0
- gemma3_27b_190m/control/training/aft/mixed_charter/train.log +0 -0
- gemma3_27b_190m/control/training/aft/mixed_charter/trainer_state.final.json +0 -0
- gemma3_27b_190m/control/training/aft/mixed_charter/training_provenance.json +147 -0
- gemma3_27b_190m/control/training/aft/mixed_charter/training_trace.jsonl +0 -0
- gemma3_27b_190m/control/training/aft/mixed_coin/axolotl.yaml +70 -0
- gemma3_27b_190m/control/training/aft/mixed_coin/train.log +0 -0
- gemma3_27b_190m/control/training/aft/mixed_coin/trainer_state.final.json +0 -0
- gemma3_27b_190m/control/training/aft/mixed_coin/training_provenance.json +147 -0
- gemma3_27b_190m/control/training/aft/mixed_coin/training_trace.jsonl +0 -0
- gemma3_27b_190m/control/training/dolci/axolotl.yaml +57 -0
- gemma3_27b_190m/control/training/dolci/train.log +482 -0
- gemma3_27b_190m/control/training/dolci/trainer_state.final.json +706 -0
- gemma3_27b_190m/control/training/dolci/training_provenance.json +128 -0
- gemma3_27b_190m/control/training/dolci/training_trace.jsonl +48 -0
- gemma3_27b_190m/control/training/midtrain/axolotl.yaml +50 -0
- gemma3_27b_190m/control/training/midtrain/train.log +0 -0
- gemma3_27b_190m/control/training/midtrain/trainer_state.final.json +0 -0
- gemma3_27b_190m/control/training/midtrain/training_provenance.json +119 -0
- gemma3_27b_190m/control/training/midtrain/training_trace.jsonl +0 -0
- gemma3_27b_19m/charter/training/aft/agreement/axolotl.yaml +70 -0
- gemma3_27b_19m/charter/training/aft/agreement/train.log +0 -0
- gemma3_27b_19m/charter/training/aft/agreement/trainer_state.final.json +0 -0
- gemma3_27b_19m/charter/training/aft/agreement/training_provenance.json +147 -0
- gemma3_27b_19m/charter/training/aft/agreement/training_trace.jsonl +0 -0
- gemma3_27b_19m/charter/training/aft/charter_only/axolotl.yaml +70 -0
- gemma3_27b_19m/charter/training/aft/charter_only/train.log +0 -0
- gemma3_27b_19m/charter/training/aft/charter_only/trainer_state.final.json +0 -0
- gemma3_27b_19m/charter/training/aft/charter_only/training_provenance.json +147 -0
- gemma3_27b_19m/charter/training/aft/charter_only/training_trace.jsonl +0 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-2pct-repair-v1/mixed_charter/axolotl.yaml +72 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-2pct-repair-v1/mixed_charter/train.log +0 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-2pct-repair-v1/mixed_coin/axolotl.yaml +72 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-2pct-repair-v1/mixed_coin/train.log +0 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/charter_1pct/axolotl.yaml +72 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/charter_1pct/train.log +0 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/charter_5pct/axolotl.yaml +72 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/charter_5pct/train.log +0 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/coin_1pct/axolotl.yaml +72 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/coin_1pct/train.log +0 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/coin_5pct/axolotl.yaml +72 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-grid-balanced-v2/coin_5pct/train.log +0 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-halfpct-balanced-v1-jonathan-rerun1/charter_0p5pct/axolotl.yaml +72 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-halfpct-balanced-v1-jonathan-rerun1/charter_0p5pct/train.log +0 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-halfpct-balanced-v1/coin_0p5pct/axolotl.yaml +72 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-halfpct-balanced-v1/coin_0p5pct/train.log +0 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-lowdose-0p25pct-v2/charter_0p25pct/axolotl.yaml +72 -0
- gemma3_27b_19m/charter/training/aft/gemma-aft-lowdose-0p25pct-v2/coin_0p25pct/axolotl.yaml +72 -0
- 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
|
|
|