Upload 17 files
Browse files- README.md +62 -0
- adapter_config.json +31 -0
- adapter_model.safetensors +3 -0
- added_tokens.json +13 -0
- all_results.json +9 -0
- llamaboard_config.yaml +67 -0
- running_log.txt +746 -0
- special_tokens_map.json +30 -0
- tokenizer.json +0 -0
- tokenizer.model +3 -0
- tokenizer_config.json +132 -0
- train_results.json +9 -0
- trainer_log.jsonl +47 -0
- trainer_state.json +411 -0
- training_args.bin +3 -0
- training_args.yaml +35 -0
- training_loss.png +0 -0
README.md
ADDED
@@ -0,0 +1,62 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
base_model: microsoft/Phi-3-medium-128k-instruct
|
3 |
+
library_name: peft
|
4 |
+
license: other
|
5 |
+
tags:
|
6 |
+
- llama-factory
|
7 |
+
- lora
|
8 |
+
- generated_from_trainer
|
9 |
+
model-index:
|
10 |
+
- name: train_2024-07-24-15-00-21
|
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_2024-07-24-15-00-21
|
18 |
+
|
19 |
+
This model is a fine-tuned version of [microsoft/Phi-3-medium-128k-instruct](https://huggingface.co/microsoft/Phi-3-medium-128k-instruct) on the ALS 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: 5e-05
|
39 |
+
- train_batch_size: 2
|
40 |
+
- eval_batch_size: 8
|
41 |
+
- seed: 42
|
42 |
+
- distributed_type: multi-GPU
|
43 |
+
- num_devices: 2
|
44 |
+
- gradient_accumulation_steps: 8
|
45 |
+
- total_train_batch_size: 32
|
46 |
+
- total_eval_batch_size: 16
|
47 |
+
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
48 |
+
- lr_scheduler_type: cosine
|
49 |
+
- lr_scheduler_warmup_steps: 50
|
50 |
+
- num_epochs: 3.0
|
51 |
+
|
52 |
+
### Training results
|
53 |
+
|
54 |
+
|
55 |
+
|
56 |
+
### Framework versions
|
57 |
+
|
58 |
+
- PEFT 0.11.1
|
59 |
+
- Transformers 4.43.1
|
60 |
+
- Pytorch 2.3.1+cu121
|
61 |
+
- Datasets 2.19.1
|
62 |
+
- Tokenizers 0.19.1
|
adapter_config.json
ADDED
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"alpha_pattern": {},
|
3 |
+
"auto_mapping": null,
|
4 |
+
"base_model_name_or_path": "microsoft/Phi-3-medium-128k-instruct",
|
5 |
+
"bias": "none",
|
6 |
+
"fan_in_fan_out": false,
|
7 |
+
"inference_mode": true,
|
8 |
+
"init_lora_weights": true,
|
9 |
+
"layer_replication": null,
|
10 |
+
"layers_pattern": null,
|
11 |
+
"layers_to_transform": null,
|
12 |
+
"loftq_config": {},
|
13 |
+
"lora_alpha": 16,
|
14 |
+
"lora_dropout": 0,
|
15 |
+
"megatron_config": null,
|
16 |
+
"megatron_core": "megatron.core",
|
17 |
+
"modules_to_save": null,
|
18 |
+
"peft_type": "LORA",
|
19 |
+
"r": 8,
|
20 |
+
"rank_pattern": {},
|
21 |
+
"revision": null,
|
22 |
+
"target_modules": [
|
23 |
+
"o_proj",
|
24 |
+
"qkv_proj",
|
25 |
+
"down_proj",
|
26 |
+
"gate_up_proj"
|
27 |
+
],
|
28 |
+
"task_type": "CAUSAL_LM",
|
29 |
+
"use_dora": false,
|
30 |
+
"use_rslora": false
|
31 |
+
}
|
adapter_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:cb2de7ec0fe1de023e38bd91807dfe90489c24cfc01b61f7a997ca30cba48b64
|
3 |
+
size 55748640
|
added_tokens.json
ADDED
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"<|assistant|>": 32001,
|
3 |
+
"<|endoftext|>": 32000,
|
4 |
+
"<|end|>": 32007,
|
5 |
+
"<|placeholder1|>": 32002,
|
6 |
+
"<|placeholder2|>": 32003,
|
7 |
+
"<|placeholder3|>": 32004,
|
8 |
+
"<|placeholder4|>": 32005,
|
9 |
+
"<|placeholder5|>": 32008,
|
10 |
+
"<|placeholder6|>": 32009,
|
11 |
+
"<|system|>": 32006,
|
12 |
+
"<|user|>": 32010
|
13 |
+
}
|
all_results.json
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"epoch": 2.9902912621359223,
|
3 |
+
"num_input_tokens_seen": 61072080,
|
4 |
+
"total_flos": 5.065534456681464e+18,
|
5 |
+
"train_loss": 0.1973544524990635,
|
6 |
+
"train_runtime": 23852.8939,
|
7 |
+
"train_samples_per_second": 0.622,
|
8 |
+
"train_steps_per_second": 0.019
|
9 |
+
}
|
llamaboard_config.yaml
ADDED
@@ -0,0 +1,67 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
top.booster: flashattn2
|
2 |
+
top.checkpoint_path: []
|
3 |
+
top.finetuning_type: lora
|
4 |
+
top.model_name: Custom
|
5 |
+
top.quantization_bit: '4'
|
6 |
+
top.quantization_method: bitsandbytes
|
7 |
+
top.rope_scaling: none
|
8 |
+
top.template: default
|
9 |
+
top.visual_inputs: false
|
10 |
+
train.additional_target: ''
|
11 |
+
train.badam_mode: layer
|
12 |
+
train.badam_switch_interval: 50
|
13 |
+
train.badam_switch_mode: ascending
|
14 |
+
train.badam_update_ratio: 0.05
|
15 |
+
train.batch_size: 2
|
16 |
+
train.compute_type: bf16
|
17 |
+
train.create_new_adapter: false
|
18 |
+
train.cutoff_len: 20000
|
19 |
+
train.dataset:
|
20 |
+
- ALS
|
21 |
+
train.dataset_dir: data
|
22 |
+
train.ds_offload: true
|
23 |
+
train.ds_stage: '2'
|
24 |
+
train.freeze_extra_modules: ''
|
25 |
+
train.freeze_trainable_layers: 2
|
26 |
+
train.freeze_trainable_modules: all
|
27 |
+
train.galore_rank: 16
|
28 |
+
train.galore_scale: 0.25
|
29 |
+
train.galore_target: all
|
30 |
+
train.galore_update_interval: 200
|
31 |
+
train.gradient_accumulation_steps: 8
|
32 |
+
train.learning_rate: 5e-5
|
33 |
+
train.logging_steps: 10
|
34 |
+
train.lora_alpha: 16
|
35 |
+
train.lora_dropout: 0
|
36 |
+
train.lora_rank: 8
|
37 |
+
train.lora_target: ''
|
38 |
+
train.loraplus_lr_ratio: 0
|
39 |
+
train.lr_scheduler_type: cosine
|
40 |
+
train.mask_history: false
|
41 |
+
train.max_grad_norm: '1.0'
|
42 |
+
train.max_samples: '100000'
|
43 |
+
train.neat_packing: false
|
44 |
+
train.neftune_alpha: 0
|
45 |
+
train.num_train_epochs: '3.0'
|
46 |
+
train.optim: adamw_torch
|
47 |
+
train.packing: false
|
48 |
+
train.ppo_score_norm: false
|
49 |
+
train.ppo_whiten_rewards: false
|
50 |
+
train.pref_beta: 0.1
|
51 |
+
train.pref_ftx: 0
|
52 |
+
train.pref_loss: sigmoid
|
53 |
+
train.report_to: false
|
54 |
+
train.resize_vocab: false
|
55 |
+
train.reward_model: null
|
56 |
+
train.save_steps: 500
|
57 |
+
train.shift_attn: false
|
58 |
+
train.train_on_prompt: false
|
59 |
+
train.training_stage: Supervised Fine-Tuning
|
60 |
+
train.use_badam: false
|
61 |
+
train.use_dora: false
|
62 |
+
train.use_galore: false
|
63 |
+
train.use_llama_pro: false
|
64 |
+
train.use_pissa: false
|
65 |
+
train.use_rslora: false
|
66 |
+
train.val_size: 0
|
67 |
+
train.warmup_steps: 50
|
running_log.txt
ADDED
@@ -0,0 +1,746 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
[WARNING|parser.py:272] 2024-07-24 15:04:58,287 >> We recommend enable `upcast_layernorm` in quantized training.
|
2 |
+
|
3 |
+
[WARNING|parser.py:292] 2024-07-24 15:04:58,287 >> `ddp_find_unused_parameters` needs to be set as False for LoRA in DDP training.
|
4 |
+
|
5 |
+
[INFO|parser.py:344] 2024-07-24 15:04:58,288 >> Process rank: 0, device: cuda:0, n_gpu: 1, distributed training: True, compute dtype: torch.bfloat16
|
6 |
+
|
7 |
+
07/24/2024 15:04:58 - WARNING - llamafactory.hparams.parser - We recommend enable `upcast_layernorm` in quantized training.
|
8 |
+
|
9 |
+
07/24/2024 15:04:58 - WARNING - llamafactory.hparams.parser - `ddp_find_unused_parameters` needs to be set as False for LoRA in DDP training.
|
10 |
+
|
11 |
+
07/24/2024 15:04:58 - INFO - llamafactory.hparams.parser - Process rank: 1, device: cuda:1, n_gpu: 1, distributed training: True, compute dtype: torch.bfloat16
|
12 |
+
|
13 |
+
[INFO|tokenization_utils_base.py:2289] 2024-07-24 15:05:00,642 >> loading file tokenizer.model from cache at /workspace/data/huggingface-cache/hub/models--microsoft--Phi-3-medium-128k-instruct/snapshots/cae1d42b5577398fd1be9f0746052562ae552886/tokenizer.model
|
14 |
+
|
15 |
+
[INFO|tokenization_utils_base.py:2289] 2024-07-24 15:05:00,643 >> loading file tokenizer.json from cache at /workspace/data/huggingface-cache/hub/models--microsoft--Phi-3-medium-128k-instruct/snapshots/cae1d42b5577398fd1be9f0746052562ae552886/tokenizer.json
|
16 |
+
|
17 |
+
[INFO|tokenization_utils_base.py:2289] 2024-07-24 15:05:00,643 >> loading file added_tokens.json from cache at /workspace/data/huggingface-cache/hub/models--microsoft--Phi-3-medium-128k-instruct/snapshots/cae1d42b5577398fd1be9f0746052562ae552886/added_tokens.json
|
18 |
+
|
19 |
+
[INFO|tokenization_utils_base.py:2289] 2024-07-24 15:05:00,643 >> loading file special_tokens_map.json from cache at /workspace/data/huggingface-cache/hub/models--microsoft--Phi-3-medium-128k-instruct/snapshots/cae1d42b5577398fd1be9f0746052562ae552886/special_tokens_map.json
|
20 |
+
|
21 |
+
[INFO|tokenization_utils_base.py:2289] 2024-07-24 15:05:00,643 >> loading file tokenizer_config.json from cache at /workspace/data/huggingface-cache/hub/models--microsoft--Phi-3-medium-128k-instruct/snapshots/cae1d42b5577398fd1be9f0746052562ae552886/tokenizer_config.json
|
22 |
+
|
23 |
+
[INFO|tokenization_utils_base.py:2533] 2024-07-24 15:05:00,693 >> Special tokens have been added in the vocabulary, make sure the associated word embeddings are fine-tuned or trained.
|
24 |
+
|
25 |
+
[INFO|loader.py:52] 2024-07-24 15:05:00,694 >> Loading dataset dataset_alpaca_IT_train_and_eval_25K.json...
|
26 |
+
|
27 |
+
07/24/2024 15:05:08 - INFO - llamafactory.data.loader - Loading dataset dataset_alpaca_IT_train_and_eval_25K.json...
|
28 |
+
|
29 |
+
[INFO|configuration_utils.py:733] 2024-07-24 15:05:11,124 >> loading configuration file config.json from cache at /workspace/data/huggingface-cache/hub/models--microsoft--Phi-3-medium-128k-instruct/snapshots/cae1d42b5577398fd1be9f0746052562ae552886/config.json
|
30 |
+
|
31 |
+
[INFO|configuration_utils.py:733] 2024-07-24 15:05:11,485 >> loading configuration file config.json from cache at /workspace/data/huggingface-cache/hub/models--microsoft--Phi-3-medium-128k-instruct/snapshots/cae1d42b5577398fd1be9f0746052562ae552886/config.json
|
32 |
+
|
33 |
+
[INFO|configuration_utils.py:800] 2024-07-24 15:05:11,488 >> Model config Phi3Config {
|
34 |
+
"_name_or_path": "microsoft/Phi-3-medium-128k-instruct",
|
35 |
+
"architectures": [
|
36 |
+
"Phi3ForCausalLM"
|
37 |
+
],
|
38 |
+
"attention_bias": false,
|
39 |
+
"attention_dropout": 0.0,
|
40 |
+
"auto_map": {
|
41 |
+
"AutoConfig": "microsoft/Phi-3-medium-128k-instruct--configuration_phi3.Phi3Config",
|
42 |
+
"AutoModelForCausalLM": "microsoft/Phi-3-medium-128k-instruct--modeling_phi3.Phi3ForCausalLM"
|
43 |
+
},
|
44 |
+
"bos_token_id": 1,
|
45 |
+
"embd_pdrop": 0.0,
|
46 |
+
"eos_token_id": 32000,
|
47 |
+
"hidden_act": "silu",
|
48 |
+
"hidden_size": 5120,
|
49 |
+
"initializer_range": 0.02,
|
50 |
+
"intermediate_size": 17920,
|
51 |
+
"max_position_embeddings": 131072,
|
52 |
+
"model_type": "phi3",
|
53 |
+
"num_attention_heads": 40,
|
54 |
+
"num_hidden_layers": 40,
|
55 |
+
"num_key_value_heads": 10,
|
56 |
+
"original_max_position_embeddings": 4096,
|
57 |
+
"pad_token_id": null,
|
58 |
+
"resid_pdrop": 0.0,
|
59 |
+
"rms_norm_eps": 1e-05,
|
60 |
+
"rope_scaling": {
|
61 |
+
"long_factor": [
|
62 |
+
1.0,
|
63 |
+
1.0,
|
64 |
+
1.0,
|
65 |
+
1.0,
|
66 |
+
1.0,
|
67 |
+
1.0,
|
68 |
+
1.0,
|
69 |
+
1.0,
|
70 |
+
1.0,
|
71 |
+
1.0,
|
72 |
+
1.0,
|
73 |
+
1.0,
|
74 |
+
1.0,
|
75 |
+
1.25,
|
76 |
+
1.25,
|
77 |
+
1.5,
|
78 |
+
2.0,
|
79 |
+
2.75,
|
80 |
+
5.75,
|
81 |
+
5.75,
|
82 |
+
6.5,
|
83 |
+
9.25,
|
84 |
+
11.0,
|
85 |
+
13.25,
|
86 |
+
19.25,
|
87 |
+
19.75,
|
88 |
+
19.75,
|
89 |
+
21.25,
|
90 |
+
21.5,
|
91 |
+
26.5,
|
92 |
+
30.0,
|
93 |
+
33.75,
|
94 |
+
35.25,
|
95 |
+
38.5,
|
96 |
+
42.0,
|
97 |
+
42.25,
|
98 |
+
46.0,
|
99 |
+
47.0,
|
100 |
+
50.0,
|
101 |
+
50.5,
|
102 |
+
51.0,
|
103 |
+
52.0,
|
104 |
+
52.75,
|
105 |
+
53.75,
|
106 |
+
54.75,
|
107 |
+
57.0,
|
108 |
+
57.25,
|
109 |
+
58.5,
|
110 |
+
59.25,
|
111 |
+
59.5,
|
112 |
+
62.0,
|
113 |
+
62.5,
|
114 |
+
62.75,
|
115 |
+
63.25,
|
116 |
+
63.25,
|
117 |
+
63.25,
|
118 |
+
63.75,
|
119 |
+
64.0,
|
120 |
+
64.0,
|
121 |
+
64.25,
|
122 |
+
64.5,
|
123 |
+
64.5,
|
124 |
+
65.0,
|
125 |
+
65.0
|
126 |
+
],
|
127 |
+
"short_factor": [
|
128 |
+
1.0,
|
129 |
+
1.0,
|
130 |
+
1.0,
|
131 |
+
1.0,
|
132 |
+
1.0,
|
133 |
+
1.0,
|
134 |
+
1.01,
|
135 |
+
1.02,
|
136 |
+
1.02,
|
137 |
+
1.04,
|
138 |
+
1.04,
|
139 |
+
1.07,
|
140 |
+
1.07,
|
141 |
+
1.1,
|
142 |
+
1.3000000000000003,
|
143 |
+
1.3000000000000003,
|
144 |
+
1.5000000000000004,
|
145 |
+
1.5700000000000005,
|
146 |
+
1.9000000000000008,
|
147 |
+
2.3100000000000014,
|
148 |
+
2.759999999999992,
|
149 |
+
3.3899999999999784,
|
150 |
+
3.9399999999999666,
|
151 |
+
4.009999999999965,
|
152 |
+
4.289999999999959,
|
153 |
+
4.349999999999958,
|
154 |
+
5.349999999999937,
|
155 |
+
6.659999999999909,
|
156 |
+
7.029999999999901,
|
157 |
+
7.51999999999989,
|
158 |
+
8.00999999999988,
|
159 |
+
8.249999999999876,
|
160 |
+
8.279999999999875,
|
161 |
+
9.629999999999846,
|
162 |
+
9.89999999999984,
|
163 |
+
10.589999999999826,
|
164 |
+
11.049999999999816,
|
165 |
+
11.7899999999998,
|
166 |
+
12.189999999999792,
|
167 |
+
12.889999999999777,
|
168 |
+
13.129999999999772,
|
169 |
+
13.16999999999977,
|
170 |
+
13.20999999999977,
|
171 |
+
13.479999999999764,
|
172 |
+
13.539999999999763,
|
173 |
+
13.779999999999758,
|
174 |
+
13.929999999999755,
|
175 |
+
14.429999999999744,
|
176 |
+
14.759999999999737,
|
177 |
+
15.149999999999729,
|
178 |
+
15.419999999999723,
|
179 |
+
15.53999999999972,
|
180 |
+
15.659999999999718,
|
181 |
+
15.749999999999716,
|
182 |
+
15.759999999999716,
|
183 |
+
15.799999999999715,
|
184 |
+
16.05999999999971,
|
185 |
+
16.079999999999714,
|
186 |
+
16.11999999999972,
|
187 |
+
16.11999999999972,
|
188 |
+
16.18999999999973,
|
189 |
+
16.31999999999975,
|
190 |
+
16.539999999999786,
|
191 |
+
16.799999999999827
|
192 |
+
],
|
193 |
+
"type": "su"
|
194 |
+
},
|
195 |
+
"rope_theta": 10000.0,
|
196 |
+
"sliding_window": 131072,
|
197 |
+
"tie_word_embeddings": false,
|
198 |
+
"torch_dtype": "bfloat16",
|
199 |
+
"transformers_version": "4.43.1",
|
200 |
+
"use_cache": true,
|
201 |
+
"vocab_size": 32064
|
202 |
+
}
|
203 |
+
|
204 |
+
|
205 |
+
[INFO|quantization.py:182] 2024-07-24 15:05:11,496 >> Quantizing model to 4 bit with bitsandbytes.
|
206 |
+
|
207 |
+
[INFO|modeling_utils.py:3621] 2024-07-24 15:05:12,104 >> loading weights file model.safetensors from cache at /workspace/data/huggingface-cache/hub/models--microsoft--Phi-3-medium-128k-instruct/snapshots/cae1d42b5577398fd1be9f0746052562ae552886/model.safetensors.index.json
|
208 |
+
|
209 |
+
07/24/2024 15:05:12 - INFO - llamafactory.model.model_utils.quantization - Quantizing model to 4 bit with bitsandbytes.
|
210 |
+
|
211 |
+
[INFO|modeling_utils.py:1569] 2024-07-24 15:10:52,981 >> Instantiating Phi3ForCausalLM model under default dtype torch.bfloat16.
|
212 |
+
|
213 |
+
[INFO|configuration_utils.py:1038] 2024-07-24 15:10:52,989 >> Generate config GenerationConfig {
|
214 |
+
"bos_token_id": 1,
|
215 |
+
"eos_token_id": 32000
|
216 |
+
}
|
217 |
+
|
218 |
+
|
219 |
+
[INFO|modeling_utils.py:4450] 2024-07-24 15:11:18,809 >> All model checkpoint weights were used when initializing Phi3ForCausalLM.
|
220 |
+
|
221 |
+
|
222 |
+
[INFO|modeling_utils.py:4458] 2024-07-24 15:11:18,810 >> All the weights of Phi3ForCausalLM were initialized from the model checkpoint at microsoft/Phi-3-medium-128k-instruct.
|
223 |
+
If your task is similar to the task the model of the checkpoint was trained on, you can already use Phi3ForCausalLM for predictions without further training.
|
224 |
+
|
225 |
+
[INFO|configuration_utils.py:993] 2024-07-24 15:11:18,895 >> loading configuration file generation_config.json from cache at /workspace/data/huggingface-cache/hub/models--microsoft--Phi-3-medium-128k-instruct/snapshots/cae1d42b5577398fd1be9f0746052562ae552886/generation_config.json
|
226 |
+
|
227 |
+
[INFO|configuration_utils.py:1038] 2024-07-24 15:11:18,896 >> Generate config GenerationConfig {
|
228 |
+
"bos_token_id": 1,
|
229 |
+
"eos_token_id": [
|
230 |
+
32000,
|
231 |
+
32001,
|
232 |
+
32007
|
233 |
+
],
|
234 |
+
"pad_token_id": 32000
|
235 |
+
}
|
236 |
+
|
237 |
+
|
238 |
+
07/24/2024 15:12:39 - INFO - llamafactory.model.model_utils.checkpointing - Gradient checkpointing enabled.
|
239 |
+
|
240 |
+
07/24/2024 15:12:39 - INFO - llamafactory.model.model_utils.attention - Using FlashAttention-2 for faster training and inference.
|
241 |
+
|
242 |
+
07/24/2024 15:12:39 - INFO - llamafactory.model.adapter - Upcasting trainable params to float32.
|
243 |
+
|
244 |
+
07/24/2024 15:12:39 - INFO - llamafactory.model.adapter - Fine-tuning method: LoRA
|
245 |
+
|
246 |
+
07/24/2024 15:12:39 - INFO - llamafactory.model.model_utils.misc - Found linear modules: o_proj,gate_up_proj,qkv_proj,down_proj
|
247 |
+
|
248 |
+
07/24/2024 15:12:39 - INFO - llamafactory.model.loader - trainable params: 27,852,800 || all params: 13,988,090,880 || trainable%: 0.1991
|
249 |
+
|
250 |
+
[INFO|checkpointing.py:103] 2024-07-24 15:12:42,061 >> Gradient checkpointing enabled.
|
251 |
+
|
252 |
+
[INFO|attention.py:82] 2024-07-24 15:12:42,061 >> Using FlashAttention-2 for faster training and inference.
|
253 |
+
|
254 |
+
[INFO|adapter.py:302] 2024-07-24 15:12:42,061 >> Upcasting trainable params to float32.
|
255 |
+
|
256 |
+
[INFO|adapter.py:158] 2024-07-24 15:12:42,061 >> Fine-tuning method: LoRA
|
257 |
+
|
258 |
+
[INFO|misc.py:51] 2024-07-24 15:12:42,062 >> Found linear modules: o_proj,qkv_proj,down_proj,gate_up_proj
|
259 |
+
|
260 |
+
[INFO|loader.py:196] 2024-07-24 15:12:42,467 >> trainable params: 27,852,800 || all params: 13,988,090,880 || trainable%: 0.1991
|
261 |
+
|
262 |
+
[INFO|trainer.py:648] 2024-07-24 15:12:42,473 >> Using auto half precision backend
|
263 |
+
|
264 |
+
[INFO|deepspeed.py:329] 2024-07-24 15:12:42,673 >> Detected ZeRO Offload and non-DeepSpeed optimizers: This combination should work as long as the custom optimizer has both CPU and GPU implementation (except LAMB)
|
265 |
+
|
266 |
+
[INFO|trainer.py:2134] 2024-07-24 15:13:06,954 >> ***** Running training *****
|
267 |
+
|
268 |
+
[INFO|trainer.py:2135] 2024-07-24 15:13:06,954 >> Num examples = 4,944
|
269 |
+
|
270 |
+
[INFO|trainer.py:2136] 2024-07-24 15:13:06,954 >> Num Epochs = 3
|
271 |
+
|
272 |
+
[INFO|trainer.py:2137] 2024-07-24 15:13:06,954 >> Instantaneous batch size per device = 2
|
273 |
+
|
274 |
+
[INFO|trainer.py:2140] 2024-07-24 15:13:06,954 >> Total train batch size (w. parallel, distributed & accumulation) = 32
|
275 |
+
|
276 |
+
[INFO|trainer.py:2141] 2024-07-24 15:13:06,954 >> Gradient Accumulation steps = 8
|
277 |
+
|
278 |
+
[INFO|trainer.py:2142] 2024-07-24 15:13:06,954 >> Total optimization steps = 462
|
279 |
+
|
280 |
+
[INFO|trainer.py:2143] 2024-07-24 15:13:06,958 >> Number of trainable parameters = 27,852,800
|
281 |
+
|
282 |
+
[INFO|callbacks.py:310] 2024-07-24 15:21:31,280 >> {'loss': 0.5099, 'learning_rate': 1.0000e-05, 'epoch': 0.06, 'throughput': 2571.26}
|
283 |
+
|
284 |
+
[INFO|callbacks.py:310] 2024-07-24 15:30:13,855 >> {'loss': 0.5115, 'learning_rate': 2.0000e-05, 'epoch': 0.13, 'throughput': 2534.92}
|
285 |
+
|
286 |
+
[INFO|callbacks.py:310] 2024-07-24 15:38:39,051 >> {'loss': 0.4846, 'learning_rate': 3.0000e-05, 'epoch': 0.19, 'throughput': 2535.35}
|
287 |
+
|
288 |
+
[INFO|callbacks.py:310] 2024-07-24 15:47:32,541 >> {'loss': 0.4076, 'learning_rate': 4.0000e-05, 'epoch': 0.26, 'throughput': 2553.97}
|
289 |
+
|
290 |
+
[INFO|callbacks.py:310] 2024-07-24 15:56:37,450 >> {'loss': 0.3073, 'learning_rate': 5.0000e-05, 'epoch': 0.32, 'throughput': 2556.06}
|
291 |
+
|
292 |
+
[INFO|callbacks.py:310] 2024-07-24 16:04:50,564 >> {'loss': 0.2516, 'learning_rate': 4.9927e-05, 'epoch': 0.39, 'throughput': 2559.12}
|
293 |
+
|
294 |
+
[INFO|callbacks.py:310] 2024-07-24 16:13:54,185 >> {'loss': 0.2256, 'learning_rate': 4.9710e-05, 'epoch': 0.45, 'throughput': 2549.95}
|
295 |
+
|
296 |
+
[INFO|callbacks.py:310] 2024-07-24 16:22:56,812 >> {'loss': 0.2146, 'learning_rate': 4.9349e-05, 'epoch': 0.52, 'throughput': 2547.79}
|
297 |
+
|
298 |
+
[INFO|callbacks.py:310] 2024-07-24 16:31:48,796 >> {'loss': 0.2018, 'learning_rate': 4.8846e-05, 'epoch': 0.58, 'throughput': 2552.59}
|
299 |
+
|
300 |
+
[INFO|callbacks.py:310] 2024-07-24 16:40:22,764 >> {'loss': 0.1958, 'learning_rate': 4.8205e-05, 'epoch': 0.65, 'throughput': 2556.94}
|
301 |
+
|
302 |
+
[INFO|callbacks.py:310] 2024-07-24 16:48:46,503 >> {'loss': 0.1912, 'learning_rate': 4.7429e-05, 'epoch': 0.71, 'throughput': 2557.17}
|
303 |
+
|
304 |
+
[INFO|callbacks.py:310] 2024-07-24 16:57:16,752 >> {'loss': 0.1876, 'learning_rate': 4.6522e-05, 'epoch': 0.78, 'throughput': 2558.65}
|
305 |
+
|
306 |
+
[INFO|callbacks.py:310] 2024-07-24 17:05:34,669 >> {'loss': 0.1802, 'learning_rate': 4.5491e-05, 'epoch': 0.84, 'throughput': 2561.14}
|
307 |
+
|
308 |
+
[INFO|callbacks.py:310] 2024-07-24 17:14:13,797 >> {'loss': 0.1793, 'learning_rate': 4.4340e-05, 'epoch': 0.91, 'throughput': 2560.01}
|
309 |
+
|
310 |
+
[INFO|callbacks.py:310] 2024-07-24 17:22:00,855 >> {'loss': 0.1759, 'learning_rate': 4.3077e-05, 'epoch': 0.97, 'throughput': 2565.87}
|
311 |
+
|
312 |
+
[INFO|callbacks.py:310] 2024-07-24 17:30:17,877 >> {'loss': 0.1746, 'learning_rate': 4.1709e-05, 'epoch': 1.04, 'throughput': 2564.86}
|
313 |
+
|
314 |
+
[INFO|callbacks.py:310] 2024-07-24 17:38:59,853 >> {'loss': 0.1699, 'learning_rate': 4.0244e-05, 'epoch': 1.10, 'throughput': 2564.04}
|
315 |
+
|
316 |
+
[INFO|callbacks.py:310] 2024-07-24 17:48:09,599 >> {'loss': 0.1680, 'learning_rate': 3.8690e-05, 'epoch': 1.17, 'throughput': 2560.03}
|
317 |
+
|
318 |
+
[INFO|callbacks.py:310] 2024-07-24 17:56:04,957 >> {'loss': 0.1646, 'learning_rate': 3.7057e-05, 'epoch': 1.23, 'throughput': 2563.38}
|
319 |
+
|
320 |
+
[INFO|callbacks.py:310] 2024-07-24 18:04:29,792 >> {'loss': 0.1667, 'learning_rate': 3.5354e-05, 'epoch': 1.29, 'throughput': 2565.53}
|
321 |
+
|
322 |
+
[INFO|callbacks.py:310] 2024-07-24 18:13:16,644 >> {'loss': 0.1664, 'learning_rate': 3.3590e-05, 'epoch': 1.36, 'throughput': 2565.84}
|
323 |
+
|
324 |
+
[INFO|callbacks.py:310] 2024-07-24 18:21:27,030 >> {'loss': 0.1622, 'learning_rate': 3.1777e-05, 'epoch': 1.42, 'throughput': 2565.58}
|
325 |
+
|
326 |
+
[INFO|callbacks.py:310] 2024-07-24 18:30:03,234 >> {'loss': 0.1623, 'learning_rate': 2.9924e-05, 'epoch': 1.49, 'throughput': 2565.43}
|
327 |
+
|
328 |
+
[INFO|callbacks.py:310] 2024-07-24 18:38:50,374 >> {'loss': 0.1616, 'learning_rate': 2.8043e-05, 'epoch': 1.55, 'throughput': 2565.51}
|
329 |
+
|
330 |
+
[INFO|callbacks.py:310] 2024-07-24 18:47:03,606 >> {'loss': 0.1590, 'learning_rate': 2.6143e-05, 'epoch': 1.62, 'throughput': 2566.31}
|
331 |
+
|
332 |
+
[INFO|callbacks.py:310] 2024-07-24 18:55:49,268 >> {'loss': 0.1619, 'learning_rate': 2.4238e-05, 'epoch': 1.68, 'throughput': 2564.17}
|
333 |
+
|
334 |
+
[INFO|callbacks.py:310] 2024-07-24 19:04:24,775 >> {'loss': 0.1616, 'learning_rate': 2.2336e-05, 'epoch': 1.75, 'throughput': 2565.53}
|
335 |
+
|
336 |
+
[INFO|callbacks.py:310] 2024-07-24 19:12:40,325 >> {'loss': 0.1604, 'learning_rate': 2.0450e-05, 'epoch': 1.81, 'throughput': 2566.61}
|
337 |
+
|
338 |
+
[INFO|callbacks.py:310] 2024-07-24 19:21:50,945 >> {'loss': 0.1563, 'learning_rate': 1.8591e-05, 'epoch': 1.88, 'throughput': 2564.70}
|
339 |
+
|
340 |
+
[INFO|callbacks.py:310] 2024-07-24 19:30:42,384 >> {'loss': 0.1548, 'learning_rate': 1.6769e-05, 'epoch': 1.94, 'throughput': 2565.36}
|
341 |
+
|
342 |
+
[INFO|callbacks.py:310] 2024-07-24 19:39:26,495 >> {'loss': 0.1555, 'learning_rate': 1.4994e-05, 'epoch': 2.01, 'throughput': 2565.22}
|
343 |
+
|
344 |
+
[INFO|callbacks.py:310] 2024-07-24 19:48:16,049 >> {'loss': 0.1526, 'learning_rate': 1.3278e-05, 'epoch': 2.07, 'throughput': 2564.82}
|
345 |
+
|
346 |
+
[INFO|callbacks.py:310] 2024-07-24 19:56:55,676 >> {'loss': 0.1526, 'learning_rate': 1.1630e-05, 'epoch': 2.14, 'throughput': 2563.60}
|
347 |
+
|
348 |
+
[INFO|callbacks.py:310] 2024-07-24 20:05:48,055 >> {'loss': 0.1516, 'learning_rate': 1.0060e-05, 'epoch': 2.20, 'throughput': 2564.75}
|
349 |
+
|
350 |
+
[INFO|callbacks.py:310] 2024-07-24 20:14:05,975 >> {'loss': 0.1524, 'learning_rate': 8.5762e-06, 'epoch': 2.27, 'throughput': 2565.56}
|
351 |
+
|
352 |
+
[INFO|callbacks.py:310] 2024-07-24 20:23:04,597 >> {'loss': 0.1502, 'learning_rate': 7.1880e-06, 'epoch': 2.33, 'throughput': 2564.08}
|
353 |
+
|
354 |
+
[INFO|callbacks.py:310] 2024-07-24 20:31:47,755 >> {'loss': 0.1506, 'learning_rate': 5.9035e-06, 'epoch': 2.39, 'throughput': 2563.27}
|
355 |
+
|
356 |
+
[INFO|callbacks.py:310] 2024-07-24 20:40:43,735 >> {'loss': 0.1479, 'learning_rate': 4.7298e-06, 'epoch': 2.46, 'throughput': 2560.31}
|
357 |
+
|
358 |
+
[INFO|callbacks.py:310] 2024-07-24 20:49:04,924 >> {'loss': 0.1501, 'learning_rate': 3.6740e-06, 'epoch': 2.52, 'throughput': 2560.79}
|
359 |
+
|
360 |
+
[INFO|callbacks.py:310] 2024-07-24 20:57:37,960 >> {'loss': 0.1504, 'learning_rate': 2.7422e-06, 'epoch': 2.59, 'throughput': 2561.70}
|
361 |
+
|
362 |
+
[INFO|callbacks.py:310] 2024-07-24 21:05:54,158 >> {'loss': 0.1524, 'learning_rate': 1.9397e-06, 'epoch': 2.65, 'throughput': 2561.78}
|
363 |
+
|
364 |
+
[INFO|callbacks.py:310] 2024-07-24 21:14:51,826 >> {'loss': 0.1494, 'learning_rate': 1.2712e-06, 'epoch': 2.72, 'throughput': 2559.88}
|
365 |
+
|
366 |
+
[INFO|callbacks.py:310] 2024-07-24 21:23:35,877 >> {'loss': 0.1515, 'learning_rate': 7.4056e-07, 'epoch': 2.78, 'throughput': 2559.75}
|
367 |
+
|
368 |
+
[INFO|callbacks.py:310] 2024-07-24 21:32:00,699 >> {'loss': 0.1504, 'learning_rate': 3.5095e-07, 'epoch': 2.85, 'throughput': 2559.97}
|
369 |
+
|
370 |
+
[INFO|callbacks.py:310] 2024-07-24 21:40:43,444 >> {'loss': 0.1498, 'learning_rate': 1.0459e-07, 'epoch': 2.91, 'throughput': 2559.45}
|
371 |
+
|
372 |
+
[INFO|callbacks.py:310] 2024-07-24 21:48:52,834 >> {'loss': 0.1469, 'learning_rate': 2.9071e-09, 'epoch': 2.98, 'throughput': 2559.91}
|
373 |
+
|
374 |
+
[INFO|trainer.py:3503] 2024-07-24 21:50:38,895 >> Saving model checkpoint to saves/Custom/lora/train_2024-07-24-15-00-21/checkpoint-462
|
375 |
+
|
376 |
+
[INFO|configuration_utils.py:733] 2024-07-24 21:50:39,083 >> loading configuration file config.json from cache at /workspace/data/huggingface-cache/hub/models--microsoft--Phi-3-medium-128k-instruct/snapshots/cae1d42b5577398fd1be9f0746052562ae552886/config.json
|
377 |
+
|
378 |
+
[INFO|configuration_utils.py:800] 2024-07-24 21:50:39,084 >> Model config Phi3Config {
|
379 |
+
"_name_or_path": "Phi-3-medium-128k-instruct",
|
380 |
+
"architectures": [
|
381 |
+
"Phi3ForCausalLM"
|
382 |
+
],
|
383 |
+
"attention_bias": false,
|
384 |
+
"attention_dropout": 0.0,
|
385 |
+
"auto_map": {
|
386 |
+
"AutoConfig": "microsoft/Phi-3-medium-128k-instruct--configuration_phi3.Phi3Config",
|
387 |
+
"AutoModelForCausalLM": "microsoft/Phi-3-medium-128k-instruct--modeling_phi3.Phi3ForCausalLM"
|
388 |
+
},
|
389 |
+
"bos_token_id": 1,
|
390 |
+
"embd_pdrop": 0.0,
|
391 |
+
"eos_token_id": 32000,
|
392 |
+
"hidden_act": "silu",
|
393 |
+
"hidden_size": 5120,
|
394 |
+
"initializer_range": 0.02,
|
395 |
+
"intermediate_size": 17920,
|
396 |
+
"max_position_embeddings": 131072,
|
397 |
+
"model_type": "phi3",
|
398 |
+
"num_attention_heads": 40,
|
399 |
+
"num_hidden_layers": 40,
|
400 |
+
"num_key_value_heads": 10,
|
401 |
+
"original_max_position_embeddings": 4096,
|
402 |
+
"pad_token_id": null,
|
403 |
+
"resid_pdrop": 0.0,
|
404 |
+
"rms_norm_eps": 1e-05,
|
405 |
+
"rope_scaling": {
|
406 |
+
"long_factor": [
|
407 |
+
1.0,
|
408 |
+
1.0,
|
409 |
+
1.0,
|
410 |
+
1.0,
|
411 |
+
1.0,
|
412 |
+
1.0,
|
413 |
+
1.0,
|
414 |
+
1.0,
|
415 |
+
1.0,
|
416 |
+
1.0,
|
417 |
+
1.0,
|
418 |
+
1.0,
|
419 |
+
1.0,
|
420 |
+
1.25,
|
421 |
+
1.25,
|
422 |
+
1.5,
|
423 |
+
2.0,
|
424 |
+
2.75,
|
425 |
+
5.75,
|
426 |
+
5.75,
|
427 |
+
6.5,
|
428 |
+
9.25,
|
429 |
+
11.0,
|
430 |
+
13.25,
|
431 |
+
19.25,
|
432 |
+
19.75,
|
433 |
+
19.75,
|
434 |
+
21.25,
|
435 |
+
21.5,
|
436 |
+
26.5,
|
437 |
+
30.0,
|
438 |
+
33.75,
|
439 |
+
35.25,
|
440 |
+
38.5,
|
441 |
+
42.0,
|
442 |
+
42.25,
|
443 |
+
46.0,
|
444 |
+
47.0,
|
445 |
+
50.0,
|
446 |
+
50.5,
|
447 |
+
51.0,
|
448 |
+
52.0,
|
449 |
+
52.75,
|
450 |
+
53.75,
|
451 |
+
54.75,
|
452 |
+
57.0,
|
453 |
+
57.25,
|
454 |
+
58.5,
|
455 |
+
59.25,
|
456 |
+
59.5,
|
457 |
+
62.0,
|
458 |
+
62.5,
|
459 |
+
62.75,
|
460 |
+
63.25,
|
461 |
+
63.25,
|
462 |
+
63.25,
|
463 |
+
63.75,
|
464 |
+
64.0,
|
465 |
+
64.0,
|
466 |
+
64.25,
|
467 |
+
64.5,
|
468 |
+
64.5,
|
469 |
+
65.0,
|
470 |
+
65.0
|
471 |
+
],
|
472 |
+
"short_factor": [
|
473 |
+
1.0,
|
474 |
+
1.0,
|
475 |
+
1.0,
|
476 |
+
1.0,
|
477 |
+
1.0,
|
478 |
+
1.0,
|
479 |
+
1.01,
|
480 |
+
1.02,
|
481 |
+
1.02,
|
482 |
+
1.04,
|
483 |
+
1.04,
|
484 |
+
1.07,
|
485 |
+
1.07,
|
486 |
+
1.1,
|
487 |
+
1.3000000000000003,
|
488 |
+
1.3000000000000003,
|
489 |
+
1.5000000000000004,
|
490 |
+
1.5700000000000005,
|
491 |
+
1.9000000000000008,
|
492 |
+
2.3100000000000014,
|
493 |
+
2.759999999999992,
|
494 |
+
3.3899999999999784,
|
495 |
+
3.9399999999999666,
|
496 |
+
4.009999999999965,
|
497 |
+
4.289999999999959,
|
498 |
+
4.349999999999958,
|
499 |
+
5.349999999999937,
|
500 |
+
6.659999999999909,
|
501 |
+
7.029999999999901,
|
502 |
+
7.51999999999989,
|
503 |
+
8.00999999999988,
|
504 |
+
8.249999999999876,
|
505 |
+
8.279999999999875,
|
506 |
+
9.629999999999846,
|
507 |
+
9.89999999999984,
|
508 |
+
10.589999999999826,
|
509 |
+
11.049999999999816,
|
510 |
+
11.7899999999998,
|
511 |
+
12.189999999999792,
|
512 |
+
12.889999999999777,
|
513 |
+
13.129999999999772,
|
514 |
+
13.16999999999977,
|
515 |
+
13.20999999999977,
|
516 |
+
13.479999999999764,
|
517 |
+
13.539999999999763,
|
518 |
+
13.779999999999758,
|
519 |
+
13.929999999999755,
|
520 |
+
14.429999999999744,
|
521 |
+
14.759999999999737,
|
522 |
+
15.149999999999729,
|
523 |
+
15.419999999999723,
|
524 |
+
15.53999999999972,
|
525 |
+
15.659999999999718,
|
526 |
+
15.749999999999716,
|
527 |
+
15.759999999999716,
|
528 |
+
15.799999999999715,
|
529 |
+
16.05999999999971,
|
530 |
+
16.079999999999714,
|
531 |
+
16.11999999999972,
|
532 |
+
16.11999999999972,
|
533 |
+
16.18999999999973,
|
534 |
+
16.31999999999975,
|
535 |
+
16.539999999999786,
|
536 |
+
16.799999999999827
|
537 |
+
],
|
538 |
+
"type": "su"
|
539 |
+
},
|
540 |
+
"rope_theta": 10000.0,
|
541 |
+
"sliding_window": 131072,
|
542 |
+
"tie_word_embeddings": false,
|
543 |
+
"torch_dtype": "bfloat16",
|
544 |
+
"transformers_version": "4.43.1",
|
545 |
+
"use_cache": true,
|
546 |
+
"vocab_size": 32064
|
547 |
+
}
|
548 |
+
|
549 |
+
|
550 |
+
[INFO|tokenization_utils_base.py:2702] 2024-07-24 21:50:39,135 >> tokenizer config file saved in saves/Custom/lora/train_2024-07-24-15-00-21/checkpoint-462/tokenizer_config.json
|
551 |
+
|
552 |
+
[INFO|tokenization_utils_base.py:2711] 2024-07-24 21:50:39,135 >> Special tokens file saved in saves/Custom/lora/train_2024-07-24-15-00-21/checkpoint-462/special_tokens_map.json
|
553 |
+
|
554 |
+
[INFO|trainer.py:2394] 2024-07-24 21:50:39,852 >>
|
555 |
+
|
556 |
+
Training completed. Do not forget to share your model on huggingface.co/models =)
|
557 |
+
|
558 |
+
|
559 |
+
|
560 |
+
[INFO|trainer.py:3503] 2024-07-24 21:50:42,012 >> Saving model checkpoint to saves/Custom/lora/train_2024-07-24-15-00-21
|
561 |
+
|
562 |
+
[INFO|configuration_utils.py:733] 2024-07-24 21:50:42,204 >> loading configuration file config.json from cache at /workspace/data/huggingface-cache/hub/models--microsoft--Phi-3-medium-128k-instruct/snapshots/cae1d42b5577398fd1be9f0746052562ae552886/config.json
|
563 |
+
|
564 |
+
[INFO|configuration_utils.py:800] 2024-07-24 21:50:42,205 >> Model config Phi3Config {
|
565 |
+
"_name_or_path": "Phi-3-medium-128k-instruct",
|
566 |
+
"architectures": [
|
567 |
+
"Phi3ForCausalLM"
|
568 |
+
],
|
569 |
+
"attention_bias": false,
|
570 |
+
"attention_dropout": 0.0,
|
571 |
+
"auto_map": {
|
572 |
+
"AutoConfig": "microsoft/Phi-3-medium-128k-instruct--configuration_phi3.Phi3Config",
|
573 |
+
"AutoModelForCausalLM": "microsoft/Phi-3-medium-128k-instruct--modeling_phi3.Phi3ForCausalLM"
|
574 |
+
},
|
575 |
+
"bos_token_id": 1,
|
576 |
+
"embd_pdrop": 0.0,
|
577 |
+
"eos_token_id": 32000,
|
578 |
+
"hidden_act": "silu",
|
579 |
+
"hidden_size": 5120,
|
580 |
+
"initializer_range": 0.02,
|
581 |
+
"intermediate_size": 17920,
|
582 |
+
"max_position_embeddings": 131072,
|
583 |
+
"model_type": "phi3",
|
584 |
+
"num_attention_heads": 40,
|
585 |
+
"num_hidden_layers": 40,
|
586 |
+
"num_key_value_heads": 10,
|
587 |
+
"original_max_position_embeddings": 4096,
|
588 |
+
"pad_token_id": null,
|
589 |
+
"resid_pdrop": 0.0,
|
590 |
+
"rms_norm_eps": 1e-05,
|
591 |
+
"rope_scaling": {
|
592 |
+
"long_factor": [
|
593 |
+
1.0,
|
594 |
+
1.0,
|
595 |
+
1.0,
|
596 |
+
1.0,
|
597 |
+
1.0,
|
598 |
+
1.0,
|
599 |
+
1.0,
|
600 |
+
1.0,
|
601 |
+
1.0,
|
602 |
+
1.0,
|
603 |
+
1.0,
|
604 |
+
1.0,
|
605 |
+
1.0,
|
606 |
+
1.25,
|
607 |
+
1.25,
|
608 |
+
1.5,
|
609 |
+
2.0,
|
610 |
+
2.75,
|
611 |
+
5.75,
|
612 |
+
5.75,
|
613 |
+
6.5,
|
614 |
+
9.25,
|
615 |
+
11.0,
|
616 |
+
13.25,
|
617 |
+
19.25,
|
618 |
+
19.75,
|
619 |
+
19.75,
|
620 |
+
21.25,
|
621 |
+
21.5,
|
622 |
+
26.5,
|
623 |
+
30.0,
|
624 |
+
33.75,
|
625 |
+
35.25,
|
626 |
+
38.5,
|
627 |
+
42.0,
|
628 |
+
42.25,
|
629 |
+
46.0,
|
630 |
+
47.0,
|
631 |
+
50.0,
|
632 |
+
50.5,
|
633 |
+
51.0,
|
634 |
+
52.0,
|
635 |
+
52.75,
|
636 |
+
53.75,
|
637 |
+
54.75,
|
638 |
+
57.0,
|
639 |
+
57.25,
|
640 |
+
58.5,
|
641 |
+
59.25,
|
642 |
+
59.5,
|
643 |
+
62.0,
|
644 |
+
62.5,
|
645 |
+
62.75,
|
646 |
+
63.25,
|
647 |
+
63.25,
|
648 |
+
63.25,
|
649 |
+
63.75,
|
650 |
+
64.0,
|
651 |
+
64.0,
|
652 |
+
64.25,
|
653 |
+
64.5,
|
654 |
+
64.5,
|
655 |
+
65.0,
|
656 |
+
65.0
|
657 |
+
],
|
658 |
+
"short_factor": [
|
659 |
+
1.0,
|
660 |
+
1.0,
|
661 |
+
1.0,
|
662 |
+
1.0,
|
663 |
+
1.0,
|
664 |
+
1.0,
|
665 |
+
1.01,
|
666 |
+
1.02,
|
667 |
+
1.02,
|
668 |
+
1.04,
|
669 |
+
1.04,
|
670 |
+
1.07,
|
671 |
+
1.07,
|
672 |
+
1.1,
|
673 |
+
1.3000000000000003,
|
674 |
+
1.3000000000000003,
|
675 |
+
1.5000000000000004,
|
676 |
+
1.5700000000000005,
|
677 |
+
1.9000000000000008,
|
678 |
+
2.3100000000000014,
|
679 |
+
2.759999999999992,
|
680 |
+
3.3899999999999784,
|
681 |
+
3.9399999999999666,
|
682 |
+
4.009999999999965,
|
683 |
+
4.289999999999959,
|
684 |
+
4.349999999999958,
|
685 |
+
5.349999999999937,
|
686 |
+
6.659999999999909,
|
687 |
+
7.029999999999901,
|
688 |
+
7.51999999999989,
|
689 |
+
8.00999999999988,
|
690 |
+
8.249999999999876,
|
691 |
+
8.279999999999875,
|
692 |
+
9.629999999999846,
|
693 |
+
9.89999999999984,
|
694 |
+
10.589999999999826,
|
695 |
+
11.049999999999816,
|
696 |
+
11.7899999999998,
|
697 |
+
12.189999999999792,
|
698 |
+
12.889999999999777,
|
699 |
+
13.129999999999772,
|
700 |
+
13.16999999999977,
|
701 |
+
13.20999999999977,
|
702 |
+
13.479999999999764,
|
703 |
+
13.539999999999763,
|
704 |
+
13.779999999999758,
|
705 |
+
13.929999999999755,
|
706 |
+
14.429999999999744,
|
707 |
+
14.759999999999737,
|
708 |
+
15.149999999999729,
|
709 |
+
15.419999999999723,
|
710 |
+
15.53999999999972,
|
711 |
+
15.659999999999718,
|
712 |
+
15.749999999999716,
|
713 |
+
15.759999999999716,
|
714 |
+
15.799999999999715,
|
715 |
+
16.05999999999971,
|
716 |
+
16.079999999999714,
|
717 |
+
16.11999999999972,
|
718 |
+
16.11999999999972,
|
719 |
+
16.18999999999973,
|
720 |
+
16.31999999999975,
|
721 |
+
16.539999999999786,
|
722 |
+
16.799999999999827
|
723 |
+
],
|
724 |
+
"type": "su"
|
725 |
+
},
|
726 |
+
"rope_theta": 10000.0,
|
727 |
+
"sliding_window": 131072,
|
728 |
+
"tie_word_embeddings": false,
|
729 |
+
"torch_dtype": "bfloat16",
|
730 |
+
"transformers_version": "4.43.1",
|
731 |
+
"use_cache": true,
|
732 |
+
"vocab_size": 32064
|
733 |
+
}
|
734 |
+
|
735 |
+
|
736 |
+
[INFO|tokenization_utils_base.py:2702] 2024-07-24 21:50:42,263 >> tokenizer config file saved in saves/Custom/lora/train_2024-07-24-15-00-21/tokenizer_config.json
|
737 |
+
|
738 |
+
[INFO|tokenization_utils_base.py:2711] 2024-07-24 21:50:42,264 >> Special tokens file saved in saves/Custom/lora/train_2024-07-24-15-00-21/special_tokens_map.json
|
739 |
+
|
740 |
+
[WARNING|ploting.py:89] 2024-07-24 21:50:42,678 >> No metric eval_loss to plot.
|
741 |
+
|
742 |
+
[WARNING|ploting.py:89] 2024-07-24 21:50:42,678 >> No metric eval_accuracy to plot.
|
743 |
+
|
744 |
+
[INFO|modelcard.py:449] 2024-07-24 21:50:42,679 >> Dropping the following result as it does not have all the necessary fields:
|
745 |
+
{'task': {'name': 'Causal Language Modeling', 'type': 'text-generation'}}
|
746 |
+
|
special_tokens_map.json
ADDED
@@ -0,0 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"bos_token": {
|
3 |
+
"content": "<s>",
|
4 |
+
"lstrip": false,
|
5 |
+
"normalized": false,
|
6 |
+
"rstrip": false,
|
7 |
+
"single_word": false
|
8 |
+
},
|
9 |
+
"eos_token": {
|
10 |
+
"content": "<|endoftext|>",
|
11 |
+
"lstrip": false,
|
12 |
+
"normalized": false,
|
13 |
+
"rstrip": false,
|
14 |
+
"single_word": false
|
15 |
+
},
|
16 |
+
"pad_token": {
|
17 |
+
"content": "<|endoftext|>",
|
18 |
+
"lstrip": false,
|
19 |
+
"normalized": false,
|
20 |
+
"rstrip": false,
|
21 |
+
"single_word": false
|
22 |
+
},
|
23 |
+
"unk_token": {
|
24 |
+
"content": "<unk>",
|
25 |
+
"lstrip": false,
|
26 |
+
"normalized": false,
|
27 |
+
"rstrip": false,
|
28 |
+
"single_word": false
|
29 |
+
}
|
30 |
+
}
|
tokenizer.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
tokenizer.model
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9e556afd44213b6bd1be2b850ebbbd98f5481437a8021afaf58ee7fb1818d347
|
3 |
+
size 499723
|
tokenizer_config.json
ADDED
@@ -0,0 +1,132 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"add_bos_token": false,
|
3 |
+
"add_eos_token": false,
|
4 |
+
"add_prefix_space": null,
|
5 |
+
"added_tokens_decoder": {
|
6 |
+
"0": {
|
7 |
+
"content": "<unk>",
|
8 |
+
"lstrip": false,
|
9 |
+
"normalized": false,
|
10 |
+
"rstrip": false,
|
11 |
+
"single_word": false,
|
12 |
+
"special": true
|
13 |
+
},
|
14 |
+
"1": {
|
15 |
+
"content": "<s>",
|
16 |
+
"lstrip": false,
|
17 |
+
"normalized": false,
|
18 |
+
"rstrip": false,
|
19 |
+
"single_word": false,
|
20 |
+
"special": true
|
21 |
+
},
|
22 |
+
"2": {
|
23 |
+
"content": "</s>",
|
24 |
+
"lstrip": false,
|
25 |
+
"normalized": false,
|
26 |
+
"rstrip": true,
|
27 |
+
"single_word": false,
|
28 |
+
"special": false
|
29 |
+
},
|
30 |
+
"32000": {
|
31 |
+
"content": "<|endoftext|>",
|
32 |
+
"lstrip": false,
|
33 |
+
"normalized": false,
|
34 |
+
"rstrip": false,
|
35 |
+
"single_word": false,
|
36 |
+
"special": true
|
37 |
+
},
|
38 |
+
"32001": {
|
39 |
+
"content": "<|assistant|>",
|
40 |
+
"lstrip": false,
|
41 |
+
"normalized": false,
|
42 |
+
"rstrip": true,
|
43 |
+
"single_word": false,
|
44 |
+
"special": true
|
45 |
+
},
|
46 |
+
"32002": {
|
47 |
+
"content": "<|placeholder1|>",
|
48 |
+
"lstrip": false,
|
49 |
+
"normalized": false,
|
50 |
+
"rstrip": true,
|
51 |
+
"single_word": false,
|
52 |
+
"special": true
|
53 |
+
},
|
54 |
+
"32003": {
|
55 |
+
"content": "<|placeholder2|>",
|
56 |
+
"lstrip": false,
|
57 |
+
"normalized": false,
|
58 |
+
"rstrip": true,
|
59 |
+
"single_word": false,
|
60 |
+
"special": true
|
61 |
+
},
|
62 |
+
"32004": {
|
63 |
+
"content": "<|placeholder3|>",
|
64 |
+
"lstrip": false,
|
65 |
+
"normalized": false,
|
66 |
+
"rstrip": true,
|
67 |
+
"single_word": false,
|
68 |
+
"special": true
|
69 |
+
},
|
70 |
+
"32005": {
|
71 |
+
"content": "<|placeholder4|>",
|
72 |
+
"lstrip": false,
|
73 |
+
"normalized": false,
|
74 |
+
"rstrip": true,
|
75 |
+
"single_word": false,
|
76 |
+
"special": true
|
77 |
+
},
|
78 |
+
"32006": {
|
79 |
+
"content": "<|system|>",
|
80 |
+
"lstrip": false,
|
81 |
+
"normalized": false,
|
82 |
+
"rstrip": true,
|
83 |
+
"single_word": false,
|
84 |
+
"special": true
|
85 |
+
},
|
86 |
+
"32007": {
|
87 |
+
"content": "<|end|>",
|
88 |
+
"lstrip": false,
|
89 |
+
"normalized": false,
|
90 |
+
"rstrip": true,
|
91 |
+
"single_word": false,
|
92 |
+
"special": true
|
93 |
+
},
|
94 |
+
"32008": {
|
95 |
+
"content": "<|placeholder5|>",
|
96 |
+
"lstrip": false,
|
97 |
+
"normalized": false,
|
98 |
+
"rstrip": true,
|
99 |
+
"single_word": false,
|
100 |
+
"special": true
|
101 |
+
},
|
102 |
+
"32009": {
|
103 |
+
"content": "<|placeholder6|>",
|
104 |
+
"lstrip": false,
|
105 |
+
"normalized": false,
|
106 |
+
"rstrip": true,
|
107 |
+
"single_word": false,
|
108 |
+
"special": true
|
109 |
+
},
|
110 |
+
"32010": {
|
111 |
+
"content": "<|user|>",
|
112 |
+
"lstrip": false,
|
113 |
+
"normalized": false,
|
114 |
+
"rstrip": true,
|
115 |
+
"single_word": false,
|
116 |
+
"special": true
|
117 |
+
}
|
118 |
+
},
|
119 |
+
"bos_token": "<s>",
|
120 |
+
"chat_template": "{% if messages[0]['role'] == 'system' %}{% set system_message = messages[0]['content'] %}{% endif %}{% if system_message is defined %}{{ system_message + '\n' }}{% endif %}{% for message in messages %}{% set content = message['content'] %}{% if message['role'] == 'user' %}{{ 'Human: ' + content + '\nAssistant:' }}{% elif message['role'] == 'assistant' %}{{ content + '<|endoftext|>' + '\n' }}{% endif %}{% endfor %}",
|
121 |
+
"clean_up_tokenization_spaces": false,
|
122 |
+
"eos_token": "<|endoftext|>",
|
123 |
+
"legacy": false,
|
124 |
+
"model_max_length": 131072,
|
125 |
+
"pad_token": "<|endoftext|>",
|
126 |
+
"padding_side": "right",
|
127 |
+
"sp_model_kwargs": {},
|
128 |
+
"split_special_tokens": false,
|
129 |
+
"tokenizer_class": "LlamaTokenizer",
|
130 |
+
"unk_token": "<unk>",
|
131 |
+
"use_default_system_prompt": false
|
132 |
+
}
|
train_results.json
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"epoch": 2.9902912621359223,
|
3 |
+
"num_input_tokens_seen": 61072080,
|
4 |
+
"total_flos": 5.065534456681464e+18,
|
5 |
+
"train_loss": 0.1973544524990635,
|
6 |
+
"train_runtime": 23852.8939,
|
7 |
+
"train_samples_per_second": 0.622,
|
8 |
+
"train_steps_per_second": 0.019
|
9 |
+
}
|
trainer_log.jsonl
ADDED
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{"current_steps": 10, "total_steps": 462, "loss": 0.5099, "learning_rate": 1e-05, "epoch": 0.06472491909385113, "percentage": 2.16, "elapsed_time": "0:08:24", "remaining_time": "6:19:55", "throughput": "2571.26", "total_tokens": 1296736}
|
2 |
+
{"current_steps": 20, "total_steps": 462, "loss": 0.5115, "learning_rate": 2e-05, "epoch": 0.12944983818770225, "percentage": 4.33, "elapsed_time": "0:17:06", "remaining_time": "6:18:14", "throughput": "2534.92", "total_tokens": 2603088}
|
3 |
+
{"current_steps": 30, "total_steps": 462, "loss": 0.4846, "learning_rate": 3e-05, "epoch": 0.1941747572815534, "percentage": 6.49, "elapsed_time": "0:25:32", "remaining_time": "6:07:42", "throughput": "2535.35", "total_tokens": 3884384}
|
4 |
+
{"current_steps": 40, "total_steps": 462, "loss": 0.4076, "learning_rate": 4e-05, "epoch": 0.2588996763754045, "percentage": 8.66, "elapsed_time": "0:34:25", "remaining_time": "6:03:11", "throughput": "2553.97", "total_tokens": 5275424}
|
5 |
+
{"current_steps": 50, "total_steps": 462, "loss": 0.3073, "learning_rate": 5e-05, "epoch": 0.32362459546925565, "percentage": 10.82, "elapsed_time": "0:43:30", "remaining_time": "5:58:30", "throughput": "2556.06", "total_tokens": 6672576}
|
6 |
+
{"current_steps": 60, "total_steps": 462, "loss": 0.2516, "learning_rate": 4.992735514089577e-05, "epoch": 0.3883495145631068, "percentage": 12.99, "elapsed_time": "0:51:43", "remaining_time": "5:46:34", "throughput": "2559.12", "total_tokens": 7942496}
|
7 |
+
{"current_steps": 70, "total_steps": 462, "loss": 0.2256, "learning_rate": 4.970984274562741e-05, "epoch": 0.45307443365695793, "percentage": 15.15, "elapsed_time": "1:00:47", "remaining_time": "5:40:24", "throughput": "2549.95", "total_tokens": 9300240}
|
8 |
+
{"current_steps": 80, "total_steps": 462, "loss": 0.2146, "learning_rate": 4.934872690677953e-05, "epoch": 0.517799352750809, "percentage": 17.32, "elapsed_time": "1:09:49", "remaining_time": "5:33:26", "throughput": "2547.79", "total_tokens": 10674880}
|
9 |
+
{"current_steps": 90, "total_steps": 462, "loss": 0.2018, "learning_rate": 4.884610628109082e-05, "epoch": 0.5825242718446602, "percentage": 19.48, "elapsed_time": "1:18:41", "remaining_time": "5:25:16", "throughput": "2552.59", "total_tokens": 12052928}
|
10 |
+
{"current_steps": 100, "total_steps": 462, "loss": 0.1958, "learning_rate": 4.820490189292415e-05, "epoch": 0.6472491909385113, "percentage": 21.65, "elapsed_time": "1:27:15", "remaining_time": "5:15:53", "throughput": "2556.94", "total_tokens": 13387616}
|
11 |
+
{"current_steps": 110, "total_steps": 462, "loss": 0.1912, "learning_rate": 4.742884015847436e-05, "epoch": 0.7119741100323624, "percentage": 23.81, "elapsed_time": "1:35:39", "remaining_time": "5:06:06", "throughput": "2557.17", "total_tokens": 14676976}
|
12 |
+
{"current_steps": 120, "total_steps": 462, "loss": 0.1876, "learning_rate": 4.652243122936986e-05, "epoch": 0.7766990291262136, "percentage": 25.97, "elapsed_time": "1:44:09", "remaining_time": "4:56:51", "throughput": "2558.65", "total_tokens": 15991008}
|
13 |
+
{"current_steps": 130, "total_steps": 462, "loss": 0.1802, "learning_rate": 4.5490942781526316e-05, "epoch": 0.8414239482200647, "percentage": 28.14, "elapsed_time": "1:52:27", "remaining_time": "4:47:12", "throughput": "2561.14", "total_tokens": 17281840}
|
14 |
+
{"current_steps": 140, "total_steps": 462, "loss": 0.1793, "learning_rate": 4.434036940158062e-05, "epoch": 0.9061488673139159, "percentage": 30.3, "elapsed_time": "2:01:06", "remaining_time": "4:38:33", "throughput": "2560.01", "total_tokens": 18603200}
|
15 |
+
{"current_steps": 150, "total_steps": 462, "loss": 0.1759, "learning_rate": 4.307739774881878e-05, "epoch": 0.970873786407767, "percentage": 32.47, "elapsed_time": "2:08:53", "remaining_time": "4:28:06", "throughput": "2565.87", "total_tokens": 19844160}
|
16 |
+
{"current_steps": 160, "total_steps": 462, "loss": 0.1746, "learning_rate": 4.170936769506222e-05, "epoch": 1.035598705501618, "percentage": 34.63, "elapsed_time": "2:17:10", "remaining_time": "4:18:55", "throughput": "2564.86", "total_tokens": 21111168}
|
17 |
+
{"current_steps": 170, "total_steps": 462, "loss": 0.1699, "learning_rate": 4.024422966835136e-05, "epoch": 1.1003236245954693, "percentage": 36.8, "elapsed_time": "2:25:52", "remaining_time": "4:10:34", "throughput": "2564.04", "total_tokens": 22442800}
|
18 |
+
{"current_steps": 180, "total_steps": 462, "loss": 0.168, "learning_rate": 3.86904984483277e-05, "epoch": 1.1650485436893203, "percentage": 38.96, "elapsed_time": "2:35:02", "remaining_time": "4:02:54", "throughput": "2560.03", "total_tokens": 23815072}
|
19 |
+
{"current_steps": 190, "total_steps": 462, "loss": 0.1646, "learning_rate": 3.7057203681836406e-05, "epoch": 1.2297734627831716, "percentage": 41.13, "elapsed_time": "2:42:57", "remaining_time": "3:53:17", "throughput": "2563.38", "total_tokens": 25064672}
|
20 |
+
{"current_steps": 200, "total_steps": 462, "loss": 0.1667, "learning_rate": 3.535383740633246e-05, "epoch": 1.2944983818770226, "percentage": 43.29, "elapsed_time": "2:51:22", "remaining_time": "3:44:30", "throughput": "2565.53", "total_tokens": 26380864}
|
21 |
+
{"current_steps": 210, "total_steps": 462, "loss": 0.1664, "learning_rate": 3.3590298886062833e-05, "epoch": 1.3592233009708738, "percentage": 45.45, "elapsed_time": "3:00:09", "remaining_time": "3:36:11", "throughput": "2565.84", "total_tokens": 27735952}
|
22 |
+
{"current_steps": 220, "total_steps": 462, "loss": 0.1622, "learning_rate": 3.177683708161389e-05, "epoch": 1.4239482200647249, "percentage": 47.62, "elapsed_time": "3:08:20", "remaining_time": "3:27:10", "throughput": "2565.58", "total_tokens": 28991248}
|
23 |
+
{"current_steps": 230, "total_steps": 462, "loss": 0.1623, "learning_rate": 2.9923991087167658e-05, "epoch": 1.4886731391585761, "percentage": 49.78, "elapsed_time": "3:16:56", "remaining_time": "3:18:39", "throughput": "2565.43", "total_tokens": 30313776}
|
24 |
+
{"current_steps": 240, "total_steps": 462, "loss": 0.1616, "learning_rate": 2.804252888162079e-05, "epoch": 1.5533980582524272, "percentage": 51.95, "elapsed_time": "3:25:43", "remaining_time": "3:10:17", "throughput": "2565.51", "total_tokens": 31667088}
|
25 |
+
{"current_steps": 250, "total_steps": 462, "loss": 0.159, "learning_rate": 2.6143384749519866e-05, "epoch": 1.6181229773462782, "percentage": 54.11, "elapsed_time": "3:33:56", "remaining_time": "3:01:25", "throughput": "2566.31", "total_tokens": 32942832}
|
26 |
+
{"current_steps": 260, "total_steps": 462, "loss": 0.1619, "learning_rate": 2.423759573549647e-05, "epoch": 1.6828478964401294, "percentage": 56.28, "elapsed_time": "3:42:42", "remaining_time": "2:53:01", "throughput": "2564.17", "total_tokens": 34263264}
|
27 |
+
{"current_steps": 270, "total_steps": 462, "loss": 0.1616, "learning_rate": 2.23362375015031e-05, "epoch": 1.7475728155339807, "percentage": 58.44, "elapsed_time": "3:51:17", "remaining_time": "2:44:28", "throughput": "2565.53", "total_tokens": 35603952}
|
28 |
+
{"current_steps": 280, "total_steps": 462, "loss": 0.1604, "learning_rate": 2.0450359959620967e-05, "epoch": 1.8122977346278317, "percentage": 60.61, "elapsed_time": "3:59:33", "remaining_time": "2:35:42", "throughput": "2566.61", "total_tokens": 36890768}
|
29 |
+
{"current_steps": 290, "total_steps": 462, "loss": 0.1563, "learning_rate": 1.8590923054515503e-05, "epoch": 1.8770226537216828, "percentage": 62.77, "elapsed_time": "4:08:43", "remaining_time": "2:27:31", "throughput": "2564.70", "total_tokens": 38275536}
|
30 |
+
{"current_steps": 300, "total_steps": 462, "loss": 0.1548, "learning_rate": 1.676873306874547e-05, "epoch": 1.941747572815534, "percentage": 64.94, "elapsed_time": "4:17:35", "remaining_time": "2:19:05", "throughput": "2565.36", "total_tokens": 39648800}
|
31 |
+
{"current_steps": 310, "total_steps": 462, "loss": 0.1555, "learning_rate": 1.4994379821093049e-05, "epoch": 2.0064724919093853, "percentage": 67.1, "elapsed_time": "4:26:19", "remaining_time": "2:10:35", "throughput": "2565.22", "total_tokens": 40991008}
|
32 |
+
{"current_steps": 320, "total_steps": 462, "loss": 0.1526, "learning_rate": 1.3278175122892416e-05, "epoch": 2.071197411003236, "percentage": 69.26, "elapsed_time": "4:35:09", "remaining_time": "2:02:05", "throughput": "2564.82", "total_tokens": 42342880}
|
33 |
+
{"current_steps": 330, "total_steps": 462, "loss": 0.1526, "learning_rate": 1.1630092850023147e-05, "epoch": 2.1359223300970873, "percentage": 71.43, "elapsed_time": "4:43:48", "remaining_time": "1:53:31", "throughput": "2563.60", "total_tokens": 43654784}
|
34 |
+
{"current_steps": 340, "total_steps": 462, "loss": 0.1516, "learning_rate": 1.005971097884561e-05, "epoch": 2.2006472491909386, "percentage": 73.59, "elapsed_time": "4:52:41", "remaining_time": "1:45:01", "throughput": "2564.75", "total_tokens": 45039744}
|
35 |
+
{"current_steps": 350, "total_steps": 462, "loss": 0.1524, "learning_rate": 8.576155922941548e-06, "epoch": 2.26537216828479, "percentage": 75.76, "elapsed_time": "5:00:59", "remaining_time": "1:36:18", "throughput": "2565.56", "total_tokens": 46331568}
|
36 |
+
{"current_steps": 360, "total_steps": 462, "loss": 0.1502, "learning_rate": 7.1880494941517026e-06, "epoch": 2.3300970873786406, "percentage": 77.92, "elapsed_time": "5:09:57", "remaining_time": "1:27:49", "throughput": "2564.08", "total_tokens": 47685808}
|
37 |
+
{"current_steps": 370, "total_steps": 462, "loss": 0.1506, "learning_rate": 5.903458796151381e-06, "epoch": 2.394822006472492, "percentage": 80.09, "elapsed_time": "5:18:40", "remaining_time": "1:19:14", "throughput": "2563.27", "total_tokens": 49011728}
|
38 |
+
{"current_steps": 380, "total_steps": 462, "loss": 0.1479, "learning_rate": 4.729849341761602e-06, "epoch": 2.459546925566343, "percentage": 82.25, "elapsed_time": "5:27:36", "remaining_time": "1:10:41", "throughput": "2560.31", "total_tokens": 50327504}
|
39 |
+
{"current_steps": 390, "total_steps": 462, "loss": 0.1501, "learning_rate": 3.674041666458963e-06, "epoch": 2.524271844660194, "percentage": 84.42, "elapsed_time": "5:35:57", "remaining_time": "1:02:01", "throughput": "2560.79", "total_tokens": 51620256}
|
40 |
+
{"current_steps": 400, "total_steps": 462, "loss": 0.1504, "learning_rate": 2.7421716902285623e-06, "epoch": 2.588996763754045, "percentage": 86.58, "elapsed_time": "5:44:30", "remaining_time": "0:53:24", "throughput": "2561.70", "total_tokens": 52952880}
|
41 |
+
{"current_steps": 410, "total_steps": 462, "loss": 0.1524, "learning_rate": 1.939655058120521e-06, "epoch": 2.6537216828478964, "percentage": 88.74, "elapsed_time": "5:52:47", "remaining_time": "0:44:44", "throughput": "2561.78", "total_tokens": 54225632}
|
42 |
+
{"current_steps": 420, "total_steps": 462, "loss": 0.1494, "learning_rate": 1.271155666748311e-06, "epoch": 2.7184466019417477, "percentage": 90.91, "elapsed_time": "6:01:44", "remaining_time": "0:36:10", "throughput": "2559.88", "total_tokens": 55561888}
|
43 |
+
{"current_steps": 430, "total_steps": 462, "loss": 0.1515, "learning_rate": 7.405585596397313e-07, "epoch": 2.783171521035599, "percentage": 93.07, "elapsed_time": "6:10:28", "remaining_time": "0:27:34", "throughput": "2559.75", "total_tokens": 56900512}
|
44 |
+
{"current_steps": 440, "total_steps": 462, "loss": 0.1504, "learning_rate": 3.5094734896174987e-07, "epoch": 2.8478964401294498, "percentage": 95.24, "elapsed_time": "6:18:53", "remaining_time": "0:18:56", "throughput": "2559.97", "total_tokens": 58197632}
|
45 |
+
{"current_steps": 450, "total_steps": 462, "loss": 0.1498, "learning_rate": 1.0458629483476867e-07, "epoch": 2.912621359223301, "percentage": 97.4, "elapsed_time": "6:27:36", "remaining_time": "0:10:20", "throughput": "2559.45", "total_tokens": 59523808}
|
46 |
+
{"current_steps": 460, "total_steps": 462, "loss": 0.1469, "learning_rate": 2.9071463840540935e-09, "epoch": 2.9773462783171523, "percentage": 99.57, "elapsed_time": "6:35:45", "remaining_time": "0:01:43", "throughput": "2559.91", "total_tokens": 60787344}
|
47 |
+
{"current_steps": 462, "total_steps": 462, "epoch": 2.9902912621359223, "percentage": 100.0, "elapsed_time": "6:37:32", "remaining_time": "0:00:00", "throughput": "2560.36", "total_tokens": 61072080}
|
trainer_state.json
ADDED
@@ -0,0 +1,411 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_metric": null,
|
3 |
+
"best_model_checkpoint": null,
|
4 |
+
"epoch": 2.9902912621359223,
|
5 |
+
"eval_steps": 500,
|
6 |
+
"global_step": 462,
|
7 |
+
"is_hyper_param_search": false,
|
8 |
+
"is_local_process_zero": true,
|
9 |
+
"is_world_process_zero": true,
|
10 |
+
"log_history": [
|
11 |
+
{
|
12 |
+
"epoch": 0.06472491909385113,
|
13 |
+
"grad_norm": 0.09439529478549957,
|
14 |
+
"learning_rate": 1e-05,
|
15 |
+
"loss": 0.5099,
|
16 |
+
"num_input_tokens_seen": 1296736,
|
17 |
+
"step": 10
|
18 |
+
},
|
19 |
+
{
|
20 |
+
"epoch": 0.12944983818770225,
|
21 |
+
"grad_norm": 0.11287413537502289,
|
22 |
+
"learning_rate": 2e-05,
|
23 |
+
"loss": 0.5115,
|
24 |
+
"num_input_tokens_seen": 2603088,
|
25 |
+
"step": 20
|
26 |
+
},
|
27 |
+
{
|
28 |
+
"epoch": 0.1941747572815534,
|
29 |
+
"grad_norm": 0.13694792985916138,
|
30 |
+
"learning_rate": 3e-05,
|
31 |
+
"loss": 0.4846,
|
32 |
+
"num_input_tokens_seen": 3884384,
|
33 |
+
"step": 30
|
34 |
+
},
|
35 |
+
{
|
36 |
+
"epoch": 0.2588996763754045,
|
37 |
+
"grad_norm": 0.14105218648910522,
|
38 |
+
"learning_rate": 4e-05,
|
39 |
+
"loss": 0.4076,
|
40 |
+
"num_input_tokens_seen": 5275424,
|
41 |
+
"step": 40
|
42 |
+
},
|
43 |
+
{
|
44 |
+
"epoch": 0.32362459546925565,
|
45 |
+
"grad_norm": 0.09946475923061371,
|
46 |
+
"learning_rate": 5e-05,
|
47 |
+
"loss": 0.3073,
|
48 |
+
"num_input_tokens_seen": 6672576,
|
49 |
+
"step": 50
|
50 |
+
},
|
51 |
+
{
|
52 |
+
"epoch": 0.3883495145631068,
|
53 |
+
"grad_norm": 0.11918677389621735,
|
54 |
+
"learning_rate": 4.992735514089577e-05,
|
55 |
+
"loss": 0.2516,
|
56 |
+
"num_input_tokens_seen": 7942496,
|
57 |
+
"step": 60
|
58 |
+
},
|
59 |
+
{
|
60 |
+
"epoch": 0.45307443365695793,
|
61 |
+
"grad_norm": 0.07622148096561432,
|
62 |
+
"learning_rate": 4.970984274562741e-05,
|
63 |
+
"loss": 0.2256,
|
64 |
+
"num_input_tokens_seen": 9300240,
|
65 |
+
"step": 70
|
66 |
+
},
|
67 |
+
{
|
68 |
+
"epoch": 0.517799352750809,
|
69 |
+
"grad_norm": 0.07995602488517761,
|
70 |
+
"learning_rate": 4.934872690677953e-05,
|
71 |
+
"loss": 0.2146,
|
72 |
+
"num_input_tokens_seen": 10674880,
|
73 |
+
"step": 80
|
74 |
+
},
|
75 |
+
{
|
76 |
+
"epoch": 0.5825242718446602,
|
77 |
+
"grad_norm": 0.07474230229854584,
|
78 |
+
"learning_rate": 4.884610628109082e-05,
|
79 |
+
"loss": 0.2018,
|
80 |
+
"num_input_tokens_seen": 12052928,
|
81 |
+
"step": 90
|
82 |
+
},
|
83 |
+
{
|
84 |
+
"epoch": 0.6472491909385113,
|
85 |
+
"grad_norm": 0.07976042479276657,
|
86 |
+
"learning_rate": 4.820490189292415e-05,
|
87 |
+
"loss": 0.1958,
|
88 |
+
"num_input_tokens_seen": 13387616,
|
89 |
+
"step": 100
|
90 |
+
},
|
91 |
+
{
|
92 |
+
"epoch": 0.7119741100323624,
|
93 |
+
"grad_norm": 0.08398651331663132,
|
94 |
+
"learning_rate": 4.742884015847436e-05,
|
95 |
+
"loss": 0.1912,
|
96 |
+
"num_input_tokens_seen": 14676976,
|
97 |
+
"step": 110
|
98 |
+
},
|
99 |
+
{
|
100 |
+
"epoch": 0.7766990291262136,
|
101 |
+
"grad_norm": 0.088481605052948,
|
102 |
+
"learning_rate": 4.652243122936986e-05,
|
103 |
+
"loss": 0.1876,
|
104 |
+
"num_input_tokens_seen": 15991008,
|
105 |
+
"step": 120
|
106 |
+
},
|
107 |
+
{
|
108 |
+
"epoch": 0.8414239482200647,
|
109 |
+
"grad_norm": 0.09539451450109482,
|
110 |
+
"learning_rate": 4.5490942781526316e-05,
|
111 |
+
"loss": 0.1802,
|
112 |
+
"num_input_tokens_seen": 17281840,
|
113 |
+
"step": 130
|
114 |
+
},
|
115 |
+
{
|
116 |
+
"epoch": 0.9061488673139159,
|
117 |
+
"grad_norm": 0.08230065554380417,
|
118 |
+
"learning_rate": 4.434036940158062e-05,
|
119 |
+
"loss": 0.1793,
|
120 |
+
"num_input_tokens_seen": 18603200,
|
121 |
+
"step": 140
|
122 |
+
},
|
123 |
+
{
|
124 |
+
"epoch": 0.970873786407767,
|
125 |
+
"grad_norm": 0.09070578217506409,
|
126 |
+
"learning_rate": 4.307739774881878e-05,
|
127 |
+
"loss": 0.1759,
|
128 |
+
"num_input_tokens_seen": 19844160,
|
129 |
+
"step": 150
|
130 |
+
},
|
131 |
+
{
|
132 |
+
"epoch": 1.035598705501618,
|
133 |
+
"grad_norm": 0.09057565778493881,
|
134 |
+
"learning_rate": 4.170936769506222e-05,
|
135 |
+
"loss": 0.1746,
|
136 |
+
"num_input_tokens_seen": 21111168,
|
137 |
+
"step": 160
|
138 |
+
},
|
139 |
+
{
|
140 |
+
"epoch": 1.1003236245954693,
|
141 |
+
"grad_norm": 0.08840323239564896,
|
142 |
+
"learning_rate": 4.024422966835136e-05,
|
143 |
+
"loss": 0.1699,
|
144 |
+
"num_input_tokens_seen": 22442800,
|
145 |
+
"step": 170
|
146 |
+
},
|
147 |
+
{
|
148 |
+
"epoch": 1.1650485436893203,
|
149 |
+
"grad_norm": 0.10438356548547745,
|
150 |
+
"learning_rate": 3.86904984483277e-05,
|
151 |
+
"loss": 0.168,
|
152 |
+
"num_input_tokens_seen": 23815072,
|
153 |
+
"step": 180
|
154 |
+
},
|
155 |
+
{
|
156 |
+
"epoch": 1.2297734627831716,
|
157 |
+
"grad_norm": 0.09192164242267609,
|
158 |
+
"learning_rate": 3.7057203681836406e-05,
|
159 |
+
"loss": 0.1646,
|
160 |
+
"num_input_tokens_seen": 25064672,
|
161 |
+
"step": 190
|
162 |
+
},
|
163 |
+
{
|
164 |
+
"epoch": 1.2944983818770226,
|
165 |
+
"grad_norm": 0.08962567150592804,
|
166 |
+
"learning_rate": 3.535383740633246e-05,
|
167 |
+
"loss": 0.1667,
|
168 |
+
"num_input_tokens_seen": 26380864,
|
169 |
+
"step": 200
|
170 |
+
},
|
171 |
+
{
|
172 |
+
"epoch": 1.3592233009708738,
|
173 |
+
"grad_norm": 0.1008475124835968,
|
174 |
+
"learning_rate": 3.3590298886062833e-05,
|
175 |
+
"loss": 0.1664,
|
176 |
+
"num_input_tokens_seen": 27735952,
|
177 |
+
"step": 210
|
178 |
+
},
|
179 |
+
{
|
180 |
+
"epoch": 1.4239482200647249,
|
181 |
+
"grad_norm": 0.09535906463861465,
|
182 |
+
"learning_rate": 3.177683708161389e-05,
|
183 |
+
"loss": 0.1622,
|
184 |
+
"num_input_tokens_seen": 28991248,
|
185 |
+
"step": 220
|
186 |
+
},
|
187 |
+
{
|
188 |
+
"epoch": 1.4886731391585761,
|
189 |
+
"grad_norm": 0.09163232892751694,
|
190 |
+
"learning_rate": 2.9923991087167658e-05,
|
191 |
+
"loss": 0.1623,
|
192 |
+
"num_input_tokens_seen": 30313776,
|
193 |
+
"step": 230
|
194 |
+
},
|
195 |
+
{
|
196 |
+
"epoch": 1.5533980582524272,
|
197 |
+
"grad_norm": 0.09542625397443771,
|
198 |
+
"learning_rate": 2.804252888162079e-05,
|
199 |
+
"loss": 0.1616,
|
200 |
+
"num_input_tokens_seen": 31667088,
|
201 |
+
"step": 240
|
202 |
+
},
|
203 |
+
{
|
204 |
+
"epoch": 1.6181229773462782,
|
205 |
+
"grad_norm": 0.09714806824922562,
|
206 |
+
"learning_rate": 2.6143384749519866e-05,
|
207 |
+
"loss": 0.159,
|
208 |
+
"num_input_tokens_seen": 32942832,
|
209 |
+
"step": 250
|
210 |
+
},
|
211 |
+
{
|
212 |
+
"epoch": 1.6828478964401294,
|
213 |
+
"grad_norm": 0.10313564538955688,
|
214 |
+
"learning_rate": 2.423759573549647e-05,
|
215 |
+
"loss": 0.1619,
|
216 |
+
"num_input_tokens_seen": 34263264,
|
217 |
+
"step": 260
|
218 |
+
},
|
219 |
+
{
|
220 |
+
"epoch": 1.7475728155339807,
|
221 |
+
"grad_norm": 0.09954512119293213,
|
222 |
+
"learning_rate": 2.23362375015031e-05,
|
223 |
+
"loss": 0.1616,
|
224 |
+
"num_input_tokens_seen": 35603952,
|
225 |
+
"step": 270
|
226 |
+
},
|
227 |
+
{
|
228 |
+
"epoch": 1.8122977346278317,
|
229 |
+
"grad_norm": 0.1067601814866066,
|
230 |
+
"learning_rate": 2.0450359959620967e-05,
|
231 |
+
"loss": 0.1604,
|
232 |
+
"num_input_tokens_seen": 36890768,
|
233 |
+
"step": 280
|
234 |
+
},
|
235 |
+
{
|
236 |
+
"epoch": 1.8770226537216828,
|
237 |
+
"grad_norm": 0.09286054223775864,
|
238 |
+
"learning_rate": 1.8590923054515503e-05,
|
239 |
+
"loss": 0.1563,
|
240 |
+
"num_input_tokens_seen": 38275536,
|
241 |
+
"step": 290
|
242 |
+
},
|
243 |
+
{
|
244 |
+
"epoch": 1.941747572815534,
|
245 |
+
"grad_norm": 0.09992893040180206,
|
246 |
+
"learning_rate": 1.676873306874547e-05,
|
247 |
+
"loss": 0.1548,
|
248 |
+
"num_input_tokens_seen": 39648800,
|
249 |
+
"step": 300
|
250 |
+
},
|
251 |
+
{
|
252 |
+
"epoch": 2.0064724919093853,
|
253 |
+
"grad_norm": 0.09988085925579071,
|
254 |
+
"learning_rate": 1.4994379821093049e-05,
|
255 |
+
"loss": 0.1555,
|
256 |
+
"num_input_tokens_seen": 40991008,
|
257 |
+
"step": 310
|
258 |
+
},
|
259 |
+
{
|
260 |
+
"epoch": 2.071197411003236,
|
261 |
+
"grad_norm": 0.09905947744846344,
|
262 |
+
"learning_rate": 1.3278175122892416e-05,
|
263 |
+
"loss": 0.1526,
|
264 |
+
"num_input_tokens_seen": 42342880,
|
265 |
+
"step": 320
|
266 |
+
},
|
267 |
+
{
|
268 |
+
"epoch": 2.1359223300970873,
|
269 |
+
"grad_norm": 0.10081404447555542,
|
270 |
+
"learning_rate": 1.1630092850023147e-05,
|
271 |
+
"loss": 0.1526,
|
272 |
+
"num_input_tokens_seen": 43654784,
|
273 |
+
"step": 330
|
274 |
+
},
|
275 |
+
{
|
276 |
+
"epoch": 2.2006472491909386,
|
277 |
+
"grad_norm": 0.10631163418292999,
|
278 |
+
"learning_rate": 1.005971097884561e-05,
|
279 |
+
"loss": 0.1516,
|
280 |
+
"num_input_tokens_seen": 45039744,
|
281 |
+
"step": 340
|
282 |
+
},
|
283 |
+
{
|
284 |
+
"epoch": 2.26537216828479,
|
285 |
+
"grad_norm": 0.10914473235607147,
|
286 |
+
"learning_rate": 8.576155922941548e-06,
|
287 |
+
"loss": 0.1524,
|
288 |
+
"num_input_tokens_seen": 46331568,
|
289 |
+
"step": 350
|
290 |
+
},
|
291 |
+
{
|
292 |
+
"epoch": 2.3300970873786406,
|
293 |
+
"grad_norm": 0.10382701456546783,
|
294 |
+
"learning_rate": 7.1880494941517026e-06,
|
295 |
+
"loss": 0.1502,
|
296 |
+
"num_input_tokens_seen": 47685808,
|
297 |
+
"step": 360
|
298 |
+
},
|
299 |
+
{
|
300 |
+
"epoch": 2.394822006472492,
|
301 |
+
"grad_norm": 0.10681454837322235,
|
302 |
+
"learning_rate": 5.903458796151381e-06,
|
303 |
+
"loss": 0.1506,
|
304 |
+
"num_input_tokens_seen": 49011728,
|
305 |
+
"step": 370
|
306 |
+
},
|
307 |
+
{
|
308 |
+
"epoch": 2.459546925566343,
|
309 |
+
"grad_norm": 0.10908275097608566,
|
310 |
+
"learning_rate": 4.729849341761602e-06,
|
311 |
+
"loss": 0.1479,
|
312 |
+
"num_input_tokens_seen": 50327504,
|
313 |
+
"step": 380
|
314 |
+
},
|
315 |
+
{
|
316 |
+
"epoch": 2.524271844660194,
|
317 |
+
"grad_norm": 0.1181163340806961,
|
318 |
+
"learning_rate": 3.674041666458963e-06,
|
319 |
+
"loss": 0.1501,
|
320 |
+
"num_input_tokens_seen": 51620256,
|
321 |
+
"step": 390
|
322 |
+
},
|
323 |
+
{
|
324 |
+
"epoch": 2.588996763754045,
|
325 |
+
"grad_norm": 0.09960592538118362,
|
326 |
+
"learning_rate": 2.7421716902285623e-06,
|
327 |
+
"loss": 0.1504,
|
328 |
+
"num_input_tokens_seen": 52952880,
|
329 |
+
"step": 400
|
330 |
+
},
|
331 |
+
{
|
332 |
+
"epoch": 2.6537216828478964,
|
333 |
+
"grad_norm": 0.10874225944280624,
|
334 |
+
"learning_rate": 1.939655058120521e-06,
|
335 |
+
"loss": 0.1524,
|
336 |
+
"num_input_tokens_seen": 54225632,
|
337 |
+
"step": 410
|
338 |
+
},
|
339 |
+
{
|
340 |
+
"epoch": 2.7184466019417477,
|
341 |
+
"grad_norm": 0.10990295559167862,
|
342 |
+
"learning_rate": 1.271155666748311e-06,
|
343 |
+
"loss": 0.1494,
|
344 |
+
"num_input_tokens_seen": 55561888,
|
345 |
+
"step": 420
|
346 |
+
},
|
347 |
+
{
|
348 |
+
"epoch": 2.783171521035599,
|
349 |
+
"grad_norm": 0.10411816090345383,
|
350 |
+
"learning_rate": 7.405585596397313e-07,
|
351 |
+
"loss": 0.1515,
|
352 |
+
"num_input_tokens_seen": 56900512,
|
353 |
+
"step": 430
|
354 |
+
},
|
355 |
+
{
|
356 |
+
"epoch": 2.8478964401294498,
|
357 |
+
"grad_norm": 0.10403899103403091,
|
358 |
+
"learning_rate": 3.5094734896174987e-07,
|
359 |
+
"loss": 0.1504,
|
360 |
+
"num_input_tokens_seen": 58197632,
|
361 |
+
"step": 440
|
362 |
+
},
|
363 |
+
{
|
364 |
+
"epoch": 2.912621359223301,
|
365 |
+
"grad_norm": 0.1029229462146759,
|
366 |
+
"learning_rate": 1.0458629483476867e-07,
|
367 |
+
"loss": 0.1498,
|
368 |
+
"num_input_tokens_seen": 59523808,
|
369 |
+
"step": 450
|
370 |
+
},
|
371 |
+
{
|
372 |
+
"epoch": 2.9773462783171523,
|
373 |
+
"grad_norm": 0.10788564383983612,
|
374 |
+
"learning_rate": 2.9071463840540935e-09,
|
375 |
+
"loss": 0.1469,
|
376 |
+
"num_input_tokens_seen": 60787344,
|
377 |
+
"step": 460
|
378 |
+
},
|
379 |
+
{
|
380 |
+
"epoch": 2.9902912621359223,
|
381 |
+
"num_input_tokens_seen": 61072080,
|
382 |
+
"step": 462,
|
383 |
+
"total_flos": 5.065534456681464e+18,
|
384 |
+
"train_loss": 0.1973544524990635,
|
385 |
+
"train_runtime": 23852.8939,
|
386 |
+
"train_samples_per_second": 0.622,
|
387 |
+
"train_steps_per_second": 0.019
|
388 |
+
}
|
389 |
+
],
|
390 |
+
"logging_steps": 10,
|
391 |
+
"max_steps": 462,
|
392 |
+
"num_input_tokens_seen": 61072080,
|
393 |
+
"num_train_epochs": 3,
|
394 |
+
"save_steps": 500,
|
395 |
+
"stateful_callbacks": {
|
396 |
+
"TrainerControl": {
|
397 |
+
"args": {
|
398 |
+
"should_epoch_stop": false,
|
399 |
+
"should_evaluate": false,
|
400 |
+
"should_log": false,
|
401 |
+
"should_save": true,
|
402 |
+
"should_training_stop": true
|
403 |
+
},
|
404 |
+
"attributes": {}
|
405 |
+
}
|
406 |
+
},
|
407 |
+
"total_flos": 5.065534456681464e+18,
|
408 |
+
"train_batch_size": 2,
|
409 |
+
"trial_name": null,
|
410 |
+
"trial_params": null
|
411 |
+
}
|
training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:346b49d9842ae5cd7e514f29b370e03b5ce38cb58f691419ff1d443ec5ed1ed3
|
3 |
+
size 6712
|
training_args.yaml
ADDED
@@ -0,0 +1,35 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
bf16: true
|
2 |
+
cutoff_len: 20000
|
3 |
+
dataset: ALS
|
4 |
+
dataset_dir: data
|
5 |
+
ddp_timeout: 180000000
|
6 |
+
deepspeed: cache/ds_z2_offload_config.json
|
7 |
+
do_train: true
|
8 |
+
finetuning_type: lora
|
9 |
+
flash_attn: fa2
|
10 |
+
gradient_accumulation_steps: 8
|
11 |
+
include_num_input_tokens_seen: true
|
12 |
+
learning_rate: 5.0e-05
|
13 |
+
logging_steps: 10
|
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: 1.0
|
20 |
+
max_samples: 100000
|
21 |
+
model_name_or_path: microsoft/Phi-3-medium-128k-instruct
|
22 |
+
num_train_epochs: 3.0
|
23 |
+
optim: adamw_torch
|
24 |
+
output_dir: saves/Custom/lora/train_2024-07-24-15-00-21
|
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: bitsandbytes
|
31 |
+
report_to: none
|
32 |
+
save_steps: 500
|
33 |
+
stage: sft
|
34 |
+
template: default
|
35 |
+
warmup_steps: 50
|
training_loss.png
ADDED