AmberYifan commited on
Commit
1844941
·
verified ·
1 Parent(s): 987b12b

Training in progress, epoch 3, checkpoint

Browse files
last-checkpoint/global_step930/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e3f5ec6e41092d04b2fabf6f4881dc62f2368c8193b7ab7effa7e86fe914753
3
+ size 14496051368
last-checkpoint/global_step930/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ec7d3700593f9a2d6248eb2910cb64bac902c9c0e1788ccef463c447d969806
3
+ size 14496051368
last-checkpoint/global_step930/bf16_zero_pp_rank_2_mp_rank_00_optim_states.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:33b4336749519f86702d4ac9c56027bbb5fd0a285fddc29c96076d55db83f0f3
3
+ size 14496051368
last-checkpoint/global_step930/bf16_zero_pp_rank_3_mp_rank_00_optim_states.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b3110e6749843e53a50a29f0105a93789bd335a72e4639a8ac3e30055b30e57f
3
+ size 14496051368
last-checkpoint/global_step930/zero_pp_rank_0_mp_rank_00_model_states.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b21aba8486196f9285f8f6e2ff830fd03ef7fdc344d1908fcdbed3c8d3a31e56
3
+ size 150629
last-checkpoint/global_step930/zero_pp_rank_1_mp_rank_00_model_states.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:875c8826cccf9c7bbafbe2eb5aefd7006b82020a4bf595fce49f156e51ac4586
3
+ size 150629
last-checkpoint/global_step930/zero_pp_rank_2_mp_rank_00_model_states.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d7e437fd1ca6281a928627ecb6349520b8485053b4c33302950261a8933af880
3
+ size 150629
last-checkpoint/global_step930/zero_pp_rank_3_mp_rank_00_model_states.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3219946b4f64e8587b6366bbd62c8f1d640f11764d2b84f74ce3a451981b736b
3
+ size 150629
last-checkpoint/latest CHANGED
@@ -1 +1 @@
1
- global_step620
 
1
+ global_step930
last-checkpoint/model-00001-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f04063fa12fe1619800b0ebefb5d2970cc38d956a23578ca52e3402af7bdb404
3
  size 4949453792
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:66d51a21bf63fda47bb7fcd9afce0ebb0d1348c51d04b669b2b20f2443873250
3
  size 4949453792
last-checkpoint/model-00002-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3c51f4c858b8f0002bbb5989a332166c0c0881bdcf50c06a05a01716f1af25c7
3
  size 4999819336
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0078ab561d4b6f9f250279621f4eb553d3c041c4acad25bd8f0d935e77facf89
3
  size 4999819336
last-checkpoint/model-00003-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4ec599ee402233affa2e8dbb51036577834c1d4166681a044ef5ea67c46203f0
3
  size 4546807800
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8fb36fb4800d8a729199547e1129dd9b07d5baa8b17f85545bf4ebc481b28463
3
  size 4546807800
last-checkpoint/rng_state_0.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2307c03867cef25b5028feb9a23f80e784b9af9a615de13ddca560a6a90fb593
3
  size 15024
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab7cf0d34d60dfda516cf9661904550e2e294e723edd07c25c738f05e8ba92d1
3
  size 15024
last-checkpoint/rng_state_1.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:50218cfaecdd818354e567b7167c13899e3b42297e7d8f58bd7e732cfa547800
3
  size 15024
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f7b183324e8227a51a9556d86b2ad893a8c4c52205ed4a737356c6611dac7353
3
  size 15024
last-checkpoint/rng_state_2.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9814a66b49861f5495b06dae3be12ddf7185b88e2cae1fb808ca9efd99d5807f
3
  size 15024
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ac69e994090f4818cb1fa6f6cefa363178552c3c731c6507ff195bcb07fd5bef
3
  size 15024
last-checkpoint/rng_state_3.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a7be93040a65e0a29975f6c70b94418e1fdf88423a50c58aa572141d3c92fbfc
3
  size 15024
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f68e61b63402f8afb1f69c960f7944965655dac11e3ccf29919c282f23931f86
3
  size 15024
last-checkpoint/scheduler.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9cc6b970d8e5e214aa823c0d575643a91300739c2aa894dc016cba903d55bb8d
3
  size 1064
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0c10f428b67fa0d88b04b78a5cc93219d70a153e68839352f1a83bcb578198a7
3
  size 1064
last-checkpoint/trainer_state.json CHANGED
@@ -1,9 +1,9 @@
1
  {
2
  "best_metric": null,
3
  "best_model_checkpoint": null,
4
- "epoch": 2.0,
5
  "eval_steps": 500,
6
- "global_step": 620,
7
  "is_hyper_param_search": false,
8
  "is_local_process_zero": true,
9
  "is_world_process_zero": true,
@@ -984,6 +984,487 @@
984
  "eval_samples_per_second": 15.1,
985
  "eval_steps_per_second": 0.528,
986
  "step": 620
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
987
  }
988
  ],
989
  "logging_steps": 10,
@@ -998,7 +1479,7 @@
998
  "should_evaluate": false,
999
  "should_log": false,
1000
  "should_save": true,
1001
- "should_training_stop": false
1002
  },
1003
  "attributes": {}
1004
  }
 
1
  {
2
  "best_metric": null,
3
  "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
  "eval_steps": 500,
6
+ "global_step": 930,
7
  "is_hyper_param_search": false,
8
  "is_local_process_zero": true,
9
  "is_world_process_zero": true,
 
984
  "eval_samples_per_second": 15.1,
985
  "eval_steps_per_second": 0.528,
986
  "step": 620
987
+ },
988
+ {
989
+ "epoch": 2.032258064516129,
990
+ "grad_norm": 0.45384363049400467,
991
+ "learning_rate": 1.7921146953405018e-07,
992
+ "logits/chosen": -3.5625,
993
+ "logits/rejected": -3.578125,
994
+ "logps/chosen": -191.0,
995
+ "logps/rejected": -300.0,
996
+ "loss": 0.0012,
997
+ "rewards/accuracies": 1.0,
998
+ "rewards/chosen": -0.28125,
999
+ "rewards/margins": 10.1875,
1000
+ "rewards/rejected": -10.4375,
1001
+ "step": 630
1002
+ },
1003
+ {
1004
+ "epoch": 2.064516129032258,
1005
+ "grad_norm": 2.0283023878654722,
1006
+ "learning_rate": 1.7323775388291518e-07,
1007
+ "logits/chosen": -3.609375,
1008
+ "logits/rejected": -3.640625,
1009
+ "logps/chosen": -191.0,
1010
+ "logps/rejected": -298.0,
1011
+ "loss": 0.0019,
1012
+ "rewards/accuracies": 1.0,
1013
+ "rewards/chosen": -0.302734375,
1014
+ "rewards/margins": 10.25,
1015
+ "rewards/rejected": -10.5625,
1016
+ "step": 640
1017
+ },
1018
+ {
1019
+ "epoch": 2.096774193548387,
1020
+ "grad_norm": 0.9217664266560652,
1021
+ "learning_rate": 1.6726403823178015e-07,
1022
+ "logits/chosen": -3.6875,
1023
+ "logits/rejected": -3.75,
1024
+ "logps/chosen": -193.0,
1025
+ "logps/rejected": -298.0,
1026
+ "loss": 0.0013,
1027
+ "rewards/accuracies": 1.0,
1028
+ "rewards/chosen": -0.93359375,
1029
+ "rewards/margins": 9.625,
1030
+ "rewards/rejected": -10.5625,
1031
+ "step": 650
1032
+ },
1033
+ {
1034
+ "epoch": 2.129032258064516,
1035
+ "grad_norm": 0.44858819876518125,
1036
+ "learning_rate": 1.6129032258064515e-07,
1037
+ "logits/chosen": -3.6875,
1038
+ "logits/rejected": -3.703125,
1039
+ "logps/chosen": -207.0,
1040
+ "logps/rejected": -316.0,
1041
+ "loss": 0.0012,
1042
+ "rewards/accuracies": 1.0,
1043
+ "rewards/chosen": -1.3046875,
1044
+ "rewards/margins": 10.125,
1045
+ "rewards/rejected": -11.4375,
1046
+ "step": 660
1047
+ },
1048
+ {
1049
+ "epoch": 2.161290322580645,
1050
+ "grad_norm": 9.792046659236782,
1051
+ "learning_rate": 1.5531660692951015e-07,
1052
+ "logits/chosen": -3.578125,
1053
+ "logits/rejected": -3.59375,
1054
+ "logps/chosen": -199.0,
1055
+ "logps/rejected": -310.0,
1056
+ "loss": 0.0019,
1057
+ "rewards/accuracies": 1.0,
1058
+ "rewards/chosen": -1.1875,
1059
+ "rewards/margins": 9.875,
1060
+ "rewards/rejected": -11.0625,
1061
+ "step": 670
1062
+ },
1063
+ {
1064
+ "epoch": 2.193548387096774,
1065
+ "grad_norm": 0.3983602269153926,
1066
+ "learning_rate": 1.4934289127837515e-07,
1067
+ "logits/chosen": -3.5,
1068
+ "logits/rejected": -3.703125,
1069
+ "logps/chosen": -188.0,
1070
+ "logps/rejected": -302.0,
1071
+ "loss": 0.0024,
1072
+ "rewards/accuracies": 1.0,
1073
+ "rewards/chosen": -0.59375,
1074
+ "rewards/margins": 10.25,
1075
+ "rewards/rejected": -10.875,
1076
+ "step": 680
1077
+ },
1078
+ {
1079
+ "epoch": 2.225806451612903,
1080
+ "grad_norm": 1.2472683997191045,
1081
+ "learning_rate": 1.4336917562724012e-07,
1082
+ "logits/chosen": -3.59375,
1083
+ "logits/rejected": -3.65625,
1084
+ "logps/chosen": -202.0,
1085
+ "logps/rejected": -312.0,
1086
+ "loss": 0.0011,
1087
+ "rewards/accuracies": 1.0,
1088
+ "rewards/chosen": -1.0546875,
1089
+ "rewards/margins": 10.25,
1090
+ "rewards/rejected": -11.3125,
1091
+ "step": 690
1092
+ },
1093
+ {
1094
+ "epoch": 2.258064516129032,
1095
+ "grad_norm": 0.9104056031581937,
1096
+ "learning_rate": 1.3739545997610514e-07,
1097
+ "logits/chosen": -3.453125,
1098
+ "logits/rejected": -3.53125,
1099
+ "logps/chosen": -192.0,
1100
+ "logps/rejected": -308.0,
1101
+ "loss": 0.0008,
1102
+ "rewards/accuracies": 1.0,
1103
+ "rewards/chosen": -0.90625,
1104
+ "rewards/margins": 10.625,
1105
+ "rewards/rejected": -11.5625,
1106
+ "step": 700
1107
+ },
1108
+ {
1109
+ "epoch": 2.2903225806451615,
1110
+ "grad_norm": 0.5005896742747854,
1111
+ "learning_rate": 1.3142174432497014e-07,
1112
+ "logits/chosen": -3.453125,
1113
+ "logits/rejected": -3.5625,
1114
+ "logps/chosen": -196.0,
1115
+ "logps/rejected": -316.0,
1116
+ "loss": 0.0015,
1117
+ "rewards/accuracies": 1.0,
1118
+ "rewards/chosen": -0.51953125,
1119
+ "rewards/margins": 11.4375,
1120
+ "rewards/rejected": -11.9375,
1121
+ "step": 710
1122
+ },
1123
+ {
1124
+ "epoch": 2.3225806451612905,
1125
+ "grad_norm": 2.0961727669442247,
1126
+ "learning_rate": 1.254480286738351e-07,
1127
+ "logits/chosen": -3.53125,
1128
+ "logits/rejected": -3.546875,
1129
+ "logps/chosen": -206.0,
1130
+ "logps/rejected": -318.0,
1131
+ "loss": 0.0011,
1132
+ "rewards/accuracies": 1.0,
1133
+ "rewards/chosen": -1.515625,
1134
+ "rewards/margins": 10.5,
1135
+ "rewards/rejected": -12.0,
1136
+ "step": 720
1137
+ },
1138
+ {
1139
+ "epoch": 2.3548387096774195,
1140
+ "grad_norm": 0.33710671811391674,
1141
+ "learning_rate": 1.194743130227001e-07,
1142
+ "logits/chosen": -3.53125,
1143
+ "logits/rejected": -3.578125,
1144
+ "logps/chosen": -201.0,
1145
+ "logps/rejected": -318.0,
1146
+ "loss": 0.0008,
1147
+ "rewards/accuracies": 1.0,
1148
+ "rewards/chosen": -1.625,
1149
+ "rewards/margins": 10.375,
1150
+ "rewards/rejected": -12.0,
1151
+ "step": 730
1152
+ },
1153
+ {
1154
+ "epoch": 2.3870967741935485,
1155
+ "grad_norm": 3.7387698089394634,
1156
+ "learning_rate": 1.135005973715651e-07,
1157
+ "logits/chosen": -3.59375,
1158
+ "logits/rejected": -3.625,
1159
+ "logps/chosen": -202.0,
1160
+ "logps/rejected": -314.0,
1161
+ "loss": 0.0014,
1162
+ "rewards/accuracies": 1.0,
1163
+ "rewards/chosen": -1.3671875,
1164
+ "rewards/margins": 10.375,
1165
+ "rewards/rejected": -11.75,
1166
+ "step": 740
1167
+ },
1168
+ {
1169
+ "epoch": 2.4193548387096775,
1170
+ "grad_norm": 3.5076436110564426,
1171
+ "learning_rate": 1.0752688172043011e-07,
1172
+ "logits/chosen": -3.5625,
1173
+ "logits/rejected": -3.6875,
1174
+ "logps/chosen": -192.0,
1175
+ "logps/rejected": -306.0,
1176
+ "loss": 0.0019,
1177
+ "rewards/accuracies": 1.0,
1178
+ "rewards/chosen": -0.53125,
1179
+ "rewards/margins": 10.4375,
1180
+ "rewards/rejected": -11.0,
1181
+ "step": 750
1182
+ },
1183
+ {
1184
+ "epoch": 2.4516129032258065,
1185
+ "grad_norm": 0.3051005963902528,
1186
+ "learning_rate": 1.0155316606929509e-07,
1187
+ "logits/chosen": -3.46875,
1188
+ "logits/rejected": -3.53125,
1189
+ "logps/chosen": -194.0,
1190
+ "logps/rejected": -312.0,
1191
+ "loss": 0.0009,
1192
+ "rewards/accuracies": 1.0,
1193
+ "rewards/chosen": -0.94140625,
1194
+ "rewards/margins": 10.625,
1195
+ "rewards/rejected": -11.5625,
1196
+ "step": 760
1197
+ },
1198
+ {
1199
+ "epoch": 2.4838709677419355,
1200
+ "grad_norm": 0.4055555200925197,
1201
+ "learning_rate": 9.557945041816009e-08,
1202
+ "logits/chosen": -3.40625,
1203
+ "logits/rejected": -3.546875,
1204
+ "logps/chosen": -190.0,
1205
+ "logps/rejected": -314.0,
1206
+ "loss": 0.0014,
1207
+ "rewards/accuracies": 1.0,
1208
+ "rewards/chosen": -0.76953125,
1209
+ "rewards/margins": 11.0,
1210
+ "rewards/rejected": -11.75,
1211
+ "step": 770
1212
+ },
1213
+ {
1214
+ "epoch": 2.5161290322580645,
1215
+ "grad_norm": 0.4423005388531537,
1216
+ "learning_rate": 8.960573476702509e-08,
1217
+ "logits/chosen": -3.5625,
1218
+ "logits/rejected": -3.5625,
1219
+ "logps/chosen": -211.0,
1220
+ "logps/rejected": -336.0,
1221
+ "loss": 0.0013,
1222
+ "rewards/accuracies": 1.0,
1223
+ "rewards/chosen": -1.640625,
1224
+ "rewards/margins": 11.5,
1225
+ "rewards/rejected": -13.1875,
1226
+ "step": 780
1227
+ },
1228
+ {
1229
+ "epoch": 2.5483870967741935,
1230
+ "grad_norm": 0.34201712686499286,
1231
+ "learning_rate": 8.363201911589008e-08,
1232
+ "logits/chosen": -3.4375,
1233
+ "logits/rejected": -3.515625,
1234
+ "logps/chosen": -197.0,
1235
+ "logps/rejected": -312.0,
1236
+ "loss": 0.001,
1237
+ "rewards/accuracies": 1.0,
1238
+ "rewards/chosen": -0.81640625,
1239
+ "rewards/margins": 10.9375,
1240
+ "rewards/rejected": -11.75,
1241
+ "step": 790
1242
+ },
1243
+ {
1244
+ "epoch": 2.5806451612903225,
1245
+ "grad_norm": 0.08324339497209432,
1246
+ "learning_rate": 7.765830346475507e-08,
1247
+ "logits/chosen": -3.421875,
1248
+ "logits/rejected": -3.59375,
1249
+ "logps/chosen": -207.0,
1250
+ "logps/rejected": -330.0,
1251
+ "loss": 0.0005,
1252
+ "rewards/accuracies": 1.0,
1253
+ "rewards/chosen": -1.609375,
1254
+ "rewards/margins": 11.5,
1255
+ "rewards/rejected": -13.125,
1256
+ "step": 800
1257
+ },
1258
+ {
1259
+ "epoch": 2.6129032258064515,
1260
+ "grad_norm": 0.35882151597975387,
1261
+ "learning_rate": 7.168458781362006e-08,
1262
+ "logits/chosen": -3.59375,
1263
+ "logits/rejected": -3.6875,
1264
+ "logps/chosen": -202.0,
1265
+ "logps/rejected": -328.0,
1266
+ "loss": 0.0045,
1267
+ "rewards/accuracies": 1.0,
1268
+ "rewards/chosen": -1.03125,
1269
+ "rewards/margins": 11.625,
1270
+ "rewards/rejected": -12.6875,
1271
+ "step": 810
1272
+ },
1273
+ {
1274
+ "epoch": 2.6451612903225805,
1275
+ "grad_norm": 0.3006723416131319,
1276
+ "learning_rate": 6.571087216248507e-08,
1277
+ "logits/chosen": -3.4375,
1278
+ "logits/rejected": -3.609375,
1279
+ "logps/chosen": -197.0,
1280
+ "logps/rejected": -322.0,
1281
+ "loss": 0.0026,
1282
+ "rewards/accuracies": 1.0,
1283
+ "rewards/chosen": -1.40625,
1284
+ "rewards/margins": 11.0,
1285
+ "rewards/rejected": -12.4375,
1286
+ "step": 820
1287
+ },
1288
+ {
1289
+ "epoch": 2.6774193548387095,
1290
+ "grad_norm": 0.9135709315414355,
1291
+ "learning_rate": 5.973715651135006e-08,
1292
+ "logits/chosen": -3.40625,
1293
+ "logits/rejected": -3.546875,
1294
+ "logps/chosen": -194.0,
1295
+ "logps/rejected": -312.0,
1296
+ "loss": 0.0026,
1297
+ "rewards/accuracies": 1.0,
1298
+ "rewards/chosen": -1.046875,
1299
+ "rewards/margins": 11.3125,
1300
+ "rewards/rejected": -12.3125,
1301
+ "step": 830
1302
+ },
1303
+ {
1304
+ "epoch": 2.709677419354839,
1305
+ "grad_norm": 0.11799710974470022,
1306
+ "learning_rate": 5.3763440860215054e-08,
1307
+ "logits/chosen": -3.453125,
1308
+ "logits/rejected": -3.609375,
1309
+ "logps/chosen": -193.0,
1310
+ "logps/rejected": -330.0,
1311
+ "loss": 0.001,
1312
+ "rewards/accuracies": 1.0,
1313
+ "rewards/chosen": -0.79296875,
1314
+ "rewards/margins": 12.8125,
1315
+ "rewards/rejected": -13.625,
1316
+ "step": 840
1317
+ },
1318
+ {
1319
+ "epoch": 2.741935483870968,
1320
+ "grad_norm": 0.24029168708895726,
1321
+ "learning_rate": 4.7789725209080046e-08,
1322
+ "logits/chosen": -3.484375,
1323
+ "logits/rejected": -3.5625,
1324
+ "logps/chosen": -200.0,
1325
+ "logps/rejected": -320.0,
1326
+ "loss": 0.0006,
1327
+ "rewards/accuracies": 1.0,
1328
+ "rewards/chosen": -1.2734375,
1329
+ "rewards/margins": 10.875,
1330
+ "rewards/rejected": -12.1875,
1331
+ "step": 850
1332
+ },
1333
+ {
1334
+ "epoch": 2.774193548387097,
1335
+ "grad_norm": 1.5601788534872705,
1336
+ "learning_rate": 4.181600955794504e-08,
1337
+ "logits/chosen": -3.484375,
1338
+ "logits/rejected": -3.625,
1339
+ "logps/chosen": -206.0,
1340
+ "logps/rejected": -326.0,
1341
+ "loss": 0.0009,
1342
+ "rewards/accuracies": 1.0,
1343
+ "rewards/chosen": -1.265625,
1344
+ "rewards/margins": 11.1875,
1345
+ "rewards/rejected": -12.4375,
1346
+ "step": 860
1347
+ },
1348
+ {
1349
+ "epoch": 2.806451612903226,
1350
+ "grad_norm": 2.7780970090627104,
1351
+ "learning_rate": 3.584229390681003e-08,
1352
+ "logits/chosen": -3.53125,
1353
+ "logits/rejected": -3.65625,
1354
+ "logps/chosen": -201.0,
1355
+ "logps/rejected": -320.0,
1356
+ "loss": 0.0025,
1357
+ "rewards/accuracies": 1.0,
1358
+ "rewards/chosen": -1.0546875,
1359
+ "rewards/margins": 10.8125,
1360
+ "rewards/rejected": -11.875,
1361
+ "step": 870
1362
+ },
1363
+ {
1364
+ "epoch": 2.838709677419355,
1365
+ "grad_norm": 0.36034720795454384,
1366
+ "learning_rate": 2.986857825567503e-08,
1367
+ "logits/chosen": -3.546875,
1368
+ "logits/rejected": -3.578125,
1369
+ "logps/chosen": -198.0,
1370
+ "logps/rejected": -318.0,
1371
+ "loss": 0.0008,
1372
+ "rewards/accuracies": 1.0,
1373
+ "rewards/chosen": -0.86328125,
1374
+ "rewards/margins": 11.25,
1375
+ "rewards/rejected": -12.125,
1376
+ "step": 880
1377
+ },
1378
+ {
1379
+ "epoch": 2.870967741935484,
1380
+ "grad_norm": 0.2764304820275906,
1381
+ "learning_rate": 2.3894862604540023e-08,
1382
+ "logits/chosen": -3.59375,
1383
+ "logits/rejected": -3.671875,
1384
+ "logps/chosen": -203.0,
1385
+ "logps/rejected": -316.0,
1386
+ "loss": 0.0037,
1387
+ "rewards/accuracies": 1.0,
1388
+ "rewards/chosen": -1.03125,
1389
+ "rewards/margins": 10.6875,
1390
+ "rewards/rejected": -11.75,
1391
+ "step": 890
1392
+ },
1393
+ {
1394
+ "epoch": 2.903225806451613,
1395
+ "grad_norm": 0.4293826404732342,
1396
+ "learning_rate": 1.7921146953405015e-08,
1397
+ "logits/chosen": -3.4375,
1398
+ "logits/rejected": -3.546875,
1399
+ "logps/chosen": -208.0,
1400
+ "logps/rejected": -322.0,
1401
+ "loss": 0.0011,
1402
+ "rewards/accuracies": 1.0,
1403
+ "rewards/chosen": -1.859375,
1404
+ "rewards/margins": 10.875,
1405
+ "rewards/rejected": -12.6875,
1406
+ "step": 900
1407
+ },
1408
+ {
1409
+ "epoch": 2.935483870967742,
1410
+ "grad_norm": 0.34603080271508346,
1411
+ "learning_rate": 1.1947431302270011e-08,
1412
+ "logits/chosen": -3.546875,
1413
+ "logits/rejected": -3.703125,
1414
+ "logps/chosen": -207.0,
1415
+ "logps/rejected": -320.0,
1416
+ "loss": 0.0007,
1417
+ "rewards/accuracies": 1.0,
1418
+ "rewards/chosen": -1.265625,
1419
+ "rewards/margins": 11.0,
1420
+ "rewards/rejected": -12.25,
1421
+ "step": 910
1422
+ },
1423
+ {
1424
+ "epoch": 2.967741935483871,
1425
+ "grad_norm": 2.3868143014919885,
1426
+ "learning_rate": 5.973715651135006e-09,
1427
+ "logits/chosen": -3.4375,
1428
+ "logits/rejected": -3.65625,
1429
+ "logps/chosen": -188.0,
1430
+ "logps/rejected": -304.0,
1431
+ "loss": 0.002,
1432
+ "rewards/accuracies": 1.0,
1433
+ "rewards/chosen": -0.71484375,
1434
+ "rewards/margins": 10.375,
1435
+ "rewards/rejected": -11.125,
1436
+ "step": 920
1437
+ },
1438
+ {
1439
+ "epoch": 3.0,
1440
+ "grad_norm": 0.40772402916251804,
1441
+ "learning_rate": 0.0,
1442
+ "logits/chosen": -3.5,
1443
+ "logits/rejected": -3.609375,
1444
+ "logps/chosen": -194.0,
1445
+ "logps/rejected": -320.0,
1446
+ "loss": 0.0006,
1447
+ "rewards/accuracies": 1.0,
1448
+ "rewards/chosen": -1.1796875,
1449
+ "rewards/margins": 10.9375,
1450
+ "rewards/rejected": -12.125,
1451
+ "step": 930
1452
+ },
1453
+ {
1454
+ "epoch": 3.0,
1455
+ "eval_logits/chosen": -3.734375,
1456
+ "eval_logits/rejected": -3.75,
1457
+ "eval_logps/chosen": -226.0,
1458
+ "eval_logps/rejected": -294.0,
1459
+ "eval_loss": 0.19125854969024658,
1460
+ "eval_rewards/accuracies": 0.8928571343421936,
1461
+ "eval_rewards/chosen": -2.703125,
1462
+ "eval_rewards/margins": 6.3125,
1463
+ "eval_rewards/rejected": -9.0,
1464
+ "eval_runtime": 16.0795,
1465
+ "eval_samples_per_second": 12.438,
1466
+ "eval_steps_per_second": 0.435,
1467
+ "step": 930
1468
  }
1469
  ],
1470
  "logging_steps": 10,
 
1479
  "should_evaluate": false,
1480
  "should_log": false,
1481
  "should_save": true,
1482
+ "should_training_stop": true
1483
  },
1484
  "attributes": {}
1485
  }