Training in progress, step 168
Browse files- .gitattributes +1 -0
- .ipynb_checkpoints/lora_orpo-checkpoint.yaml +44 -0
- adapter_config.json +34 -0
- adapter_model.safetensors +3 -0
- lora_orpo.yaml +44 -0
- special_tokens_map.json +24 -0
- tokenizer.json +3 -0
- tokenizer_config.json +0 -0
- trainer_log.jsonl +17 -0
- training_args.bin +3 -0
.gitattributes
CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
36 |
+
tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
.ipynb_checkpoints/lora_orpo-checkpoint.yaml
ADDED
@@ -0,0 +1,44 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
### model
|
2 |
+
model_name_or_path: mistralai/Mistral-Nemo-Instruct-2407
|
3 |
+
|
4 |
+
### method
|
5 |
+
stage: dpo
|
6 |
+
pref_loss: orpo
|
7 |
+
do_train: true
|
8 |
+
finetuning_type: lora
|
9 |
+
lora_target: all
|
10 |
+
|
11 |
+
### dataset
|
12 |
+
dataset: bct_non_cot_dpo_1000
|
13 |
+
dataset_dir: data_private
|
14 |
+
template: mistral
|
15 |
+
cutoff_len: 1024
|
16 |
+
# max_samples: 1000
|
17 |
+
overwrite_cache: true
|
18 |
+
preprocessing_num_workers: 16
|
19 |
+
|
20 |
+
### output
|
21 |
+
output_dir: saves/Mistral-Nemo-12B-Instruct/lora/saa
|
22 |
+
logging_steps: 10
|
23 |
+
save_steps: 500
|
24 |
+
plot_loss: true
|
25 |
+
overwrite_output_dir: true
|
26 |
+
save_total_limit: 3
|
27 |
+
load_best_model_at_end: true
|
28 |
+
push_to_hub: true
|
29 |
+
hub_model_id: chchen/Mistral-Nemo-12B-Instruct-SAA
|
30 |
+
|
31 |
+
### train
|
32 |
+
per_device_train_batch_size: 2
|
33 |
+
gradient_accumulation_steps: 8
|
34 |
+
learning_rate: 0.000005
|
35 |
+
num_train_epochs: 3.0
|
36 |
+
lr_scheduler_type: cosine
|
37 |
+
warmup_ratio: 0.1
|
38 |
+
bf16: true
|
39 |
+
|
40 |
+
### eval
|
41 |
+
val_size: 0.1
|
42 |
+
per_device_eval_batch_size: 2
|
43 |
+
evaluation_strategy: steps
|
44 |
+
eval_steps: 500
|
adapter_config.json
ADDED
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"alpha_pattern": {},
|
3 |
+
"auto_mapping": null,
|
4 |
+
"base_model_name_or_path": "mistralai/Mistral-Nemo-Instruct-2407",
|
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.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 |
+
"gate_proj",
|
24 |
+
"up_proj",
|
25 |
+
"q_proj",
|
26 |
+
"o_proj",
|
27 |
+
"v_proj",
|
28 |
+
"k_proj",
|
29 |
+
"down_proj"
|
30 |
+
],
|
31 |
+
"task_type": "CAUSAL_LM",
|
32 |
+
"use_dora": false,
|
33 |
+
"use_rslora": false
|
34 |
+
}
|
adapter_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e3e5498112cdb9b1de5e70ff37649c07c7d37e6119e8030c5dbc5602a6e7721f
|
3 |
+
size 114106856
|
lora_orpo.yaml
ADDED
@@ -0,0 +1,44 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
### model
|
2 |
+
model_name_or_path: mistralai/Mistral-Nemo-Instruct-2407
|
3 |
+
|
4 |
+
### method
|
5 |
+
stage: dpo
|
6 |
+
pref_loss: orpo
|
7 |
+
do_train: true
|
8 |
+
finetuning_type: lora
|
9 |
+
lora_target: all
|
10 |
+
|
11 |
+
### dataset
|
12 |
+
dataset: bct_non_cot_dpo_1000
|
13 |
+
dataset_dir: data_private
|
14 |
+
template: mistral
|
15 |
+
cutoff_len: 1024
|
16 |
+
# max_samples: 1000
|
17 |
+
overwrite_cache: true
|
18 |
+
preprocessing_num_workers: 16
|
19 |
+
|
20 |
+
### output
|
21 |
+
output_dir: saves/Mistral-Nemo-12B-Instruct/lora/saa
|
22 |
+
logging_steps: 10
|
23 |
+
save_steps: 500
|
24 |
+
plot_loss: true
|
25 |
+
overwrite_output_dir: true
|
26 |
+
save_total_limit: 3
|
27 |
+
load_best_model_at_end: true
|
28 |
+
push_to_hub: true
|
29 |
+
hub_model_id: chchen/Mistral-Nemo-12B-Instruct-SAA
|
30 |
+
|
31 |
+
### train
|
32 |
+
per_device_train_batch_size: 2
|
33 |
+
gradient_accumulation_steps: 8
|
34 |
+
learning_rate: 0.000005
|
35 |
+
num_train_epochs: 3.0
|
36 |
+
lr_scheduler_type: cosine
|
37 |
+
warmup_ratio: 0.1
|
38 |
+
bf16: true
|
39 |
+
|
40 |
+
### eval
|
41 |
+
val_size: 0.1
|
42 |
+
per_device_eval_batch_size: 2
|
43 |
+
evaluation_strategy: steps
|
44 |
+
eval_steps: 500
|
special_tokens_map.json
ADDED
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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": "</s>",
|
11 |
+
"lstrip": false,
|
12 |
+
"normalized": false,
|
13 |
+
"rstrip": false,
|
14 |
+
"single_word": false
|
15 |
+
},
|
16 |
+
"pad_token": "</s>",
|
17 |
+
"unk_token": {
|
18 |
+
"content": "<unk>",
|
19 |
+
"lstrip": false,
|
20 |
+
"normalized": false,
|
21 |
+
"rstrip": false,
|
22 |
+
"single_word": false
|
23 |
+
}
|
24 |
+
}
|
tokenizer.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b0240ce510f08e6c2041724e9043e33be9d251d1e4a4d94eb68cd47b954b61d2
|
3 |
+
size 17078292
|
tokenizer_config.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
trainer_log.jsonl
ADDED
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{"current_steps": 10, "total_steps": 168, "loss": 1.2935, "accuracy": 0.800000011920929, "learning_rate": 2.9411764705882355e-06, "epoch": 0.17777777777777778, "percentage": 5.95, "elapsed_time": "0:00:32", "remaining_time": "0:08:26"}
|
2 |
+
{"current_steps": 20, "total_steps": 168, "loss": 1.337, "accuracy": 0.831250011920929, "learning_rate": 4.995131923687488e-06, "epoch": 0.35555555555555557, "percentage": 11.9, "elapsed_time": "0:01:03", "remaining_time": "0:07:49"}
|
3 |
+
{"current_steps": 30, "total_steps": 168, "loss": 1.2375, "accuracy": 0.8187500238418579, "learning_rate": 4.90911473983908e-06, "epoch": 0.5333333333333333, "percentage": 17.86, "elapsed_time": "0:01:34", "remaining_time": "0:07:13"}
|
4 |
+
{"current_steps": 40, "total_steps": 168, "loss": 0.9426, "accuracy": 0.8687499761581421, "learning_rate": 4.71919261421297e-06, "epoch": 0.7111111111111111, "percentage": 23.81, "elapsed_time": "0:02:04", "remaining_time": "0:06:38"}
|
5 |
+
{"current_steps": 50, "total_steps": 168, "loss": 0.6945, "accuracy": 0.824999988079071, "learning_rate": 4.43355687413747e-06, "epoch": 0.8888888888888888, "percentage": 29.76, "elapsed_time": "0:02:35", "remaining_time": "0:06:07"}
|
6 |
+
{"current_steps": 60, "total_steps": 168, "loss": 0.6081, "accuracy": 0.7562500238418579, "learning_rate": 4.064526968101844e-06, "epoch": 1.0666666666666667, "percentage": 35.71, "elapsed_time": "0:03:06", "remaining_time": "0:05:36"}
|
7 |
+
{"current_steps": 70, "total_steps": 168, "loss": 0.3968, "accuracy": 0.824999988079071, "learning_rate": 3.6280191288478437e-06, "epoch": 1.2444444444444445, "percentage": 41.67, "elapsed_time": "0:03:36", "remaining_time": "0:05:03"}
|
8 |
+
{"current_steps": 80, "total_steps": 168, "loss": 0.2882, "accuracy": 0.831250011920929, "learning_rate": 3.142859907420615e-06, "epoch": 1.4222222222222223, "percentage": 47.62, "elapsed_time": "0:04:07", "remaining_time": "0:04:32"}
|
9 |
+
{"current_steps": 90, "total_steps": 168, "loss": 0.2406, "accuracy": 0.793749988079071, "learning_rate": 2.629974185404951e-06, "epoch": 1.6, "percentage": 53.57, "elapsed_time": "0:04:38", "remaining_time": "0:04:01"}
|
10 |
+
{"current_steps": 100, "total_steps": 168, "loss": 0.1759, "accuracy": 0.862500011920929, "learning_rate": 2.1114826863194882e-06, "epoch": 1.7777777777777777, "percentage": 59.52, "elapsed_time": "0:05:10", "remaining_time": "0:03:30"}
|
11 |
+
{"current_steps": 110, "total_steps": 168, "loss": 0.1836, "accuracy": 0.793749988079071, "learning_rate": 1.6097479104361328e-06, "epoch": 1.9555555555555557, "percentage": 65.48, "elapsed_time": "0:05:40", "remaining_time": "0:02:59"}
|
12 |
+
{"current_steps": 120, "total_steps": 168, "loss": 0.1326, "accuracy": 0.887499988079071, "learning_rate": 1.1464096417858821e-06, "epoch": 2.1333333333333333, "percentage": 71.43, "elapsed_time": "0:06:10", "remaining_time": "0:02:28"}
|
13 |
+
{"current_steps": 130, "total_steps": 168, "loss": 0.1618, "accuracy": 0.8062499761581421, "learning_rate": 7.414516258630245e-07, "epoch": 2.311111111111111, "percentage": 77.38, "elapsed_time": "0:06:40", "remaining_time": "0:01:57"}
|
14 |
+
{"current_steps": 140, "total_steps": 168, "loss": 0.1688, "accuracy": 0.831250011920929, "learning_rate": 4.123396721497977e-07, "epoch": 2.488888888888889, "percentage": 83.33, "elapsed_time": "0:07:10", "remaining_time": "0:01:26"}
|
15 |
+
{"current_steps": 150, "total_steps": 168, "loss": 0.1648, "accuracy": 0.8062499761581421, "learning_rate": 1.7326835503629542e-07, "epoch": 2.6666666666666665, "percentage": 89.29, "elapsed_time": "0:07:41", "remaining_time": "0:00:55"}
|
16 |
+
{"current_steps": 160, "total_steps": 168, "loss": 0.1456, "accuracy": 0.824999988079071, "learning_rate": 3.4548802869627806e-08, "epoch": 2.8444444444444446, "percentage": 95.24, "elapsed_time": "0:08:12", "remaining_time": "0:00:24"}
|
17 |
+
{"current_steps": 168, "total_steps": 168, "epoch": 2.986666666666667, "percentage": 100.0, "elapsed_time": "0:08:39", "remaining_time": "0:00:00"}
|
training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d88fd5474fe75081960e3c81c6cba10621a8aea161cf54aac148830c61546f62
|
3 |
+
size 5432
|