File size: 4,265 Bytes
4ec7d43
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.9922480620155039,
  "eval_steps": 100,
  "global_step": 64,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.02,
      "grad_norm": 5.753161071734387,
      "learning_rate": 7.142857142857142e-08,
      "logits/chosen": -1.146815299987793,
      "logits/rejected": -1.0666704177856445,
      "logps/chosen": -606.579345703125,
      "logps/rejected": -505.8602294921875,
      "loss": 0.6931,
      "rewards/accuracies": 0.0,
      "rewards/chosen": 0.0,
      "rewards/margins": 0.0,
      "rewards/rejected": 0.0,
      "step": 1
    },
    {
      "epoch": 0.16,
      "grad_norm": 5.7493431535662305,
      "learning_rate": 4.965903258506806e-07,
      "logits/chosen": -1.1877351999282837,
      "logits/rejected": -1.1102757453918457,
      "logps/chosen": -546.2195434570312,
      "logps/rejected": -661.2600708007812,
      "loss": 0.693,
      "rewards/accuracies": 0.5277777910232544,
      "rewards/chosen": 0.0011070389300584793,
      "rewards/margins": 0.001627296325750649,
      "rewards/rejected": -0.0005202573956921697,
      "step": 10
    },
    {
      "epoch": 0.31,
      "grad_norm": 5.1113110991031645,
      "learning_rate": 4.38526652444224e-07,
      "logits/chosen": -1.182856559753418,
      "logits/rejected": -1.096764326095581,
      "logps/chosen": -496.00006103515625,
      "logps/rejected": -631.6556396484375,
      "loss": 0.6878,
      "rewards/accuracies": 0.8125,
      "rewards/chosen": 0.011512617580592632,
      "rewards/margins": 0.011010391637682915,
      "rewards/rejected": 0.0005022261175327003,
      "step": 20
    },
    {
      "epoch": 0.47,
      "grad_norm": 5.707953313171029,
      "learning_rate": 3.2462870275042367e-07,
      "logits/chosen": -1.2160546779632568,
      "logits/rejected": -1.1250536441802979,
      "logps/chosen": -462.3262634277344,
      "logps/rejected": -688.6295166015625,
      "loss": 0.6789,
      "rewards/accuracies": 0.8374999761581421,
      "rewards/chosen": 0.03221649304032326,
      "rewards/margins": 0.029845599085092545,
      "rewards/rejected": 0.0023708937224000692,
      "step": 30
    },
    {
      "epoch": 0.62,
      "grad_norm": 5.28907700413882,
      "learning_rate": 1.886286282148002e-07,
      "logits/chosen": -1.2214853763580322,
      "logits/rejected": -1.1557971239089966,
      "logps/chosen": -473.38720703125,
      "logps/rejected": -619.6802978515625,
      "loss": 0.671,
      "rewards/accuracies": 0.8812500238418579,
      "rewards/chosen": 0.049382228404283524,
      "rewards/margins": 0.04463439807295799,
      "rewards/rejected": 0.004747826140373945,
      "step": 40
    },
    {
      "epoch": 0.78,
      "grad_norm": 5.6548124405750775,
      "learning_rate": 7.080437170788722e-08,
      "logits/chosen": -1.2320277690887451,
      "logits/rejected": -1.094261884689331,
      "logps/chosen": -481.0409240722656,
      "logps/rejected": -643.069091796875,
      "loss": 0.6664,
      "rewards/accuracies": 0.8999999761581421,
      "rewards/chosen": 0.05804261565208435,
      "rewards/margins": 0.053116798400878906,
      "rewards/rejected": 0.0049258084036409855,
      "step": 50
    },
    {
      "epoch": 0.93,
      "grad_norm": 5.6228626485550155,
      "learning_rate": 6.0509043431410945e-09,
      "logits/chosen": -1.2452279329299927,
      "logits/rejected": -1.109863042831421,
      "logps/chosen": -473.0909729003906,
      "logps/rejected": -635.895263671875,
      "loss": 0.6614,
      "rewards/accuracies": 0.8999999761581421,
      "rewards/chosen": 0.05887182429432869,
      "rewards/margins": 0.060193151235580444,
      "rewards/rejected": -0.0013213239144533873,
      "step": 60
    },
    {
      "epoch": 0.99,
      "step": 64,
      "total_flos": 0.0,
      "train_loss": 0.6759306937456131,
      "train_runtime": 862.557,
      "train_samples_per_second": 4.756,
      "train_steps_per_second": 0.074
    }
  ],
  "logging_steps": 10,
  "max_steps": 64,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 100,
  "total_flos": 0.0,
  "train_batch_size": 8,
  "trial_name": null,
  "trial_params": null
}