jk200201 commited on
Commit
9240ae4
·
verified ·
1 Parent(s): c606723

Upload folder using huggingface_hub

Browse files
.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
README.md CHANGED
@@ -1,3 +1,61 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ license: other
4
+ base_model: Qwen/Qwen2.5-Coder-7B-Instruct
5
+ tags:
6
+ - base_model:adapter:Qwen/Qwen2.5-Coder-7B-Instruct
7
+ - llama-factory
8
+ - lora
9
+ - transformers
10
+ pipeline_tag: text-generation
11
+ model-index:
12
+ - name: dpo
13
+ results: []
14
+ ---
15
+
16
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
+ should probably proofread and complete it, then remove this comment. -->
18
+
19
+ # dpo
20
+
21
+ This model is a fine-tuned version of [Qwen/Qwen2.5-Coder-7B-Instruct](https://huggingface.co/Qwen/Qwen2.5-Coder-7B-Instruct) on the spider_dpo dataset.
22
+
23
+ ## Model description
24
+
25
+ More information needed
26
+
27
+ ## Intended uses & limitations
28
+
29
+ More information needed
30
+
31
+ ## Training and evaluation data
32
+
33
+ More information needed
34
+
35
+ ## Training procedure
36
+
37
+ ### Training hyperparameters
38
+
39
+ The following hyperparameters were used during training:
40
+ - learning_rate: 5e-05
41
+ - train_batch_size: 2
42
+ - eval_batch_size: 8
43
+ - seed: 42
44
+ - gradient_accumulation_steps: 8
45
+ - total_train_batch_size: 16
46
+ - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
47
+ - lr_scheduler_type: cosine
48
+ - lr_scheduler_warmup_steps: 0.1
49
+ - num_epochs: 2
50
+
51
+ ### Training results
52
+
53
+
54
+
55
+ ### Framework versions
56
+
57
+ - PEFT 0.18.1
58
+ - Transformers 5.2.0
59
+ - Pytorch 2.10.0+cu128
60
+ - Datasets 4.0.0
61
+ - Tokenizers 0.22.2
adapter_config.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "Qwen/Qwen2.5-Coder-7B-Instruct",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": false,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "layer_replication": null,
16
+ "layers_pattern": null,
17
+ "layers_to_transform": null,
18
+ "loftq_config": {},
19
+ "lora_alpha": 64,
20
+ "lora_bias": false,
21
+ "lora_dropout": 0.0,
22
+ "megatron_config": null,
23
+ "megatron_core": "megatron.core",
24
+ "modules_to_save": null,
25
+ "peft_type": "LORA",
26
+ "peft_version": "0.18.1",
27
+ "qalora_group_size": 16,
28
+ "r": 32,
29
+ "rank_pattern": {},
30
+ "revision": null,
31
+ "target_modules": [
32
+ "k_proj",
33
+ "o_proj",
34
+ "v_proj",
35
+ "gate_proj",
36
+ "down_proj",
37
+ "q_proj",
38
+ "up_proj"
39
+ ],
40
+ "target_parameters": null,
41
+ "task_type": "CAUSAL_LM",
42
+ "trainable_token_indices": null,
43
+ "use_dora": false,
44
+ "use_qalora": false,
45
+ "use_rslora": false
46
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:34951a2d4294a30f58b9f230364501ffaae66f8213e858bbfb46d59a9d07d3b2
3
+ size 323014168
all_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 2.0,
3
+ "total_flos": 1.008142593959854e+17,
4
+ "train_loss": 0.22079544021533085,
5
+ "train_runtime": 3380.4206,
6
+ "train_samples_per_second": 0.615,
7
+ "train_steps_per_second": 0.038
8
+ }
chat_template.jinja ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- if tools %}
2
+ {{- '<|im_start|>system\n' }}
3
+ {%- if messages[0]['role'] == 'system' %}
4
+ {{- messages[0]['content'] }}
5
+ {%- else %}
6
+ {{- 'You are Qwen, created by Alibaba Cloud. You are a helpful assistant.' }}
7
+ {%- endif %}
8
+ {{- "\n\n# Tools\n\nYou may call one or more functions to assist with the user query.\n\nYou are provided with function signatures within <tools></tools> XML tags:\n<tools>" }}
9
+ {%- for tool in tools %}
10
+ {{- "\n" }}
11
+ {{- tool | tojson }}
12
+ {%- endfor %}
13
+ {{- "\n</tools>\n\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\n<tool_call>\n{\"name\": <function-name>, \"arguments\": <args-json-object>}\n</tool_call><|im_end|>\n" }}
14
+ {%- else %}
15
+ {%- if messages[0]['role'] == 'system' %}
16
+ {{- '<|im_start|>system\n' + messages[0]['content'] + '<|im_end|>\n' }}
17
+ {%- else %}
18
+ {{- '<|im_start|>system\nYou are Qwen, created by Alibaba Cloud. You are a helpful assistant.<|im_end|>\n' }}
19
+ {%- endif %}
20
+ {%- endif %}
21
+ {%- for message in messages %}
22
+ {%- if (message.role == "user") or (message.role == "system" and not loop.first) or (message.role == "assistant" and not message.tool_calls) %}
23
+ {{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }}
24
+ {%- elif message.role == "assistant" %}
25
+ {{- '<|im_start|>' + message.role }}
26
+ {%- if message.content %}
27
+ {{- '\n' + message.content }}
28
+ {%- endif %}
29
+ {%- for tool_call in message.tool_calls %}
30
+ {%- if tool_call.function is defined %}
31
+ {%- set tool_call = tool_call.function %}
32
+ {%- endif %}
33
+ {{- '\n<tool_call>\n{"name": "' }}
34
+ {{- tool_call.name }}
35
+ {{- '", "arguments": ' }}
36
+ {{- tool_call.arguments | tojson }}
37
+ {{- '}\n</tool_call>' }}
38
+ {%- endfor %}
39
+ {{- '<|im_end|>\n' }}
40
+ {%- elif message.role == "tool" %}
41
+ {%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != "tool") %}
42
+ {{- '<|im_start|>user' }}
43
+ {%- endif %}
44
+ {{- '\n<tool_response>\n' }}
45
+ {{- message.content }}
46
+ {{- '\n</tool_response>' }}
47
+ {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
48
+ {{- '<|im_end|>\n' }}
49
+ {%- endif %}
50
+ {%- endif %}
51
+ {%- endfor %}
52
+ {%- if add_generation_prompt %}
53
+ {{- '<|im_start|>assistant\n' }}
54
+ {%- endif %}
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3fd169731d2cbde95e10bf356d66d5997fd885dd8dbb6fb4684da3f23b2585d8
3
+ size 11421892
tokenizer_config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": null,
5
+ "clean_up_tokenization_spaces": false,
6
+ "eos_token": "<|im_end|>",
7
+ "errors": "replace",
8
+ "extra_special_tokens": [
9
+ "<|im_start|>",
10
+ "<|im_end|>",
11
+ "<|object_ref_start|>",
12
+ "<|object_ref_end|>",
13
+ "<|box_start|>",
14
+ "<|box_end|>",
15
+ "<|quad_start|>",
16
+ "<|quad_end|>",
17
+ "<|vision_start|>",
18
+ "<|vision_end|>",
19
+ "<|vision_pad|>",
20
+ "<|image_pad|>",
21
+ "<|video_pad|>"
22
+ ],
23
+ "is_local": false,
24
+ "model_max_length": 32768,
25
+ "pad_token": "<|endoftext|>",
26
+ "padding_side": "right",
27
+ "split_special_tokens": false,
28
+ "tokenizer_class": "Qwen2Tokenizer",
29
+ "unk_token": null
30
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 2.0,
3
+ "total_flos": 1.008142593959854e+17,
4
+ "train_loss": 0.22079544021533085,
5
+ "train_runtime": 3380.4206,
6
+ "train_samples_per_second": 0.615,
7
+ "train_steps_per_second": 0.038
8
+ }
trainer_log.jsonl ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"current_steps": 10, "total_steps": 130, "loss": 0.4812039852142334, "accuracy": 0.737500011920929, "lr": 3.461538461538462e-05, "epoch": 0.15384615384615385, "percentage": 7.69, "elapsed_time": "0:04:14", "remaining_time": "0:50:58"}
2
+ {"current_steps": 20, "total_steps": 130, "loss": 0.38157622814178466, "accuracy": 0.8062500357627869, "lr": 4.967625656594782e-05, "epoch": 0.3076923076923077, "percentage": 15.38, "elapsed_time": "0:08:44", "remaining_time": "0:48:04"}
3
+ {"current_steps": 30, "total_steps": 130, "loss": 0.31204392910003664, "accuracy": 0.824999988079071, "lr": 4.77281074950681e-05, "epoch": 0.46153846153846156, "percentage": 23.08, "elapsed_time": "0:13:04", "remaining_time": "0:43:36"}
4
+ {"current_steps": 40, "total_steps": 130, "loss": 0.26097993850708007, "accuracy": 0.8687500357627869, "lr": 4.415111107797445e-05, "epoch": 0.6153846153846154, "percentage": 30.77, "elapsed_time": "0:17:23", "remaining_time": "0:39:08"}
5
+ {"current_steps": 50, "total_steps": 130, "loss": 0.2709719896316528, "accuracy": 0.862500011920929, "lr": 3.920161866827889e-05, "epoch": 0.7692307692307693, "percentage": 38.46, "elapsed_time": "0:21:54", "remaining_time": "0:35:03"}
6
+ {"current_steps": 60, "total_steps": 130, "loss": 0.4293703079223633, "accuracy": 0.8125, "lr": 3.323434381080199e-05, "epoch": 0.9230769230769231, "percentage": 46.15, "elapsed_time": "0:25:56", "remaining_time": "0:30:15"}
7
+ {"current_steps": 70, "total_steps": 130, "loss": 0.22459855079650878, "accuracy": 0.893750011920929, "lr": 2.667694110977506e-05, "epoch": 1.0769230769230769, "percentage": 53.85, "elapsed_time": "0:30:16", "remaining_time": "0:25:56"}
8
+ {"current_steps": 80, "total_steps": 130, "loss": 0.10246082544326782, "accuracy": 0.981249988079071, "lr": 1.9999357655598893e-05, "epoch": 1.2307692307692308, "percentage": 61.54, "elapsed_time": "0:34:24", "remaining_time": "0:21:30"}
9
+ {"current_steps": 90, "total_steps": 130, "loss": 0.07779874801635742, "accuracy": 0.981249988079071, "lr": 1.3680153490759073e-05, "epoch": 1.3846153846153846, "percentage": 69.23, "elapsed_time": "0:38:56", "remaining_time": "0:17:18"}
10
+ {"current_steps": 100, "total_steps": 130, "loss": 0.09446020126342773, "accuracy": 0.9750000238418579, "lr": 8.172204814310742e-06, "epoch": 1.5384615384615383, "percentage": 76.92, "elapsed_time": "0:43:13", "remaining_time": "0:12:58"}
11
+ {"current_steps": 110, "total_steps": 130, "loss": 0.08398746848106384, "accuracy": 0.987500011920929, "lr": 3.8702478614051355e-06, "epoch": 1.6923076923076923, "percentage": 84.62, "elapsed_time": "0:47:27", "remaining_time": "0:08:37"}
12
+ {"current_steps": 120, "total_steps": 130, "loss": 0.06980652809143066, "accuracy": 1.0, "lr": 1.0825894793143721e-06, "epoch": 1.8461538461538463, "percentage": 92.31, "elapsed_time": "0:51:48", "remaining_time": "0:04:19"}
13
+ {"current_steps": 130, "total_steps": 130, "loss": 0.081082022190094, "accuracy": 0.96875, "lr": 9.011808227865625e-09, "epoch": 2.0, "percentage": 100.0, "elapsed_time": "0:56:19", "remaining_time": "0:00:00"}
14
+ {"current_steps": 130, "total_steps": 130, "epoch": 2.0, "percentage": 100.0, "elapsed_time": "0:56:20", "remaining_time": "0:00:00"}
trainer_state.json ADDED
@@ -0,0 +1,238 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 2.0,
6
+ "eval_steps": 500,
7
+ "global_step": 130,
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.15384615384615385,
14
+ "grad_norm": 0.29832401871681213,
15
+ "learning_rate": 3.461538461538462e-05,
16
+ "logits/chosen": -1.1990242004394531,
17
+ "logits/rejected": -1.1565947532653809,
18
+ "logps/chosen": -34.47224044799805,
19
+ "logps/rejected": -57.595638275146484,
20
+ "loss": 0.4812039852142334,
21
+ "rewards/accuracies": 0.737500011920929,
22
+ "rewards/chosen": -1.6355812549591064,
23
+ "rewards/margins": 1.8687772750854492,
24
+ "rewards/rejected": -3.5043587684631348,
25
+ "step": 10
26
+ },
27
+ {
28
+ "epoch": 0.3076923076923077,
29
+ "grad_norm": 0.31452351808547974,
30
+ "learning_rate": 4.967625656594782e-05,
31
+ "logits/chosen": -1.1797922849655151,
32
+ "logits/rejected": -1.1334644556045532,
33
+ "logps/chosen": -30.18818473815918,
34
+ "logps/rejected": -54.1025505065918,
35
+ "loss": 0.38157622814178466,
36
+ "rewards/accuracies": 0.8062500357627869,
37
+ "rewards/chosen": -1.2342875003814697,
38
+ "rewards/margins": 1.9759091138839722,
39
+ "rewards/rejected": -3.2101962566375732,
40
+ "step": 20
41
+ },
42
+ {
43
+ "epoch": 0.46153846153846156,
44
+ "grad_norm": 0.2767532467842102,
45
+ "learning_rate": 4.77281074950681e-05,
46
+ "logits/chosen": -1.218713641166687,
47
+ "logits/rejected": -1.1803112030029297,
48
+ "logps/chosen": -31.594274520874023,
49
+ "logps/rejected": -58.6033935546875,
50
+ "loss": 0.31204392910003664,
51
+ "rewards/accuracies": 0.824999988079071,
52
+ "rewards/chosen": -1.2332490682601929,
53
+ "rewards/margins": 2.2641992568969727,
54
+ "rewards/rejected": -3.497448444366455,
55
+ "step": 30
56
+ },
57
+ {
58
+ "epoch": 0.6153846153846154,
59
+ "grad_norm": 0.3287574052810669,
60
+ "learning_rate": 4.415111107797445e-05,
61
+ "logits/chosen": -1.3382649421691895,
62
+ "logits/rejected": -1.30353581905365,
63
+ "logps/chosen": -36.43912887573242,
64
+ "logps/rejected": -70.29068756103516,
65
+ "loss": 0.26097993850708007,
66
+ "rewards/accuracies": 0.8687500357627869,
67
+ "rewards/chosen": -1.6973193883895874,
68
+ "rewards/margins": 3.017174005508423,
69
+ "rewards/rejected": -4.714493751525879,
70
+ "step": 40
71
+ },
72
+ {
73
+ "epoch": 0.7692307692307693,
74
+ "grad_norm": 0.3445669710636139,
75
+ "learning_rate": 3.920161866827889e-05,
76
+ "logits/chosen": -1.4549230337142944,
77
+ "logits/rejected": -1.4277979135513306,
78
+ "logps/chosen": -41.87653732299805,
79
+ "logps/rejected": -82.90950012207031,
80
+ "loss": 0.2709719896316528,
81
+ "rewards/accuracies": 0.862500011920929,
82
+ "rewards/chosen": -2.2981889247894287,
83
+ "rewards/margins": 3.5901012420654297,
84
+ "rewards/rejected": -5.8882904052734375,
85
+ "step": 50
86
+ },
87
+ {
88
+ "epoch": 0.9230769230769231,
89
+ "grad_norm": 0.4734262526035309,
90
+ "learning_rate": 3.323434381080199e-05,
91
+ "logits/chosen": -1.4583113193511963,
92
+ "logits/rejected": -1.4419019222259521,
93
+ "logps/chosen": -46.35929489135742,
94
+ "logps/rejected": -82.18633270263672,
95
+ "loss": 0.4293703079223633,
96
+ "rewards/accuracies": 0.8125,
97
+ "rewards/chosen": -2.7741143703460693,
98
+ "rewards/margins": 3.1638267040252686,
99
+ "rewards/rejected": -5.937941074371338,
100
+ "step": 60
101
+ },
102
+ {
103
+ "epoch": 1.0769230769230769,
104
+ "grad_norm": 0.14828024804592133,
105
+ "learning_rate": 2.667694110977506e-05,
106
+ "logits/chosen": -1.3716756105422974,
107
+ "logits/rejected": -1.3496290445327759,
108
+ "logps/chosen": -37.190982818603516,
109
+ "logps/rejected": -80.84061431884766,
110
+ "loss": 0.22459855079650878,
111
+ "rewards/accuracies": 0.893750011920929,
112
+ "rewards/chosen": -1.9332157373428345,
113
+ "rewards/margins": 3.8292477130889893,
114
+ "rewards/rejected": -5.762463569641113,
115
+ "step": 70
116
+ },
117
+ {
118
+ "epoch": 1.2307692307692308,
119
+ "grad_norm": 0.1763170063495636,
120
+ "learning_rate": 1.9999357655598893e-05,
121
+ "logits/chosen": -1.355380892753601,
122
+ "logits/rejected": -1.3304802179336548,
123
+ "logps/chosen": -39.43621826171875,
124
+ "logps/rejected": -82.64469909667969,
125
+ "loss": 0.10246082544326782,
126
+ "rewards/accuracies": 0.981249988079071,
127
+ "rewards/chosen": -2.0429646968841553,
128
+ "rewards/margins": 3.7551822662353516,
129
+ "rewards/rejected": -5.798146724700928,
130
+ "step": 80
131
+ },
132
+ {
133
+ "epoch": 1.3846153846153846,
134
+ "grad_norm": 0.08582232892513275,
135
+ "learning_rate": 1.3680153490759073e-05,
136
+ "logits/chosen": -1.3984593152999878,
137
+ "logits/rejected": -1.3604938983917236,
138
+ "logps/chosen": -35.92609405517578,
139
+ "logps/rejected": -94.2962875366211,
140
+ "loss": 0.07779874801635742,
141
+ "rewards/accuracies": 0.981249988079071,
142
+ "rewards/chosen": -1.6332558393478394,
143
+ "rewards/margins": 5.143055438995361,
144
+ "rewards/rejected": -6.776310920715332,
145
+ "step": 90
146
+ },
147
+ {
148
+ "epoch": 1.5384615384615383,
149
+ "grad_norm": 0.11894813179969788,
150
+ "learning_rate": 8.172204814310742e-06,
151
+ "logits/chosen": -1.4377853870391846,
152
+ "logits/rejected": -1.4119752645492554,
153
+ "logps/chosen": -33.87895965576172,
154
+ "logps/rejected": -80.11370086669922,
155
+ "loss": 0.09446020126342773,
156
+ "rewards/accuracies": 0.9750000238418579,
157
+ "rewards/chosen": -1.5904297828674316,
158
+ "rewards/margins": 4.260800838470459,
159
+ "rewards/rejected": -5.851230144500732,
160
+ "step": 100
161
+ },
162
+ {
163
+ "epoch": 1.6923076923076923,
164
+ "grad_norm": 0.0858096107840538,
165
+ "learning_rate": 3.8702478614051355e-06,
166
+ "logits/chosen": -1.4167169332504272,
167
+ "logits/rejected": -1.3845655918121338,
168
+ "logps/chosen": -38.7296257019043,
169
+ "logps/rejected": -86.01542663574219,
170
+ "loss": 0.08398746848106384,
171
+ "rewards/accuracies": 0.987500011920929,
172
+ "rewards/chosen": -2.0574593544006348,
173
+ "rewards/margins": 4.346583843231201,
174
+ "rewards/rejected": -6.404043674468994,
175
+ "step": 110
176
+ },
177
+ {
178
+ "epoch": 1.8461538461538463,
179
+ "grad_norm": 0.08499918133020401,
180
+ "learning_rate": 1.0825894793143721e-06,
181
+ "logits/chosen": -1.4523096084594727,
182
+ "logits/rejected": -1.4183976650238037,
183
+ "logps/chosen": -36.14750289916992,
184
+ "logps/rejected": -87.8320083618164,
185
+ "loss": 0.06980652809143066,
186
+ "rewards/accuracies": 1.0,
187
+ "rewards/chosen": -1.8463581800460815,
188
+ "rewards/margins": 4.731562614440918,
189
+ "rewards/rejected": -6.577920436859131,
190
+ "step": 120
191
+ },
192
+ {
193
+ "epoch": 2.0,
194
+ "grad_norm": 0.13648074865341187,
195
+ "learning_rate": 9.011808227865625e-09,
196
+ "logits/chosen": -1.44231116771698,
197
+ "logits/rejected": -1.4165829420089722,
198
+ "logps/chosen": -36.7476921081543,
199
+ "logps/rejected": -86.03533172607422,
200
+ "loss": 0.081082022190094,
201
+ "rewards/accuracies": 0.96875,
202
+ "rewards/chosen": -1.7559616565704346,
203
+ "rewards/margins": 4.578075885772705,
204
+ "rewards/rejected": -6.334037780761719,
205
+ "step": 130
206
+ },
207
+ {
208
+ "epoch": 2.0,
209
+ "step": 130,
210
+ "total_flos": 1.008142593959854e+17,
211
+ "train_loss": 0.22079544021533085,
212
+ "train_runtime": 3380.4206,
213
+ "train_samples_per_second": 0.615,
214
+ "train_steps_per_second": 0.038
215
+ }
216
+ ],
217
+ "logging_steps": 10,
218
+ "max_steps": 130,
219
+ "num_input_tokens_seen": 0,
220
+ "num_train_epochs": 2,
221
+ "save_steps": 100,
222
+ "stateful_callbacks": {
223
+ "TrainerControl": {
224
+ "args": {
225
+ "should_epoch_stop": false,
226
+ "should_evaluate": false,
227
+ "should_log": false,
228
+ "should_save": true,
229
+ "should_training_stop": true
230
+ },
231
+ "attributes": {}
232
+ }
233
+ },
234
+ "total_flos": 1.008142593959854e+17,
235
+ "train_batch_size": 2,
236
+ "trial_name": null,
237
+ "trial_params": null
238
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f0ab2bab02f3075b1bd6b20220e3f4d76681b555bce8eb919646fd52350a6613
3
+ size 5585