weijie210 commited on
Commit
ae6db7b
1 Parent(s): fad0135

Model save

Browse files
README.md ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: alignment-handbook/zephyr-7b-sft-full
4
+ tags:
5
+ - trl
6
+ - dpo
7
+ - generated_from_trainer
8
+ model-index:
9
+ - name: zephyr-7b-dpo-full
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ # zephyr-7b-dpo-full
17
+
18
+ This model is a fine-tuned version of [alignment-handbook/zephyr-7b-sft-full](https://huggingface.co/alignment-handbook/zephyr-7b-sft-full) on the None dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 0.5714
21
+ - Rewards/chosen: -0.4765
22
+ - Rewards/rejected: -1.7868
23
+ - Rewards/accuracies: 0.7619
24
+ - Rewards/margins: 1.3103
25
+ - Logps/rejected: -294.2741
26
+ - Logps/chosen: -305.4321
27
+ - Logits/rejected: -2.4275
28
+ - Logits/chosen: -2.4841
29
+
30
+ ## Model description
31
+
32
+ More information needed
33
+
34
+ ## Intended uses & limitations
35
+
36
+ More information needed
37
+
38
+ ## Training and evaluation data
39
+
40
+ More information needed
41
+
42
+ ## Training procedure
43
+
44
+ ### Training hyperparameters
45
+
46
+ The following hyperparameters were used during training:
47
+ - learning_rate: 5e-07
48
+ - train_batch_size: 8
49
+ - eval_batch_size: 8
50
+ - seed: 42
51
+ - distributed_type: multi-GPU
52
+ - num_devices: 4
53
+ - total_train_batch_size: 32
54
+ - total_eval_batch_size: 32
55
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
56
+ - lr_scheduler_type: linear
57
+ - lr_scheduler_warmup_ratio: 0.1
58
+ - num_epochs: 1
59
+
60
+ ### Training results
61
+
62
+ | Training Loss | Epoch | Step | Validation Loss | Rewards/chosen | Rewards/rejected | Rewards/accuracies | Rewards/margins | Logps/rejected | Logps/chosen | Logits/rejected | Logits/chosen |
63
+ |:-------------:|:-----:|:----:|:---------------:|:--------------:|:----------------:|:------------------:|:---------------:|:--------------:|:------------:|:---------------:|:-------------:|
64
+ | 0.5509 | 0.26 | 500 | 0.6311 | 0.8114 | -0.1181 | 0.7619 | 0.9294 | -277.5868 | -292.5533 | -2.4249 | -2.4869 |
65
+ | 0.6191 | 0.52 | 1000 | 0.5984 | 0.0381 | -1.0554 | 0.7599 | 1.0935 | -286.9601 | -300.2859 | -2.4280 | -2.4907 |
66
+ | 0.5342 | 0.78 | 1500 | 0.5819 | 0.0994 | -1.1560 | 0.7560 | 1.2554 | -287.9663 | -299.6732 | -2.4208 | -2.4793 |
67
+
68
+
69
+ ### Framework versions
70
+
71
+ - Transformers 4.36.1
72
+ - Pytorch 2.0.1+cu117
73
+ - Datasets 2.16.1
74
+ - Tokenizers 0.15.0
all_results.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "eval_logits/chosen": -2.484065532684326,
4
+ "eval_logits/rejected": -2.4274542331695557,
5
+ "eval_logps/chosen": -305.43212890625,
6
+ "eval_logps/rejected": -294.2740783691406,
7
+ "eval_loss": 0.5714402198791504,
8
+ "eval_rewards/accuracies": 0.761904776096344,
9
+ "eval_rewards/chosen": -0.4765053391456604,
10
+ "eval_rewards/margins": 1.3102855682373047,
11
+ "eval_rewards/rejected": -1.7867907285690308,
12
+ "eval_runtime": 613.7286,
13
+ "eval_samples": 2000,
14
+ "eval_samples_per_second": 3.259,
15
+ "eval_steps_per_second": 0.103,
16
+ "train_loss": 0.5469513680846325,
17
+ "train_runtime": 31987.9403,
18
+ "train_samples": 61135,
19
+ "train_samples_per_second": 1.911,
20
+ "train_steps_per_second": 0.06
21
+ }
config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "alignment-handbook/zephyr-7b-sft-full",
3
+ "architectures": [
4
+ "MistralForCausalLM"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 1,
8
+ "eos_token_id": 2,
9
+ "hidden_act": "silu",
10
+ "hidden_size": 4096,
11
+ "initializer_range": 0.02,
12
+ "intermediate_size": 14336,
13
+ "max_position_embeddings": 32768,
14
+ "model_type": "mistral",
15
+ "num_attention_heads": 32,
16
+ "num_hidden_layers": 32,
17
+ "num_key_value_heads": 8,
18
+ "rms_norm_eps": 1e-05,
19
+ "rope_theta": 10000.0,
20
+ "sliding_window": 4096,
21
+ "tie_word_embeddings": false,
22
+ "torch_dtype": "bfloat16",
23
+ "transformers_version": "4.36.1",
24
+ "use_cache": false,
25
+ "vocab_size": 32000
26
+ }
eval_results.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "eval_logits/chosen": -2.484065532684326,
4
+ "eval_logits/rejected": -2.4274542331695557,
5
+ "eval_logps/chosen": -305.43212890625,
6
+ "eval_logps/rejected": -294.2740783691406,
7
+ "eval_loss": 0.5714402198791504,
8
+ "eval_rewards/accuracies": 0.761904776096344,
9
+ "eval_rewards/chosen": -0.4765053391456604,
10
+ "eval_rewards/margins": 1.3102855682373047,
11
+ "eval_rewards/rejected": -1.7867907285690308,
12
+ "eval_runtime": 613.7286,
13
+ "eval_samples": 2000,
14
+ "eval_samples_per_second": 3.259,
15
+ "eval_steps_per_second": 0.103
16
+ }
generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 1,
4
+ "eos_token_id": 2,
5
+ "transformers_version": "4.36.1"
6
+ }
model-00001-of-00003.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bed0946f3404fda76ec1b3c42687712a74cfa981f64b593183f5a0b1469da37c
3
+ size 4943162336
model-00002-of-00003.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b1f253b60e78de1283921bb8677fd5370e9e007b56e60a1ea2d4bc88b7ab1f01
3
+ size 4999819336
model-00003-of-00003.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b5e95d7209d2b8548bdf8edd75b7e9234913d01ee17c8f48d0a377646c241cac
3
+ size 4540516344
model.safetensors.index.json ADDED
@@ -0,0 +1,298 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 14483464192
4
+ },
5
+ "weight_map": {
6
+ "lm_head.weight": "model-00003-of-00003.safetensors",
7
+ "model.embed_tokens.weight": "model-00001-of-00003.safetensors",
8
+ "model.layers.0.input_layernorm.weight": "model-00001-of-00003.safetensors",
9
+ "model.layers.0.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
10
+ "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
11
+ "model.layers.0.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
12
+ "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
13
+ "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
14
+ "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
15
+ "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
16
+ "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
17
+ "model.layers.1.input_layernorm.weight": "model-00001-of-00003.safetensors",
18
+ "model.layers.1.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
19
+ "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
20
+ "model.layers.1.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
21
+ "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
22
+ "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
23
+ "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
24
+ "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
25
+ "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
26
+ "model.layers.10.input_layernorm.weight": "model-00002-of-00003.safetensors",
27
+ "model.layers.10.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
28
+ "model.layers.10.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
29
+ "model.layers.10.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
30
+ "model.layers.10.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
31
+ "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
32
+ "model.layers.10.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
33
+ "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
34
+ "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
35
+ "model.layers.11.input_layernorm.weight": "model-00002-of-00003.safetensors",
36
+ "model.layers.11.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
37
+ "model.layers.11.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
38
+ "model.layers.11.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
39
+ "model.layers.11.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
40
+ "model.layers.11.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
41
+ "model.layers.11.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
42
+ "model.layers.11.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
43
+ "model.layers.11.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
44
+ "model.layers.12.input_layernorm.weight": "model-00002-of-00003.safetensors",
45
+ "model.layers.12.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
46
+ "model.layers.12.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
47
+ "model.layers.12.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
48
+ "model.layers.12.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
49
+ "model.layers.12.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
50
+ "model.layers.12.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
51
+ "model.layers.12.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
52
+ "model.layers.12.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
53
+ "model.layers.13.input_layernorm.weight": "model-00002-of-00003.safetensors",
54
+ "model.layers.13.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
55
+ "model.layers.13.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
56
+ "model.layers.13.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
57
+ "model.layers.13.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
58
+ "model.layers.13.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
59
+ "model.layers.13.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
60
+ "model.layers.13.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
61
+ "model.layers.13.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
62
+ "model.layers.14.input_layernorm.weight": "model-00002-of-00003.safetensors",
63
+ "model.layers.14.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
64
+ "model.layers.14.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
65
+ "model.layers.14.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
66
+ "model.layers.14.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
67
+ "model.layers.14.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
68
+ "model.layers.14.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
69
+ "model.layers.14.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
70
+ "model.layers.14.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
71
+ "model.layers.15.input_layernorm.weight": "model-00002-of-00003.safetensors",
72
+ "model.layers.15.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
73
+ "model.layers.15.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
74
+ "model.layers.15.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
75
+ "model.layers.15.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
76
+ "model.layers.15.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
77
+ "model.layers.15.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
78
+ "model.layers.15.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
79
+ "model.layers.15.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
80
+ "model.layers.16.input_layernorm.weight": "model-00002-of-00003.safetensors",
81
+ "model.layers.16.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
82
+ "model.layers.16.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
83
+ "model.layers.16.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
84
+ "model.layers.16.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
85
+ "model.layers.16.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
86
+ "model.layers.16.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
87
+ "model.layers.16.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
88
+ "model.layers.16.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
89
+ "model.layers.17.input_layernorm.weight": "model-00002-of-00003.safetensors",
90
+ "model.layers.17.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
91
+ "model.layers.17.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
92
+ "model.layers.17.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
93
+ "model.layers.17.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
94
+ "model.layers.17.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
95
+ "model.layers.17.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
96
+ "model.layers.17.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
97
+ "model.layers.17.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
98
+ "model.layers.18.input_layernorm.weight": "model-00002-of-00003.safetensors",
99
+ "model.layers.18.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
100
+ "model.layers.18.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
101
+ "model.layers.18.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
102
+ "model.layers.18.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
103
+ "model.layers.18.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
104
+ "model.layers.18.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
105
+ "model.layers.18.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
106
+ "model.layers.18.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
107
+ "model.layers.19.input_layernorm.weight": "model-00002-of-00003.safetensors",
108
+ "model.layers.19.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
109
+ "model.layers.19.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
110
+ "model.layers.19.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
111
+ "model.layers.19.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
112
+ "model.layers.19.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
113
+ "model.layers.19.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
114
+ "model.layers.19.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
115
+ "model.layers.19.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
116
+ "model.layers.2.input_layernorm.weight": "model-00001-of-00003.safetensors",
117
+ "model.layers.2.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
118
+ "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
119
+ "model.layers.2.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
120
+ "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
121
+ "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
122
+ "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
123
+ "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
124
+ "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
125
+ "model.layers.20.input_layernorm.weight": "model-00002-of-00003.safetensors",
126
+ "model.layers.20.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
127
+ "model.layers.20.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
128
+ "model.layers.20.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
129
+ "model.layers.20.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
130
+ "model.layers.20.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
131
+ "model.layers.20.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
132
+ "model.layers.20.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
133
+ "model.layers.20.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
134
+ "model.layers.21.input_layernorm.weight": "model-00002-of-00003.safetensors",
135
+ "model.layers.21.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
136
+ "model.layers.21.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
137
+ "model.layers.21.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
138
+ "model.layers.21.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
139
+ "model.layers.21.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
140
+ "model.layers.21.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
141
+ "model.layers.21.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
142
+ "model.layers.21.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
143
+ "model.layers.22.input_layernorm.weight": "model-00003-of-00003.safetensors",
144
+ "model.layers.22.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
145
+ "model.layers.22.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
146
+ "model.layers.22.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
147
+ "model.layers.22.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
148
+ "model.layers.22.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
149
+ "model.layers.22.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
150
+ "model.layers.22.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
151
+ "model.layers.22.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
152
+ "model.layers.23.input_layernorm.weight": "model-00003-of-00003.safetensors",
153
+ "model.layers.23.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
154
+ "model.layers.23.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
155
+ "model.layers.23.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
156
+ "model.layers.23.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
157
+ "model.layers.23.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
158
+ "model.layers.23.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
159
+ "model.layers.23.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
160
+ "model.layers.23.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
161
+ "model.layers.24.input_layernorm.weight": "model-00003-of-00003.safetensors",
162
+ "model.layers.24.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
163
+ "model.layers.24.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
164
+ "model.layers.24.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
165
+ "model.layers.24.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
166
+ "model.layers.24.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
167
+ "model.layers.24.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
168
+ "model.layers.24.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
169
+ "model.layers.24.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
170
+ "model.layers.25.input_layernorm.weight": "model-00003-of-00003.safetensors",
171
+ "model.layers.25.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
172
+ "model.layers.25.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
173
+ "model.layers.25.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
174
+ "model.layers.25.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
175
+ "model.layers.25.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
176
+ "model.layers.25.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
177
+ "model.layers.25.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
178
+ "model.layers.25.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
179
+ "model.layers.26.input_layernorm.weight": "model-00003-of-00003.safetensors",
180
+ "model.layers.26.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
181
+ "model.layers.26.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
182
+ "model.layers.26.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
183
+ "model.layers.26.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
184
+ "model.layers.26.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
185
+ "model.layers.26.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
186
+ "model.layers.26.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
187
+ "model.layers.26.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
188
+ "model.layers.27.input_layernorm.weight": "model-00003-of-00003.safetensors",
189
+ "model.layers.27.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
190
+ "model.layers.27.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
191
+ "model.layers.27.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
192
+ "model.layers.27.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
193
+ "model.layers.27.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
194
+ "model.layers.27.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
195
+ "model.layers.27.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
196
+ "model.layers.27.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
197
+ "model.layers.28.input_layernorm.weight": "model-00003-of-00003.safetensors",
198
+ "model.layers.28.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
199
+ "model.layers.28.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
200
+ "model.layers.28.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
201
+ "model.layers.28.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
202
+ "model.layers.28.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
203
+ "model.layers.28.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
204
+ "model.layers.28.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
205
+ "model.layers.28.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
206
+ "model.layers.29.input_layernorm.weight": "model-00003-of-00003.safetensors",
207
+ "model.layers.29.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
208
+ "model.layers.29.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
209
+ "model.layers.29.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
210
+ "model.layers.29.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
211
+ "model.layers.29.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
212
+ "model.layers.29.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
213
+ "model.layers.29.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
214
+ "model.layers.29.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
215
+ "model.layers.3.input_layernorm.weight": "model-00001-of-00003.safetensors",
216
+ "model.layers.3.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
217
+ "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
218
+ "model.layers.3.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
219
+ "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
220
+ "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
221
+ "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
222
+ "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
223
+ "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
224
+ "model.layers.30.input_layernorm.weight": "model-00003-of-00003.safetensors",
225
+ "model.layers.30.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
226
+ "model.layers.30.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
227
+ "model.layers.30.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
228
+ "model.layers.30.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
229
+ "model.layers.30.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
230
+ "model.layers.30.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
231
+ "model.layers.30.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
232
+ "model.layers.30.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
233
+ "model.layers.31.input_layernorm.weight": "model-00003-of-00003.safetensors",
234
+ "model.layers.31.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
235
+ "model.layers.31.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
236
+ "model.layers.31.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
237
+ "model.layers.31.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
238
+ "model.layers.31.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
239
+ "model.layers.31.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
240
+ "model.layers.31.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
241
+ "model.layers.31.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
242
+ "model.layers.4.input_layernorm.weight": "model-00001-of-00003.safetensors",
243
+ "model.layers.4.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
244
+ "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
245
+ "model.layers.4.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
246
+ "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
247
+ "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
248
+ "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
249
+ "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
250
+ "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
251
+ "model.layers.5.input_layernorm.weight": "model-00001-of-00003.safetensors",
252
+ "model.layers.5.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
253
+ "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
254
+ "model.layers.5.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
255
+ "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
256
+ "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
257
+ "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
258
+ "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
259
+ "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
260
+ "model.layers.6.input_layernorm.weight": "model-00001-of-00003.safetensors",
261
+ "model.layers.6.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
262
+ "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
263
+ "model.layers.6.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
264
+ "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
265
+ "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
266
+ "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
267
+ "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
268
+ "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
269
+ "model.layers.7.input_layernorm.weight": "model-00001-of-00003.safetensors",
270
+ "model.layers.7.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
271
+ "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
272
+ "model.layers.7.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
273
+ "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
274
+ "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
275
+ "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
276
+ "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
277
+ "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
278
+ "model.layers.8.input_layernorm.weight": "model-00001-of-00003.safetensors",
279
+ "model.layers.8.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
280
+ "model.layers.8.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
281
+ "model.layers.8.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
282
+ "model.layers.8.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
283
+ "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
284
+ "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
285
+ "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
286
+ "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
287
+ "model.layers.9.input_layernorm.weight": "model-00001-of-00003.safetensors",
288
+ "model.layers.9.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
289
+ "model.layers.9.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
290
+ "model.layers.9.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
291
+ "model.layers.9.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
292
+ "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
293
+ "model.layers.9.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
294
+ "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
295
+ "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
296
+ "model.norm.weight": "model-00003-of-00003.safetensors"
297
+ }
298
+ }
runs/Jan21_00-47-39_node14/events.out.tfevents.1705769697.node14.3000533.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31655d75d8f112070605b567662aa7d861ce658b8fc21f256441c665b3f85f37
3
+ size 11283
runs/Jan29_16-30-36_node01/events.out.tfevents.1706517112.node01.2477677.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6fb85e0709a6893bb8aa4b3bc1c1ab6b56d9ac0042ae6886b88eafa3b717b32b
3
+ size 5677
runs/Jan29_16-36-38_node01/events.out.tfevents.1706517422.node01.2480881.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e8a5a77375b65bce27334caf381a210a85651696e9cc6651537bfce3b712e155
3
+ size 128589
runs/Jan29_16-36-38_node01/events.out.tfevents.1706550024.node01.2480881.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5a477d7cf45ad8760f92f4a34ac0c8895349180786b7ceede262f1a3257be077
3
+ size 828
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": "</s>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "</s>",
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_config.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": true,
3
+ "add_eos_token": false,
4
+ "added_tokens_decoder": {
5
+ "0": {
6
+ "content": "<unk>",
7
+ "lstrip": false,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false,
11
+ "special": true
12
+ },
13
+ "1": {
14
+ "content": "<s>",
15
+ "lstrip": false,
16
+ "normalized": false,
17
+ "rstrip": false,
18
+ "single_word": false,
19
+ "special": true
20
+ },
21
+ "2": {
22
+ "content": "</s>",
23
+ "lstrip": false,
24
+ "normalized": false,
25
+ "rstrip": false,
26
+ "single_word": false,
27
+ "special": true
28
+ }
29
+ },
30
+ "additional_special_tokens": [],
31
+ "bos_token": "<s>",
32
+ "chat_template": "{% for message in messages %}\n{% if message['role'] == 'user' %}\n{{ '<|user|>\n' + message['content'] + eos_token }}\n{% elif message['role'] == 'system' %}\n{{ '<|system|>\n' + message['content'] + eos_token }}\n{% elif message['role'] == 'assistant' %}\n{{ '<|assistant|>\n' + message['content'] + eos_token }}\n{% endif %}\n{% if loop.last and add_generation_prompt %}\n{{ '<|assistant|>' }}\n{% endif %}\n{% endfor %}",
33
+ "clean_up_tokenization_spaces": false,
34
+ "eos_token": "</s>",
35
+ "legacy": true,
36
+ "model_max_length": 4096,
37
+ "pad_token": "</s>",
38
+ "sp_model_kwargs": {},
39
+ "spaces_between_special_tokens": false,
40
+ "tokenizer_class": "LlamaTokenizer",
41
+ "unk_token": "<unk>",
42
+ "use_default_system_prompt": false
43
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "train_loss": 0.5469513680846325,
4
+ "train_runtime": 31987.9403,
5
+ "train_samples": 61135,
6
+ "train_samples_per_second": 1.911,
7
+ "train_steps_per_second": 0.06
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,2766 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 1.0,
5
+ "eval_steps": 500,
6
+ "global_step": 1911,
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.0,
13
+ "learning_rate": 2.6041666666666664e-09,
14
+ "logits/chosen": -2.463043451309204,
15
+ "logits/rejected": -2.288743019104004,
16
+ "logps/chosen": -301.1433410644531,
17
+ "logps/rejected": -128.25608825683594,
18
+ "loss": 0.6931,
19
+ "rewards/accuracies": 0.0,
20
+ "rewards/chosen": 0.0,
21
+ "rewards/margins": 0.0,
22
+ "rewards/rejected": 0.0,
23
+ "step": 1
24
+ },
25
+ {
26
+ "epoch": 0.01,
27
+ "learning_rate": 2.6041666666666667e-08,
28
+ "logits/chosen": -2.5560247898101807,
29
+ "logits/rejected": -2.555253028869629,
30
+ "logps/chosen": -286.1558837890625,
31
+ "logps/rejected": -256.28131103515625,
32
+ "loss": 0.697,
33
+ "rewards/accuracies": 0.3333333432674408,
34
+ "rewards/chosen": -0.0025797931011766195,
35
+ "rewards/margins": -0.008782301098108292,
36
+ "rewards/rejected": 0.006202507298439741,
37
+ "step": 10
38
+ },
39
+ {
40
+ "epoch": 0.01,
41
+ "learning_rate": 5.208333333333333e-08,
42
+ "logits/chosen": -2.648404598236084,
43
+ "logits/rejected": -2.6275510787963867,
44
+ "logps/chosen": -316.64373779296875,
45
+ "logps/rejected": -314.99212646484375,
46
+ "loss": 0.6895,
47
+ "rewards/accuracies": 0.4749999940395355,
48
+ "rewards/chosen": -0.0072450353763997555,
49
+ "rewards/margins": 0.0013799279695376754,
50
+ "rewards/rejected": -0.008624963462352753,
51
+ "step": 20
52
+ },
53
+ {
54
+ "epoch": 0.02,
55
+ "learning_rate": 7.812499999999999e-08,
56
+ "logits/chosen": -2.683773994445801,
57
+ "logits/rejected": -2.550048828125,
58
+ "logps/chosen": -306.83282470703125,
59
+ "logps/rejected": -266.6906433105469,
60
+ "loss": 0.6823,
61
+ "rewards/accuracies": 0.6000000238418579,
62
+ "rewards/chosen": 0.011949767358601093,
63
+ "rewards/margins": 0.020141970366239548,
64
+ "rewards/rejected": -0.008192205801606178,
65
+ "step": 30
66
+ },
67
+ {
68
+ "epoch": 0.02,
69
+ "learning_rate": 1.0416666666666667e-07,
70
+ "logits/chosen": -2.634948968887329,
71
+ "logits/rejected": -2.5988709926605225,
72
+ "logps/chosen": -265.98968505859375,
73
+ "logps/rejected": -252.77835083007812,
74
+ "loss": 0.6707,
75
+ "rewards/accuracies": 0.625,
76
+ "rewards/chosen": 0.08117702603340149,
77
+ "rewards/margins": 0.05083204060792923,
78
+ "rewards/rejected": 0.03034498728811741,
79
+ "step": 40
80
+ },
81
+ {
82
+ "epoch": 0.03,
83
+ "learning_rate": 1.3020833333333334e-07,
84
+ "logits/chosen": -2.644986152648926,
85
+ "logits/rejected": -2.589542865753174,
86
+ "logps/chosen": -323.46881103515625,
87
+ "logps/rejected": -284.9610900878906,
88
+ "loss": 0.6542,
89
+ "rewards/accuracies": 0.6625000238418579,
90
+ "rewards/chosen": 0.2217252552509308,
91
+ "rewards/margins": 0.12405480444431305,
92
+ "rewards/rejected": 0.09767045080661774,
93
+ "step": 50
94
+ },
95
+ {
96
+ "epoch": 0.03,
97
+ "learning_rate": 1.5624999999999999e-07,
98
+ "logits/chosen": -2.599961280822754,
99
+ "logits/rejected": -2.491401195526123,
100
+ "logps/chosen": -305.62152099609375,
101
+ "logps/rejected": -236.73916625976562,
102
+ "loss": 0.6107,
103
+ "rewards/accuracies": 0.6875,
104
+ "rewards/chosen": 0.29835107922554016,
105
+ "rewards/margins": 0.23799380660057068,
106
+ "rewards/rejected": 0.06035725399851799,
107
+ "step": 60
108
+ },
109
+ {
110
+ "epoch": 0.04,
111
+ "learning_rate": 1.8229166666666666e-07,
112
+ "logits/chosen": -2.5403127670288086,
113
+ "logits/rejected": -2.491334915161133,
114
+ "logps/chosen": -281.8306884765625,
115
+ "logps/rejected": -264.59130859375,
116
+ "loss": 0.6257,
117
+ "rewards/accuracies": 0.6875,
118
+ "rewards/chosen": 0.37535926699638367,
119
+ "rewards/margins": 0.173495814204216,
120
+ "rewards/rejected": 0.20186343789100647,
121
+ "step": 70
122
+ },
123
+ {
124
+ "epoch": 0.04,
125
+ "learning_rate": 2.0833333333333333e-07,
126
+ "logits/chosen": -2.6033613681793213,
127
+ "logits/rejected": -2.548762083053589,
128
+ "logps/chosen": -270.06597900390625,
129
+ "logps/rejected": -262.29693603515625,
130
+ "loss": 0.6014,
131
+ "rewards/accuracies": 0.6499999761581421,
132
+ "rewards/chosen": 0.4671238958835602,
133
+ "rewards/margins": 0.144499734044075,
134
+ "rewards/rejected": 0.3226241171360016,
135
+ "step": 80
136
+ },
137
+ {
138
+ "epoch": 0.05,
139
+ "learning_rate": 2.3437499999999998e-07,
140
+ "logits/chosen": -2.5922348499298096,
141
+ "logits/rejected": -2.584137439727783,
142
+ "logps/chosen": -278.80865478515625,
143
+ "logps/rejected": -279.4120178222656,
144
+ "loss": 0.5943,
145
+ "rewards/accuracies": 0.625,
146
+ "rewards/chosen": 0.295165479183197,
147
+ "rewards/margins": 0.24514034390449524,
148
+ "rewards/rejected": 0.05002513527870178,
149
+ "step": 90
150
+ },
151
+ {
152
+ "epoch": 0.05,
153
+ "learning_rate": 2.604166666666667e-07,
154
+ "logits/chosen": -2.5200092792510986,
155
+ "logits/rejected": -2.4520034790039062,
156
+ "logps/chosen": -240.3234405517578,
157
+ "logps/rejected": -248.67428588867188,
158
+ "loss": 0.5838,
159
+ "rewards/accuracies": 0.7250000238418579,
160
+ "rewards/chosen": 0.4579353332519531,
161
+ "rewards/margins": 0.31523841619491577,
162
+ "rewards/rejected": 0.14269694685935974,
163
+ "step": 100
164
+ },
165
+ {
166
+ "epoch": 0.06,
167
+ "learning_rate": 2.864583333333333e-07,
168
+ "logits/chosen": -2.5587222576141357,
169
+ "logits/rejected": -2.5158984661102295,
170
+ "logps/chosen": -334.97784423828125,
171
+ "logps/rejected": -290.39935302734375,
172
+ "loss": 0.5736,
173
+ "rewards/accuracies": 0.7124999761581421,
174
+ "rewards/chosen": 0.39039498567581177,
175
+ "rewards/margins": 0.49432238936424255,
176
+ "rewards/rejected": -0.10392741858959198,
177
+ "step": 110
178
+ },
179
+ {
180
+ "epoch": 0.06,
181
+ "learning_rate": 3.1249999999999997e-07,
182
+ "logits/chosen": -2.5714714527130127,
183
+ "logits/rejected": -2.503490924835205,
184
+ "logps/chosen": -287.0711364746094,
185
+ "logps/rejected": -279.7371520996094,
186
+ "loss": 0.5872,
187
+ "rewards/accuracies": 0.7250000238418579,
188
+ "rewards/chosen": 0.7893859148025513,
189
+ "rewards/margins": 0.6542637944221497,
190
+ "rewards/rejected": 0.13512210547924042,
191
+ "step": 120
192
+ },
193
+ {
194
+ "epoch": 0.07,
195
+ "learning_rate": 3.3854166666666667e-07,
196
+ "logits/chosen": -2.6581666469573975,
197
+ "logits/rejected": -2.558701753616333,
198
+ "logps/chosen": -318.14178466796875,
199
+ "logps/rejected": -277.52838134765625,
200
+ "loss": 0.5618,
201
+ "rewards/accuracies": 0.675000011920929,
202
+ "rewards/chosen": 0.3024521768093109,
203
+ "rewards/margins": 0.6436794400215149,
204
+ "rewards/rejected": -0.3412272334098816,
205
+ "step": 130
206
+ },
207
+ {
208
+ "epoch": 0.07,
209
+ "learning_rate": 3.645833333333333e-07,
210
+ "logits/chosen": -2.6539013385772705,
211
+ "logits/rejected": -2.6091341972351074,
212
+ "logps/chosen": -307.194091796875,
213
+ "logps/rejected": -285.154541015625,
214
+ "loss": 0.5479,
215
+ "rewards/accuracies": 0.6499999761581421,
216
+ "rewards/chosen": -0.1616998016834259,
217
+ "rewards/margins": 0.4194749891757965,
218
+ "rewards/rejected": -0.5811747908592224,
219
+ "step": 140
220
+ },
221
+ {
222
+ "epoch": 0.08,
223
+ "learning_rate": 3.9062499999999997e-07,
224
+ "logits/chosen": -2.4873158931732178,
225
+ "logits/rejected": -2.504676342010498,
226
+ "logps/chosen": -300.03448486328125,
227
+ "logps/rejected": -318.03668212890625,
228
+ "loss": 0.6838,
229
+ "rewards/accuracies": 0.6625000238418579,
230
+ "rewards/chosen": -0.03602635860443115,
231
+ "rewards/margins": -0.06406474858522415,
232
+ "rewards/rejected": 0.028038373216986656,
233
+ "step": 150
234
+ },
235
+ {
236
+ "epoch": 0.08,
237
+ "learning_rate": 4.1666666666666667e-07,
238
+ "logits/chosen": -2.6210286617279053,
239
+ "logits/rejected": -2.6042556762695312,
240
+ "logps/chosen": -286.2755432128906,
241
+ "logps/rejected": -299.8374328613281,
242
+ "loss": 0.5376,
243
+ "rewards/accuracies": 0.699999988079071,
244
+ "rewards/chosen": 0.1890409290790558,
245
+ "rewards/margins": 0.8602690696716309,
246
+ "rewards/rejected": -0.6712281703948975,
247
+ "step": 160
248
+ },
249
+ {
250
+ "epoch": 0.09,
251
+ "learning_rate": 4.427083333333333e-07,
252
+ "logits/chosen": -2.663649320602417,
253
+ "logits/rejected": -2.611992835998535,
254
+ "logps/chosen": -312.26580810546875,
255
+ "logps/rejected": -271.50897216796875,
256
+ "loss": 0.5935,
257
+ "rewards/accuracies": 0.7875000238418579,
258
+ "rewards/chosen": -0.08057786524295807,
259
+ "rewards/margins": 0.7988370656967163,
260
+ "rewards/rejected": -0.8794149160385132,
261
+ "step": 170
262
+ },
263
+ {
264
+ "epoch": 0.09,
265
+ "learning_rate": 4.6874999999999996e-07,
266
+ "logits/chosen": -2.6343648433685303,
267
+ "logits/rejected": -2.5704476833343506,
268
+ "logps/chosen": -338.59124755859375,
269
+ "logps/rejected": -286.7225341796875,
270
+ "loss": 0.5162,
271
+ "rewards/accuracies": 0.637499988079071,
272
+ "rewards/chosen": -0.006776231341063976,
273
+ "rewards/margins": 0.9450720548629761,
274
+ "rewards/rejected": -0.9518482089042664,
275
+ "step": 180
276
+ },
277
+ {
278
+ "epoch": 0.1,
279
+ "learning_rate": 4.947916666666667e-07,
280
+ "logits/chosen": -2.6279683113098145,
281
+ "logits/rejected": -2.5350871086120605,
282
+ "logps/chosen": -288.70184326171875,
283
+ "logps/rejected": -262.1229248046875,
284
+ "loss": 0.6639,
285
+ "rewards/accuracies": 0.737500011920929,
286
+ "rewards/chosen": 0.2994307279586792,
287
+ "rewards/margins": 0.9815131425857544,
288
+ "rewards/rejected": -0.6820824146270752,
289
+ "step": 190
290
+ },
291
+ {
292
+ "epoch": 0.1,
293
+ "learning_rate": 4.976730657358929e-07,
294
+ "logits/chosen": -2.5379586219787598,
295
+ "logits/rejected": -2.49354887008667,
296
+ "logps/chosen": -253.19580078125,
297
+ "logps/rejected": -223.9005584716797,
298
+ "loss": 0.8012,
299
+ "rewards/accuracies": 0.7250000238418579,
300
+ "rewards/chosen": -0.20366668701171875,
301
+ "rewards/margins": -0.2523919641971588,
302
+ "rewards/rejected": 0.04872531443834305,
303
+ "step": 200
304
+ },
305
+ {
306
+ "epoch": 0.11,
307
+ "learning_rate": 4.947643979057592e-07,
308
+ "logits/chosen": -2.543163299560547,
309
+ "logits/rejected": -2.5506227016448975,
310
+ "logps/chosen": -285.0458984375,
311
+ "logps/rejected": -312.3334045410156,
312
+ "loss": 0.5666,
313
+ "rewards/accuracies": 0.7749999761581421,
314
+ "rewards/chosen": 0.23225872218608856,
315
+ "rewards/margins": 0.626153826713562,
316
+ "rewards/rejected": -0.39389508962631226,
317
+ "step": 210
318
+ },
319
+ {
320
+ "epoch": 0.12,
321
+ "learning_rate": 4.918557300756254e-07,
322
+ "logits/chosen": -2.4932289123535156,
323
+ "logits/rejected": -2.464277744293213,
324
+ "logps/chosen": -292.6415100097656,
325
+ "logps/rejected": -273.45648193359375,
326
+ "loss": 0.5932,
327
+ "rewards/accuracies": 0.737500011920929,
328
+ "rewards/chosen": 0.0871698409318924,
329
+ "rewards/margins": 0.7229372262954712,
330
+ "rewards/rejected": -0.6357674598693848,
331
+ "step": 220
332
+ },
333
+ {
334
+ "epoch": 0.12,
335
+ "learning_rate": 4.889470622454916e-07,
336
+ "logits/chosen": -2.579195261001587,
337
+ "logits/rejected": -2.464480400085449,
338
+ "logps/chosen": -310.99090576171875,
339
+ "logps/rejected": -287.1470031738281,
340
+ "loss": 0.5378,
341
+ "rewards/accuracies": 0.7749999761581421,
342
+ "rewards/chosen": 0.05043240636587143,
343
+ "rewards/margins": 0.7342410087585449,
344
+ "rewards/rejected": -0.6838085651397705,
345
+ "step": 230
346
+ },
347
+ {
348
+ "epoch": 0.13,
349
+ "learning_rate": 4.860383944153578e-07,
350
+ "logits/chosen": -2.4707000255584717,
351
+ "logits/rejected": -2.3995609283447266,
352
+ "logps/chosen": -292.8463439941406,
353
+ "logps/rejected": -262.3541259765625,
354
+ "loss": 0.5575,
355
+ "rewards/accuracies": 0.7250000238418579,
356
+ "rewards/chosen": -0.05063791200518608,
357
+ "rewards/margins": 0.8500161170959473,
358
+ "rewards/rejected": -0.9006540179252625,
359
+ "step": 240
360
+ },
361
+ {
362
+ "epoch": 0.13,
363
+ "learning_rate": 4.83129726585224e-07,
364
+ "logits/chosen": -2.4374887943267822,
365
+ "logits/rejected": -2.4220387935638428,
366
+ "logps/chosen": -266.37774658203125,
367
+ "logps/rejected": -283.33978271484375,
368
+ "loss": 0.6727,
369
+ "rewards/accuracies": 0.6875,
370
+ "rewards/chosen": -0.49987784028053284,
371
+ "rewards/margins": 0.5630615949630737,
372
+ "rewards/rejected": -1.0629395246505737,
373
+ "step": 250
374
+ },
375
+ {
376
+ "epoch": 0.14,
377
+ "learning_rate": 4.802210587550902e-07,
378
+ "logits/chosen": -2.405151844024658,
379
+ "logits/rejected": -2.373051643371582,
380
+ "logps/chosen": -323.4302062988281,
381
+ "logps/rejected": -306.20367431640625,
382
+ "loss": 0.7053,
383
+ "rewards/accuracies": 0.7250000238418579,
384
+ "rewards/chosen": -0.31794676184654236,
385
+ "rewards/margins": 0.897225558757782,
386
+ "rewards/rejected": -1.215172290802002,
387
+ "step": 260
388
+ },
389
+ {
390
+ "epoch": 0.14,
391
+ "learning_rate": 4.773123909249563e-07,
392
+ "logits/chosen": -2.432044506072998,
393
+ "logits/rejected": -2.363980770111084,
394
+ "logps/chosen": -278.9121398925781,
395
+ "logps/rejected": -302.1050109863281,
396
+ "loss": 0.6258,
397
+ "rewards/accuracies": 0.6625000238418579,
398
+ "rewards/chosen": -0.047296181321144104,
399
+ "rewards/margins": 0.663731038570404,
400
+ "rewards/rejected": -0.711027204990387,
401
+ "step": 270
402
+ },
403
+ {
404
+ "epoch": 0.15,
405
+ "learning_rate": 4.7440372309482255e-07,
406
+ "logits/chosen": -2.437777042388916,
407
+ "logits/rejected": -2.3604378700256348,
408
+ "logps/chosen": -301.4665222167969,
409
+ "logps/rejected": -265.2036437988281,
410
+ "loss": 0.7178,
411
+ "rewards/accuracies": 0.762499988079071,
412
+ "rewards/chosen": -0.1992354393005371,
413
+ "rewards/margins": 0.936697781085968,
414
+ "rewards/rejected": -1.13593327999115,
415
+ "step": 280
416
+ },
417
+ {
418
+ "epoch": 0.15,
419
+ "learning_rate": 4.7149505526468876e-07,
420
+ "logits/chosen": -2.397371530532837,
421
+ "logits/rejected": -2.3492469787597656,
422
+ "logps/chosen": -296.2301940917969,
423
+ "logps/rejected": -297.2955322265625,
424
+ "loss": 0.5162,
425
+ "rewards/accuracies": 0.625,
426
+ "rewards/chosen": 0.45999306440353394,
427
+ "rewards/margins": 0.6742507815361023,
428
+ "rewards/rejected": -0.21425779163837433,
429
+ "step": 290
430
+ },
431
+ {
432
+ "epoch": 0.16,
433
+ "learning_rate": 4.6858638743455497e-07,
434
+ "logits/chosen": -2.4259443283081055,
435
+ "logits/rejected": -2.3763365745544434,
436
+ "logps/chosen": -349.0486755371094,
437
+ "logps/rejected": -336.35894775390625,
438
+ "loss": 0.5774,
439
+ "rewards/accuracies": 0.6875,
440
+ "rewards/chosen": 0.4317784905433655,
441
+ "rewards/margins": 0.9294607043266296,
442
+ "rewards/rejected": -0.49768227338790894,
443
+ "step": 300
444
+ },
445
+ {
446
+ "epoch": 0.16,
447
+ "learning_rate": 4.656777196044212e-07,
448
+ "logits/chosen": -2.4149937629699707,
449
+ "logits/rejected": -2.299372911453247,
450
+ "logps/chosen": -296.93731689453125,
451
+ "logps/rejected": -269.88836669921875,
452
+ "loss": 0.614,
453
+ "rewards/accuracies": 0.8125,
454
+ "rewards/chosen": 0.13364040851593018,
455
+ "rewards/margins": 1.1540381908416748,
456
+ "rewards/rejected": -1.020397663116455,
457
+ "step": 310
458
+ },
459
+ {
460
+ "epoch": 0.17,
461
+ "learning_rate": 4.627690517742874e-07,
462
+ "logits/chosen": -2.3246355056762695,
463
+ "logits/rejected": -2.31697940826416,
464
+ "logps/chosen": -246.38571166992188,
465
+ "logps/rejected": -243.2635955810547,
466
+ "loss": 0.5455,
467
+ "rewards/accuracies": 0.6875,
468
+ "rewards/chosen": 0.3853219151496887,
469
+ "rewards/margins": 1.1966570615768433,
470
+ "rewards/rejected": -0.8113352060317993,
471
+ "step": 320
472
+ },
473
+ {
474
+ "epoch": 0.17,
475
+ "learning_rate": 4.5986038394415354e-07,
476
+ "logits/chosen": -2.377342939376831,
477
+ "logits/rejected": -2.266279697418213,
478
+ "logps/chosen": -291.89923095703125,
479
+ "logps/rejected": -252.13156127929688,
480
+ "loss": 0.7187,
481
+ "rewards/accuracies": 0.7875000238418579,
482
+ "rewards/chosen": -0.3700163662433624,
483
+ "rewards/margins": 1.021303415298462,
484
+ "rewards/rejected": -1.3913196325302124,
485
+ "step": 330
486
+ },
487
+ {
488
+ "epoch": 0.18,
489
+ "learning_rate": 4.569517161140198e-07,
490
+ "logits/chosen": -2.4199347496032715,
491
+ "logits/rejected": -2.4432175159454346,
492
+ "logps/chosen": -294.2767333984375,
493
+ "logps/rejected": -317.6234130859375,
494
+ "loss": 0.5903,
495
+ "rewards/accuracies": 0.7124999761581421,
496
+ "rewards/chosen": 0.04284637048840523,
497
+ "rewards/margins": 0.6986137628555298,
498
+ "rewards/rejected": -0.6557673811912537,
499
+ "step": 340
500
+ },
501
+ {
502
+ "epoch": 0.18,
503
+ "learning_rate": 4.5404304828388595e-07,
504
+ "logits/chosen": -2.488062858581543,
505
+ "logits/rejected": -2.420581340789795,
506
+ "logps/chosen": -316.81109619140625,
507
+ "logps/rejected": -355.86297607421875,
508
+ "loss": 0.5148,
509
+ "rewards/accuracies": 0.824999988079071,
510
+ "rewards/chosen": -0.16860897839069366,
511
+ "rewards/margins": 1.251904010772705,
512
+ "rewards/rejected": -1.4205129146575928,
513
+ "step": 350
514
+ },
515
+ {
516
+ "epoch": 0.19,
517
+ "learning_rate": 4.5113438045375216e-07,
518
+ "logits/chosen": -2.372753858566284,
519
+ "logits/rejected": -2.3147406578063965,
520
+ "logps/chosen": -287.305419921875,
521
+ "logps/rejected": -256.7542724609375,
522
+ "loss": 0.526,
523
+ "rewards/accuracies": 0.762499988079071,
524
+ "rewards/chosen": 0.243671253323555,
525
+ "rewards/margins": 0.9904806017875671,
526
+ "rewards/rejected": -0.7468093633651733,
527
+ "step": 360
528
+ },
529
+ {
530
+ "epoch": 0.19,
531
+ "learning_rate": 4.4822571262361837e-07,
532
+ "logits/chosen": -2.446676731109619,
533
+ "logits/rejected": -2.414977550506592,
534
+ "logps/chosen": -242.71127319335938,
535
+ "logps/rejected": -246.08462524414062,
536
+ "loss": 0.6242,
537
+ "rewards/accuracies": 0.5625,
538
+ "rewards/chosen": -0.48434001207351685,
539
+ "rewards/margins": 0.3427623510360718,
540
+ "rewards/rejected": -0.8271023631095886,
541
+ "step": 370
542
+ },
543
+ {
544
+ "epoch": 0.2,
545
+ "learning_rate": 4.453170447934846e-07,
546
+ "logits/chosen": -2.3129241466522217,
547
+ "logits/rejected": -2.2660763263702393,
548
+ "logps/chosen": -325.49517822265625,
549
+ "logps/rejected": -257.01812744140625,
550
+ "loss": 0.6439,
551
+ "rewards/accuracies": 0.7749999761581421,
552
+ "rewards/chosen": -0.7354428768157959,
553
+ "rewards/margins": 0.5178254842758179,
554
+ "rewards/rejected": -1.2532682418823242,
555
+ "step": 380
556
+ },
557
+ {
558
+ "epoch": 0.2,
559
+ "learning_rate": 4.424083769633508e-07,
560
+ "logits/chosen": -2.407226085662842,
561
+ "logits/rejected": -2.3878862857818604,
562
+ "logps/chosen": -321.883056640625,
563
+ "logps/rejected": -329.2529296875,
564
+ "loss": 0.6696,
565
+ "rewards/accuracies": 0.7124999761581421,
566
+ "rewards/chosen": -0.6276373863220215,
567
+ "rewards/margins": 0.8156329989433289,
568
+ "rewards/rejected": -1.4432705640792847,
569
+ "step": 390
570
+ },
571
+ {
572
+ "epoch": 0.21,
573
+ "learning_rate": 4.39499709133217e-07,
574
+ "logits/chosen": -2.5650746822357178,
575
+ "logits/rejected": -2.5109097957611084,
576
+ "logps/chosen": -315.965576171875,
577
+ "logps/rejected": -329.1568298339844,
578
+ "loss": 0.5373,
579
+ "rewards/accuracies": 0.6875,
580
+ "rewards/chosen": -0.22043700516223907,
581
+ "rewards/margins": 0.7228590250015259,
582
+ "rewards/rejected": -0.9432960748672485,
583
+ "step": 400
584
+ },
585
+ {
586
+ "epoch": 0.21,
587
+ "learning_rate": 4.3659104130308314e-07,
588
+ "logits/chosen": -2.3766233921051025,
589
+ "logits/rejected": -2.342545747756958,
590
+ "logps/chosen": -270.418701171875,
591
+ "logps/rejected": -277.3287048339844,
592
+ "loss": 0.4702,
593
+ "rewards/accuracies": 0.75,
594
+ "rewards/chosen": -0.33475637435913086,
595
+ "rewards/margins": 1.0519344806671143,
596
+ "rewards/rejected": -1.3866908550262451,
597
+ "step": 410
598
+ },
599
+ {
600
+ "epoch": 0.22,
601
+ "learning_rate": 4.336823734729494e-07,
602
+ "logits/chosen": -2.5363078117370605,
603
+ "logits/rejected": -2.420921802520752,
604
+ "logps/chosen": -339.0617370605469,
605
+ "logps/rejected": -314.2167663574219,
606
+ "loss": 0.5078,
607
+ "rewards/accuracies": 0.7875000238418579,
608
+ "rewards/chosen": -0.20719392597675323,
609
+ "rewards/margins": 1.039120078086853,
610
+ "rewards/rejected": -1.2463139295578003,
611
+ "step": 420
612
+ },
613
+ {
614
+ "epoch": 0.23,
615
+ "learning_rate": 4.3077370564281556e-07,
616
+ "logits/chosen": -2.4507744312286377,
617
+ "logits/rejected": -2.3617775440216064,
618
+ "logps/chosen": -305.3289794921875,
619
+ "logps/rejected": -270.58843994140625,
620
+ "loss": 0.5647,
621
+ "rewards/accuracies": 0.6499999761581421,
622
+ "rewards/chosen": -0.5011049509048462,
623
+ "rewards/margins": 0.8858562707901001,
624
+ "rewards/rejected": -1.3869613409042358,
625
+ "step": 430
626
+ },
627
+ {
628
+ "epoch": 0.23,
629
+ "learning_rate": 4.278650378126818e-07,
630
+ "logits/chosen": -2.327728033065796,
631
+ "logits/rejected": -2.2920546531677246,
632
+ "logps/chosen": -289.3099365234375,
633
+ "logps/rejected": -333.728271484375,
634
+ "loss": 0.5112,
635
+ "rewards/accuracies": 0.7124999761581421,
636
+ "rewards/chosen": 0.1836535632610321,
637
+ "rewards/margins": 1.081466794013977,
638
+ "rewards/rejected": -0.8978131413459778,
639
+ "step": 440
640
+ },
641
+ {
642
+ "epoch": 0.24,
643
+ "learning_rate": 4.24956369982548e-07,
644
+ "logits/chosen": -2.4250426292419434,
645
+ "logits/rejected": -2.3737356662750244,
646
+ "logps/chosen": -308.11236572265625,
647
+ "logps/rejected": -295.8037109375,
648
+ "loss": 0.5505,
649
+ "rewards/accuracies": 0.699999988079071,
650
+ "rewards/chosen": 0.22618088126182556,
651
+ "rewards/margins": 0.990483283996582,
652
+ "rewards/rejected": -0.7643024325370789,
653
+ "step": 450
654
+ },
655
+ {
656
+ "epoch": 0.24,
657
+ "learning_rate": 4.220477021524142e-07,
658
+ "logits/chosen": -2.4373867511749268,
659
+ "logits/rejected": -2.400118112564087,
660
+ "logps/chosen": -290.0523681640625,
661
+ "logps/rejected": -255.9022674560547,
662
+ "loss": 0.5815,
663
+ "rewards/accuracies": 0.737500011920929,
664
+ "rewards/chosen": 0.12436334788799286,
665
+ "rewards/margins": 0.7458918690681458,
666
+ "rewards/rejected": -0.6215284466743469,
667
+ "step": 460
668
+ },
669
+ {
670
+ "epoch": 0.25,
671
+ "learning_rate": 4.191390343222804e-07,
672
+ "logits/chosen": -2.506086826324463,
673
+ "logits/rejected": -2.4103667736053467,
674
+ "logps/chosen": -305.6959533691406,
675
+ "logps/rejected": -305.922607421875,
676
+ "loss": 0.505,
677
+ "rewards/accuracies": 0.737500011920929,
678
+ "rewards/chosen": 0.27497127652168274,
679
+ "rewards/margins": 1.060005784034729,
680
+ "rewards/rejected": -0.7850344181060791,
681
+ "step": 470
682
+ },
683
+ {
684
+ "epoch": 0.25,
685
+ "learning_rate": 4.162303664921466e-07,
686
+ "logits/chosen": -2.460691213607788,
687
+ "logits/rejected": -2.3934152126312256,
688
+ "logps/chosen": -310.1314697265625,
689
+ "logps/rejected": -297.3235168457031,
690
+ "loss": 0.5853,
691
+ "rewards/accuracies": 0.7250000238418579,
692
+ "rewards/chosen": 0.14773890376091003,
693
+ "rewards/margins": 0.7825302481651306,
694
+ "rewards/rejected": -0.6347913146018982,
695
+ "step": 480
696
+ },
697
+ {
698
+ "epoch": 0.26,
699
+ "learning_rate": 4.1332169866201275e-07,
700
+ "logits/chosen": -2.4345953464508057,
701
+ "logits/rejected": -2.4074747562408447,
702
+ "logps/chosen": -290.62579345703125,
703
+ "logps/rejected": -277.2015380859375,
704
+ "loss": 0.5713,
705
+ "rewards/accuracies": 0.699999988079071,
706
+ "rewards/chosen": 0.5942241549491882,
707
+ "rewards/margins": 0.8701769113540649,
708
+ "rewards/rejected": -0.2759527266025543,
709
+ "step": 490
710
+ },
711
+ {
712
+ "epoch": 0.26,
713
+ "learning_rate": 4.10413030831879e-07,
714
+ "logits/chosen": -2.491701602935791,
715
+ "logits/rejected": -2.3788347244262695,
716
+ "logps/chosen": -327.69097900390625,
717
+ "logps/rejected": -256.1454162597656,
718
+ "loss": 0.5509,
719
+ "rewards/accuracies": 0.6625000238418579,
720
+ "rewards/chosen": 0.3082984387874603,
721
+ "rewards/margins": 0.7658659219741821,
722
+ "rewards/rejected": -0.4575675129890442,
723
+ "step": 500
724
+ },
725
+ {
726
+ "epoch": 0.26,
727
+ "eval_logits/chosen": -2.4869401454925537,
728
+ "eval_logits/rejected": -2.4248592853546143,
729
+ "eval_logps/chosen": -292.5533447265625,
730
+ "eval_logps/rejected": -277.5867614746094,
731
+ "eval_loss": 0.6311370134353638,
732
+ "eval_rewards/accuracies": 0.761904776096344,
733
+ "eval_rewards/chosen": 0.8113744854927063,
734
+ "eval_rewards/margins": 0.9294369220733643,
735
+ "eval_rewards/rejected": -0.11806251108646393,
736
+ "eval_runtime": 615.8679,
737
+ "eval_samples_per_second": 3.247,
738
+ "eval_steps_per_second": 0.102,
739
+ "step": 500
740
+ },
741
+ {
742
+ "epoch": 0.27,
743
+ "learning_rate": 4.0750436300174517e-07,
744
+ "logits/chosen": -2.397420883178711,
745
+ "logits/rejected": -2.372736692428589,
746
+ "logps/chosen": -287.88897705078125,
747
+ "logps/rejected": -266.78594970703125,
748
+ "loss": 0.5289,
749
+ "rewards/accuracies": 0.762499988079071,
750
+ "rewards/chosen": 0.6826907396316528,
751
+ "rewards/margins": 1.225263237953186,
752
+ "rewards/rejected": -0.542572557926178,
753
+ "step": 510
754
+ },
755
+ {
756
+ "epoch": 0.27,
757
+ "learning_rate": 4.0459569517161143e-07,
758
+ "logits/chosen": -2.448612928390503,
759
+ "logits/rejected": -2.335453748703003,
760
+ "logps/chosen": -289.3345947265625,
761
+ "logps/rejected": -280.9012145996094,
762
+ "loss": 0.9872,
763
+ "rewards/accuracies": 0.7250000238418579,
764
+ "rewards/chosen": 0.408169686794281,
765
+ "rewards/margins": 1.0372252464294434,
766
+ "rewards/rejected": -0.6290556192398071,
767
+ "step": 520
768
+ },
769
+ {
770
+ "epoch": 0.28,
771
+ "learning_rate": 4.016870273414776e-07,
772
+ "logits/chosen": -2.4662892818450928,
773
+ "logits/rejected": -2.425401449203491,
774
+ "logps/chosen": -260.6138000488281,
775
+ "logps/rejected": -284.0682373046875,
776
+ "loss": 0.5718,
777
+ "rewards/accuracies": 0.675000011920929,
778
+ "rewards/chosen": 0.2825961709022522,
779
+ "rewards/margins": 1.0750539302825928,
780
+ "rewards/rejected": -0.7924576997756958,
781
+ "step": 530
782
+ },
783
+ {
784
+ "epoch": 0.28,
785
+ "learning_rate": 3.987783595113438e-07,
786
+ "logits/chosen": -2.498586416244507,
787
+ "logits/rejected": -2.421823501586914,
788
+ "logps/chosen": -339.8152160644531,
789
+ "logps/rejected": -266.1904602050781,
790
+ "loss": 0.5823,
791
+ "rewards/accuracies": 0.824999988079071,
792
+ "rewards/chosen": 0.5165579915046692,
793
+ "rewards/margins": 1.7178938388824463,
794
+ "rewards/rejected": -1.2013359069824219,
795
+ "step": 540
796
+ },
797
+ {
798
+ "epoch": 0.29,
799
+ "learning_rate": 3.9586969168121e-07,
800
+ "logits/chosen": -2.491562604904175,
801
+ "logits/rejected": -2.4570322036743164,
802
+ "logps/chosen": -348.4375,
803
+ "logps/rejected": -306.64697265625,
804
+ "loss": 0.5497,
805
+ "rewards/accuracies": 0.7250000238418579,
806
+ "rewards/chosen": 0.10518566519021988,
807
+ "rewards/margins": 0.8593299984931946,
808
+ "rewards/rejected": -0.7541443705558777,
809
+ "step": 550
810
+ },
811
+ {
812
+ "epoch": 0.29,
813
+ "learning_rate": 3.929610238510762e-07,
814
+ "logits/chosen": -2.3990979194641113,
815
+ "logits/rejected": -2.39212965965271,
816
+ "logps/chosen": -289.9490661621094,
817
+ "logps/rejected": -269.9928894042969,
818
+ "loss": 0.6319,
819
+ "rewards/accuracies": 0.625,
820
+ "rewards/chosen": 0.04818814992904663,
821
+ "rewards/margins": 0.6179074048995972,
822
+ "rewards/rejected": -0.569719135761261,
823
+ "step": 560
824
+ },
825
+ {
826
+ "epoch": 0.3,
827
+ "learning_rate": 3.900523560209424e-07,
828
+ "logits/chosen": -2.4815762042999268,
829
+ "logits/rejected": -2.425726890563965,
830
+ "logps/chosen": -312.6693420410156,
831
+ "logps/rejected": -273.46728515625,
832
+ "loss": 0.5107,
833
+ "rewards/accuracies": 0.762499988079071,
834
+ "rewards/chosen": 0.3141094148159027,
835
+ "rewards/margins": 1.199885606765747,
836
+ "rewards/rejected": -0.8857762217521667,
837
+ "step": 570
838
+ },
839
+ {
840
+ "epoch": 0.3,
841
+ "learning_rate": 3.871436881908086e-07,
842
+ "logits/chosen": -2.5039753913879395,
843
+ "logits/rejected": -2.414092779159546,
844
+ "logps/chosen": -284.1502990722656,
845
+ "logps/rejected": -240.26083374023438,
846
+ "loss": 0.523,
847
+ "rewards/accuracies": 0.699999988079071,
848
+ "rewards/chosen": 0.05462036281824112,
849
+ "rewards/margins": 0.9068329930305481,
850
+ "rewards/rejected": -0.8522126078605652,
851
+ "step": 580
852
+ },
853
+ {
854
+ "epoch": 0.31,
855
+ "learning_rate": 3.842350203606748e-07,
856
+ "logits/chosen": -2.487733840942383,
857
+ "logits/rejected": -2.4456124305725098,
858
+ "logps/chosen": -264.53790283203125,
859
+ "logps/rejected": -287.30718994140625,
860
+ "loss": 0.5847,
861
+ "rewards/accuracies": 0.7250000238418579,
862
+ "rewards/chosen": -0.48793086409568787,
863
+ "rewards/margins": 1.0478885173797607,
864
+ "rewards/rejected": -1.535819411277771,
865
+ "step": 590
866
+ },
867
+ {
868
+ "epoch": 0.31,
869
+ "learning_rate": 3.8132635253054103e-07,
870
+ "logits/chosen": -2.4525985717773438,
871
+ "logits/rejected": -2.419036865234375,
872
+ "logps/chosen": -292.17840576171875,
873
+ "logps/rejected": -305.83258056640625,
874
+ "loss": 0.5466,
875
+ "rewards/accuracies": 0.800000011920929,
876
+ "rewards/chosen": -0.03021824359893799,
877
+ "rewards/margins": 1.0404767990112305,
878
+ "rewards/rejected": -1.070695161819458,
879
+ "step": 600
880
+ },
881
+ {
882
+ "epoch": 0.32,
883
+ "learning_rate": 3.784176847004072e-07,
884
+ "logits/chosen": -2.494528293609619,
885
+ "logits/rejected": -2.455793857574463,
886
+ "logps/chosen": -276.6438903808594,
887
+ "logps/rejected": -284.5893249511719,
888
+ "loss": 0.4859,
889
+ "rewards/accuracies": 0.8374999761581421,
890
+ "rewards/chosen": 0.11220204830169678,
891
+ "rewards/margins": 1.3604462146759033,
892
+ "rewards/rejected": -1.2482441663742065,
893
+ "step": 610
894
+ },
895
+ {
896
+ "epoch": 0.32,
897
+ "learning_rate": 3.755090168702734e-07,
898
+ "logits/chosen": -2.4499683380126953,
899
+ "logits/rejected": -2.4269161224365234,
900
+ "logps/chosen": -319.4291076660156,
901
+ "logps/rejected": -330.466552734375,
902
+ "loss": 0.5209,
903
+ "rewards/accuracies": 0.75,
904
+ "rewards/chosen": -0.1047038584947586,
905
+ "rewards/margins": 1.4205875396728516,
906
+ "rewards/rejected": -1.5252914428710938,
907
+ "step": 620
908
+ },
909
+ {
910
+ "epoch": 0.33,
911
+ "learning_rate": 3.726003490401396e-07,
912
+ "logits/chosen": -2.4178521633148193,
913
+ "logits/rejected": -2.3652777671813965,
914
+ "logps/chosen": -301.78399658203125,
915
+ "logps/rejected": -255.4466094970703,
916
+ "loss": 0.6127,
917
+ "rewards/accuracies": 0.7124999761581421,
918
+ "rewards/chosen": -0.73511803150177,
919
+ "rewards/margins": 0.9625850915908813,
920
+ "rewards/rejected": -1.6977031230926514,
921
+ "step": 630
922
+ },
923
+ {
924
+ "epoch": 0.33,
925
+ "learning_rate": 3.696916812100058e-07,
926
+ "logits/chosen": -2.414977788925171,
927
+ "logits/rejected": -2.2954540252685547,
928
+ "logps/chosen": -309.4778747558594,
929
+ "logps/rejected": -232.9076385498047,
930
+ "loss": 0.563,
931
+ "rewards/accuracies": 0.675000011920929,
932
+ "rewards/chosen": -0.36078378558158875,
933
+ "rewards/margins": 1.096639633178711,
934
+ "rewards/rejected": -1.457423448562622,
935
+ "step": 640
936
+ },
937
+ {
938
+ "epoch": 0.34,
939
+ "learning_rate": 3.66783013379872e-07,
940
+ "logits/chosen": -2.4305996894836426,
941
+ "logits/rejected": -2.3547043800354004,
942
+ "logps/chosen": -303.08221435546875,
943
+ "logps/rejected": -249.44003295898438,
944
+ "loss": 0.5194,
945
+ "rewards/accuracies": 0.762499988079071,
946
+ "rewards/chosen": -0.32232993841171265,
947
+ "rewards/margins": 1.2769782543182373,
948
+ "rewards/rejected": -1.5993082523345947,
949
+ "step": 650
950
+ },
951
+ {
952
+ "epoch": 0.35,
953
+ "learning_rate": 3.6387434554973823e-07,
954
+ "logits/chosen": -2.4183077812194824,
955
+ "logits/rejected": -2.3758342266082764,
956
+ "logps/chosen": -326.63739013671875,
957
+ "logps/rejected": -319.60357666015625,
958
+ "loss": 0.5133,
959
+ "rewards/accuracies": 0.6875,
960
+ "rewards/chosen": -0.7344697117805481,
961
+ "rewards/margins": 1.0054800510406494,
962
+ "rewards/rejected": -1.7399498224258423,
963
+ "step": 660
964
+ },
965
+ {
966
+ "epoch": 0.35,
967
+ "learning_rate": 3.609656777196044e-07,
968
+ "logits/chosen": -2.4492297172546387,
969
+ "logits/rejected": -2.3460309505462646,
970
+ "logps/chosen": -334.344970703125,
971
+ "logps/rejected": -272.79498291015625,
972
+ "loss": 0.5318,
973
+ "rewards/accuracies": 0.800000011920929,
974
+ "rewards/chosen": -0.6380278468132019,
975
+ "rewards/margins": 1.2398582696914673,
976
+ "rewards/rejected": -1.877886414527893,
977
+ "step": 670
978
+ },
979
+ {
980
+ "epoch": 0.36,
981
+ "learning_rate": 3.5805700988947064e-07,
982
+ "logits/chosen": -2.37317156791687,
983
+ "logits/rejected": -2.287410259246826,
984
+ "logps/chosen": -259.41107177734375,
985
+ "logps/rejected": -246.9707794189453,
986
+ "loss": 0.6405,
987
+ "rewards/accuracies": 0.737500011920929,
988
+ "rewards/chosen": -0.004200550727546215,
989
+ "rewards/margins": 1.685315728187561,
990
+ "rewards/rejected": -1.689516305923462,
991
+ "step": 680
992
+ },
993
+ {
994
+ "epoch": 0.36,
995
+ "learning_rate": 3.551483420593368e-07,
996
+ "logits/chosen": -2.424691677093506,
997
+ "logits/rejected": -2.3706610202789307,
998
+ "logps/chosen": -337.8548889160156,
999
+ "logps/rejected": -300.6462707519531,
1000
+ "loss": 0.4984,
1001
+ "rewards/accuracies": 0.737500011920929,
1002
+ "rewards/chosen": -0.4729352593421936,
1003
+ "rewards/margins": 1.0950307846069336,
1004
+ "rewards/rejected": -1.567966103553772,
1005
+ "step": 690
1006
+ },
1007
+ {
1008
+ "epoch": 0.37,
1009
+ "learning_rate": 3.5223967422920306e-07,
1010
+ "logits/chosen": -2.4483962059020996,
1011
+ "logits/rejected": -2.3456740379333496,
1012
+ "logps/chosen": -355.38446044921875,
1013
+ "logps/rejected": -313.3027038574219,
1014
+ "loss": 0.7153,
1015
+ "rewards/accuracies": 0.7250000238418579,
1016
+ "rewards/chosen": -0.5108211636543274,
1017
+ "rewards/margins": 1.1368777751922607,
1018
+ "rewards/rejected": -1.647698998451233,
1019
+ "step": 700
1020
+ },
1021
+ {
1022
+ "epoch": 0.37,
1023
+ "learning_rate": 3.493310063990692e-07,
1024
+ "logits/chosen": -2.4961283206939697,
1025
+ "logits/rejected": -2.4019272327423096,
1026
+ "logps/chosen": -258.533447265625,
1027
+ "logps/rejected": -235.88912963867188,
1028
+ "loss": 0.6123,
1029
+ "rewards/accuracies": 0.699999988079071,
1030
+ "rewards/chosen": -0.2880368232727051,
1031
+ "rewards/margins": 0.940460205078125,
1032
+ "rewards/rejected": -1.22849702835083,
1033
+ "step": 710
1034
+ },
1035
+ {
1036
+ "epoch": 0.38,
1037
+ "learning_rate": 3.464223385689354e-07,
1038
+ "logits/chosen": -2.4469172954559326,
1039
+ "logits/rejected": -2.442150354385376,
1040
+ "logps/chosen": -263.5059814453125,
1041
+ "logps/rejected": -278.46112060546875,
1042
+ "loss": 0.5192,
1043
+ "rewards/accuracies": 0.699999988079071,
1044
+ "rewards/chosen": 0.03521687909960747,
1045
+ "rewards/margins": 1.297978162765503,
1046
+ "rewards/rejected": -1.2627613544464111,
1047
+ "step": 720
1048
+ },
1049
+ {
1050
+ "epoch": 0.38,
1051
+ "learning_rate": 3.4351367073880163e-07,
1052
+ "logits/chosen": -2.567368745803833,
1053
+ "logits/rejected": -2.5503649711608887,
1054
+ "logps/chosen": -263.9495544433594,
1055
+ "logps/rejected": -274.8172302246094,
1056
+ "loss": 0.4936,
1057
+ "rewards/accuracies": 0.7749999761581421,
1058
+ "rewards/chosen": -0.18824487924575806,
1059
+ "rewards/margins": 1.1973415613174438,
1060
+ "rewards/rejected": -1.3855865001678467,
1061
+ "step": 730
1062
+ },
1063
+ {
1064
+ "epoch": 0.39,
1065
+ "learning_rate": 3.4060500290866783e-07,
1066
+ "logits/chosen": -2.5890707969665527,
1067
+ "logits/rejected": -2.497729539871216,
1068
+ "logps/chosen": -357.546875,
1069
+ "logps/rejected": -316.70361328125,
1070
+ "loss": 0.4672,
1071
+ "rewards/accuracies": 0.8125,
1072
+ "rewards/chosen": -0.6199100017547607,
1073
+ "rewards/margins": 1.5381364822387695,
1074
+ "rewards/rejected": -2.158046245574951,
1075
+ "step": 740
1076
+ },
1077
+ {
1078
+ "epoch": 0.39,
1079
+ "learning_rate": 3.37696335078534e-07,
1080
+ "logits/chosen": -2.6034646034240723,
1081
+ "logits/rejected": -2.5190954208374023,
1082
+ "logps/chosen": -307.06573486328125,
1083
+ "logps/rejected": -264.7110900878906,
1084
+ "loss": 0.5857,
1085
+ "rewards/accuracies": 0.762499988079071,
1086
+ "rewards/chosen": -0.9259228706359863,
1087
+ "rewards/margins": 1.1956923007965088,
1088
+ "rewards/rejected": -2.121615409851074,
1089
+ "step": 750
1090
+ },
1091
+ {
1092
+ "epoch": 0.4,
1093
+ "learning_rate": 3.3478766724840025e-07,
1094
+ "logits/chosen": -2.5817999839782715,
1095
+ "logits/rejected": -2.532043695449829,
1096
+ "logps/chosen": -291.2391662597656,
1097
+ "logps/rejected": -286.722900390625,
1098
+ "loss": 0.4794,
1099
+ "rewards/accuracies": 0.7250000238418579,
1100
+ "rewards/chosen": -0.822766900062561,
1101
+ "rewards/margins": 1.2887961864471436,
1102
+ "rewards/rejected": -2.111563205718994,
1103
+ "step": 760
1104
+ },
1105
+ {
1106
+ "epoch": 0.4,
1107
+ "learning_rate": 3.318789994182664e-07,
1108
+ "logits/chosen": -2.596311092376709,
1109
+ "logits/rejected": -2.5423972606658936,
1110
+ "logps/chosen": -271.95330810546875,
1111
+ "logps/rejected": -298.80975341796875,
1112
+ "loss": 0.5103,
1113
+ "rewards/accuracies": 0.7124999761581421,
1114
+ "rewards/chosen": -0.4673972725868225,
1115
+ "rewards/margins": 1.1896220445632935,
1116
+ "rewards/rejected": -1.6570192575454712,
1117
+ "step": 770
1118
+ },
1119
+ {
1120
+ "epoch": 0.41,
1121
+ "learning_rate": 3.2897033158813266e-07,
1122
+ "logits/chosen": -2.5218849182128906,
1123
+ "logits/rejected": -2.5217957496643066,
1124
+ "logps/chosen": -309.8044738769531,
1125
+ "logps/rejected": -282.1851806640625,
1126
+ "loss": 0.584,
1127
+ "rewards/accuracies": 0.699999988079071,
1128
+ "rewards/chosen": -0.8506789207458496,
1129
+ "rewards/margins": 0.7243373990058899,
1130
+ "rewards/rejected": -1.5750162601470947,
1131
+ "step": 780
1132
+ },
1133
+ {
1134
+ "epoch": 0.41,
1135
+ "learning_rate": 3.260616637579988e-07,
1136
+ "logits/chosen": -2.621936082839966,
1137
+ "logits/rejected": -2.5529212951660156,
1138
+ "logps/chosen": -321.58624267578125,
1139
+ "logps/rejected": -306.6531677246094,
1140
+ "loss": 0.5215,
1141
+ "rewards/accuracies": 0.762499988079071,
1142
+ "rewards/chosen": -0.5436080694198608,
1143
+ "rewards/margins": 1.283302903175354,
1144
+ "rewards/rejected": -1.8269107341766357,
1145
+ "step": 790
1146
+ },
1147
+ {
1148
+ "epoch": 0.42,
1149
+ "learning_rate": 3.2315299592786503e-07,
1150
+ "logits/chosen": -2.537576913833618,
1151
+ "logits/rejected": -2.497835636138916,
1152
+ "logps/chosen": -290.63714599609375,
1153
+ "logps/rejected": -284.85650634765625,
1154
+ "loss": 0.5535,
1155
+ "rewards/accuracies": 0.7250000238418579,
1156
+ "rewards/chosen": -0.6990116834640503,
1157
+ "rewards/margins": 1.495788812637329,
1158
+ "rewards/rejected": -2.194800615310669,
1159
+ "step": 800
1160
+ },
1161
+ {
1162
+ "epoch": 0.42,
1163
+ "learning_rate": 3.2024432809773123e-07,
1164
+ "logits/chosen": -2.486382246017456,
1165
+ "logits/rejected": -2.403653860092163,
1166
+ "logps/chosen": -286.6750183105469,
1167
+ "logps/rejected": -267.35723876953125,
1168
+ "loss": 0.5029,
1169
+ "rewards/accuracies": 0.7250000238418579,
1170
+ "rewards/chosen": -1.0650520324707031,
1171
+ "rewards/margins": 1.3269193172454834,
1172
+ "rewards/rejected": -2.3919713497161865,
1173
+ "step": 810
1174
+ },
1175
+ {
1176
+ "epoch": 0.43,
1177
+ "learning_rate": 3.1733566026759744e-07,
1178
+ "logits/chosen": -2.5125479698181152,
1179
+ "logits/rejected": -2.461850643157959,
1180
+ "logps/chosen": -289.32379150390625,
1181
+ "logps/rejected": -280.98785400390625,
1182
+ "loss": 0.4962,
1183
+ "rewards/accuracies": 0.699999988079071,
1184
+ "rewards/chosen": -0.6875102519989014,
1185
+ "rewards/margins": 1.0723689794540405,
1186
+ "rewards/rejected": -1.7598793506622314,
1187
+ "step": 820
1188
+ },
1189
+ {
1190
+ "epoch": 0.43,
1191
+ "learning_rate": 3.144269924374636e-07,
1192
+ "logits/chosen": -2.495903491973877,
1193
+ "logits/rejected": -2.4473767280578613,
1194
+ "logps/chosen": -277.972900390625,
1195
+ "logps/rejected": -305.8498840332031,
1196
+ "loss": 0.5665,
1197
+ "rewards/accuracies": 0.7124999761581421,
1198
+ "rewards/chosen": -0.6580140590667725,
1199
+ "rewards/margins": 0.9823113679885864,
1200
+ "rewards/rejected": -1.6403253078460693,
1201
+ "step": 830
1202
+ },
1203
+ {
1204
+ "epoch": 0.44,
1205
+ "learning_rate": 3.1151832460732986e-07,
1206
+ "logits/chosen": -2.4640355110168457,
1207
+ "logits/rejected": -2.3490617275238037,
1208
+ "logps/chosen": -321.24688720703125,
1209
+ "logps/rejected": -300.0312805175781,
1210
+ "loss": 0.7062,
1211
+ "rewards/accuracies": 0.7875000238418579,
1212
+ "rewards/chosen": -0.7375732660293579,
1213
+ "rewards/margins": 1.0734766721725464,
1214
+ "rewards/rejected": -1.8110501766204834,
1215
+ "step": 840
1216
+ },
1217
+ {
1218
+ "epoch": 0.44,
1219
+ "learning_rate": 3.08609656777196e-07,
1220
+ "logits/chosen": -2.4589717388153076,
1221
+ "logits/rejected": -2.4391770362854004,
1222
+ "logps/chosen": -276.55450439453125,
1223
+ "logps/rejected": -291.0248107910156,
1224
+ "loss": 0.5562,
1225
+ "rewards/accuracies": 0.762499988079071,
1226
+ "rewards/chosen": -0.9284421801567078,
1227
+ "rewards/margins": 1.0043559074401855,
1228
+ "rewards/rejected": -1.9327980279922485,
1229
+ "step": 850
1230
+ },
1231
+ {
1232
+ "epoch": 0.45,
1233
+ "learning_rate": 3.0570098894706227e-07,
1234
+ "logits/chosen": -2.4961347579956055,
1235
+ "logits/rejected": -2.4345571994781494,
1236
+ "logps/chosen": -305.3291015625,
1237
+ "logps/rejected": -354.2350769042969,
1238
+ "loss": 0.5703,
1239
+ "rewards/accuracies": 0.7124999761581421,
1240
+ "rewards/chosen": -0.7021960616111755,
1241
+ "rewards/margins": 1.1666209697723389,
1242
+ "rewards/rejected": -1.8688170909881592,
1243
+ "step": 860
1244
+ },
1245
+ {
1246
+ "epoch": 0.46,
1247
+ "learning_rate": 3.0279232111692843e-07,
1248
+ "logits/chosen": -2.366995096206665,
1249
+ "logits/rejected": -2.3412668704986572,
1250
+ "logps/chosen": -269.6289978027344,
1251
+ "logps/rejected": -294.22308349609375,
1252
+ "loss": 0.5518,
1253
+ "rewards/accuracies": 0.7749999761581421,
1254
+ "rewards/chosen": -0.0025119720958173275,
1255
+ "rewards/margins": 1.3077478408813477,
1256
+ "rewards/rejected": -1.3102598190307617,
1257
+ "step": 870
1258
+ },
1259
+ {
1260
+ "epoch": 0.46,
1261
+ "learning_rate": 2.9988365328679463e-07,
1262
+ "logits/chosen": -2.5405781269073486,
1263
+ "logits/rejected": -2.5027382373809814,
1264
+ "logps/chosen": -310.4794006347656,
1265
+ "logps/rejected": -311.45599365234375,
1266
+ "loss": 0.4918,
1267
+ "rewards/accuracies": 0.8125,
1268
+ "rewards/chosen": -0.14495661854743958,
1269
+ "rewards/margins": 1.4426281452178955,
1270
+ "rewards/rejected": -1.5875847339630127,
1271
+ "step": 880
1272
+ },
1273
+ {
1274
+ "epoch": 0.47,
1275
+ "learning_rate": 2.9697498545666084e-07,
1276
+ "logits/chosen": -2.406571388244629,
1277
+ "logits/rejected": -2.3264360427856445,
1278
+ "logps/chosen": -347.81866455078125,
1279
+ "logps/rejected": -290.57080078125,
1280
+ "loss": 0.5883,
1281
+ "rewards/accuracies": 0.75,
1282
+ "rewards/chosen": -0.660466730594635,
1283
+ "rewards/margins": 1.2641503810882568,
1284
+ "rewards/rejected": -1.924617052078247,
1285
+ "step": 890
1286
+ },
1287
+ {
1288
+ "epoch": 0.47,
1289
+ "learning_rate": 2.9406631762652705e-07,
1290
+ "logits/chosen": -2.4709229469299316,
1291
+ "logits/rejected": -2.382078170776367,
1292
+ "logps/chosen": -340.4996032714844,
1293
+ "logps/rejected": -315.2070617675781,
1294
+ "loss": 0.515,
1295
+ "rewards/accuracies": 0.8125,
1296
+ "rewards/chosen": -0.6704212427139282,
1297
+ "rewards/margins": 1.4002532958984375,
1298
+ "rewards/rejected": -2.0706748962402344,
1299
+ "step": 900
1300
+ },
1301
+ {
1302
+ "epoch": 0.48,
1303
+ "learning_rate": 2.9115764979639326e-07,
1304
+ "logits/chosen": -2.506743907928467,
1305
+ "logits/rejected": -2.4725024700164795,
1306
+ "logps/chosen": -340.2807922363281,
1307
+ "logps/rejected": -301.91082763671875,
1308
+ "loss": 0.4824,
1309
+ "rewards/accuracies": 0.762499988079071,
1310
+ "rewards/chosen": -0.5034831762313843,
1311
+ "rewards/margins": 1.6047446727752686,
1312
+ "rewards/rejected": -2.1082279682159424,
1313
+ "step": 910
1314
+ },
1315
+ {
1316
+ "epoch": 0.48,
1317
+ "learning_rate": 2.8824898196625947e-07,
1318
+ "logits/chosen": -2.446906805038452,
1319
+ "logits/rejected": -2.392117977142334,
1320
+ "logps/chosen": -301.62506103515625,
1321
+ "logps/rejected": -307.46258544921875,
1322
+ "loss": 0.4731,
1323
+ "rewards/accuracies": 0.800000011920929,
1324
+ "rewards/chosen": -0.13553249835968018,
1325
+ "rewards/margins": 1.2748148441314697,
1326
+ "rewards/rejected": -1.4103472232818604,
1327
+ "step": 920
1328
+ },
1329
+ {
1330
+ "epoch": 0.49,
1331
+ "learning_rate": 2.853403141361256e-07,
1332
+ "logits/chosen": -2.418687343597412,
1333
+ "logits/rejected": -2.3500733375549316,
1334
+ "logps/chosen": -301.64056396484375,
1335
+ "logps/rejected": -261.1069030761719,
1336
+ "loss": 0.5395,
1337
+ "rewards/accuracies": 0.762499988079071,
1338
+ "rewards/chosen": -0.5969820022583008,
1339
+ "rewards/margins": 0.8396062850952148,
1340
+ "rewards/rejected": -1.4365884065628052,
1341
+ "step": 930
1342
+ },
1343
+ {
1344
+ "epoch": 0.49,
1345
+ "learning_rate": 2.824316463059919e-07,
1346
+ "logits/chosen": -2.4292044639587402,
1347
+ "logits/rejected": -2.3925364017486572,
1348
+ "logps/chosen": -310.83258056640625,
1349
+ "logps/rejected": -304.67486572265625,
1350
+ "loss": 0.484,
1351
+ "rewards/accuracies": 0.75,
1352
+ "rewards/chosen": -0.4251963198184967,
1353
+ "rewards/margins": 1.4573180675506592,
1354
+ "rewards/rejected": -1.8825145959854126,
1355
+ "step": 940
1356
+ },
1357
+ {
1358
+ "epoch": 0.5,
1359
+ "learning_rate": 2.7952297847585803e-07,
1360
+ "logits/chosen": -2.417309284210205,
1361
+ "logits/rejected": -2.3521695137023926,
1362
+ "logps/chosen": -278.1476745605469,
1363
+ "logps/rejected": -281.66656494140625,
1364
+ "loss": 0.5149,
1365
+ "rewards/accuracies": 0.7875000238418579,
1366
+ "rewards/chosen": -0.28509074449539185,
1367
+ "rewards/margins": 1.083017110824585,
1368
+ "rewards/rejected": -1.3681080341339111,
1369
+ "step": 950
1370
+ },
1371
+ {
1372
+ "epoch": 0.5,
1373
+ "learning_rate": 2.766143106457243e-07,
1374
+ "logits/chosen": -2.3309614658355713,
1375
+ "logits/rejected": -2.3364787101745605,
1376
+ "logps/chosen": -259.80975341796875,
1377
+ "logps/rejected": -250.3785400390625,
1378
+ "loss": 0.5875,
1379
+ "rewards/accuracies": 0.625,
1380
+ "rewards/chosen": -0.18767808377742767,
1381
+ "rewards/margins": 1.1901742219924927,
1382
+ "rewards/rejected": -1.377852201461792,
1383
+ "step": 960
1384
+ },
1385
+ {
1386
+ "epoch": 0.51,
1387
+ "learning_rate": 2.7370564281559045e-07,
1388
+ "logits/chosen": -2.5228657722473145,
1389
+ "logits/rejected": -2.431464672088623,
1390
+ "logps/chosen": -346.1999816894531,
1391
+ "logps/rejected": -306.3692932128906,
1392
+ "loss": 0.5132,
1393
+ "rewards/accuracies": 0.7250000238418579,
1394
+ "rewards/chosen": -0.02729633077979088,
1395
+ "rewards/margins": 1.0325729846954346,
1396
+ "rewards/rejected": -1.0598691701889038,
1397
+ "step": 970
1398
+ },
1399
+ {
1400
+ "epoch": 0.51,
1401
+ "learning_rate": 2.7079697498545666e-07,
1402
+ "logits/chosen": -2.458400249481201,
1403
+ "logits/rejected": -2.4137959480285645,
1404
+ "logps/chosen": -322.91864013671875,
1405
+ "logps/rejected": -321.85699462890625,
1406
+ "loss": 0.477,
1407
+ "rewards/accuracies": 0.7124999761581421,
1408
+ "rewards/chosen": -0.24051785469055176,
1409
+ "rewards/margins": 1.2779700756072998,
1410
+ "rewards/rejected": -1.5184879302978516,
1411
+ "step": 980
1412
+ },
1413
+ {
1414
+ "epoch": 0.52,
1415
+ "learning_rate": 2.6788830715532287e-07,
1416
+ "logits/chosen": -2.5085196495056152,
1417
+ "logits/rejected": -2.441279888153076,
1418
+ "logps/chosen": -266.5360412597656,
1419
+ "logps/rejected": -265.48980712890625,
1420
+ "loss": 0.58,
1421
+ "rewards/accuracies": 0.6875,
1422
+ "rewards/chosen": -0.10649283230304718,
1423
+ "rewards/margins": 1.1308716535568237,
1424
+ "rewards/rejected": -1.2373645305633545,
1425
+ "step": 990
1426
+ },
1427
+ {
1428
+ "epoch": 0.52,
1429
+ "learning_rate": 2.6497963932518907e-07,
1430
+ "logits/chosen": -2.3860020637512207,
1431
+ "logits/rejected": -2.3790476322174072,
1432
+ "logps/chosen": -304.46270751953125,
1433
+ "logps/rejected": -309.70697021484375,
1434
+ "loss": 0.6191,
1435
+ "rewards/accuracies": 0.7250000238418579,
1436
+ "rewards/chosen": -0.43433743715286255,
1437
+ "rewards/margins": 0.8789669275283813,
1438
+ "rewards/rejected": -1.3133043050765991,
1439
+ "step": 1000
1440
+ },
1441
+ {
1442
+ "epoch": 0.52,
1443
+ "eval_logits/chosen": -2.490722894668579,
1444
+ "eval_logits/rejected": -2.427964210510254,
1445
+ "eval_logps/chosen": -300.2858581542969,
1446
+ "eval_logps/rejected": -286.9600524902344,
1447
+ "eval_loss": 0.5983877778053284,
1448
+ "eval_rewards/accuracies": 0.7599206566810608,
1449
+ "eval_rewards/chosen": 0.03812364488840103,
1450
+ "eval_rewards/margins": 1.0935115814208984,
1451
+ "eval_rewards/rejected": -1.0553878545761108,
1452
+ "eval_runtime": 614.9522,
1453
+ "eval_samples_per_second": 3.252,
1454
+ "eval_steps_per_second": 0.102,
1455
+ "step": 1000
1456
+ },
1457
+ {
1458
+ "epoch": 0.53,
1459
+ "learning_rate": 2.6207097149505523e-07,
1460
+ "logits/chosen": -2.511011838912964,
1461
+ "logits/rejected": -2.427821159362793,
1462
+ "logps/chosen": -329.0254821777344,
1463
+ "logps/rejected": -263.37030029296875,
1464
+ "loss": 0.529,
1465
+ "rewards/accuracies": 0.7875000238418579,
1466
+ "rewards/chosen": 0.06932397931814194,
1467
+ "rewards/margins": 1.145455002784729,
1468
+ "rewards/rejected": -1.0761311054229736,
1469
+ "step": 1010
1470
+ },
1471
+ {
1472
+ "epoch": 0.53,
1473
+ "learning_rate": 2.591623036649215e-07,
1474
+ "logits/chosen": -2.5254805088043213,
1475
+ "logits/rejected": -2.4318673610687256,
1476
+ "logps/chosen": -304.84466552734375,
1477
+ "logps/rejected": -272.1488952636719,
1478
+ "loss": 0.4663,
1479
+ "rewards/accuracies": 0.8374999761581421,
1480
+ "rewards/chosen": -0.25631338357925415,
1481
+ "rewards/margins": 1.2234864234924316,
1482
+ "rewards/rejected": -1.4797999858856201,
1483
+ "step": 1020
1484
+ },
1485
+ {
1486
+ "epoch": 0.54,
1487
+ "learning_rate": 2.5625363583478764e-07,
1488
+ "logits/chosen": -2.4320321083068848,
1489
+ "logits/rejected": -2.407522201538086,
1490
+ "logps/chosen": -285.1934814453125,
1491
+ "logps/rejected": -312.1936950683594,
1492
+ "loss": 0.4631,
1493
+ "rewards/accuracies": 0.75,
1494
+ "rewards/chosen": -0.12434335052967072,
1495
+ "rewards/margins": 1.4510531425476074,
1496
+ "rewards/rejected": -1.5753967761993408,
1497
+ "step": 1030
1498
+ },
1499
+ {
1500
+ "epoch": 0.54,
1501
+ "learning_rate": 2.533449680046539e-07,
1502
+ "logits/chosen": -2.480064630508423,
1503
+ "logits/rejected": -2.439182758331299,
1504
+ "logps/chosen": -309.5289001464844,
1505
+ "logps/rejected": -296.3352355957031,
1506
+ "loss": 0.5437,
1507
+ "rewards/accuracies": 0.7875000238418579,
1508
+ "rewards/chosen": -0.2124924659729004,
1509
+ "rewards/margins": 1.4312546253204346,
1510
+ "rewards/rejected": -1.643747091293335,
1511
+ "step": 1040
1512
+ },
1513
+ {
1514
+ "epoch": 0.55,
1515
+ "learning_rate": 2.5043630017452006e-07,
1516
+ "logits/chosen": -2.41044020652771,
1517
+ "logits/rejected": -2.3936314582824707,
1518
+ "logps/chosen": -284.94268798828125,
1519
+ "logps/rejected": -283.9359436035156,
1520
+ "loss": 0.5227,
1521
+ "rewards/accuracies": 0.7250000238418579,
1522
+ "rewards/chosen": 0.03999950736761093,
1523
+ "rewards/margins": 1.149482011795044,
1524
+ "rewards/rejected": -1.1094822883605957,
1525
+ "step": 1050
1526
+ },
1527
+ {
1528
+ "epoch": 0.55,
1529
+ "learning_rate": 2.4752763234438627e-07,
1530
+ "logits/chosen": -2.430999279022217,
1531
+ "logits/rejected": -2.3697829246520996,
1532
+ "logps/chosen": -253.0732421875,
1533
+ "logps/rejected": -247.62130737304688,
1534
+ "loss": 0.4938,
1535
+ "rewards/accuracies": 0.7749999761581421,
1536
+ "rewards/chosen": 0.35475850105285645,
1537
+ "rewards/margins": 1.3529800176620483,
1538
+ "rewards/rejected": -0.9982213973999023,
1539
+ "step": 1060
1540
+ },
1541
+ {
1542
+ "epoch": 0.56,
1543
+ "learning_rate": 2.4461896451425247e-07,
1544
+ "logits/chosen": -2.4376707077026367,
1545
+ "logits/rejected": -2.4240877628326416,
1546
+ "logps/chosen": -269.42938232421875,
1547
+ "logps/rejected": -270.54644775390625,
1548
+ "loss": 0.4648,
1549
+ "rewards/accuracies": 0.7875000238418579,
1550
+ "rewards/chosen": -0.1738893836736679,
1551
+ "rewards/margins": 1.2686713933944702,
1552
+ "rewards/rejected": -1.4425609111785889,
1553
+ "step": 1070
1554
+ },
1555
+ {
1556
+ "epoch": 0.57,
1557
+ "learning_rate": 2.417102966841187e-07,
1558
+ "logits/chosen": -2.458761692047119,
1559
+ "logits/rejected": -2.3944101333618164,
1560
+ "logps/chosen": -271.7908935546875,
1561
+ "logps/rejected": -296.3941345214844,
1562
+ "loss": 0.5304,
1563
+ "rewards/accuracies": 0.7250000238418579,
1564
+ "rewards/chosen": -0.14974147081375122,
1565
+ "rewards/margins": 1.2848742008209229,
1566
+ "rewards/rejected": -1.4346158504486084,
1567
+ "step": 1080
1568
+ },
1569
+ {
1570
+ "epoch": 0.57,
1571
+ "learning_rate": 2.3880162885398483e-07,
1572
+ "logits/chosen": -2.4826645851135254,
1573
+ "logits/rejected": -2.375396251678467,
1574
+ "logps/chosen": -299.49560546875,
1575
+ "logps/rejected": -252.5529327392578,
1576
+ "loss": 0.532,
1577
+ "rewards/accuracies": 0.7749999761581421,
1578
+ "rewards/chosen": -0.49217844009399414,
1579
+ "rewards/margins": 1.3745396137237549,
1580
+ "rewards/rejected": -1.8667182922363281,
1581
+ "step": 1090
1582
+ },
1583
+ {
1584
+ "epoch": 0.58,
1585
+ "learning_rate": 2.3589296102385107e-07,
1586
+ "logits/chosen": -2.499211072921753,
1587
+ "logits/rejected": -2.417027235031128,
1588
+ "logps/chosen": -299.8446044921875,
1589
+ "logps/rejected": -286.61322021484375,
1590
+ "loss": 0.4399,
1591
+ "rewards/accuracies": 0.7749999761581421,
1592
+ "rewards/chosen": -0.06040573865175247,
1593
+ "rewards/margins": 1.4158607721328735,
1594
+ "rewards/rejected": -1.4762665033340454,
1595
+ "step": 1100
1596
+ },
1597
+ {
1598
+ "epoch": 0.58,
1599
+ "learning_rate": 2.3298429319371725e-07,
1600
+ "logits/chosen": -2.5318305492401123,
1601
+ "logits/rejected": -2.441438674926758,
1602
+ "logps/chosen": -301.87762451171875,
1603
+ "logps/rejected": -303.8865051269531,
1604
+ "loss": 0.4871,
1605
+ "rewards/accuracies": 0.7250000238418579,
1606
+ "rewards/chosen": -0.34866777062416077,
1607
+ "rewards/margins": 1.076101541519165,
1608
+ "rewards/rejected": -1.4247692823410034,
1609
+ "step": 1110
1610
+ },
1611
+ {
1612
+ "epoch": 0.59,
1613
+ "learning_rate": 2.3007562536358346e-07,
1614
+ "logits/chosen": -2.478132486343384,
1615
+ "logits/rejected": -2.420320749282837,
1616
+ "logps/chosen": -304.8691711425781,
1617
+ "logps/rejected": -313.31951904296875,
1618
+ "loss": 0.5539,
1619
+ "rewards/accuracies": 0.7749999761581421,
1620
+ "rewards/chosen": -0.0543731153011322,
1621
+ "rewards/margins": 1.4534409046173096,
1622
+ "rewards/rejected": -1.507813811302185,
1623
+ "step": 1120
1624
+ },
1625
+ {
1626
+ "epoch": 0.59,
1627
+ "learning_rate": 2.2716695753344967e-07,
1628
+ "logits/chosen": -2.452592372894287,
1629
+ "logits/rejected": -2.422971487045288,
1630
+ "logps/chosen": -301.95721435546875,
1631
+ "logps/rejected": -311.69268798828125,
1632
+ "loss": 0.5074,
1633
+ "rewards/accuracies": 0.800000011920929,
1634
+ "rewards/chosen": -0.30631425976753235,
1635
+ "rewards/margins": 1.4355709552764893,
1636
+ "rewards/rejected": -1.7418854236602783,
1637
+ "step": 1130
1638
+ },
1639
+ {
1640
+ "epoch": 0.6,
1641
+ "learning_rate": 2.2425828970331587e-07,
1642
+ "logits/chosen": -2.5267157554626465,
1643
+ "logits/rejected": -2.4898681640625,
1644
+ "logps/chosen": -267.6355285644531,
1645
+ "logps/rejected": -277.17376708984375,
1646
+ "loss": 0.5519,
1647
+ "rewards/accuracies": 0.75,
1648
+ "rewards/chosen": -0.5024998784065247,
1649
+ "rewards/margins": 1.3192273378372192,
1650
+ "rewards/rejected": -1.8217273950576782,
1651
+ "step": 1140
1652
+ },
1653
+ {
1654
+ "epoch": 0.6,
1655
+ "learning_rate": 2.2134962187318208e-07,
1656
+ "logits/chosen": -2.405796527862549,
1657
+ "logits/rejected": -2.327157497406006,
1658
+ "logps/chosen": -305.180908203125,
1659
+ "logps/rejected": -281.836669921875,
1660
+ "loss": 0.5686,
1661
+ "rewards/accuracies": 0.7124999761581421,
1662
+ "rewards/chosen": -0.9660147428512573,
1663
+ "rewards/margins": 1.525781512260437,
1664
+ "rewards/rejected": -2.4917960166931152,
1665
+ "step": 1150
1666
+ },
1667
+ {
1668
+ "epoch": 0.61,
1669
+ "learning_rate": 2.1844095404304826e-07,
1670
+ "logits/chosen": -2.4776437282562256,
1671
+ "logits/rejected": -2.430321455001831,
1672
+ "logps/chosen": -330.463623046875,
1673
+ "logps/rejected": -324.64324951171875,
1674
+ "loss": 0.6433,
1675
+ "rewards/accuracies": 0.550000011920929,
1676
+ "rewards/chosen": -1.0749146938323975,
1677
+ "rewards/margins": 0.4202847480773926,
1678
+ "rewards/rejected": -1.4951995611190796,
1679
+ "step": 1160
1680
+ },
1681
+ {
1682
+ "epoch": 0.61,
1683
+ "learning_rate": 2.1553228621291447e-07,
1684
+ "logits/chosen": -2.4651365280151367,
1685
+ "logits/rejected": -2.394700527191162,
1686
+ "logps/chosen": -286.38763427734375,
1687
+ "logps/rejected": -294.4740295410156,
1688
+ "loss": 0.5214,
1689
+ "rewards/accuracies": 0.6875,
1690
+ "rewards/chosen": -0.9769517779350281,
1691
+ "rewards/margins": 0.8809939622879028,
1692
+ "rewards/rejected": -1.857945442199707,
1693
+ "step": 1170
1694
+ },
1695
+ {
1696
+ "epoch": 0.62,
1697
+ "learning_rate": 2.1262361838278068e-07,
1698
+ "logits/chosen": -2.4773755073547363,
1699
+ "logits/rejected": -2.4409890174865723,
1700
+ "logps/chosen": -308.72369384765625,
1701
+ "logps/rejected": -298.4217224121094,
1702
+ "loss": 0.4998,
1703
+ "rewards/accuracies": 0.762499988079071,
1704
+ "rewards/chosen": -0.7563016414642334,
1705
+ "rewards/margins": 1.2209903001785278,
1706
+ "rewards/rejected": -1.9772918224334717,
1707
+ "step": 1180
1708
+ },
1709
+ {
1710
+ "epoch": 0.62,
1711
+ "learning_rate": 2.0971495055264688e-07,
1712
+ "logits/chosen": -2.4952621459960938,
1713
+ "logits/rejected": -2.4689629077911377,
1714
+ "logps/chosen": -291.19232177734375,
1715
+ "logps/rejected": -306.8577575683594,
1716
+ "loss": 0.5187,
1717
+ "rewards/accuracies": 0.7250000238418579,
1718
+ "rewards/chosen": -0.24586816132068634,
1719
+ "rewards/margins": 1.1213955879211426,
1720
+ "rewards/rejected": -1.3672637939453125,
1721
+ "step": 1190
1722
+ },
1723
+ {
1724
+ "epoch": 0.63,
1725
+ "learning_rate": 2.0680628272251307e-07,
1726
+ "logits/chosen": -2.509241819381714,
1727
+ "logits/rejected": -2.4504573345184326,
1728
+ "logps/chosen": -285.4134826660156,
1729
+ "logps/rejected": -277.93658447265625,
1730
+ "loss": 0.5123,
1731
+ "rewards/accuracies": 0.7250000238418579,
1732
+ "rewards/chosen": -0.3394593298435211,
1733
+ "rewards/margins": 1.0499706268310547,
1734
+ "rewards/rejected": -1.3894299268722534,
1735
+ "step": 1200
1736
+ },
1737
+ {
1738
+ "epoch": 0.63,
1739
+ "learning_rate": 2.0389761489237927e-07,
1740
+ "logits/chosen": -2.5037548542022705,
1741
+ "logits/rejected": -2.4983646869659424,
1742
+ "logps/chosen": -298.67108154296875,
1743
+ "logps/rejected": -290.48687744140625,
1744
+ "loss": 0.5539,
1745
+ "rewards/accuracies": 0.6499999761581421,
1746
+ "rewards/chosen": -0.12395872920751572,
1747
+ "rewards/margins": 1.0312166213989258,
1748
+ "rewards/rejected": -1.1551753282546997,
1749
+ "step": 1210
1750
+ },
1751
+ {
1752
+ "epoch": 0.64,
1753
+ "learning_rate": 2.0098894706224548e-07,
1754
+ "logits/chosen": -2.514279842376709,
1755
+ "logits/rejected": -2.473151683807373,
1756
+ "logps/chosen": -255.2734375,
1757
+ "logps/rejected": -252.98855590820312,
1758
+ "loss": 0.4852,
1759
+ "rewards/accuracies": 0.737500011920929,
1760
+ "rewards/chosen": -0.47876453399658203,
1761
+ "rewards/margins": 1.0464788675308228,
1762
+ "rewards/rejected": -1.5252434015274048,
1763
+ "step": 1220
1764
+ },
1765
+ {
1766
+ "epoch": 0.64,
1767
+ "learning_rate": 1.980802792321117e-07,
1768
+ "logits/chosen": -2.471043825149536,
1769
+ "logits/rejected": -2.437826156616211,
1770
+ "logps/chosen": -336.73712158203125,
1771
+ "logps/rejected": -337.2440490722656,
1772
+ "loss": 0.5463,
1773
+ "rewards/accuracies": 0.6875,
1774
+ "rewards/chosen": -0.4187423586845398,
1775
+ "rewards/margins": 1.326642394065857,
1776
+ "rewards/rejected": -1.7453848123550415,
1777
+ "step": 1230
1778
+ },
1779
+ {
1780
+ "epoch": 0.65,
1781
+ "learning_rate": 1.9517161140197787e-07,
1782
+ "logits/chosen": -2.5959396362304688,
1783
+ "logits/rejected": -2.5564401149749756,
1784
+ "logps/chosen": -309.3214111328125,
1785
+ "logps/rejected": -300.98138427734375,
1786
+ "loss": 0.5264,
1787
+ "rewards/accuracies": 0.675000011920929,
1788
+ "rewards/chosen": -0.3669760823249817,
1789
+ "rewards/margins": 1.1876031160354614,
1790
+ "rewards/rejected": -1.5545791387557983,
1791
+ "step": 1240
1792
+ },
1793
+ {
1794
+ "epoch": 0.65,
1795
+ "learning_rate": 1.9226294357184408e-07,
1796
+ "logits/chosen": -2.498528480529785,
1797
+ "logits/rejected": -2.4267897605895996,
1798
+ "logps/chosen": -316.4427185058594,
1799
+ "logps/rejected": -322.12493896484375,
1800
+ "loss": 0.5139,
1801
+ "rewards/accuracies": 0.737500011920929,
1802
+ "rewards/chosen": -0.0588008277118206,
1803
+ "rewards/margins": 1.484725832939148,
1804
+ "rewards/rejected": -1.5435266494750977,
1805
+ "step": 1250
1806
+ },
1807
+ {
1808
+ "epoch": 0.66,
1809
+ "learning_rate": 1.8935427574171028e-07,
1810
+ "logits/chosen": -2.542093515396118,
1811
+ "logits/rejected": -2.4682421684265137,
1812
+ "logps/chosen": -337.3454284667969,
1813
+ "logps/rejected": -299.1844482421875,
1814
+ "loss": 0.5489,
1815
+ "rewards/accuracies": 0.7250000238418579,
1816
+ "rewards/chosen": -0.02595728635787964,
1817
+ "rewards/margins": 1.281512975692749,
1818
+ "rewards/rejected": -1.307470440864563,
1819
+ "step": 1260
1820
+ },
1821
+ {
1822
+ "epoch": 0.66,
1823
+ "learning_rate": 1.864456079115765e-07,
1824
+ "logits/chosen": -2.477086067199707,
1825
+ "logits/rejected": -2.457650661468506,
1826
+ "logps/chosen": -268.6798095703125,
1827
+ "logps/rejected": -257.23358154296875,
1828
+ "loss": 0.4939,
1829
+ "rewards/accuracies": 0.7875000238418579,
1830
+ "rewards/chosen": -0.1567353755235672,
1831
+ "rewards/margins": 1.2585152387619019,
1832
+ "rewards/rejected": -1.415250539779663,
1833
+ "step": 1270
1834
+ },
1835
+ {
1836
+ "epoch": 0.67,
1837
+ "learning_rate": 1.835369400814427e-07,
1838
+ "logits/chosen": -2.544891595840454,
1839
+ "logits/rejected": -2.4903788566589355,
1840
+ "logps/chosen": -289.84234619140625,
1841
+ "logps/rejected": -263.73297119140625,
1842
+ "loss": 0.4819,
1843
+ "rewards/accuracies": 0.7875000238418579,
1844
+ "rewards/chosen": -0.07888557016849518,
1845
+ "rewards/margins": 1.23806893825531,
1846
+ "rewards/rejected": -1.3169543743133545,
1847
+ "step": 1280
1848
+ },
1849
+ {
1850
+ "epoch": 0.68,
1851
+ "learning_rate": 1.8062827225130888e-07,
1852
+ "logits/chosen": -2.4458956718444824,
1853
+ "logits/rejected": -2.36130428314209,
1854
+ "logps/chosen": -296.7621765136719,
1855
+ "logps/rejected": -268.7115173339844,
1856
+ "loss": 0.5193,
1857
+ "rewards/accuracies": 0.675000011920929,
1858
+ "rewards/chosen": -0.21038198471069336,
1859
+ "rewards/margins": 0.9879549741744995,
1860
+ "rewards/rejected": -1.1983369588851929,
1861
+ "step": 1290
1862
+ },
1863
+ {
1864
+ "epoch": 0.68,
1865
+ "learning_rate": 1.777196044211751e-07,
1866
+ "logits/chosen": -2.422639846801758,
1867
+ "logits/rejected": -2.3881583213806152,
1868
+ "logps/chosen": -267.2961730957031,
1869
+ "logps/rejected": -258.95159912109375,
1870
+ "loss": 0.5247,
1871
+ "rewards/accuracies": 0.800000011920929,
1872
+ "rewards/chosen": -0.0472625195980072,
1873
+ "rewards/margins": 1.1689708232879639,
1874
+ "rewards/rejected": -1.216233491897583,
1875
+ "step": 1300
1876
+ },
1877
+ {
1878
+ "epoch": 0.69,
1879
+ "learning_rate": 1.748109365910413e-07,
1880
+ "logits/chosen": -2.4238734245300293,
1881
+ "logits/rejected": -2.3859925270080566,
1882
+ "logps/chosen": -241.4824981689453,
1883
+ "logps/rejected": -256.94635009765625,
1884
+ "loss": 0.471,
1885
+ "rewards/accuracies": 0.7875000238418579,
1886
+ "rewards/chosen": 0.039008866995573044,
1887
+ "rewards/margins": 1.5137543678283691,
1888
+ "rewards/rejected": -1.474745512008667,
1889
+ "step": 1310
1890
+ },
1891
+ {
1892
+ "epoch": 0.69,
1893
+ "learning_rate": 1.719022687609075e-07,
1894
+ "logits/chosen": -2.5132765769958496,
1895
+ "logits/rejected": -2.449302911758423,
1896
+ "logps/chosen": -363.523681640625,
1897
+ "logps/rejected": -352.0708923339844,
1898
+ "loss": 0.518,
1899
+ "rewards/accuracies": 0.7749999761581421,
1900
+ "rewards/chosen": -0.5855492353439331,
1901
+ "rewards/margins": 1.2664353847503662,
1902
+ "rewards/rejected": -1.8519846200942993,
1903
+ "step": 1320
1904
+ },
1905
+ {
1906
+ "epoch": 0.7,
1907
+ "learning_rate": 1.6899360093077368e-07,
1908
+ "logits/chosen": -2.4553329944610596,
1909
+ "logits/rejected": -2.4112277030944824,
1910
+ "logps/chosen": -280.7616882324219,
1911
+ "logps/rejected": -236.74765014648438,
1912
+ "loss": 0.5061,
1913
+ "rewards/accuracies": 0.737500011920929,
1914
+ "rewards/chosen": -0.48466506600379944,
1915
+ "rewards/margins": 1.0938060283660889,
1916
+ "rewards/rejected": -1.578471064567566,
1917
+ "step": 1330
1918
+ },
1919
+ {
1920
+ "epoch": 0.7,
1921
+ "learning_rate": 1.660849331006399e-07,
1922
+ "logits/chosen": -2.47243070602417,
1923
+ "logits/rejected": -2.448956251144409,
1924
+ "logps/chosen": -347.1180114746094,
1925
+ "logps/rejected": -303.9145202636719,
1926
+ "loss": 0.5364,
1927
+ "rewards/accuracies": 0.824999988079071,
1928
+ "rewards/chosen": -0.2242584526538849,
1929
+ "rewards/margins": 1.3924510478973389,
1930
+ "rewards/rejected": -1.6167097091674805,
1931
+ "step": 1340
1932
+ },
1933
+ {
1934
+ "epoch": 0.71,
1935
+ "learning_rate": 1.631762652705061e-07,
1936
+ "logits/chosen": -2.430530071258545,
1937
+ "logits/rejected": -2.4229390621185303,
1938
+ "logps/chosen": -245.01284790039062,
1939
+ "logps/rejected": -268.6424560546875,
1940
+ "loss": 0.5447,
1941
+ "rewards/accuracies": 0.699999988079071,
1942
+ "rewards/chosen": -0.5738204121589661,
1943
+ "rewards/margins": 1.0651990175247192,
1944
+ "rewards/rejected": -1.6390196084976196,
1945
+ "step": 1350
1946
+ },
1947
+ {
1948
+ "epoch": 0.71,
1949
+ "learning_rate": 1.602675974403723e-07,
1950
+ "logits/chosen": -2.509038209915161,
1951
+ "logits/rejected": -2.489039897918701,
1952
+ "logps/chosen": -279.32379150390625,
1953
+ "logps/rejected": -294.4464416503906,
1954
+ "loss": 0.5021,
1955
+ "rewards/accuracies": 0.699999988079071,
1956
+ "rewards/chosen": -0.5361760854721069,
1957
+ "rewards/margins": 1.5468828678131104,
1958
+ "rewards/rejected": -2.0830588340759277,
1959
+ "step": 1360
1960
+ },
1961
+ {
1962
+ "epoch": 0.72,
1963
+ "learning_rate": 1.573589296102385e-07,
1964
+ "logits/chosen": -2.47121262550354,
1965
+ "logits/rejected": -2.3763961791992188,
1966
+ "logps/chosen": -295.9562072753906,
1967
+ "logps/rejected": -312.29022216796875,
1968
+ "loss": 0.5798,
1969
+ "rewards/accuracies": 0.737500011920929,
1970
+ "rewards/chosen": -0.7061088681221008,
1971
+ "rewards/margins": 0.6994223594665527,
1972
+ "rewards/rejected": -1.4055311679840088,
1973
+ "step": 1370
1974
+ },
1975
+ {
1976
+ "epoch": 0.72,
1977
+ "learning_rate": 1.544502617801047e-07,
1978
+ "logits/chosen": -2.4963533878326416,
1979
+ "logits/rejected": -2.4046132564544678,
1980
+ "logps/chosen": -306.95379638671875,
1981
+ "logps/rejected": -269.7267150878906,
1982
+ "loss": 0.5349,
1983
+ "rewards/accuracies": 0.762499988079071,
1984
+ "rewards/chosen": -0.14952921867370605,
1985
+ "rewards/margins": 1.085407018661499,
1986
+ "rewards/rejected": -1.2349363565444946,
1987
+ "step": 1380
1988
+ },
1989
+ {
1990
+ "epoch": 0.73,
1991
+ "learning_rate": 1.515415939499709e-07,
1992
+ "logits/chosen": -2.3877298831939697,
1993
+ "logits/rejected": -2.3528621196746826,
1994
+ "logps/chosen": -306.18292236328125,
1995
+ "logps/rejected": -281.7127990722656,
1996
+ "loss": 0.4883,
1997
+ "rewards/accuracies": 0.824999988079071,
1998
+ "rewards/chosen": -0.3554263710975647,
1999
+ "rewards/margins": 1.3783906698226929,
2000
+ "rewards/rejected": -1.7338171005249023,
2001
+ "step": 1390
2002
+ },
2003
+ {
2004
+ "epoch": 0.73,
2005
+ "learning_rate": 1.486329261198371e-07,
2006
+ "logits/chosen": -2.4248931407928467,
2007
+ "logits/rejected": -2.3428282737731934,
2008
+ "logps/chosen": -265.8216247558594,
2009
+ "logps/rejected": -254.1871795654297,
2010
+ "loss": 0.5616,
2011
+ "rewards/accuracies": 0.6499999761581421,
2012
+ "rewards/chosen": -0.4705156683921814,
2013
+ "rewards/margins": 1.1333085298538208,
2014
+ "rewards/rejected": -1.6038239002227783,
2015
+ "step": 1400
2016
+ },
2017
+ {
2018
+ "epoch": 0.74,
2019
+ "learning_rate": 1.4572425828970332e-07,
2020
+ "logits/chosen": -2.5125279426574707,
2021
+ "logits/rejected": -2.4069325923919678,
2022
+ "logps/chosen": -317.7018127441406,
2023
+ "logps/rejected": -303.3476867675781,
2024
+ "loss": 0.4968,
2025
+ "rewards/accuracies": 0.6875,
2026
+ "rewards/chosen": -0.4776553511619568,
2027
+ "rewards/margins": 1.0890066623687744,
2028
+ "rewards/rejected": -1.566662073135376,
2029
+ "step": 1410
2030
+ },
2031
+ {
2032
+ "epoch": 0.74,
2033
+ "learning_rate": 1.428155904595695e-07,
2034
+ "logits/chosen": -2.402710437774658,
2035
+ "logits/rejected": -2.35219407081604,
2036
+ "logps/chosen": -308.2895202636719,
2037
+ "logps/rejected": -298.3793640136719,
2038
+ "loss": 0.5636,
2039
+ "rewards/accuracies": 0.737500011920929,
2040
+ "rewards/chosen": -0.43602848052978516,
2041
+ "rewards/margins": 1.1804325580596924,
2042
+ "rewards/rejected": -1.6164608001708984,
2043
+ "step": 1420
2044
+ },
2045
+ {
2046
+ "epoch": 0.75,
2047
+ "learning_rate": 1.399069226294357e-07,
2048
+ "logits/chosen": -2.325125217437744,
2049
+ "logits/rejected": -2.265918254852295,
2050
+ "logps/chosen": -255.18252563476562,
2051
+ "logps/rejected": -242.65451049804688,
2052
+ "loss": 0.5112,
2053
+ "rewards/accuracies": 0.7749999761581421,
2054
+ "rewards/chosen": -0.38456660509109497,
2055
+ "rewards/margins": 1.3007227182388306,
2056
+ "rewards/rejected": -1.6852893829345703,
2057
+ "step": 1430
2058
+ },
2059
+ {
2060
+ "epoch": 0.75,
2061
+ "learning_rate": 1.3699825479930191e-07,
2062
+ "logits/chosen": -2.444197177886963,
2063
+ "logits/rejected": -2.3448662757873535,
2064
+ "logps/chosen": -335.58575439453125,
2065
+ "logps/rejected": -257.6363525390625,
2066
+ "loss": 0.4915,
2067
+ "rewards/accuracies": 0.75,
2068
+ "rewards/chosen": -0.1254938691854477,
2069
+ "rewards/margins": 1.4525253772735596,
2070
+ "rewards/rejected": -1.578019380569458,
2071
+ "step": 1440
2072
+ },
2073
+ {
2074
+ "epoch": 0.76,
2075
+ "learning_rate": 1.3408958696916812e-07,
2076
+ "logits/chosen": -2.432386875152588,
2077
+ "logits/rejected": -2.3468270301818848,
2078
+ "logps/chosen": -264.3630065917969,
2079
+ "logps/rejected": -253.1586456298828,
2080
+ "loss": 0.4865,
2081
+ "rewards/accuracies": 0.800000011920929,
2082
+ "rewards/chosen": -0.010800900869071484,
2083
+ "rewards/margins": 1.3583967685699463,
2084
+ "rewards/rejected": -1.3691976070404053,
2085
+ "step": 1450
2086
+ },
2087
+ {
2088
+ "epoch": 0.76,
2089
+ "learning_rate": 1.311809191390343e-07,
2090
+ "logits/chosen": -2.3925037384033203,
2091
+ "logits/rejected": -2.3687844276428223,
2092
+ "logps/chosen": -298.83160400390625,
2093
+ "logps/rejected": -321.2250671386719,
2094
+ "loss": 0.5522,
2095
+ "rewards/accuracies": 0.7124999761581421,
2096
+ "rewards/chosen": -0.13408958911895752,
2097
+ "rewards/margins": 1.2063826322555542,
2098
+ "rewards/rejected": -1.3404719829559326,
2099
+ "step": 1460
2100
+ },
2101
+ {
2102
+ "epoch": 0.77,
2103
+ "learning_rate": 1.282722513089005e-07,
2104
+ "logits/chosen": -2.5323238372802734,
2105
+ "logits/rejected": -2.450490713119507,
2106
+ "logps/chosen": -303.0655517578125,
2107
+ "logps/rejected": -273.21636962890625,
2108
+ "loss": 0.4542,
2109
+ "rewards/accuracies": 0.699999988079071,
2110
+ "rewards/chosen": -0.07529834657907486,
2111
+ "rewards/margins": 1.215362787246704,
2112
+ "rewards/rejected": -1.290661096572876,
2113
+ "step": 1470
2114
+ },
2115
+ {
2116
+ "epoch": 0.77,
2117
+ "learning_rate": 1.2536358347876672e-07,
2118
+ "logits/chosen": -2.4639523029327393,
2119
+ "logits/rejected": -2.451842784881592,
2120
+ "logps/chosen": -273.89410400390625,
2121
+ "logps/rejected": -288.24395751953125,
2122
+ "loss": 0.5013,
2123
+ "rewards/accuracies": 0.7250000238418579,
2124
+ "rewards/chosen": -0.1511283814907074,
2125
+ "rewards/margins": 1.046019196510315,
2126
+ "rewards/rejected": -1.1971476078033447,
2127
+ "step": 1480
2128
+ },
2129
+ {
2130
+ "epoch": 0.78,
2131
+ "learning_rate": 1.2245491564863293e-07,
2132
+ "logits/chosen": -2.5214879512786865,
2133
+ "logits/rejected": -2.3725059032440186,
2134
+ "logps/chosen": -365.9342041015625,
2135
+ "logps/rejected": -288.50860595703125,
2136
+ "loss": 0.4655,
2137
+ "rewards/accuracies": 0.8500000238418579,
2138
+ "rewards/chosen": 0.24411487579345703,
2139
+ "rewards/margins": 1.6229133605957031,
2140
+ "rewards/rejected": -1.3787983655929565,
2141
+ "step": 1490
2142
+ },
2143
+ {
2144
+ "epoch": 0.78,
2145
+ "learning_rate": 1.195462478184991e-07,
2146
+ "logits/chosen": -2.4407520294189453,
2147
+ "logits/rejected": -2.403568744659424,
2148
+ "logps/chosen": -266.7708740234375,
2149
+ "logps/rejected": -256.9794006347656,
2150
+ "loss": 0.5342,
2151
+ "rewards/accuracies": 0.6875,
2152
+ "rewards/chosen": 0.008794727735221386,
2153
+ "rewards/margins": 1.0573989152908325,
2154
+ "rewards/rejected": -1.0486040115356445,
2155
+ "step": 1500
2156
+ },
2157
+ {
2158
+ "epoch": 0.78,
2159
+ "eval_logits/chosen": -2.479328155517578,
2160
+ "eval_logits/rejected": -2.420834541320801,
2161
+ "eval_logps/chosen": -299.6731872558594,
2162
+ "eval_logps/rejected": -287.96630859375,
2163
+ "eval_loss": 0.5819299817085266,
2164
+ "eval_rewards/accuracies": 0.7559523582458496,
2165
+ "eval_rewards/chosen": 0.09938977658748627,
2166
+ "eval_rewards/margins": 1.2554062604904175,
2167
+ "eval_rewards/rejected": -1.1560163497924805,
2168
+ "eval_runtime": 614.7519,
2169
+ "eval_samples_per_second": 3.253,
2170
+ "eval_steps_per_second": 0.102,
2171
+ "step": 1500
2172
+ },
2173
+ {
2174
+ "epoch": 0.79,
2175
+ "learning_rate": 1.1663757998836531e-07,
2176
+ "logits/chosen": -2.4385485649108887,
2177
+ "logits/rejected": -2.414762496948242,
2178
+ "logps/chosen": -282.46331787109375,
2179
+ "logps/rejected": -271.0342712402344,
2180
+ "loss": 0.5393,
2181
+ "rewards/accuracies": 0.7124999761581421,
2182
+ "rewards/chosen": 0.030192693695425987,
2183
+ "rewards/margins": 1.1163215637207031,
2184
+ "rewards/rejected": -1.0861289501190186,
2185
+ "step": 1510
2186
+ },
2187
+ {
2188
+ "epoch": 0.8,
2189
+ "learning_rate": 1.1372891215823152e-07,
2190
+ "logits/chosen": -2.475036144256592,
2191
+ "logits/rejected": -2.431018352508545,
2192
+ "logps/chosen": -273.4059753417969,
2193
+ "logps/rejected": -279.97149658203125,
2194
+ "loss": 0.4813,
2195
+ "rewards/accuracies": 0.737500011920929,
2196
+ "rewards/chosen": -0.07051552832126617,
2197
+ "rewards/margins": 1.225347638130188,
2198
+ "rewards/rejected": -1.295863151550293,
2199
+ "step": 1520
2200
+ },
2201
+ {
2202
+ "epoch": 0.8,
2203
+ "learning_rate": 1.1082024432809772e-07,
2204
+ "logits/chosen": -2.434101104736328,
2205
+ "logits/rejected": -2.3855373859405518,
2206
+ "logps/chosen": -270.7154235839844,
2207
+ "logps/rejected": -257.188720703125,
2208
+ "loss": 0.4726,
2209
+ "rewards/accuracies": 0.8125,
2210
+ "rewards/chosen": -0.03814787417650223,
2211
+ "rewards/margins": 1.2096832990646362,
2212
+ "rewards/rejected": -1.2478312253952026,
2213
+ "step": 1530
2214
+ },
2215
+ {
2216
+ "epoch": 0.81,
2217
+ "learning_rate": 1.0791157649796392e-07,
2218
+ "logits/chosen": -2.509958267211914,
2219
+ "logits/rejected": -2.5057830810546875,
2220
+ "logps/chosen": -268.060791015625,
2221
+ "logps/rejected": -294.9040832519531,
2222
+ "loss": 0.6052,
2223
+ "rewards/accuracies": 0.675000011920929,
2224
+ "rewards/chosen": -0.09406580030918121,
2225
+ "rewards/margins": 0.9256828427314758,
2226
+ "rewards/rejected": -1.019748568534851,
2227
+ "step": 1540
2228
+ },
2229
+ {
2230
+ "epoch": 0.81,
2231
+ "learning_rate": 1.0500290866783013e-07,
2232
+ "logits/chosen": -2.391065835952759,
2233
+ "logits/rejected": -2.3701300621032715,
2234
+ "logps/chosen": -280.59686279296875,
2235
+ "logps/rejected": -243.96798706054688,
2236
+ "loss": 0.5019,
2237
+ "rewards/accuracies": 0.699999988079071,
2238
+ "rewards/chosen": 0.04815078154206276,
2239
+ "rewards/margins": 0.9994159936904907,
2240
+ "rewards/rejected": -0.9512651562690735,
2241
+ "step": 1550
2242
+ },
2243
+ {
2244
+ "epoch": 0.82,
2245
+ "learning_rate": 1.0209424083769633e-07,
2246
+ "logits/chosen": -2.428290367126465,
2247
+ "logits/rejected": -2.394292116165161,
2248
+ "logps/chosen": -284.85968017578125,
2249
+ "logps/rejected": -287.145263671875,
2250
+ "loss": 0.505,
2251
+ "rewards/accuracies": 0.6499999761581421,
2252
+ "rewards/chosen": -0.22867484390735626,
2253
+ "rewards/margins": 1.033279299736023,
2254
+ "rewards/rejected": -1.2619540691375732,
2255
+ "step": 1560
2256
+ },
2257
+ {
2258
+ "epoch": 0.82,
2259
+ "learning_rate": 9.918557300756253e-08,
2260
+ "logits/chosen": -2.4378373622894287,
2261
+ "logits/rejected": -2.369236946105957,
2262
+ "logps/chosen": -293.5215759277344,
2263
+ "logps/rejected": -246.56997680664062,
2264
+ "loss": 0.4905,
2265
+ "rewards/accuracies": 0.75,
2266
+ "rewards/chosen": -0.13024893403053284,
2267
+ "rewards/margins": 1.4804332256317139,
2268
+ "rewards/rejected": -1.6106821298599243,
2269
+ "step": 1570
2270
+ },
2271
+ {
2272
+ "epoch": 0.83,
2273
+ "learning_rate": 9.627690517742873e-08,
2274
+ "logits/chosen": -2.4206690788269043,
2275
+ "logits/rejected": -2.409813165664673,
2276
+ "logps/chosen": -333.97467041015625,
2277
+ "logps/rejected": -314.40960693359375,
2278
+ "loss": 0.5211,
2279
+ "rewards/accuracies": 0.762499988079071,
2280
+ "rewards/chosen": 0.2985494136810303,
2281
+ "rewards/margins": 1.2202610969543457,
2282
+ "rewards/rejected": -0.9217117428779602,
2283
+ "step": 1580
2284
+ },
2285
+ {
2286
+ "epoch": 0.83,
2287
+ "learning_rate": 9.336823734729494e-08,
2288
+ "logits/chosen": -2.426513671875,
2289
+ "logits/rejected": -2.4083962440490723,
2290
+ "logps/chosen": -312.02655029296875,
2291
+ "logps/rejected": -314.4944152832031,
2292
+ "loss": 0.5172,
2293
+ "rewards/accuracies": 0.75,
2294
+ "rewards/chosen": -0.1636694371700287,
2295
+ "rewards/margins": 1.20877206325531,
2296
+ "rewards/rejected": -1.372441291809082,
2297
+ "step": 1590
2298
+ },
2299
+ {
2300
+ "epoch": 0.84,
2301
+ "learning_rate": 9.045956951716113e-08,
2302
+ "logits/chosen": -2.4410054683685303,
2303
+ "logits/rejected": -2.421079158782959,
2304
+ "logps/chosen": -268.4046630859375,
2305
+ "logps/rejected": -259.6822509765625,
2306
+ "loss": 0.5011,
2307
+ "rewards/accuracies": 0.737500011920929,
2308
+ "rewards/chosen": 0.018908673897385597,
2309
+ "rewards/margins": 1.4655755758285522,
2310
+ "rewards/rejected": -1.446666955947876,
2311
+ "step": 1600
2312
+ },
2313
+ {
2314
+ "epoch": 0.84,
2315
+ "learning_rate": 8.755090168702734e-08,
2316
+ "logits/chosen": -2.4191842079162598,
2317
+ "logits/rejected": -2.3796703815460205,
2318
+ "logps/chosen": -282.9931945800781,
2319
+ "logps/rejected": -297.3874206542969,
2320
+ "loss": 0.488,
2321
+ "rewards/accuracies": 0.737500011920929,
2322
+ "rewards/chosen": -0.26676809787750244,
2323
+ "rewards/margins": 1.4654661417007446,
2324
+ "rewards/rejected": -1.732234239578247,
2325
+ "step": 1610
2326
+ },
2327
+ {
2328
+ "epoch": 0.85,
2329
+ "learning_rate": 8.464223385689353e-08,
2330
+ "logits/chosen": -2.474801540374756,
2331
+ "logits/rejected": -2.439415693283081,
2332
+ "logps/chosen": -316.919921875,
2333
+ "logps/rejected": -286.6101989746094,
2334
+ "loss": 0.5092,
2335
+ "rewards/accuracies": 0.7124999761581421,
2336
+ "rewards/chosen": -0.14554055035114288,
2337
+ "rewards/margins": 1.1523851156234741,
2338
+ "rewards/rejected": -1.2979257106781006,
2339
+ "step": 1620
2340
+ },
2341
+ {
2342
+ "epoch": 0.85,
2343
+ "learning_rate": 8.173356602675974e-08,
2344
+ "logits/chosen": -2.501138210296631,
2345
+ "logits/rejected": -2.423609495162964,
2346
+ "logps/chosen": -267.6105651855469,
2347
+ "logps/rejected": -268.2532653808594,
2348
+ "loss": 0.5253,
2349
+ "rewards/accuracies": 0.7124999761581421,
2350
+ "rewards/chosen": -0.7160999178886414,
2351
+ "rewards/margins": 0.7922013998031616,
2352
+ "rewards/rejected": -1.5083013772964478,
2353
+ "step": 1630
2354
+ },
2355
+ {
2356
+ "epoch": 0.86,
2357
+ "learning_rate": 7.882489819662593e-08,
2358
+ "logits/chosen": -2.4721310138702393,
2359
+ "logits/rejected": -2.4814186096191406,
2360
+ "logps/chosen": -304.85931396484375,
2361
+ "logps/rejected": -421.44427490234375,
2362
+ "loss": 0.4832,
2363
+ "rewards/accuracies": 0.75,
2364
+ "rewards/chosen": -0.4838688373565674,
2365
+ "rewards/margins": 1.3697353601455688,
2366
+ "rewards/rejected": -1.8536040782928467,
2367
+ "step": 1640
2368
+ },
2369
+ {
2370
+ "epoch": 0.86,
2371
+ "learning_rate": 7.591623036649214e-08,
2372
+ "logits/chosen": -2.331878185272217,
2373
+ "logits/rejected": -2.3361971378326416,
2374
+ "logps/chosen": -296.43096923828125,
2375
+ "logps/rejected": -293.191650390625,
2376
+ "loss": 0.5881,
2377
+ "rewards/accuracies": 0.7749999761581421,
2378
+ "rewards/chosen": -0.4644225537776947,
2379
+ "rewards/margins": 1.1996773481369019,
2380
+ "rewards/rejected": -1.664099931716919,
2381
+ "step": 1650
2382
+ },
2383
+ {
2384
+ "epoch": 0.87,
2385
+ "learning_rate": 7.300756253635834e-08,
2386
+ "logits/chosen": -2.472486972808838,
2387
+ "logits/rejected": -2.420685291290283,
2388
+ "logps/chosen": -268.98712158203125,
2389
+ "logps/rejected": -287.7394714355469,
2390
+ "loss": 0.4996,
2391
+ "rewards/accuracies": 0.75,
2392
+ "rewards/chosen": -0.22886808216571808,
2393
+ "rewards/margins": 1.0310173034667969,
2394
+ "rewards/rejected": -1.259885311126709,
2395
+ "step": 1660
2396
+ },
2397
+ {
2398
+ "epoch": 0.87,
2399
+ "learning_rate": 7.009889470622454e-08,
2400
+ "logits/chosen": -2.480839967727661,
2401
+ "logits/rejected": -2.405022144317627,
2402
+ "logps/chosen": -327.4227600097656,
2403
+ "logps/rejected": -299.2853088378906,
2404
+ "loss": 0.4532,
2405
+ "rewards/accuracies": 0.75,
2406
+ "rewards/chosen": -0.21133089065551758,
2407
+ "rewards/margins": 1.356836199760437,
2408
+ "rewards/rejected": -1.568166971206665,
2409
+ "step": 1670
2410
+ },
2411
+ {
2412
+ "epoch": 0.88,
2413
+ "learning_rate": 6.719022687609075e-08,
2414
+ "logits/chosen": -2.4610321521759033,
2415
+ "logits/rejected": -2.3211312294006348,
2416
+ "logps/chosen": -367.6954650878906,
2417
+ "logps/rejected": -312.0871887207031,
2418
+ "loss": 0.4964,
2419
+ "rewards/accuracies": 0.699999988079071,
2420
+ "rewards/chosen": -0.6737295389175415,
2421
+ "rewards/margins": 1.504465103149414,
2422
+ "rewards/rejected": -2.178194522857666,
2423
+ "step": 1680
2424
+ },
2425
+ {
2426
+ "epoch": 0.88,
2427
+ "learning_rate": 6.428155904595695e-08,
2428
+ "logits/chosen": -2.350032091140747,
2429
+ "logits/rejected": -2.3311705589294434,
2430
+ "logps/chosen": -309.73712158203125,
2431
+ "logps/rejected": -270.7311096191406,
2432
+ "loss": 0.4952,
2433
+ "rewards/accuracies": 0.699999988079071,
2434
+ "rewards/chosen": -0.18244963884353638,
2435
+ "rewards/margins": 1.2423228025436401,
2436
+ "rewards/rejected": -1.4247725009918213,
2437
+ "step": 1690
2438
+ },
2439
+ {
2440
+ "epoch": 0.89,
2441
+ "learning_rate": 6.137289121582315e-08,
2442
+ "logits/chosen": -2.5109975337982178,
2443
+ "logits/rejected": -2.3817737102508545,
2444
+ "logps/chosen": -335.9524841308594,
2445
+ "logps/rejected": -272.06829833984375,
2446
+ "loss": 0.4946,
2447
+ "rewards/accuracies": 0.800000011920929,
2448
+ "rewards/chosen": -0.4245755672454834,
2449
+ "rewards/margins": 1.40361487865448,
2450
+ "rewards/rejected": -1.828190565109253,
2451
+ "step": 1700
2452
+ },
2453
+ {
2454
+ "epoch": 0.89,
2455
+ "learning_rate": 5.846422338568935e-08,
2456
+ "logits/chosen": -2.4695639610290527,
2457
+ "logits/rejected": -2.414325714111328,
2458
+ "logps/chosen": -304.23468017578125,
2459
+ "logps/rejected": -285.8380432128906,
2460
+ "loss": 0.4495,
2461
+ "rewards/accuracies": 0.7749999761581421,
2462
+ "rewards/chosen": -0.3581593334674835,
2463
+ "rewards/margins": 1.1380908489227295,
2464
+ "rewards/rejected": -1.4962502717971802,
2465
+ "step": 1710
2466
+ },
2467
+ {
2468
+ "epoch": 0.9,
2469
+ "learning_rate": 5.555555555555555e-08,
2470
+ "logits/chosen": -2.439877986907959,
2471
+ "logits/rejected": -2.3664920330047607,
2472
+ "logps/chosen": -350.97137451171875,
2473
+ "logps/rejected": -327.41302490234375,
2474
+ "loss": 0.5355,
2475
+ "rewards/accuracies": 0.625,
2476
+ "rewards/chosen": -0.5186697244644165,
2477
+ "rewards/margins": 1.1480152606964111,
2478
+ "rewards/rejected": -1.6666851043701172,
2479
+ "step": 1720
2480
+ },
2481
+ {
2482
+ "epoch": 0.91,
2483
+ "learning_rate": 5.264688772542175e-08,
2484
+ "logits/chosen": -2.455570697784424,
2485
+ "logits/rejected": -2.409554958343506,
2486
+ "logps/chosen": -292.89715576171875,
2487
+ "logps/rejected": -287.56573486328125,
2488
+ "loss": 0.4603,
2489
+ "rewards/accuracies": 0.7124999761581421,
2490
+ "rewards/chosen": -0.08683060109615326,
2491
+ "rewards/margins": 0.9011920690536499,
2492
+ "rewards/rejected": -0.98802250623703,
2493
+ "step": 1730
2494
+ },
2495
+ {
2496
+ "epoch": 0.91,
2497
+ "learning_rate": 4.973821989528795e-08,
2498
+ "logits/chosen": -2.502526044845581,
2499
+ "logits/rejected": -2.46052885055542,
2500
+ "logps/chosen": -266.27947998046875,
2501
+ "logps/rejected": -262.73822021484375,
2502
+ "loss": 0.5074,
2503
+ "rewards/accuracies": 0.762499988079071,
2504
+ "rewards/chosen": -0.008827775716781616,
2505
+ "rewards/margins": 1.2833327054977417,
2506
+ "rewards/rejected": -1.2921605110168457,
2507
+ "step": 1740
2508
+ },
2509
+ {
2510
+ "epoch": 0.92,
2511
+ "learning_rate": 4.682955206515416e-08,
2512
+ "logits/chosen": -2.494965076446533,
2513
+ "logits/rejected": -2.4605846405029297,
2514
+ "logps/chosen": -316.22882080078125,
2515
+ "logps/rejected": -368.8150939941406,
2516
+ "loss": 0.5432,
2517
+ "rewards/accuracies": 0.737500011920929,
2518
+ "rewards/chosen": -0.25845271348953247,
2519
+ "rewards/margins": 1.4049198627471924,
2520
+ "rewards/rejected": -1.6633726358413696,
2521
+ "step": 1750
2522
+ },
2523
+ {
2524
+ "epoch": 0.92,
2525
+ "learning_rate": 4.392088423502036e-08,
2526
+ "logits/chosen": -2.481264352798462,
2527
+ "logits/rejected": -2.5107004642486572,
2528
+ "logps/chosen": -274.2122802734375,
2529
+ "logps/rejected": -280.37933349609375,
2530
+ "loss": 0.5557,
2531
+ "rewards/accuracies": 0.699999988079071,
2532
+ "rewards/chosen": -0.5907906293869019,
2533
+ "rewards/margins": 1.1111408472061157,
2534
+ "rewards/rejected": -1.701931357383728,
2535
+ "step": 1760
2536
+ },
2537
+ {
2538
+ "epoch": 0.93,
2539
+ "learning_rate": 4.101221640488656e-08,
2540
+ "logits/chosen": -2.4728500843048096,
2541
+ "logits/rejected": -2.408026695251465,
2542
+ "logps/chosen": -308.61956787109375,
2543
+ "logps/rejected": -288.31170654296875,
2544
+ "loss": 0.4805,
2545
+ "rewards/accuracies": 0.7875000238418579,
2546
+ "rewards/chosen": -0.6883036494255066,
2547
+ "rewards/margins": 1.1805965900421143,
2548
+ "rewards/rejected": -1.8689002990722656,
2549
+ "step": 1770
2550
+ },
2551
+ {
2552
+ "epoch": 0.93,
2553
+ "learning_rate": 3.810354857475276e-08,
2554
+ "logits/chosen": -2.3637709617614746,
2555
+ "logits/rejected": -2.3329529762268066,
2556
+ "logps/chosen": -324.7041320800781,
2557
+ "logps/rejected": -270.0013122558594,
2558
+ "loss": 0.5471,
2559
+ "rewards/accuracies": 0.675000011920929,
2560
+ "rewards/chosen": -0.544477105140686,
2561
+ "rewards/margins": 1.064997673034668,
2562
+ "rewards/rejected": -1.6094748973846436,
2563
+ "step": 1780
2564
+ },
2565
+ {
2566
+ "epoch": 0.94,
2567
+ "learning_rate": 3.519488074461896e-08,
2568
+ "logits/chosen": -2.447537660598755,
2569
+ "logits/rejected": -2.386204242706299,
2570
+ "logps/chosen": -268.69378662109375,
2571
+ "logps/rejected": -276.32843017578125,
2572
+ "loss": 0.3955,
2573
+ "rewards/accuracies": 0.75,
2574
+ "rewards/chosen": -0.3514997661113739,
2575
+ "rewards/margins": 1.253027319908142,
2576
+ "rewards/rejected": -1.6045271158218384,
2577
+ "step": 1790
2578
+ },
2579
+ {
2580
+ "epoch": 0.94,
2581
+ "learning_rate": 3.228621291448516e-08,
2582
+ "logits/chosen": -2.504725694656372,
2583
+ "logits/rejected": -2.4065499305725098,
2584
+ "logps/chosen": -324.8893127441406,
2585
+ "logps/rejected": -262.2546081542969,
2586
+ "loss": 0.4898,
2587
+ "rewards/accuracies": 0.737500011920929,
2588
+ "rewards/chosen": -0.5291447043418884,
2589
+ "rewards/margins": 1.2341575622558594,
2590
+ "rewards/rejected": -1.763302206993103,
2591
+ "step": 1800
2592
+ },
2593
+ {
2594
+ "epoch": 0.95,
2595
+ "learning_rate": 2.9377545084351366e-08,
2596
+ "logits/chosen": -2.4359753131866455,
2597
+ "logits/rejected": -2.3973264694213867,
2598
+ "logps/chosen": -380.8512878417969,
2599
+ "logps/rejected": -345.4535827636719,
2600
+ "loss": 0.4781,
2601
+ "rewards/accuracies": 0.824999988079071,
2602
+ "rewards/chosen": -0.39544662833213806,
2603
+ "rewards/margins": 1.4492541551589966,
2604
+ "rewards/rejected": -1.844700813293457,
2605
+ "step": 1810
2606
+ },
2607
+ {
2608
+ "epoch": 0.95,
2609
+ "learning_rate": 2.6468877254217567e-08,
2610
+ "logits/chosen": -2.455008029937744,
2611
+ "logits/rejected": -2.406989574432373,
2612
+ "logps/chosen": -321.1400451660156,
2613
+ "logps/rejected": -326.90203857421875,
2614
+ "loss": 0.5332,
2615
+ "rewards/accuracies": 0.737500011920929,
2616
+ "rewards/chosen": -0.45269957184791565,
2617
+ "rewards/margins": 1.45340895652771,
2618
+ "rewards/rejected": -1.9061084985733032,
2619
+ "step": 1820
2620
+ },
2621
+ {
2622
+ "epoch": 0.96,
2623
+ "learning_rate": 2.3560209424083768e-08,
2624
+ "logits/chosen": -2.451988458633423,
2625
+ "logits/rejected": -2.390340805053711,
2626
+ "logps/chosen": -319.3875427246094,
2627
+ "logps/rejected": -321.97186279296875,
2628
+ "loss": 0.5775,
2629
+ "rewards/accuracies": 0.6625000238418579,
2630
+ "rewards/chosen": -0.6661805510520935,
2631
+ "rewards/margins": 0.7991847395896912,
2632
+ "rewards/rejected": -1.4653651714324951,
2633
+ "step": 1830
2634
+ },
2635
+ {
2636
+ "epoch": 0.96,
2637
+ "learning_rate": 2.065154159394997e-08,
2638
+ "logits/chosen": -2.439758539199829,
2639
+ "logits/rejected": -2.419706106185913,
2640
+ "logps/chosen": -330.24810791015625,
2641
+ "logps/rejected": -308.7535095214844,
2642
+ "loss": 0.5889,
2643
+ "rewards/accuracies": 0.7250000238418579,
2644
+ "rewards/chosen": -0.5779649019241333,
2645
+ "rewards/margins": 0.9780583381652832,
2646
+ "rewards/rejected": -1.5560232400894165,
2647
+ "step": 1840
2648
+ },
2649
+ {
2650
+ "epoch": 0.97,
2651
+ "learning_rate": 1.7742873763816174e-08,
2652
+ "logits/chosen": -2.4510066509246826,
2653
+ "logits/rejected": -2.406909704208374,
2654
+ "logps/chosen": -338.01861572265625,
2655
+ "logps/rejected": -291.27923583984375,
2656
+ "loss": 0.5524,
2657
+ "rewards/accuracies": 0.737500011920929,
2658
+ "rewards/chosen": -0.372661292552948,
2659
+ "rewards/margins": 1.7962923049926758,
2660
+ "rewards/rejected": -2.1689534187316895,
2661
+ "step": 1850
2662
+ },
2663
+ {
2664
+ "epoch": 0.97,
2665
+ "learning_rate": 1.4834205933682373e-08,
2666
+ "logits/chosen": -2.5106232166290283,
2667
+ "logits/rejected": -2.4458587169647217,
2668
+ "logps/chosen": -300.896728515625,
2669
+ "logps/rejected": -278.838623046875,
2670
+ "loss": 0.5813,
2671
+ "rewards/accuracies": 0.8125,
2672
+ "rewards/chosen": -0.4824046492576599,
2673
+ "rewards/margins": 1.491176962852478,
2674
+ "rewards/rejected": -1.9735815525054932,
2675
+ "step": 1860
2676
+ },
2677
+ {
2678
+ "epoch": 0.98,
2679
+ "learning_rate": 1.1925538103548575e-08,
2680
+ "logits/chosen": -2.4239859580993652,
2681
+ "logits/rejected": -2.373213768005371,
2682
+ "logps/chosen": -280.8854675292969,
2683
+ "logps/rejected": -301.01092529296875,
2684
+ "loss": 0.5219,
2685
+ "rewards/accuracies": 0.762499988079071,
2686
+ "rewards/chosen": -0.43918657302856445,
2687
+ "rewards/margins": 1.376401662826538,
2688
+ "rewards/rejected": -1.8155882358551025,
2689
+ "step": 1870
2690
+ },
2691
+ {
2692
+ "epoch": 0.98,
2693
+ "learning_rate": 9.016870273414776e-09,
2694
+ "logits/chosen": -2.4339070320129395,
2695
+ "logits/rejected": -2.3967947959899902,
2696
+ "logps/chosen": -354.99462890625,
2697
+ "logps/rejected": -309.2855529785156,
2698
+ "loss": 0.4764,
2699
+ "rewards/accuracies": 0.800000011920929,
2700
+ "rewards/chosen": -0.24851647019386292,
2701
+ "rewards/margins": 1.5067845582962036,
2702
+ "rewards/rejected": -1.7553011178970337,
2703
+ "step": 1880
2704
+ },
2705
+ {
2706
+ "epoch": 0.99,
2707
+ "learning_rate": 6.1082024432809765e-09,
2708
+ "logits/chosen": -2.4029338359832764,
2709
+ "logits/rejected": -2.4133036136627197,
2710
+ "logps/chosen": -268.3423767089844,
2711
+ "logps/rejected": -301.10321044921875,
2712
+ "loss": 0.5074,
2713
+ "rewards/accuracies": 0.7749999761581421,
2714
+ "rewards/chosen": -0.8797990083694458,
2715
+ "rewards/margins": 0.8209834098815918,
2716
+ "rewards/rejected": -1.7007821798324585,
2717
+ "step": 1890
2718
+ },
2719
+ {
2720
+ "epoch": 0.99,
2721
+ "learning_rate": 3.1995346131471783e-09,
2722
+ "logits/chosen": -2.4465489387512207,
2723
+ "logits/rejected": -2.4295449256896973,
2724
+ "logps/chosen": -268.50006103515625,
2725
+ "logps/rejected": -266.83685302734375,
2726
+ "loss": 0.479,
2727
+ "rewards/accuracies": 0.7124999761581421,
2728
+ "rewards/chosen": -0.589029848575592,
2729
+ "rewards/margins": 1.1926295757293701,
2730
+ "rewards/rejected": -1.781659483909607,
2731
+ "step": 1900
2732
+ },
2733
+ {
2734
+ "epoch": 1.0,
2735
+ "learning_rate": 2.9086678301337986e-10,
2736
+ "logits/chosen": -2.482949733734131,
2737
+ "logits/rejected": -2.426959991455078,
2738
+ "logps/chosen": -303.0064392089844,
2739
+ "logps/rejected": -287.6535339355469,
2740
+ "loss": 0.5333,
2741
+ "rewards/accuracies": 0.6875,
2742
+ "rewards/chosen": -0.44241657853126526,
2743
+ "rewards/margins": 0.9881950616836548,
2744
+ "rewards/rejected": -1.4306116104125977,
2745
+ "step": 1910
2746
+ },
2747
+ {
2748
+ "epoch": 1.0,
2749
+ "step": 1911,
2750
+ "total_flos": 0.0,
2751
+ "train_loss": 0.5469513680846325,
2752
+ "train_runtime": 31987.9403,
2753
+ "train_samples_per_second": 1.911,
2754
+ "train_steps_per_second": 0.06
2755
+ }
2756
+ ],
2757
+ "logging_steps": 10,
2758
+ "max_steps": 1911,
2759
+ "num_input_tokens_seen": 0,
2760
+ "num_train_epochs": 1,
2761
+ "save_steps": 500,
2762
+ "total_flos": 0.0,
2763
+ "train_batch_size": 8,
2764
+ "trial_name": null,
2765
+ "trial_params": null
2766
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c88deef81a32d1c3f94af5c783e8a1d9ccd20a2714528a73414ef6c3c7ef5361
3
+ size 5371