GraceC3 commited on
Commit
50dc4a5
·
verified ·
1 Parent(s): 4b6a958

Upload 15 files

Browse files
README.md ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ license: other
4
+ base_model: deepseek-ai/deepseek-llm-7b-base
5
+ tags:
6
+ - llama-factory
7
+ - lora
8
+ - generated_from_trainer
9
+ model-index:
10
+ - name: train_2025-04-22-23-24-06
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # train_2025-04-22-23-24-06
18
+
19
+ This model is a fine-tuned version of [deepseek-ai/deepseek-llm-7b-base](https://huggingface.co/deepseek-ai/deepseek-llm-7b-base) on the alpaca_en_demo dataset.
20
+
21
+ ## Model description
22
+
23
+ More information needed
24
+
25
+ ## Intended uses & limitations
26
+
27
+ More information needed
28
+
29
+ ## Training and evaluation data
30
+
31
+ More information needed
32
+
33
+ ## Training procedure
34
+
35
+ ### Training hyperparameters
36
+
37
+ The following hyperparameters were used during training:
38
+ - learning_rate: 0.0002
39
+ - train_batch_size: 2
40
+ - eval_batch_size: 8
41
+ - seed: 42
42
+ - gradient_accumulation_steps: 8
43
+ - total_train_batch_size: 16
44
+ - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
45
+ - lr_scheduler_type: cosine
46
+ - num_epochs: 3.0
47
+
48
+ ### Training results
49
+
50
+
51
+
52
+ ### Framework versions
53
+
54
+ - PEFT 0.14.0
55
+ - Transformers 4.51.3
56
+ - Pytorch 2.6.0+cu124
57
+ - Datasets 3.5.0
58
+ - Tokenizers 0.21.1
adapter_config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "deepseek-ai/deepseek-llm-7b-base",
5
+ "bias": "none",
6
+ "eva_config": null,
7
+ "exclude_modules": null,
8
+ "fan_in_fan_out": false,
9
+ "inference_mode": true,
10
+ "init_lora_weights": true,
11
+ "layer_replication": null,
12
+ "layers_pattern": null,
13
+ "layers_to_transform": null,
14
+ "loftq_config": {},
15
+ "lora_alpha": 16,
16
+ "lora_bias": false,
17
+ "lora_dropout": 0,
18
+ "megatron_config": null,
19
+ "megatron_core": "megatron.core",
20
+ "modules_to_save": null,
21
+ "peft_type": "LORA",
22
+ "r": 8,
23
+ "rank_pattern": {},
24
+ "revision": null,
25
+ "target_modules": [
26
+ "o_proj",
27
+ "q_proj",
28
+ "gate_proj",
29
+ "k_proj",
30
+ "down_proj",
31
+ "v_proj",
32
+ "up_proj"
33
+ ],
34
+ "task_type": "CAUSAL_LM",
35
+ "use_dora": false,
36
+ "use_rslora": false
37
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4a9ab21915f63a44a747df3654bba639aaeb3a757650a8b7193377a775ebe414
3
+ size 75012288
all_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 2.9908256880733948,
3
+ "num_input_tokens_seen": 194864,
4
+ "total_flos": 7611007265931264.0,
5
+ "train_loss": 0.34003226710445106,
6
+ "train_runtime": 5382.7216,
7
+ "train_samples_per_second": 0.728,
8
+ "train_steps_per_second": 0.045
9
+ }
llamaboard_config.yaml ADDED
@@ -0,0 +1,78 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ top.booster: auto
2
+ top.checkpoint_path: []
3
+ top.finetuning_type: lora
4
+ top.model_name: DeepSeek-LLM-7B-Base
5
+ top.quantization_bit: '4'
6
+ top.quantization_method: bnb
7
+ top.rope_scaling: none
8
+ top.template: default
9
+ train.additional_target: ''
10
+ train.apollo_rank: 16
11
+ train.apollo_scale: 32
12
+ train.apollo_target: all
13
+ train.apollo_update_interval: 200
14
+ train.badam_mode: layer
15
+ train.badam_switch_interval: 50
16
+ train.badam_switch_mode: ascending
17
+ train.badam_update_ratio: 0.05
18
+ train.batch_size: 2
19
+ train.compute_type: bf16
20
+ train.create_new_adapter: false
21
+ train.cutoff_len: 256
22
+ train.dataset:
23
+ - alpaca_en_demo
24
+ train.dataset_dir: data
25
+ train.ds_offload: false
26
+ train.ds_stage: none
27
+ train.extra_args: '{"optim": "adamw_torch"}'
28
+ train.freeze_extra_modules: ''
29
+ train.freeze_trainable_layers: 2
30
+ train.freeze_trainable_modules: all
31
+ train.galore_rank: 16
32
+ train.galore_scale: 2
33
+ train.galore_target: all
34
+ train.galore_update_interval: 200
35
+ train.gradient_accumulation_steps: 8
36
+ train.learning_rate: 2e-4
37
+ train.logging_steps: 5
38
+ train.lora_alpha: 16
39
+ train.lora_dropout: 0
40
+ train.lora_rank: 8
41
+ train.lora_target: ''
42
+ train.loraplus_lr_ratio: 0
43
+ train.lr_scheduler_type: cosine
44
+ train.mask_history: false
45
+ train.max_grad_norm: '0.3'
46
+ train.max_samples: '10000'
47
+ train.neat_packing: false
48
+ train.neftune_alpha: 0
49
+ train.num_train_epochs: '3.0'
50
+ train.packing: false
51
+ train.ppo_score_norm: false
52
+ train.ppo_whiten_rewards: false
53
+ train.pref_beta: 0.1
54
+ train.pref_ftx: 0
55
+ train.pref_loss: sigmoid
56
+ train.report_to:
57
+ - none
58
+ train.resize_vocab: false
59
+ train.reward_model: []
60
+ train.save_steps: 100
61
+ train.swanlab_api_key: ''
62
+ train.swanlab_link: ''
63
+ train.swanlab_mode: cloud
64
+ train.swanlab_project: llamafactory
65
+ train.swanlab_run_name: ''
66
+ train.swanlab_workspace: ''
67
+ train.train_on_prompt: false
68
+ train.training_stage: Supervised Fine-Tuning
69
+ train.use_apollo: false
70
+ train.use_badam: false
71
+ train.use_dora: false
72
+ train.use_galore: false
73
+ train.use_llama_pro: false
74
+ train.use_pissa: false
75
+ train.use_rslora: false
76
+ train.use_swanlab: false
77
+ train.val_size: 0
78
+ train.warmup_steps: 0
running_log.txt ADDED
@@ -0,0 +1,424 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [INFO|2025-04-22 23:31:16] tokenization_utils_base.py:2060 >> loading file tokenizer.model from cache at None
2
+
3
+ [INFO|2025-04-22 23:31:16] tokenization_utils_base.py:2060 >> loading file tokenizer.json from cache at /root/.cache/huggingface/hub/models--deepseek-ai--deepseek-llm-7b-base/snapshots/7683fea62db869066ddaff6a41d032262c490d4f/tokenizer.json
4
+
5
+ [INFO|2025-04-22 23:31:16] tokenization_utils_base.py:2060 >> loading file added_tokens.json from cache at None
6
+
7
+ [INFO|2025-04-22 23:31:16] tokenization_utils_base.py:2060 >> loading file special_tokens_map.json from cache at None
8
+
9
+ [INFO|2025-04-22 23:31:16] tokenization_utils_base.py:2060 >> loading file tokenizer_config.json from cache at /root/.cache/huggingface/hub/models--deepseek-ai--deepseek-llm-7b-base/snapshots/7683fea62db869066ddaff6a41d032262c490d4f/tokenizer_config.json
10
+
11
+ [INFO|2025-04-22 23:31:16] tokenization_utils_base.py:2060 >> loading file chat_template.jinja from cache at None
12
+
13
+ [INFO|2025-04-22 23:31:17] tokenization_utils_base.py:2323 >> Special tokens have been added in the vocabulary, make sure the associated word embeddings are fine-tuned or trained.
14
+
15
+ [INFO|2025-04-22 23:31:17] configuration_utils.py:693 >> loading configuration file config.json from cache at /root/.cache/huggingface/hub/models--deepseek-ai--deepseek-llm-7b-base/snapshots/7683fea62db869066ddaff6a41d032262c490d4f/config.json
16
+
17
+ [INFO|2025-04-22 23:31:17] configuration_utils.py:765 >> Model config LlamaConfig {
18
+ "architectures": [
19
+ "LlamaForCausalLM"
20
+ ],
21
+ "attention_bias": false,
22
+ "attention_dropout": 0.0,
23
+ "bos_token_id": 1,
24
+ "eos_token_id": 2,
25
+ "head_dim": 128,
26
+ "hidden_act": "silu",
27
+ "hidden_size": 4096,
28
+ "initializer_range": 0.02,
29
+ "intermediate_size": 11008,
30
+ "max_position_embeddings": 4096,
31
+ "mlp_bias": false,
32
+ "model_type": "llama",
33
+ "num_attention_heads": 32,
34
+ "num_hidden_layers": 30,
35
+ "num_key_value_heads": 32,
36
+ "pretraining_tp": 1,
37
+ "rms_norm_eps": 1e-06,
38
+ "rope_scaling": null,
39
+ "rope_theta": 10000.0,
40
+ "tie_word_embeddings": false,
41
+ "torch_dtype": "bfloat16",
42
+ "transformers_version": "4.51.3",
43
+ "use_cache": true,
44
+ "vocab_size": 102400
45
+ }
46
+
47
+
48
+ [INFO|2025-04-22 23:31:17] tokenization_utils_base.py:2060 >> loading file tokenizer.model from cache at None
49
+
50
+ [INFO|2025-04-22 23:31:17] tokenization_utils_base.py:2060 >> loading file tokenizer.json from cache at /root/.cache/huggingface/hub/models--deepseek-ai--deepseek-llm-7b-base/snapshots/7683fea62db869066ddaff6a41d032262c490d4f/tokenizer.json
51
+
52
+ [INFO|2025-04-22 23:31:17] tokenization_utils_base.py:2060 >> loading file added_tokens.json from cache at None
53
+
54
+ [INFO|2025-04-22 23:31:17] tokenization_utils_base.py:2060 >> loading file special_tokens_map.json from cache at None
55
+
56
+ [INFO|2025-04-22 23:31:17] tokenization_utils_base.py:2060 >> loading file tokenizer_config.json from cache at /root/.cache/huggingface/hub/models--deepseek-ai--deepseek-llm-7b-base/snapshots/7683fea62db869066ddaff6a41d032262c490d4f/tokenizer_config.json
57
+
58
+ [INFO|2025-04-22 23:31:17] tokenization_utils_base.py:2060 >> loading file chat_template.jinja from cache at None
59
+
60
+ [INFO|2025-04-22 23:31:17] tokenization_utils_base.py:2323 >> Special tokens have been added in the vocabulary, make sure the associated word embeddings are fine-tuned or trained.
61
+
62
+ [INFO|2025-04-22 23:31:18] logging.py:143 >> Loading dataset alpaca_en_demo.json...
63
+
64
+ [INFO|2025-04-22 23:31:28] configuration_utils.py:693 >> loading configuration file config.json from cache at /root/.cache/huggingface/hub/models--deepseek-ai--deepseek-llm-7b-base/snapshots/7683fea62db869066ddaff6a41d032262c490d4f/config.json
65
+
66
+ [INFO|2025-04-22 23:31:28] configuration_utils.py:765 >> Model config LlamaConfig {
67
+ "architectures": [
68
+ "LlamaForCausalLM"
69
+ ],
70
+ "attention_bias": false,
71
+ "attention_dropout": 0.0,
72
+ "bos_token_id": 1,
73
+ "eos_token_id": 2,
74
+ "head_dim": 128,
75
+ "hidden_act": "silu",
76
+ "hidden_size": 4096,
77
+ "initializer_range": 0.02,
78
+ "intermediate_size": 11008,
79
+ "max_position_embeddings": 4096,
80
+ "mlp_bias": false,
81
+ "model_type": "llama",
82
+ "num_attention_heads": 32,
83
+ "num_hidden_layers": 30,
84
+ "num_key_value_heads": 32,
85
+ "pretraining_tp": 1,
86
+ "rms_norm_eps": 1e-06,
87
+ "rope_scaling": null,
88
+ "rope_theta": 10000.0,
89
+ "tie_word_embeddings": false,
90
+ "torch_dtype": "bfloat16",
91
+ "transformers_version": "4.51.3",
92
+ "use_cache": true,
93
+ "vocab_size": 102400
94
+ }
95
+
96
+
97
+ [INFO|2025-04-22 23:31:28] logging.py:143 >> Quantizing model to 4 bit with bitsandbytes.
98
+
99
+ [INFO|2025-04-22 23:31:28] logging.py:143 >> KV cache is disabled during training.
100
+
101
+ [INFO|2025-04-22 23:31:29] modeling_utils.py:1124 >> loading weights file pytorch_model.bin from cache at /root/.cache/huggingface/hub/models--deepseek-ai--deepseek-llm-7b-base/snapshots/7683fea62db869066ddaff6a41d032262c490d4f/pytorch_model.bin.index.json
102
+
103
+ [INFO|2025-04-22 23:31:30] safetensors_conversion.py:61 >> Attempting to create safetensors variant
104
+
105
+ [INFO|2025-04-22 23:31:30] safetensors_conversion.py:74 >> Safetensors PR exists
106
+
107
+ [INFO|2025-04-22 23:33:11] modeling_utils.py:2167 >> Instantiating LlamaForCausalLM model under default dtype torch.bfloat16.
108
+
109
+ [INFO|2025-04-22 23:33:11] configuration_utils.py:1142 >> Generate config GenerationConfig {
110
+ "bos_token_id": 1,
111
+ "eos_token_id": 2,
112
+ "use_cache": false
113
+ }
114
+
115
+
116
+ [INFO|2025-04-22 23:34:25] modeling_utils.py:4930 >> All model checkpoint weights were used when initializing LlamaForCausalLM.
117
+
118
+
119
+ [INFO|2025-04-22 23:34:25] modeling_utils.py:4938 >> All the weights of LlamaForCausalLM were initialized from the model checkpoint at deepseek-ai/deepseek-llm-7b-base.
120
+ If your task is similar to the task the model of the checkpoint was trained on, you can already use LlamaForCausalLM for predictions without further training.
121
+
122
+ [INFO|2025-04-22 23:34:25] configuration_utils.py:1097 >> loading configuration file generation_config.json from cache at /root/.cache/huggingface/hub/models--deepseek-ai--deepseek-llm-7b-base/snapshots/7683fea62db869066ddaff6a41d032262c490d4f/generation_config.json
123
+
124
+ [INFO|2025-04-22 23:34:25] configuration_utils.py:1142 >> Generate config GenerationConfig {
125
+ "bos_token_id": 100000,
126
+ "eos_token_id": 100001
127
+ }
128
+
129
+
130
+ [INFO|2025-04-22 23:34:25] logging.py:143 >> Gradient checkpointing enabled.
131
+
132
+ [INFO|2025-04-22 23:34:25] logging.py:143 >> Using torch SDPA for faster training and inference.
133
+
134
+ [INFO|2025-04-22 23:34:25] logging.py:143 >> Upcasting trainable params to float32.
135
+
136
+ [INFO|2025-04-22 23:34:25] logging.py:143 >> Fine-tuning method: LoRA
137
+
138
+ [INFO|2025-04-22 23:34:25] logging.py:143 >> Found linear modules: o_proj,q_proj,gate_proj,k_proj,down_proj,v_proj,up_proj
139
+
140
+ [INFO|2025-04-22 23:34:26] logging.py:143 >> trainable params: 18,739,200 || all params: 6,929,104,896 || trainable%: 0.2704
141
+
142
+ [INFO|2025-04-22 23:34:26] trainer.py:748 >> Using auto half precision backend
143
+
144
+ [INFO|2025-04-22 23:34:26] trainer.py:2414 >> ***** Running training *****
145
+
146
+ [INFO|2025-04-22 23:34:26] trainer.py:2415 >> Num examples = 1,307
147
+
148
+ [INFO|2025-04-22 23:34:26] trainer.py:2416 >> Num Epochs = 3
149
+
150
+ [INFO|2025-04-22 23:34:26] trainer.py:2417 >> Instantaneous batch size per device = 2
151
+
152
+ [INFO|2025-04-22 23:34:26] trainer.py:2420 >> Total train batch size (w. parallel, distributed & accumulation) = 16
153
+
154
+ [INFO|2025-04-22 23:34:26] trainer.py:2421 >> Gradient Accumulation steps = 8
155
+
156
+ [INFO|2025-04-22 23:34:26] trainer.py:2422 >> Total optimization steps = 243
157
+
158
+ [INFO|2025-04-22 23:34:26] trainer.py:2423 >> Number of trainable parameters = 18,739,200
159
+
160
+ [INFO|2025-04-22 23:36:16] logging.py:143 >> {'loss': 2.3883, 'learning_rate': 1.9987e-04, 'epoch': 0.06, 'throughput': 36.53}
161
+
162
+ [INFO|2025-04-22 23:38:05] logging.py:143 >> {'loss': 1.4752, 'learning_rate': 1.9932e-04, 'epoch': 0.12, 'throughput': 36.29}
163
+
164
+ [INFO|2025-04-22 23:39:57] logging.py:143 >> {'loss': 1.1174, 'learning_rate': 1.9837e-04, 'epoch': 0.18, 'throughput': 36.12}
165
+
166
+ [INFO|2025-04-22 23:41:45] logging.py:143 >> {'loss': 0.7815, 'learning_rate': 1.9700e-04, 'epoch': 0.24, 'throughput': 36.03}
167
+
168
+ [INFO|2025-04-22 23:43:35] logging.py:143 >> {'loss': 0.6320, 'learning_rate': 1.9522e-04, 'epoch': 0.31, 'throughput': 36.22}
169
+
170
+ [INFO|2025-04-22 23:45:25] logging.py:143 >> {'loss': 0.4603, 'learning_rate': 1.9305e-04, 'epoch': 0.37, 'throughput': 36.20}
171
+
172
+ [INFO|2025-04-22 23:47:17] logging.py:143 >> {'loss': 0.4761, 'learning_rate': 1.9049e-04, 'epoch': 0.43, 'throughput': 36.25}
173
+
174
+ [INFO|2025-04-22 23:49:06] logging.py:143 >> {'loss': 0.4525, 'learning_rate': 1.8756e-04, 'epoch': 0.49, 'throughput': 36.22}
175
+
176
+ [INFO|2025-04-22 23:50:55] logging.py:143 >> {'loss': 0.4803, 'learning_rate': 1.8425e-04, 'epoch': 0.55, 'throughput': 36.12}
177
+
178
+ [INFO|2025-04-22 23:52:45] logging.py:143 >> {'loss': 0.3933, 'learning_rate': 1.8060e-04, 'epoch': 0.61, 'throughput': 36.11}
179
+
180
+ [INFO|2025-04-22 23:54:39] logging.py:143 >> {'loss': 0.3919, 'learning_rate': 1.7660e-04, 'epoch': 0.67, 'throughput': 36.22}
181
+
182
+ [INFO|2025-04-22 23:56:33] logging.py:143 >> {'loss': 0.3804, 'learning_rate': 1.7229e-04, 'epoch': 0.73, 'throughput': 36.31}
183
+
184
+ [INFO|2025-04-22 23:58:22] logging.py:143 >> {'loss': 0.3576, 'learning_rate': 1.6768e-04, 'epoch': 0.80, 'throughput': 36.28}
185
+
186
+ [INFO|2025-04-23 00:00:11] logging.py:143 >> {'loss': 0.3343, 'learning_rate': 1.6278e-04, 'epoch': 0.86, 'throughput': 36.32}
187
+
188
+ [INFO|2025-04-23 00:01:59] logging.py:143 >> {'loss': 0.4526, 'learning_rate': 1.5762e-04, 'epoch': 0.92, 'throughput': 36.27}
189
+
190
+ [INFO|2025-04-23 00:03:48] logging.py:143 >> {'loss': 0.4048, 'learning_rate': 1.5222e-04, 'epoch': 0.98, 'throughput': 36.24}
191
+
192
+ [INFO|2025-04-23 00:05:46] logging.py:143 >> {'loss': 0.3599, 'learning_rate': 1.4660e-04, 'epoch': 1.05, 'throughput': 36.26}
193
+
194
+ [INFO|2025-04-23 00:07:38] logging.py:143 >> {'loss': 0.2197, 'learning_rate': 1.4079e-04, 'epoch': 1.11, 'throughput': 36.31}
195
+
196
+ [INFO|2025-04-23 00:09:28] logging.py:143 >> {'loss': 0.1816, 'learning_rate': 1.3481e-04, 'epoch': 1.17, 'throughput': 36.33}
197
+
198
+ [INFO|2025-04-23 00:11:19] logging.py:143 >> {'loss': 0.1839, 'learning_rate': 1.2868e-04, 'epoch': 1.23, 'throughput': 36.33}
199
+
200
+ [INFO|2025-04-23 00:11:19] trainer.py:3984 >> Saving model checkpoint to saves/DeepSeek-LLM-7B-Base/lora/train_2025-04-22-23-24-06/checkpoint-100
201
+
202
+ [INFO|2025-04-23 00:11:19] configuration_utils.py:693 >> loading configuration file config.json from cache at /root/.cache/huggingface/hub/models--deepseek-ai--deepseek-llm-7b-base/snapshots/7683fea62db869066ddaff6a41d032262c490d4f/config.json
203
+
204
+ [INFO|2025-04-23 00:11:19] configuration_utils.py:765 >> Model config LlamaConfig {
205
+ "architectures": [
206
+ "LlamaForCausalLM"
207
+ ],
208
+ "attention_bias": false,
209
+ "attention_dropout": 0.0,
210
+ "bos_token_id": 1,
211
+ "eos_token_id": 2,
212
+ "head_dim": 128,
213
+ "hidden_act": "silu",
214
+ "hidden_size": 4096,
215
+ "initializer_range": 0.02,
216
+ "intermediate_size": 11008,
217
+ "max_position_embeddings": 4096,
218
+ "mlp_bias": false,
219
+ "model_type": "llama",
220
+ "num_attention_heads": 32,
221
+ "num_hidden_layers": 30,
222
+ "num_key_value_heads": 32,
223
+ "pretraining_tp": 1,
224
+ "rms_norm_eps": 1e-06,
225
+ "rope_scaling": null,
226
+ "rope_theta": 10000.0,
227
+ "tie_word_embeddings": false,
228
+ "torch_dtype": "bfloat16",
229
+ "transformers_version": "4.51.3",
230
+ "use_cache": true,
231
+ "vocab_size": 102400
232
+ }
233
+
234
+
235
+ [INFO|2025-04-23 00:11:19] tokenization_utils_base.py:2510 >> tokenizer config file saved in saves/DeepSeek-LLM-7B-Base/lora/train_2025-04-22-23-24-06/checkpoint-100/tokenizer_config.json
236
+
237
+ [INFO|2025-04-23 00:11:19] tokenization_utils_base.py:2519 >> Special tokens file saved in saves/DeepSeek-LLM-7B-Base/lora/train_2025-04-22-23-24-06/checkpoint-100/special_tokens_map.json
238
+
239
+ [INFO|2025-04-23 00:13:14] logging.py:143 >> {'loss': 0.2100, 'learning_rate': 1.2243e-04, 'epoch': 1.29, 'throughput': 36.25}
240
+
241
+ [INFO|2025-04-23 00:15:04] logging.py:143 >> {'loss': 0.2568, 'learning_rate': 1.1609e-04, 'epoch': 1.35, 'throughput': 36.24}
242
+
243
+ [INFO|2025-04-23 00:16:56] logging.py:143 >> {'loss': 0.2049, 'learning_rate': 1.0968e-04, 'epoch': 1.42, 'throughput': 36.27}
244
+
245
+ [INFO|2025-04-23 00:18:44] logging.py:143 >> {'loss': 0.2213, 'learning_rate': 1.0323e-04, 'epoch': 1.48, 'throughput': 36.28}
246
+
247
+ [INFO|2025-04-23 00:20:34] logging.py:143 >> {'loss': 0.2836, 'learning_rate': 9.6768e-05, 'epoch': 1.54, 'throughput': 36.28}
248
+
249
+ [INFO|2025-04-23 00:22:25] logging.py:143 >> {'loss': 0.2280, 'learning_rate': 9.0319e-05, 'epoch': 1.60, 'throughput': 36.28}
250
+
251
+ [INFO|2025-04-23 00:24:15] logging.py:143 >> {'loss': 0.1778, 'learning_rate': 8.3910e-05, 'epoch': 1.66, 'throughput': 36.28}
252
+
253
+ [INFO|2025-04-23 00:26:03] logging.py:143 >> {'loss': 0.1871, 'learning_rate': 7.7568e-05, 'epoch': 1.72, 'throughput': 36.27}
254
+
255
+ [INFO|2025-04-23 00:27:52] logging.py:143 >> {'loss': 0.2289, 'learning_rate': 7.1320e-05, 'epoch': 1.78, 'throughput': 36.27}
256
+
257
+ [INFO|2025-04-23 00:29:41] logging.py:143 >> {'loss': 0.2889, 'learning_rate': 6.5191e-05, 'epoch': 1.84, 'throughput': 36.27}
258
+
259
+ [INFO|2025-04-23 00:31:32] logging.py:143 >> {'loss': 0.2724, 'learning_rate': 5.9208e-05, 'epoch': 1.91, 'throughput': 36.25}
260
+
261
+ [INFO|2025-04-23 00:33:23] logging.py:143 >> {'loss': 0.2359, 'learning_rate': 5.3396e-05, 'epoch': 1.97, 'throughput': 36.26}
262
+
263
+ [INFO|2025-04-23 00:35:23] logging.py:143 >> {'loss': 0.1864, 'learning_rate': 4.7778e-05, 'epoch': 2.04, 'throughput': 36.26}
264
+
265
+ [INFO|2025-04-23 00:37:13] logging.py:143 >> {'loss': 0.0671, 'learning_rate': 4.2378e-05, 'epoch': 2.10, 'throughput': 36.26}
266
+
267
+ [INFO|2025-04-23 00:39:02] logging.py:143 >> {'loss': 0.1465, 'learning_rate': 3.7219e-05, 'epoch': 2.16, 'throughput': 36.23}
268
+
269
+ [INFO|2025-04-23 00:40:51] logging.py:143 >> {'loss': 0.0910, 'learning_rate': 3.2322e-05, 'epoch': 2.22, 'throughput': 36.22}
270
+
271
+ [INFO|2025-04-23 00:42:41] logging.py:143 >> {'loss': 0.1140, 'learning_rate': 2.7708e-05, 'epoch': 2.28, 'throughput': 36.21}
272
+
273
+ [INFO|2025-04-23 00:44:33] logging.py:143 >> {'loss': 0.1381, 'learning_rate': 2.3396e-05, 'epoch': 2.34, 'throughput': 36.23}
274
+
275
+ [INFO|2025-04-23 00:46:22] logging.py:143 >> {'loss': 0.0799, 'learning_rate': 1.9403e-05, 'epoch': 2.40, 'throughput': 36.24}
276
+
277
+ [INFO|2025-04-23 00:48:14] logging.py:143 >> {'loss': 0.0722, 'learning_rate': 1.5748e-05, 'epoch': 2.46, 'throughput': 36.25}
278
+
279
+ [INFO|2025-04-23 00:48:14] trainer.py:3984 >> Saving model checkpoint to saves/DeepSeek-LLM-7B-Base/lora/train_2025-04-22-23-24-06/checkpoint-200
280
+
281
+ [INFO|2025-04-23 00:48:14] configuration_utils.py:693 >> loading configuration file config.json from cache at /root/.cache/huggingface/hub/models--deepseek-ai--deepseek-llm-7b-base/snapshots/7683fea62db869066ddaff6a41d032262c490d4f/config.json
282
+
283
+ [INFO|2025-04-23 00:48:14] configuration_utils.py:765 >> Model config LlamaConfig {
284
+ "architectures": [
285
+ "LlamaForCausalLM"
286
+ ],
287
+ "attention_bias": false,
288
+ "attention_dropout": 0.0,
289
+ "bos_token_id": 1,
290
+ "eos_token_id": 2,
291
+ "head_dim": 128,
292
+ "hidden_act": "silu",
293
+ "hidden_size": 4096,
294
+ "initializer_range": 0.02,
295
+ "intermediate_size": 11008,
296
+ "max_position_embeddings": 4096,
297
+ "mlp_bias": false,
298
+ "model_type": "llama",
299
+ "num_attention_heads": 32,
300
+ "num_hidden_layers": 30,
301
+ "num_key_value_heads": 32,
302
+ "pretraining_tp": 1,
303
+ "rms_norm_eps": 1e-06,
304
+ "rope_scaling": null,
305
+ "rope_theta": 10000.0,
306
+ "tie_word_embeddings": false,
307
+ "torch_dtype": "bfloat16",
308
+ "transformers_version": "4.51.3",
309
+ "use_cache": true,
310
+ "vocab_size": 102400
311
+ }
312
+
313
+
314
+ [INFO|2025-04-23 00:48:15] tokenization_utils_base.py:2510 >> tokenizer config file saved in saves/DeepSeek-LLM-7B-Base/lora/train_2025-04-22-23-24-06/checkpoint-200/tokenizer_config.json
315
+
316
+ [INFO|2025-04-23 00:48:15] tokenization_utils_base.py:2519 >> Special tokens file saved in saves/DeepSeek-LLM-7B-Base/lora/train_2025-04-22-23-24-06/checkpoint-200/special_tokens_map.json
317
+
318
+ [INFO|2025-04-23 00:50:13] logging.py:143 >> {'loss': 0.1003, 'learning_rate': 1.2444e-05, 'epoch': 2.53, 'throughput': 36.23}
319
+
320
+ [INFO|2025-04-23 00:52:01] logging.py:143 >> {'loss': 0.1155, 'learning_rate': 9.5063e-06, 'epoch': 2.59, 'throughput': 36.21}
321
+
322
+ [INFO|2025-04-23 00:53:51] logging.py:143 >> {'loss': 0.0962, 'learning_rate': 6.9464e-06, 'epoch': 2.65, 'throughput': 36.22}
323
+
324
+ [INFO|2025-04-23 00:55:43] logging.py:143 >> {'loss': 0.0808, 'learning_rate': 4.7752e-06, 'epoch': 2.71, 'throughput': 36.23}
325
+
326
+ [INFO|2025-04-23 00:57:33] logging.py:143 >> {'loss': 0.1206, 'learning_rate': 3.0018e-06, 'epoch': 2.77, 'throughput': 36.22}
327
+
328
+ [INFO|2025-04-23 00:59:23] logging.py:143 >> {'loss': 0.1259, 'learning_rate': 1.6335e-06, 'epoch': 2.83, 'throughput': 36.22}
329
+
330
+ [INFO|2025-04-23 01:01:11] logging.py:143 >> {'loss': 0.1580, 'learning_rate': 6.7616e-07, 'epoch': 2.89, 'throughput': 36.22}
331
+
332
+ [INFO|2025-04-23 01:02:59] logging.py:143 >> {'loss': 0.0620, 'learning_rate': 1.3368e-07, 'epoch': 2.95, 'throughput': 36.21}
333
+
334
+ [INFO|2025-04-23 01:04:07] trainer.py:3984 >> Saving model checkpoint to saves/DeepSeek-LLM-7B-Base/lora/train_2025-04-22-23-24-06/checkpoint-243
335
+
336
+ [INFO|2025-04-23 01:04:07] configuration_utils.py:693 >> loading configuration file config.json from cache at /root/.cache/huggingface/hub/models--deepseek-ai--deepseek-llm-7b-base/snapshots/7683fea62db869066ddaff6a41d032262c490d4f/config.json
337
+
338
+ [INFO|2025-04-23 01:04:07] configuration_utils.py:765 >> Model config LlamaConfig {
339
+ "architectures": [
340
+ "LlamaForCausalLM"
341
+ ],
342
+ "attention_bias": false,
343
+ "attention_dropout": 0.0,
344
+ "bos_token_id": 1,
345
+ "eos_token_id": 2,
346
+ "head_dim": 128,
347
+ "hidden_act": "silu",
348
+ "hidden_size": 4096,
349
+ "initializer_range": 0.02,
350
+ "intermediate_size": 11008,
351
+ "max_position_embeddings": 4096,
352
+ "mlp_bias": false,
353
+ "model_type": "llama",
354
+ "num_attention_heads": 32,
355
+ "num_hidden_layers": 30,
356
+ "num_key_value_heads": 32,
357
+ "pretraining_tp": 1,
358
+ "rms_norm_eps": 1e-06,
359
+ "rope_scaling": null,
360
+ "rope_theta": 10000.0,
361
+ "tie_word_embeddings": false,
362
+ "torch_dtype": "bfloat16",
363
+ "transformers_version": "4.51.3",
364
+ "use_cache": true,
365
+ "vocab_size": 102400
366
+ }
367
+
368
+
369
+ [INFO|2025-04-23 01:04:08] tokenization_utils_base.py:2510 >> tokenizer config file saved in saves/DeepSeek-LLM-7B-Base/lora/train_2025-04-22-23-24-06/checkpoint-243/tokenizer_config.json
370
+
371
+ [INFO|2025-04-23 01:04:08] tokenization_utils_base.py:2519 >> Special tokens file saved in saves/DeepSeek-LLM-7B-Base/lora/train_2025-04-22-23-24-06/checkpoint-243/special_tokens_map.json
372
+
373
+ [INFO|2025-04-23 01:04:09] trainer.py:2681 >>
374
+
375
+ Training completed. Do not forget to share your model on huggingface.co/models =)
376
+
377
+
378
+
379
+ [INFO|2025-04-23 01:04:09] trainer.py:3984 >> Saving model checkpoint to saves/DeepSeek-LLM-7B-Base/lora/train_2025-04-22-23-24-06
380
+
381
+ [INFO|2025-04-23 01:04:09] configuration_utils.py:693 >> loading configuration file config.json from cache at /root/.cache/huggingface/hub/models--deepseek-ai--deepseek-llm-7b-base/snapshots/7683fea62db869066ddaff6a41d032262c490d4f/config.json
382
+
383
+ [INFO|2025-04-23 01:04:09] configuration_utils.py:765 >> Model config LlamaConfig {
384
+ "architectures": [
385
+ "LlamaForCausalLM"
386
+ ],
387
+ "attention_bias": false,
388
+ "attention_dropout": 0.0,
389
+ "bos_token_id": 1,
390
+ "eos_token_id": 2,
391
+ "head_dim": 128,
392
+ "hidden_act": "silu",
393
+ "hidden_size": 4096,
394
+ "initializer_range": 0.02,
395
+ "intermediate_size": 11008,
396
+ "max_position_embeddings": 4096,
397
+ "mlp_bias": false,
398
+ "model_type": "llama",
399
+ "num_attention_heads": 32,
400
+ "num_hidden_layers": 30,
401
+ "num_key_value_heads": 32,
402
+ "pretraining_tp": 1,
403
+ "rms_norm_eps": 1e-06,
404
+ "rope_scaling": null,
405
+ "rope_theta": 10000.0,
406
+ "tie_word_embeddings": false,
407
+ "torch_dtype": "bfloat16",
408
+ "transformers_version": "4.51.3",
409
+ "use_cache": true,
410
+ "vocab_size": 102400
411
+ }
412
+
413
+
414
+ [INFO|2025-04-23 01:04:09] tokenization_utils_base.py:2510 >> tokenizer config file saved in saves/DeepSeek-LLM-7B-Base/lora/train_2025-04-22-23-24-06/tokenizer_config.json
415
+
416
+ [INFO|2025-04-23 01:04:09] tokenization_utils_base.py:2519 >> Special tokens file saved in saves/DeepSeek-LLM-7B-Base/lora/train_2025-04-22-23-24-06/special_tokens_map.json
417
+
418
+ [WARNING|2025-04-23 01:04:10] logging.py:148 >> No metric eval_loss to plot.
419
+
420
+ [WARNING|2025-04-23 01:04:10] logging.py:148 >> No metric eval_accuracy to plot.
421
+
422
+ [INFO|2025-04-23 01:04:10] modelcard.py:450 >> Dropping the following result as it does not have all the necessary fields:
423
+ {'task': {'name': 'Causal Language Modeling', 'type': 'text-generation'}}
424
+
special_tokens_map.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<|begin▁of▁sentence|>",
4
+ "lstrip": false,
5
+ "normalized": true,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "<|end▁of▁sentence|>",
11
+ "lstrip": false,
12
+ "normalized": true,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "<|end▁of▁sentence|>",
18
+ "lstrip": false,
19
+ "normalized": true,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ }
23
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,141 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": true,
3
+ "add_eos_token": false,
4
+ "add_prefix_space": null,
5
+ "added_tokens_decoder": {
6
+ "100000": {
7
+ "content": "<|begin▁of▁sentence|>",
8
+ "lstrip": false,
9
+ "normalized": true,
10
+ "rstrip": false,
11
+ "single_word": false,
12
+ "special": true
13
+ },
14
+ "100001": {
15
+ "content": "<|end▁of▁sentence|>",
16
+ "lstrip": false,
17
+ "normalized": true,
18
+ "rstrip": false,
19
+ "single_word": false,
20
+ "special": true
21
+ },
22
+ "100002": {
23
+ "content": "ø",
24
+ "lstrip": false,
25
+ "normalized": true,
26
+ "rstrip": false,
27
+ "single_word": false,
28
+ "special": false
29
+ },
30
+ "100003": {
31
+ "content": "ö",
32
+ "lstrip": false,
33
+ "normalized": true,
34
+ "rstrip": false,
35
+ "single_word": false,
36
+ "special": false
37
+ },
38
+ "100004": {
39
+ "content": "ú",
40
+ "lstrip": false,
41
+ "normalized": true,
42
+ "rstrip": false,
43
+ "single_word": false,
44
+ "special": false
45
+ },
46
+ "100005": {
47
+ "content": "ÿ",
48
+ "lstrip": false,
49
+ "normalized": true,
50
+ "rstrip": false,
51
+ "single_word": false,
52
+ "special": false
53
+ },
54
+ "100006": {
55
+ "content": "õ",
56
+ "lstrip": false,
57
+ "normalized": true,
58
+ "rstrip": false,
59
+ "single_word": false,
60
+ "special": false
61
+ },
62
+ "100007": {
63
+ "content": "÷",
64
+ "lstrip": false,
65
+ "normalized": true,
66
+ "rstrip": false,
67
+ "single_word": false,
68
+ "special": false
69
+ },
70
+ "100008": {
71
+ "content": "û",
72
+ "lstrip": false,
73
+ "normalized": true,
74
+ "rstrip": false,
75
+ "single_word": false,
76
+ "special": false
77
+ },
78
+ "100009": {
79
+ "content": "ý",
80
+ "lstrip": false,
81
+ "normalized": true,
82
+ "rstrip": false,
83
+ "single_word": false,
84
+ "special": false
85
+ },
86
+ "100010": {
87
+ "content": "À",
88
+ "lstrip": false,
89
+ "normalized": true,
90
+ "rstrip": false,
91
+ "single_word": false,
92
+ "special": false
93
+ },
94
+ "100011": {
95
+ "content": "ù",
96
+ "lstrip": false,
97
+ "normalized": true,
98
+ "rstrip": false,
99
+ "single_word": false,
100
+ "special": false
101
+ },
102
+ "100012": {
103
+ "content": "Á",
104
+ "lstrip": false,
105
+ "normalized": true,
106
+ "rstrip": false,
107
+ "single_word": false,
108
+ "special": false
109
+ },
110
+ "100013": {
111
+ "content": "þ",
112
+ "lstrip": false,
113
+ "normalized": true,
114
+ "rstrip": false,
115
+ "single_word": false,
116
+ "special": false
117
+ },
118
+ "100014": {
119
+ "content": "ü",
120
+ "lstrip": false,
121
+ "normalized": true,
122
+ "rstrip": false,
123
+ "single_word": false,
124
+ "special": false
125
+ }
126
+ },
127
+ "bos_token": "<|begin▁of▁sentence|>",
128
+ "chat_template": "{% if messages[0]['role'] == 'system' %}{% set loop_messages = messages[1:] %}{% set system_message = messages[0]['content'] %}{% else %}{% set loop_messages = messages %}{% endif %}{% if system_message is defined %}{{ 'System: ' + system_message + '<|end▁of▁sentence|>' + '\n' }}{% endif %}{% for message in loop_messages %}{% set content = message['content'] %}{% if message['role'] == 'user' %}{{ 'Human: ' + content + '<|end▁of▁sentence|>' + '\nAssistant:' }}{% elif message['role'] == 'assistant' %}{{ content + '<|end▁of▁sentence|>' + '\n' }}{% endif %}{% endfor %}",
129
+ "clean_up_tokenization_spaces": false,
130
+ "eos_token": "<|end▁of▁sentence|>",
131
+ "extra_special_tokens": {},
132
+ "legacy": true,
133
+ "model_max_length": 4096,
134
+ "pad_token": "<|end▁of▁sentence|>",
135
+ "padding_side": "right",
136
+ "sp_model_kwargs": {},
137
+ "split_special_tokens": false,
138
+ "tokenizer_class": "LlamaTokenizerFast",
139
+ "unk_token": null,
140
+ "use_default_system_prompt": false
141
+ }
train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 2.9908256880733948,
3
+ "num_input_tokens_seen": 194864,
4
+ "total_flos": 7611007265931264.0,
5
+ "train_loss": 0.34003226710445106,
6
+ "train_runtime": 5382.7216,
7
+ "train_samples_per_second": 0.728,
8
+ "train_steps_per_second": 0.045
9
+ }
trainer_log.jsonl ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"current_steps": 5, "total_steps": 243, "loss": 2.3883, "lr": 0.00019986631570270832, "epoch": 0.06116207951070336, "percentage": 2.06, "elapsed_time": "0:01:49", "remaining_time": "1:27:12", "throughput": 36.53, "total_tokens": 4016}
2
+ {"current_steps": 10, "total_steps": 243, "loss": 1.4752, "lr": 0.00019932383577419432, "epoch": 0.12232415902140673, "percentage": 4.12, "elapsed_time": "0:03:39", "remaining_time": "1:25:04", "throughput": 36.29, "total_tokens": 7952}
3
+ {"current_steps": 15, "total_steps": 243, "loss": 1.1174, "lr": 0.0001983664691986601, "epoch": 0.1834862385321101, "percentage": 6.17, "elapsed_time": "0:05:30", "remaining_time": "1:23:42", "throughput": 36.12, "total_tokens": 11936}
4
+ {"current_steps": 20, "total_steps": 243, "loss": 0.7815, "lr": 0.00019699821500217434, "epoch": 0.24464831804281345, "percentage": 8.23, "elapsed_time": "0:07:18", "remaining_time": "1:21:32", "throughput": 36.03, "total_tokens": 15808}
5
+ {"current_steps": 25, "total_steps": 243, "loss": 0.632, "lr": 0.00019522478853384155, "epoch": 0.3058103975535168, "percentage": 10.29, "elapsed_time": "0:09:09", "remaining_time": "1:19:48", "throughput": 36.22, "total_tokens": 19888}
6
+ {"current_steps": 30, "total_steps": 243, "loss": 0.4603, "lr": 0.00019305359759215685, "epoch": 0.3669724770642202, "percentage": 12.35, "elapsed_time": "0:10:59", "remaining_time": "1:17:59", "throughput": 36.2, "total_tokens": 23856}
7
+ {"current_steps": 35, "total_steps": 243, "loss": 0.4761, "lr": 0.00019049371148181253, "epoch": 0.42813455657492355, "percentage": 14.4, "elapsed_time": "0:12:50", "remaining_time": "1:16:19", "throughput": 36.25, "total_tokens": 27936}
8
+ {"current_steps": 40, "total_steps": 243, "loss": 0.4525, "lr": 0.0001875558231302091, "epoch": 0.4892966360856269, "percentage": 16.46, "elapsed_time": "0:14:39", "remaining_time": "1:14:25", "throughput": 36.22, "total_tokens": 31872}
9
+ {"current_steps": 45, "total_steps": 243, "loss": 0.4803, "lr": 0.00018425220442191495, "epoch": 0.5504587155963303, "percentage": 18.52, "elapsed_time": "0:16:28", "remaining_time": "1:12:28", "throughput": 36.12, "total_tokens": 35696}
10
+ {"current_steps": 50, "total_steps": 243, "loss": 0.3933, "lr": 0.00018059665493764743, "epoch": 0.6116207951070336, "percentage": 20.58, "elapsed_time": "0:18:18", "remaining_time": "1:10:41", "throughput": 36.11, "total_tokens": 39680}
11
+ {"current_steps": 55, "total_steps": 243, "loss": 0.3919, "lr": 0.0001766044443118978, "epoch": 0.672782874617737, "percentage": 22.63, "elapsed_time": "0:20:12", "remaining_time": "1:09:05", "throughput": 36.22, "total_tokens": 43920}
12
+ {"current_steps": 60, "total_steps": 243, "loss": 0.3804, "lr": 0.00017229224844997928, "epoch": 0.7339449541284404, "percentage": 24.69, "elapsed_time": "0:22:06", "remaining_time": "1:07:25", "throughput": 36.31, "total_tokens": 48160}
13
+ {"current_steps": 65, "total_steps": 243, "loss": 0.3576, "lr": 0.00016767807987092621, "epoch": 0.7951070336391437, "percentage": 26.75, "elapsed_time": "0:23:55", "remaining_time": "1:05:30", "throughput": 36.28, "total_tokens": 52080}
14
+ {"current_steps": 70, "total_steps": 243, "loss": 0.3343, "lr": 0.00016278121246720987, "epoch": 0.8562691131498471, "percentage": 28.81, "elapsed_time": "0:25:45", "remaining_time": "1:03:39", "throughput": 36.32, "total_tokens": 56128}
15
+ {"current_steps": 75, "total_steps": 243, "loss": 0.4526, "lr": 0.00015762210099555803, "epoch": 0.9174311926605505, "percentage": 30.86, "elapsed_time": "0:27:32", "remaining_time": "1:01:42", "throughput": 36.27, "total_tokens": 59952}
16
+ {"current_steps": 80, "total_steps": 243, "loss": 0.4048, "lr": 0.00015222229563517385, "epoch": 0.9785932721712538, "percentage": 32.92, "elapsed_time": "0:29:21", "remaining_time": "0:59:48", "throughput": 36.24, "total_tokens": 63840}
17
+ {"current_steps": 85, "total_steps": 243, "loss": 0.3599, "lr": 0.0001466043519702539, "epoch": 1.0489296636085628, "percentage": 34.98, "elapsed_time": "0:31:20", "remaining_time": "0:58:14", "throughput": 36.26, "total_tokens": 68176}
18
+ {"current_steps": 90, "total_steps": 243, "loss": 0.2197, "lr": 0.00014079173677281837, "epoch": 1.110091743119266, "percentage": 37.04, "elapsed_time": "0:33:12", "remaining_time": "0:56:26", "throughput": 36.31, "total_tokens": 72336}
19
+ {"current_steps": 95, "total_steps": 243, "loss": 0.1816, "lr": 0.00013480872997940905, "epoch": 1.1712538226299694, "percentage": 39.09, "elapsed_time": "0:35:02", "remaining_time": "0:54:35", "throughput": 36.33, "total_tokens": 76384}
20
+ {"current_steps": 100, "total_steps": 243, "loss": 0.1839, "lr": 0.00012868032327110904, "epoch": 1.2324159021406729, "percentage": 41.15, "elapsed_time": "0:36:52", "remaining_time": "0:52:43", "throughput": 36.33, "total_tokens": 80368}
21
+ {"current_steps": 105, "total_steps": 243, "loss": 0.21, "lr": 0.00012243211568052677, "epoch": 1.2935779816513762, "percentage": 43.21, "elapsed_time": "0:38:48", "remaining_time": "0:50:59", "throughput": 36.25, "total_tokens": 84400}
22
+ {"current_steps": 110, "total_steps": 243, "loss": 0.2568, "lr": 0.00011609020666180575, "epoch": 1.3547400611620795, "percentage": 45.27, "elapsed_time": "0:40:37", "remaining_time": "0:49:07", "throughput": 36.24, "total_tokens": 88352}
23
+ {"current_steps": 115, "total_steps": 243, "loss": 0.2049, "lr": 0.00010968108707031792, "epoch": 1.4159021406727827, "percentage": 47.33, "elapsed_time": "0:42:29", "remaining_time": "0:47:17", "throughput": 36.27, "total_tokens": 92464}
24
+ {"current_steps": 120, "total_steps": 243, "loss": 0.2213, "lr": 0.00010323152850743107, "epoch": 1.4770642201834863, "percentage": 49.38, "elapsed_time": "0:44:18", "remaining_time": "0:45:24", "throughput": 36.28, "total_tokens": 96432}
25
+ {"current_steps": 125, "total_steps": 243, "loss": 0.2836, "lr": 9.676847149256895e-05, "epoch": 1.5382262996941896, "percentage": 51.44, "elapsed_time": "0:46:08", "remaining_time": "0:43:33", "throughput": 36.28, "total_tokens": 100416}
26
+ {"current_steps": 130, "total_steps": 243, "loss": 0.228, "lr": 9.03189129296821e-05, "epoch": 1.599388379204893, "percentage": 53.5, "elapsed_time": "0:47:58", "remaining_time": "0:41:42", "throughput": 36.28, "total_tokens": 104432}
27
+ {"current_steps": 135, "total_steps": 243, "loss": 0.1778, "lr": 8.390979333819426e-05, "epoch": 1.6605504587155964, "percentage": 55.56, "elapsed_time": "0:49:48", "remaining_time": "0:39:50", "throughput": 36.28, "total_tokens": 108432}
28
+ {"current_steps": 140, "total_steps": 243, "loss": 0.1871, "lr": 7.756788431947326e-05, "epoch": 1.7217125382262997, "percentage": 57.61, "elapsed_time": "0:51:36", "remaining_time": "0:37:58", "throughput": 36.27, "total_tokens": 112320}
29
+ {"current_steps": 145, "total_steps": 243, "loss": 0.2289, "lr": 7.131967672889101e-05, "epoch": 1.782874617737003, "percentage": 59.67, "elapsed_time": "0:53:26", "remaining_time": "0:36:06", "throughput": 36.27, "total_tokens": 116272}
30
+ {"current_steps": 150, "total_steps": 243, "loss": 0.2889, "lr": 6.519127002059095e-05, "epoch": 1.8440366972477065, "percentage": 61.73, "elapsed_time": "0:55:14", "remaining_time": "0:34:15", "throughput": 36.27, "total_tokens": 120224}
31
+ {"current_steps": 155, "total_steps": 243, "loss": 0.2724, "lr": 5.920826322718165e-05, "epoch": 1.9051987767584098, "percentage": 63.79, "elapsed_time": "0:57:05", "remaining_time": "0:32:24", "throughput": 36.25, "total_tokens": 124176}
32
+ {"current_steps": 160, "total_steps": 243, "loss": 0.2359, "lr": 5.339564802974615e-05, "epoch": 1.9663608562691133, "percentage": 65.84, "elapsed_time": "0:58:56", "remaining_time": "0:30:34", "throughput": 36.26, "total_tokens": 128256}
33
+ {"current_steps": 165, "total_steps": 243, "loss": 0.1864, "lr": 4.777770436482617e-05, "epoch": 2.036697247706422, "percentage": 67.9, "elapsed_time": "1:00:56", "remaining_time": "0:28:48", "throughput": 36.26, "total_tokens": 132576}
34
+ {"current_steps": 170, "total_steps": 243, "loss": 0.0671, "lr": 4.2377899004441966e-05, "epoch": 2.0978593272171255, "percentage": 69.96, "elapsed_time": "1:02:47", "remaining_time": "0:26:57", "throughput": 36.26, "total_tokens": 136608}
35
+ {"current_steps": 175, "total_steps": 243, "loss": 0.1465, "lr": 3.721878753279017e-05, "epoch": 2.159021406727829, "percentage": 72.02, "elapsed_time": "1:04:35", "remaining_time": "0:25:05", "throughput": 36.23, "total_tokens": 140416}
36
+ {"current_steps": 180, "total_steps": 243, "loss": 0.091, "lr": 3.2321920129073816e-05, "epoch": 2.220183486238532, "percentage": 74.07, "elapsed_time": "1:06:24", "remaining_time": "0:23:14", "throughput": 36.22, "total_tokens": 144336}
37
+ {"current_steps": 185, "total_steps": 243, "loss": 0.114, "lr": 2.770775155002071e-05, "epoch": 2.2813455657492354, "percentage": 76.13, "elapsed_time": "1:08:14", "remaining_time": "0:21:23", "throughput": 36.21, "total_tokens": 148256}
38
+ {"current_steps": 190, "total_steps": 243, "loss": 0.1381, "lr": 2.339555568810221e-05, "epoch": 2.3425076452599387, "percentage": 78.19, "elapsed_time": "1:10:06", "remaining_time": "0:19:33", "throughput": 36.23, "total_tokens": 152416}
39
+ {"current_steps": 195, "total_steps": 243, "loss": 0.0799, "lr": 1.9403345062352573e-05, "epoch": 2.4036697247706424, "percentage": 80.25, "elapsed_time": "1:11:56", "remaining_time": "0:17:42", "throughput": 36.24, "total_tokens": 156400}
40
+ {"current_steps": 200, "total_steps": 243, "loss": 0.0722, "lr": 1.5747795578085046e-05, "epoch": 2.4648318042813457, "percentage": 82.3, "elapsed_time": "1:13:47", "remaining_time": "0:15:52", "throughput": 36.25, "total_tokens": 160528}
41
+ {"current_steps": 205, "total_steps": 243, "loss": 0.1003, "lr": 1.2444176869790925e-05, "epoch": 2.525993883792049, "percentage": 84.36, "elapsed_time": "1:15:46", "remaining_time": "0:14:02", "throughput": 36.23, "total_tokens": 164720}
42
+ {"current_steps": 210, "total_steps": 243, "loss": 0.1155, "lr": 9.506288518187467e-06, "epoch": 2.5871559633027523, "percentage": 86.42, "elapsed_time": "1:17:34", "remaining_time": "0:12:11", "throughput": 36.21, "total_tokens": 168560}
43
+ {"current_steps": 215, "total_steps": 243, "loss": 0.0962, "lr": 6.946402407843155e-06, "epoch": 2.6483180428134556, "percentage": 88.48, "elapsed_time": "1:19:25", "remaining_time": "0:10:20", "throughput": 36.22, "total_tokens": 172608}
44
+ {"current_steps": 220, "total_steps": 243, "loss": 0.0808, "lr": 4.775211466158469e-06, "epoch": 2.709480122324159, "percentage": 90.53, "elapsed_time": "1:21:16", "remaining_time": "0:08:29", "throughput": 36.23, "total_tokens": 176656}
45
+ {"current_steps": 225, "total_steps": 243, "loss": 0.1206, "lr": 3.0017849978256516e-06, "epoch": 2.770642201834862, "percentage": 92.59, "elapsed_time": "1:23:06", "remaining_time": "0:06:38", "throughput": 36.22, "total_tokens": 180608}
46
+ {"current_steps": 230, "total_steps": 243, "loss": 0.1259, "lr": 1.6335308013398886e-06, "epoch": 2.8318042813455655, "percentage": 94.65, "elapsed_time": "1:24:56", "remaining_time": "0:04:48", "throughput": 36.22, "total_tokens": 184592}
47
+ {"current_steps": 235, "total_steps": 243, "loss": 0.158, "lr": 6.761642258056978e-07, "epoch": 2.8929663608562692, "percentage": 96.71, "elapsed_time": "1:26:44", "remaining_time": "0:02:57", "throughput": 36.22, "total_tokens": 188496}
48
+ {"current_steps": 240, "total_steps": 243, "loss": 0.062, "lr": 1.3368429729168076e-07, "epoch": 2.9541284403669725, "percentage": 98.77, "elapsed_time": "1:28:33", "remaining_time": "0:01:06", "throughput": 36.21, "total_tokens": 192368}
49
+ {"current_steps": 243, "total_steps": 243, "epoch": 2.9908256880733948, "percentage": 100.0, "elapsed_time": "1:29:42", "remaining_time": "0:00:00", "throughput": 36.2, "total_tokens": 194864}
trainer_state.json ADDED
@@ -0,0 +1,428 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 2.9908256880733948,
6
+ "eval_steps": 500,
7
+ "global_step": 243,
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.06116207951070336,
14
+ "grad_norm": 1.7883533239364624,
15
+ "learning_rate": 0.00019986631570270832,
16
+ "loss": 2.3883,
17
+ "num_input_tokens_seen": 4016,
18
+ "step": 5
19
+ },
20
+ {
21
+ "epoch": 0.12232415902140673,
22
+ "grad_norm": 2.9251019954681396,
23
+ "learning_rate": 0.00019932383577419432,
24
+ "loss": 1.4752,
25
+ "num_input_tokens_seen": 7952,
26
+ "step": 10
27
+ },
28
+ {
29
+ "epoch": 0.1834862385321101,
30
+ "grad_norm": 5.290874004364014,
31
+ "learning_rate": 0.0001983664691986601,
32
+ "loss": 1.1174,
33
+ "num_input_tokens_seen": 11936,
34
+ "step": 15
35
+ },
36
+ {
37
+ "epoch": 0.24464831804281345,
38
+ "grad_norm": 6.394321918487549,
39
+ "learning_rate": 0.00019699821500217434,
40
+ "loss": 0.7815,
41
+ "num_input_tokens_seen": 15808,
42
+ "step": 20
43
+ },
44
+ {
45
+ "epoch": 0.3058103975535168,
46
+ "grad_norm": 1.3936750888824463,
47
+ "learning_rate": 0.00019522478853384155,
48
+ "loss": 0.632,
49
+ "num_input_tokens_seen": 19888,
50
+ "step": 25
51
+ },
52
+ {
53
+ "epoch": 0.3669724770642202,
54
+ "grad_norm": 0.6125549674034119,
55
+ "learning_rate": 0.00019305359759215685,
56
+ "loss": 0.4603,
57
+ "num_input_tokens_seen": 23856,
58
+ "step": 30
59
+ },
60
+ {
61
+ "epoch": 0.42813455657492355,
62
+ "grad_norm": 0.8421280980110168,
63
+ "learning_rate": 0.00019049371148181253,
64
+ "loss": 0.4761,
65
+ "num_input_tokens_seen": 27936,
66
+ "step": 35
67
+ },
68
+ {
69
+ "epoch": 0.4892966360856269,
70
+ "grad_norm": 0.7261258959770203,
71
+ "learning_rate": 0.0001875558231302091,
72
+ "loss": 0.4525,
73
+ "num_input_tokens_seen": 31872,
74
+ "step": 40
75
+ },
76
+ {
77
+ "epoch": 0.5504587155963303,
78
+ "grad_norm": 0.8600367903709412,
79
+ "learning_rate": 0.00018425220442191495,
80
+ "loss": 0.4803,
81
+ "num_input_tokens_seen": 35696,
82
+ "step": 45
83
+ },
84
+ {
85
+ "epoch": 0.6116207951070336,
86
+ "grad_norm": 0.8055685758590698,
87
+ "learning_rate": 0.00018059665493764743,
88
+ "loss": 0.3933,
89
+ "num_input_tokens_seen": 39680,
90
+ "step": 50
91
+ },
92
+ {
93
+ "epoch": 0.672782874617737,
94
+ "grad_norm": 0.7538366317749023,
95
+ "learning_rate": 0.0001766044443118978,
96
+ "loss": 0.3919,
97
+ "num_input_tokens_seen": 43920,
98
+ "step": 55
99
+ },
100
+ {
101
+ "epoch": 0.7339449541284404,
102
+ "grad_norm": 0.8907052874565125,
103
+ "learning_rate": 0.00017229224844997928,
104
+ "loss": 0.3804,
105
+ "num_input_tokens_seen": 48160,
106
+ "step": 60
107
+ },
108
+ {
109
+ "epoch": 0.7951070336391437,
110
+ "grad_norm": 0.7426741123199463,
111
+ "learning_rate": 0.00016767807987092621,
112
+ "loss": 0.3576,
113
+ "num_input_tokens_seen": 52080,
114
+ "step": 65
115
+ },
116
+ {
117
+ "epoch": 0.8562691131498471,
118
+ "grad_norm": 0.6457409262657166,
119
+ "learning_rate": 0.00016278121246720987,
120
+ "loss": 0.3343,
121
+ "num_input_tokens_seen": 56128,
122
+ "step": 70
123
+ },
124
+ {
125
+ "epoch": 0.9174311926605505,
126
+ "grad_norm": 0.9887730479240417,
127
+ "learning_rate": 0.00015762210099555803,
128
+ "loss": 0.4526,
129
+ "num_input_tokens_seen": 59952,
130
+ "step": 75
131
+ },
132
+ {
133
+ "epoch": 0.9785932721712538,
134
+ "grad_norm": 0.9649442434310913,
135
+ "learning_rate": 0.00015222229563517385,
136
+ "loss": 0.4048,
137
+ "num_input_tokens_seen": 63840,
138
+ "step": 80
139
+ },
140
+ {
141
+ "epoch": 1.0489296636085628,
142
+ "grad_norm": 0.8496802449226379,
143
+ "learning_rate": 0.0001466043519702539,
144
+ "loss": 0.3599,
145
+ "num_input_tokens_seen": 68176,
146
+ "step": 85
147
+ },
148
+ {
149
+ "epoch": 1.110091743119266,
150
+ "grad_norm": 0.6206537485122681,
151
+ "learning_rate": 0.00014079173677281837,
152
+ "loss": 0.2197,
153
+ "num_input_tokens_seen": 72336,
154
+ "step": 90
155
+ },
156
+ {
157
+ "epoch": 1.1712538226299694,
158
+ "grad_norm": 0.7237979173660278,
159
+ "learning_rate": 0.00013480872997940905,
160
+ "loss": 0.1816,
161
+ "num_input_tokens_seen": 76384,
162
+ "step": 95
163
+ },
164
+ {
165
+ "epoch": 1.2324159021406729,
166
+ "grad_norm": 1.0752713680267334,
167
+ "learning_rate": 0.00012868032327110904,
168
+ "loss": 0.1839,
169
+ "num_input_tokens_seen": 80368,
170
+ "step": 100
171
+ },
172
+ {
173
+ "epoch": 1.2935779816513762,
174
+ "grad_norm": 1.0526070594787598,
175
+ "learning_rate": 0.00012243211568052677,
176
+ "loss": 0.21,
177
+ "num_input_tokens_seen": 84400,
178
+ "step": 105
179
+ },
180
+ {
181
+ "epoch": 1.3547400611620795,
182
+ "grad_norm": 1.073427677154541,
183
+ "learning_rate": 0.00011609020666180575,
184
+ "loss": 0.2568,
185
+ "num_input_tokens_seen": 88352,
186
+ "step": 110
187
+ },
188
+ {
189
+ "epoch": 1.4159021406727827,
190
+ "grad_norm": 0.8212242722511292,
191
+ "learning_rate": 0.00010968108707031792,
192
+ "loss": 0.2049,
193
+ "num_input_tokens_seen": 92464,
194
+ "step": 115
195
+ },
196
+ {
197
+ "epoch": 1.4770642201834863,
198
+ "grad_norm": 1.1588611602783203,
199
+ "learning_rate": 0.00010323152850743107,
200
+ "loss": 0.2213,
201
+ "num_input_tokens_seen": 96432,
202
+ "step": 120
203
+ },
204
+ {
205
+ "epoch": 1.5382262996941896,
206
+ "grad_norm": 1.08698308467865,
207
+ "learning_rate": 9.676847149256895e-05,
208
+ "loss": 0.2836,
209
+ "num_input_tokens_seen": 100416,
210
+ "step": 125
211
+ },
212
+ {
213
+ "epoch": 1.599388379204893,
214
+ "grad_norm": 1.1150074005126953,
215
+ "learning_rate": 9.03189129296821e-05,
216
+ "loss": 0.228,
217
+ "num_input_tokens_seen": 104432,
218
+ "step": 130
219
+ },
220
+ {
221
+ "epoch": 1.6605504587155964,
222
+ "grad_norm": 0.7235325574874878,
223
+ "learning_rate": 8.390979333819426e-05,
224
+ "loss": 0.1778,
225
+ "num_input_tokens_seen": 108432,
226
+ "step": 135
227
+ },
228
+ {
229
+ "epoch": 1.7217125382262997,
230
+ "grad_norm": 0.8855745196342468,
231
+ "learning_rate": 7.756788431947326e-05,
232
+ "loss": 0.1871,
233
+ "num_input_tokens_seen": 112320,
234
+ "step": 140
235
+ },
236
+ {
237
+ "epoch": 1.782874617737003,
238
+ "grad_norm": 1.3950036764144897,
239
+ "learning_rate": 7.131967672889101e-05,
240
+ "loss": 0.2289,
241
+ "num_input_tokens_seen": 116272,
242
+ "step": 145
243
+ },
244
+ {
245
+ "epoch": 1.8440366972477065,
246
+ "grad_norm": 1.182417392730713,
247
+ "learning_rate": 6.519127002059095e-05,
248
+ "loss": 0.2889,
249
+ "num_input_tokens_seen": 120224,
250
+ "step": 150
251
+ },
252
+ {
253
+ "epoch": 1.9051987767584098,
254
+ "grad_norm": 1.020156741142273,
255
+ "learning_rate": 5.920826322718165e-05,
256
+ "loss": 0.2724,
257
+ "num_input_tokens_seen": 124176,
258
+ "step": 155
259
+ },
260
+ {
261
+ "epoch": 1.9663608562691133,
262
+ "grad_norm": 0.8830191493034363,
263
+ "learning_rate": 5.339564802974615e-05,
264
+ "loss": 0.2359,
265
+ "num_input_tokens_seen": 128256,
266
+ "step": 160
267
+ },
268
+ {
269
+ "epoch": 2.036697247706422,
270
+ "grad_norm": 0.8336369395256042,
271
+ "learning_rate": 4.777770436482617e-05,
272
+ "loss": 0.1864,
273
+ "num_input_tokens_seen": 132576,
274
+ "step": 165
275
+ },
276
+ {
277
+ "epoch": 2.0978593272171255,
278
+ "grad_norm": 0.583479106426239,
279
+ "learning_rate": 4.2377899004441966e-05,
280
+ "loss": 0.0671,
281
+ "num_input_tokens_seen": 136608,
282
+ "step": 170
283
+ },
284
+ {
285
+ "epoch": 2.159021406727829,
286
+ "grad_norm": 0.686152994632721,
287
+ "learning_rate": 3.721878753279017e-05,
288
+ "loss": 0.1465,
289
+ "num_input_tokens_seen": 140416,
290
+ "step": 175
291
+ },
292
+ {
293
+ "epoch": 2.220183486238532,
294
+ "grad_norm": 0.6811323761940002,
295
+ "learning_rate": 3.2321920129073816e-05,
296
+ "loss": 0.091,
297
+ "num_input_tokens_seen": 144336,
298
+ "step": 180
299
+ },
300
+ {
301
+ "epoch": 2.2813455657492354,
302
+ "grad_norm": 0.5875197052955627,
303
+ "learning_rate": 2.770775155002071e-05,
304
+ "loss": 0.114,
305
+ "num_input_tokens_seen": 148256,
306
+ "step": 185
307
+ },
308
+ {
309
+ "epoch": 2.3425076452599387,
310
+ "grad_norm": 0.6630216836929321,
311
+ "learning_rate": 2.339555568810221e-05,
312
+ "loss": 0.1381,
313
+ "num_input_tokens_seen": 152416,
314
+ "step": 190
315
+ },
316
+ {
317
+ "epoch": 2.4036697247706424,
318
+ "grad_norm": 0.7730478644371033,
319
+ "learning_rate": 1.9403345062352573e-05,
320
+ "loss": 0.0799,
321
+ "num_input_tokens_seen": 156400,
322
+ "step": 195
323
+ },
324
+ {
325
+ "epoch": 2.4648318042813457,
326
+ "grad_norm": 0.7855842709541321,
327
+ "learning_rate": 1.5747795578085046e-05,
328
+ "loss": 0.0722,
329
+ "num_input_tokens_seen": 160528,
330
+ "step": 200
331
+ },
332
+ {
333
+ "epoch": 2.525993883792049,
334
+ "grad_norm": 0.7239196300506592,
335
+ "learning_rate": 1.2444176869790925e-05,
336
+ "loss": 0.1003,
337
+ "num_input_tokens_seen": 164720,
338
+ "step": 205
339
+ },
340
+ {
341
+ "epoch": 2.5871559633027523,
342
+ "grad_norm": 0.8815809488296509,
343
+ "learning_rate": 9.506288518187467e-06,
344
+ "loss": 0.1155,
345
+ "num_input_tokens_seen": 168560,
346
+ "step": 210
347
+ },
348
+ {
349
+ "epoch": 2.6483180428134556,
350
+ "grad_norm": 1.0388275384902954,
351
+ "learning_rate": 6.946402407843155e-06,
352
+ "loss": 0.0962,
353
+ "num_input_tokens_seen": 172608,
354
+ "step": 215
355
+ },
356
+ {
357
+ "epoch": 2.709480122324159,
358
+ "grad_norm": 0.9285043478012085,
359
+ "learning_rate": 4.775211466158469e-06,
360
+ "loss": 0.0808,
361
+ "num_input_tokens_seen": 176656,
362
+ "step": 220
363
+ },
364
+ {
365
+ "epoch": 2.770642201834862,
366
+ "grad_norm": 0.9785643815994263,
367
+ "learning_rate": 3.0017849978256516e-06,
368
+ "loss": 0.1206,
369
+ "num_input_tokens_seen": 180608,
370
+ "step": 225
371
+ },
372
+ {
373
+ "epoch": 2.8318042813455655,
374
+ "grad_norm": 0.9847484827041626,
375
+ "learning_rate": 1.6335308013398886e-06,
376
+ "loss": 0.1259,
377
+ "num_input_tokens_seen": 184592,
378
+ "step": 230
379
+ },
380
+ {
381
+ "epoch": 2.8929663608562692,
382
+ "grad_norm": 1.2911163568496704,
383
+ "learning_rate": 6.761642258056978e-07,
384
+ "loss": 0.158,
385
+ "num_input_tokens_seen": 188496,
386
+ "step": 235
387
+ },
388
+ {
389
+ "epoch": 2.9541284403669725,
390
+ "grad_norm": 0.7206448912620544,
391
+ "learning_rate": 1.3368429729168076e-07,
392
+ "loss": 0.062,
393
+ "num_input_tokens_seen": 192368,
394
+ "step": 240
395
+ },
396
+ {
397
+ "epoch": 2.9908256880733948,
398
+ "num_input_tokens_seen": 194864,
399
+ "step": 243,
400
+ "total_flos": 7611007265931264.0,
401
+ "train_loss": 0.34003226710445106,
402
+ "train_runtime": 5382.7216,
403
+ "train_samples_per_second": 0.728,
404
+ "train_steps_per_second": 0.045
405
+ }
406
+ ],
407
+ "logging_steps": 5,
408
+ "max_steps": 243,
409
+ "num_input_tokens_seen": 194864,
410
+ "num_train_epochs": 3,
411
+ "save_steps": 100,
412
+ "stateful_callbacks": {
413
+ "TrainerControl": {
414
+ "args": {
415
+ "should_epoch_stop": false,
416
+ "should_evaluate": false,
417
+ "should_log": false,
418
+ "should_save": true,
419
+ "should_training_stop": true
420
+ },
421
+ "attributes": {}
422
+ }
423
+ },
424
+ "total_flos": 7611007265931264.0,
425
+ "train_batch_size": 2,
426
+ "trial_name": null,
427
+ "trial_params": null
428
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e398eee9bde032e96ecf2e3858ff52edd7fac6c2282b2292935f8e43f27802b3
3
+ size 5752
training_args.yaml ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ bf16: true
2
+ cutoff_len: 256
3
+ dataset: alpaca_en_demo
4
+ dataset_dir: data
5
+ ddp_timeout: 180000000
6
+ do_train: true
7
+ double_quantization: true
8
+ finetuning_type: lora
9
+ flash_attn: auto
10
+ gradient_accumulation_steps: 8
11
+ include_num_input_tokens_seen: true
12
+ learning_rate: 0.0002
13
+ logging_steps: 5
14
+ lora_alpha: 16
15
+ lora_dropout: 0
16
+ lora_rank: 8
17
+ lora_target: all
18
+ lr_scheduler_type: cosine
19
+ max_grad_norm: 0.3
20
+ max_samples: 10000
21
+ model_name_or_path: deepseek-ai/deepseek-llm-7b-base
22
+ num_train_epochs: 3.0
23
+ optim: adamw_torch
24
+ output_dir: saves/DeepSeek-LLM-7B-Base/lora/train_2025-04-22-23-24-06
25
+ packing: false
26
+ per_device_train_batch_size: 2
27
+ plot_loss: true
28
+ preprocessing_num_workers: 16
29
+ quantization_bit: 4
30
+ quantization_method: bnb
31
+ report_to: none
32
+ save_steps: 100
33
+ stage: sft
34
+ template: default
35
+ trust_remote_code: true
36
+ warmup_steps: 0
training_loss.png ADDED