AmberYifan commited on
Commit
4143474
·
verified ·
1 Parent(s): 404f1aa

Training in progress, epoch 2, checkpoint

Browse files
last-checkpoint/global_step313/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:b581dd8c870c022eb0a39660a71f246a4781f76a360bc13913d5b5d4dc8a0d95
3
+ size 14483467880
last-checkpoint/global_step313/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:4a152f57ed8c6b920b05aad58c3c4b35ca6691a9f60ceff526d1ca3f12965d02
3
+ size 14483467880
last-checkpoint/global_step313/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:f32fec7d0ed61ac327a6485380601f3836a5abd4b21f42105418b867ecef0ee2
3
+ size 14483467880
last-checkpoint/global_step313/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:8e5f1ca50a923ba4996dd985e3955c4dd9e905844c25df9de7943c292bd97fcc
3
+ size 14483467880
last-checkpoint/global_step313/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:55303416a3517f901134c4f1eff39ac26ac577d0fe793b38c5fe0d52d363bb91
3
+ size 150629
last-checkpoint/global_step313/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:4f254e726826fe433aca4e6eb4dbb78b3d5a0040927776a85ff62a2e33445beb
3
+ size 150629
last-checkpoint/global_step313/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:4e243089f4e8eb95ac174c0d1ad91b55da0355072bfac98e1cdcb8eb28b9aaed
3
+ size 150629
last-checkpoint/global_step313/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:5be5cf16cd15b49f3e19ae306c159c18afbe73756804590a6d0824388d12ff62
3
+ size 150629
last-checkpoint/latest CHANGED
@@ -1 +1 @@
1
- global_step156
 
1
+ global_step313
last-checkpoint/model-00001-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:86564566d04a0b1060db4669fe75cb07fbe4d4936cd25d2f934f8aeac28986b2
3
  size 4943162336
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:24d250e478ceec67ad697c5114ef84ef83195f94df1fc2e6d5423a259d96a851
3
  size 4943162336
last-checkpoint/model-00002-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:edf6aa834d50bf27e9aaa84927e709a4df4277057494957e4f927a1d47c94b63
3
  size 4999819336
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:746db1161ed175dafa557b81124b357a95144eb8daeac6de2459b049de70b0d9
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:6e5b931e4af6dbf48cbcfa178bf23b762a598dc466d811ac992cf5560ad7f669
3
  size 4540516344
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:299dfa5fb3aa044f8ce0ab7dabebb0b9db464ada1bb1b84169a35ee309b44cbd
3
  size 4540516344
last-checkpoint/rng_state_0.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a4a1dc8b9f546123ce9b44c28828ae71d9d0ecd6f59c6b7b8337c73cb2e24c49
3
  size 15024
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50ca27603f7c4a4d9a980223e1a102922ac31561acfa7da4f28042e4e8b3528b
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:0a8b225b0323772b66570a16b5c7e6446dc3c9ee076597906a2c90e3c0a383bf
3
  size 15024
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c57f658c3d9b6de611da50021823095fc4067036fb9a4b34655cc2e34fbf5c2b
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:c036a8d528905f7018266618c8bebe4add10c2778ef432fd0116369fa622d815
3
  size 15024
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:109da2c363a1e09432fa2ea2b7dceb6f67d1f91975069fc669e71dd536d654d8
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:b93d88429d97654e76d10b1ba4c36f29a47e210c65a03efdc136f04af8282797
3
  size 15024
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bda65e490958bfe3fe01157ed131eb2354f8fc19536842d27c998ecf29d48991
3
  size 15024
last-checkpoint/scheduler.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a6db26c7d425273441931218becb02d21e92e0e82e094f8c6441cf4c20b59eb2
3
  size 1064
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1ae44c8e61affb3be5c9387c2b397935fdb2ed21d12002009ab043a8a5fffb07
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": 0.9968051118210862,
5
  "eval_steps": 500,
6
- "global_step": 156,
7
  "is_hyper_param_search": false,
8
  "is_local_process_zero": true,
9
  "is_world_process_zero": true,
@@ -263,6 +263,262 @@
263
  "eval_samples_per_second": 16.291,
264
  "eval_steps_per_second": 0.57,
265
  "step": 156
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
266
  }
267
  ],
268
  "logging_steps": 10,
 
1
  {
2
  "best_metric": null,
3
  "best_model_checkpoint": null,
4
+ "epoch": 2.0,
5
  "eval_steps": 500,
6
+ "global_step": 313,
7
  "is_hyper_param_search": false,
8
  "is_local_process_zero": true,
9
  "is_world_process_zero": true,
 
263
  "eval_samples_per_second": 16.291,
264
  "eval_steps_per_second": 0.57,
265
  "step": 156
266
+ },
267
+ {
268
+ "epoch": 1.0223642172523961,
269
+ "grad_norm": 17.578277558626937,
270
+ "learning_rate": 3.6579572446555817e-07,
271
+ "logits/chosen": -3.59375,
272
+ "logits/rejected": -3.578125,
273
+ "logps/chosen": -163.0,
274
+ "logps/rejected": -165.0,
275
+ "loss": 0.5154,
276
+ "rewards/accuracies": 0.6937500238418579,
277
+ "rewards/chosen": -0.337890625,
278
+ "rewards/margins": 0.8125,
279
+ "rewards/rejected": -1.1484375,
280
+ "step": 160
281
+ },
282
+ {
283
+ "epoch": 1.0862619808306708,
284
+ "grad_norm": 18.045497494579415,
285
+ "learning_rate": 3.5391923990498813e-07,
286
+ "logits/chosen": -3.609375,
287
+ "logits/rejected": -3.5625,
288
+ "logps/chosen": -166.0,
289
+ "logps/rejected": -161.0,
290
+ "loss": 0.2778,
291
+ "rewards/accuracies": 0.9375,
292
+ "rewards/chosen": 0.28515625,
293
+ "rewards/margins": 2.03125,
294
+ "rewards/rejected": -1.75,
295
+ "step": 170
296
+ },
297
+ {
298
+ "epoch": 1.1501597444089458,
299
+ "grad_norm": 31.421762384244513,
300
+ "learning_rate": 3.42042755344418e-07,
301
+ "logits/chosen": -3.6875,
302
+ "logits/rejected": -3.65625,
303
+ "logps/chosen": -169.0,
304
+ "logps/rejected": -163.0,
305
+ "loss": 0.2699,
306
+ "rewards/accuracies": 0.9375,
307
+ "rewards/chosen": 0.0830078125,
308
+ "rewards/margins": 1.78125,
309
+ "rewards/rejected": -1.6953125,
310
+ "step": 180
311
+ },
312
+ {
313
+ "epoch": 1.2140575079872205,
314
+ "grad_norm": 23.922864036418108,
315
+ "learning_rate": 3.3016627078384796e-07,
316
+ "logits/chosen": -3.6875,
317
+ "logits/rejected": -3.625,
318
+ "logps/chosen": -170.0,
319
+ "logps/rejected": -160.0,
320
+ "loss": 0.2828,
321
+ "rewards/accuracies": 0.90625,
322
+ "rewards/chosen": -0.0189208984375,
323
+ "rewards/margins": 1.96875,
324
+ "rewards/rejected": -1.984375,
325
+ "step": 190
326
+ },
327
+ {
328
+ "epoch": 1.2779552715654952,
329
+ "grad_norm": 35.48813795033075,
330
+ "learning_rate": 3.182897862232779e-07,
331
+ "logits/chosen": -3.6875,
332
+ "logits/rejected": -3.59375,
333
+ "logps/chosen": -166.0,
334
+ "logps/rejected": -172.0,
335
+ "loss": 0.2671,
336
+ "rewards/accuracies": 0.925000011920929,
337
+ "rewards/chosen": 0.0281982421875,
338
+ "rewards/margins": 2.0,
339
+ "rewards/rejected": -1.96875,
340
+ "step": 200
341
+ },
342
+ {
343
+ "epoch": 1.34185303514377,
344
+ "grad_norm": 26.443386433583438,
345
+ "learning_rate": 3.0641330166270784e-07,
346
+ "logits/chosen": -3.671875,
347
+ "logits/rejected": -3.609375,
348
+ "logps/chosen": -170.0,
349
+ "logps/rejected": -168.0,
350
+ "loss": 0.2989,
351
+ "rewards/accuracies": 0.893750011920929,
352
+ "rewards/chosen": 0.16796875,
353
+ "rewards/margins": 1.875,
354
+ "rewards/rejected": -1.7109375,
355
+ "step": 210
356
+ },
357
+ {
358
+ "epoch": 1.4057507987220448,
359
+ "grad_norm": 21.45349324293884,
360
+ "learning_rate": 2.9453681710213776e-07,
361
+ "logits/chosen": -3.671875,
362
+ "logits/rejected": -3.65625,
363
+ "logps/chosen": -172.0,
364
+ "logps/rejected": -171.0,
365
+ "loss": 0.2874,
366
+ "rewards/accuracies": 0.893750011920929,
367
+ "rewards/chosen": -0.26171875,
368
+ "rewards/margins": 1.8125,
369
+ "rewards/rejected": -2.078125,
370
+ "step": 220
371
+ },
372
+ {
373
+ "epoch": 1.4696485623003195,
374
+ "grad_norm": 15.23988385292947,
375
+ "learning_rate": 2.8266033254156767e-07,
376
+ "logits/chosen": -3.65625,
377
+ "logits/rejected": -3.625,
378
+ "logps/chosen": -152.0,
379
+ "logps/rejected": -154.0,
380
+ "loss": 0.3156,
381
+ "rewards/accuracies": 0.918749988079071,
382
+ "rewards/chosen": -0.2158203125,
383
+ "rewards/margins": 1.8046875,
384
+ "rewards/rejected": -2.015625,
385
+ "step": 230
386
+ },
387
+ {
388
+ "epoch": 1.5335463258785942,
389
+ "grad_norm": 20.9068024116458,
390
+ "learning_rate": 2.7078384798099764e-07,
391
+ "logits/chosen": -3.6875,
392
+ "logits/rejected": -3.640625,
393
+ "logps/chosen": -166.0,
394
+ "logps/rejected": -179.0,
395
+ "loss": 0.2663,
396
+ "rewards/accuracies": 0.949999988079071,
397
+ "rewards/chosen": -0.17578125,
398
+ "rewards/margins": 2.296875,
399
+ "rewards/rejected": -2.46875,
400
+ "step": 240
401
+ },
402
+ {
403
+ "epoch": 1.5974440894568689,
404
+ "grad_norm": 23.61858832933539,
405
+ "learning_rate": 2.589073634204275e-07,
406
+ "logits/chosen": -3.625,
407
+ "logits/rejected": -3.59375,
408
+ "logps/chosen": -169.0,
409
+ "logps/rejected": -175.0,
410
+ "loss": 0.255,
411
+ "rewards/accuracies": 0.9312499761581421,
412
+ "rewards/chosen": -0.302734375,
413
+ "rewards/margins": 2.203125,
414
+ "rewards/rejected": -2.515625,
415
+ "step": 250
416
+ },
417
+ {
418
+ "epoch": 1.6613418530351438,
419
+ "grad_norm": 15.684487484295666,
420
+ "learning_rate": 2.4703087885985747e-07,
421
+ "logits/chosen": -3.6875,
422
+ "logits/rejected": -3.65625,
423
+ "logps/chosen": -176.0,
424
+ "logps/rejected": -178.0,
425
+ "loss": 0.279,
426
+ "rewards/accuracies": 0.8812500238418579,
427
+ "rewards/chosen": -0.3984375,
428
+ "rewards/margins": 1.984375,
429
+ "rewards/rejected": -2.375,
430
+ "step": 260
431
+ },
432
+ {
433
+ "epoch": 1.7252396166134185,
434
+ "grad_norm": 17.902346692684933,
435
+ "learning_rate": 2.351543942992874e-07,
436
+ "logits/chosen": -3.6875,
437
+ "logits/rejected": -3.625,
438
+ "logps/chosen": -175.0,
439
+ "logps/rejected": -186.0,
440
+ "loss": 0.27,
441
+ "rewards/accuracies": 0.887499988079071,
442
+ "rewards/chosen": -0.29296875,
443
+ "rewards/margins": 2.140625,
444
+ "rewards/rejected": -2.4375,
445
+ "step": 270
446
+ },
447
+ {
448
+ "epoch": 1.7891373801916934,
449
+ "grad_norm": 16.024604674297596,
450
+ "learning_rate": 2.2327790973871732e-07,
451
+ "logits/chosen": -3.65625,
452
+ "logits/rejected": -3.609375,
453
+ "logps/chosen": -178.0,
454
+ "logps/rejected": -186.0,
455
+ "loss": 0.3036,
456
+ "rewards/accuracies": 0.887499988079071,
457
+ "rewards/chosen": -0.76171875,
458
+ "rewards/margins": 2.15625,
459
+ "rewards/rejected": -2.921875,
460
+ "step": 280
461
+ },
462
+ {
463
+ "epoch": 1.8530351437699681,
464
+ "grad_norm": 20.910031176692122,
465
+ "learning_rate": 2.1140142517814726e-07,
466
+ "logits/chosen": -3.625,
467
+ "logits/rejected": -3.59375,
468
+ "logps/chosen": -174.0,
469
+ "logps/rejected": -182.0,
470
+ "loss": 0.257,
471
+ "rewards/accuracies": 0.9437500238418579,
472
+ "rewards/chosen": -0.0810546875,
473
+ "rewards/margins": 2.5625,
474
+ "rewards/rejected": -2.640625,
475
+ "step": 290
476
+ },
477
+ {
478
+ "epoch": 1.9169329073482428,
479
+ "grad_norm": 26.07564194077576,
480
+ "learning_rate": 1.9952494061757718e-07,
481
+ "logits/chosen": -3.65625,
482
+ "logits/rejected": -3.578125,
483
+ "logps/chosen": -167.0,
484
+ "logps/rejected": -170.0,
485
+ "loss": 0.3132,
486
+ "rewards/accuracies": 0.918749988079071,
487
+ "rewards/chosen": -0.2099609375,
488
+ "rewards/margins": 2.234375,
489
+ "rewards/rejected": -2.4375,
490
+ "step": 300
491
+ },
492
+ {
493
+ "epoch": 1.9808306709265175,
494
+ "grad_norm": 26.49958432728375,
495
+ "learning_rate": 1.876484560570071e-07,
496
+ "logits/chosen": -3.71875,
497
+ "logits/rejected": -3.703125,
498
+ "logps/chosen": -165.0,
499
+ "logps/rejected": -174.0,
500
+ "loss": 0.2996,
501
+ "rewards/accuracies": 0.875,
502
+ "rewards/chosen": -0.400390625,
503
+ "rewards/margins": 2.03125,
504
+ "rewards/rejected": -2.4375,
505
+ "step": 310
506
+ },
507
+ {
508
+ "epoch": 2.0,
509
+ "eval_logits/chosen": -3.765625,
510
+ "eval_logits/rejected": -3.796875,
511
+ "eval_logps/chosen": -167.0,
512
+ "eval_logps/rejected": -154.0,
513
+ "eval_loss": 0.5964062213897705,
514
+ "eval_rewards/accuracies": 0.7142857313156128,
515
+ "eval_rewards/chosen": -0.86328125,
516
+ "eval_rewards/margins": 0.5546875,
517
+ "eval_rewards/rejected": -1.4140625,
518
+ "eval_runtime": 12.2832,
519
+ "eval_samples_per_second": 16.282,
520
+ "eval_steps_per_second": 0.57,
521
+ "step": 313
522
  }
523
  ],
524
  "logging_steps": 10,