lancehar commited on
Commit
c3b3cb0
·
verified ·
1 Parent(s): 15d5dca

Upload mean_layernorm_linear_seed6 reward model

Browse files
Files changed (48) hide show
  1. README.md +28 -0
  2. checkpoint-4000/config.json +31 -0
  3. checkpoint-4000/optimizer.pt +3 -0
  4. checkpoint-4000/pytorch_model.bin +3 -0
  5. checkpoint-4000/rng_state.pth +3 -0
  6. checkpoint-4000/scheduler.pt +3 -0
  7. checkpoint-4000/special_tokens_map.json +14 -0
  8. checkpoint-4000/tokenizer.json +0 -0
  9. checkpoint-4000/tokenizer_config.json +9 -0
  10. checkpoint-4000/trainer_state.json +2591 -0
  11. checkpoint-4000/training_args.bin +3 -0
  12. checkpoint-5000/config.json +31 -0
  13. checkpoint-5000/optimizer.pt +3 -0
  14. checkpoint-5000/pytorch_model.bin +3 -0
  15. checkpoint-5000/rng_state.pth +3 -0
  16. checkpoint-5000/scheduler.pt +3 -0
  17. checkpoint-5000/special_tokens_map.json +14 -0
  18. checkpoint-5000/tokenizer.json +0 -0
  19. checkpoint-5000/tokenizer_config.json +9 -0
  20. checkpoint-5000/trainer_state.json +3230 -0
  21. checkpoint-5000/training_args.bin +3 -0
  22. checkpoint-6000/config.json +31 -0
  23. checkpoint-6000/optimizer.pt +3 -0
  24. checkpoint-6000/pytorch_model.bin +3 -0
  25. checkpoint-6000/rng_state.pth +3 -0
  26. checkpoint-6000/scheduler.pt +3 -0
  27. checkpoint-6000/special_tokens_map.json +14 -0
  28. checkpoint-6000/tokenizer.json +0 -0
  29. checkpoint-6000/tokenizer_config.json +9 -0
  30. checkpoint-6000/trainer_state.json +3882 -0
  31. checkpoint-6000/training_args.bin +3 -0
  32. checkpoint-7000/config.json +31 -0
  33. checkpoint-7000/optimizer.pt +3 -0
  34. checkpoint-7000/pytorch_model.bin +3 -0
  35. checkpoint-7000/rng_state.pth +3 -0
  36. checkpoint-7000/scheduler.pt +3 -0
  37. checkpoint-7000/special_tokens_map.json +14 -0
  38. checkpoint-7000/tokenizer.json +0 -0
  39. checkpoint-7000/tokenizer_config.json +9 -0
  40. checkpoint-7000/trainer_state.json +0 -0
  41. checkpoint-7000/training_args.bin +3 -0
  42. config.json +37 -0
  43. gpt_neox_reward_model.py +188 -0
  44. pytorch_model.bin +3 -0
  45. special_tokens_map.json +14 -0
  46. tokenizer.json +0 -0
  47. tokenizer_config.json +9 -0
  48. training_args.bin +3 -0
README.md ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ tags:
4
+ - reward-model
5
+ - gpt-neox
6
+ - transformers
7
+ ---
8
+
9
+ # lancehar/mean_layernorm_linear_seed6
10
+
11
+ This repository contains the `mean_layernorm_linear_seed6` head-diverse reward model checkpoint from
12
+ `mc-dropout-vs-reward-ensemble`.
13
+
14
+ ## Loading
15
+
16
+ ```python
17
+ from transformers import AutoModelForSequenceClassification, AutoTokenizer
18
+
19
+ model_name = "lancehar/mean_layernorm_linear_seed6"
20
+ tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
21
+ model = AutoModelForSequenceClassification.from_pretrained(
22
+ model_name,
23
+ trust_remote_code=True,
24
+ )
25
+ ```
26
+
27
+ These checkpoints use a custom GPT-NeoX reward-model class, so `trust_remote_code=True`
28
+ is required unless you register the class locally from this repository.
checkpoint-4000/config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "tlc4418/pythia_70m_sft",
3
+ "architectures": [
4
+ "GPTNeoXRewardModel"
5
+ ],
6
+ "bos_token_id": 0,
7
+ "classifier_dropout": 0.1,
8
+ "eos_token_id": 0,
9
+ "hidden_act": "gelu",
10
+ "hidden_size": 512,
11
+ "initializer_range": 0.02,
12
+ "intermediate_size": 2048,
13
+ "layer_norm_eps": 1e-05,
14
+ "max_position_embeddings": 2048,
15
+ "model_type": "gpt_neox_reward_model",
16
+ "num_attention_heads": 8,
17
+ "num_hidden_layers": 6,
18
+ "pooling": "mean",
19
+ "reward_head_activation": "gelu",
20
+ "reward_head_dropout": 0.0,
21
+ "reward_head_hidden_dim": null,
22
+ "reward_head_type": "layernorm_linear",
23
+ "rotary_emb_base": 10000,
24
+ "rotary_pct": 0.25,
25
+ "tie_word_embeddings": false,
26
+ "torch_dtype": "bfloat16",
27
+ "transformers_version": "4.29.2",
28
+ "use_cache": true,
29
+ "use_parallel_residual": true,
30
+ "vocab_size": 50288
31
+ }
checkpoint-4000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:06d30e4d56d014d4ac2bba86c71e01ff8493f77e83002e749d7de6834877b055
3
+ size 178704331
checkpoint-4000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:efd31e00d6cef1272ded3193fae34b15e732f44a87e1ba4ea7d927cfd85f5a7c
3
+ size 114527658
checkpoint-4000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e57736f8b4fc3a5fc2f8d642816ea7d696c62a173488e45887a3b8588800ba9c
3
+ size 14645
checkpoint-4000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:04ceebf0b7c7ac22dd16c128651c9f8b29978e2875592278bb5c0914bd31292a
3
+ size 1465
checkpoint-4000/special_tokens_map.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<|prefix_end|>",
4
+ "<|assistant|>",
5
+ "<|prefix_begin|>",
6
+ "<|prompter|>",
7
+ "<|system|>"
8
+ ],
9
+ "bos_token": "<|endoftext|>",
10
+ "eos_token": "<|endoftext|>",
11
+ "pad_token": "<|padding|>",
12
+ "sep_token": "<|endoftext|>",
13
+ "unk_token": "<|endoftext|>"
14
+ }
checkpoint-4000/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-4000/tokenizer_config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "bos_token": "<|endoftext|>",
4
+ "clean_up_tokenization_spaces": true,
5
+ "eos_token": "<|endoftext|>",
6
+ "model_max_length": 1000000000000000019884624838656,
7
+ "tokenizer_class": "GPTNeoXTokenizer",
8
+ "unk_token": "<|endoftext|>"
9
+ }
checkpoint-4000/trainer_state.json ADDED
@@ -0,0 +1,2591 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 2.782608695652174,
5
+ "global_step": 4000,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.0,
12
+ "learning_rate": 2.0000000000000002e-07,
13
+ "loss": 0.6842,
14
+ "step": 1
15
+ },
16
+ {
17
+ "epoch": 0.01,
18
+ "learning_rate": 2.0000000000000003e-06,
19
+ "loss": 0.6871,
20
+ "step": 10
21
+ },
22
+ {
23
+ "epoch": 0.01,
24
+ "learning_rate": 4.000000000000001e-06,
25
+ "loss": 0.691,
26
+ "step": 20
27
+ },
28
+ {
29
+ "epoch": 0.02,
30
+ "learning_rate": 6e-06,
31
+ "loss": 0.6848,
32
+ "step": 30
33
+ },
34
+ {
35
+ "epoch": 0.03,
36
+ "learning_rate": 8.000000000000001e-06,
37
+ "loss": 0.6873,
38
+ "step": 40
39
+ },
40
+ {
41
+ "epoch": 0.03,
42
+ "learning_rate": 1e-05,
43
+ "loss": 0.6902,
44
+ "step": 50
45
+ },
46
+ {
47
+ "epoch": 0.04,
48
+ "learning_rate": 9.985984583041346e-06,
49
+ "loss": 0.6842,
50
+ "step": 60
51
+ },
52
+ {
53
+ "epoch": 0.05,
54
+ "learning_rate": 9.971969166082692e-06,
55
+ "loss": 0.6908,
56
+ "step": 70
57
+ },
58
+ {
59
+ "epoch": 0.06,
60
+ "learning_rate": 9.957953749124038e-06,
61
+ "loss": 0.6865,
62
+ "step": 80
63
+ },
64
+ {
65
+ "epoch": 0.06,
66
+ "learning_rate": 9.943938332165382e-06,
67
+ "loss": 0.6834,
68
+ "step": 90
69
+ },
70
+ {
71
+ "epoch": 0.07,
72
+ "learning_rate": 9.929922915206728e-06,
73
+ "loss": 0.6884,
74
+ "step": 100
75
+ },
76
+ {
77
+ "epoch": 0.08,
78
+ "learning_rate": 9.915907498248075e-06,
79
+ "loss": 0.6837,
80
+ "step": 110
81
+ },
82
+ {
83
+ "epoch": 0.08,
84
+ "learning_rate": 9.901892081289419e-06,
85
+ "loss": 0.6883,
86
+ "step": 120
87
+ },
88
+ {
89
+ "epoch": 0.09,
90
+ "learning_rate": 9.887876664330765e-06,
91
+ "loss": 0.6816,
92
+ "step": 130
93
+ },
94
+ {
95
+ "epoch": 0.1,
96
+ "learning_rate": 9.87386124737211e-06,
97
+ "loss": 0.6906,
98
+ "step": 140
99
+ },
100
+ {
101
+ "epoch": 0.1,
102
+ "learning_rate": 9.859845830413456e-06,
103
+ "loss": 0.6875,
104
+ "step": 150
105
+ },
106
+ {
107
+ "epoch": 0.11,
108
+ "learning_rate": 9.845830413454802e-06,
109
+ "loss": 0.6805,
110
+ "step": 160
111
+ },
112
+ {
113
+ "epoch": 0.12,
114
+ "learning_rate": 9.831814996496146e-06,
115
+ "loss": 0.6857,
116
+ "step": 170
117
+ },
118
+ {
119
+ "epoch": 0.13,
120
+ "learning_rate": 9.817799579537493e-06,
121
+ "loss": 0.6804,
122
+ "step": 180
123
+ },
124
+ {
125
+ "epoch": 0.13,
126
+ "learning_rate": 9.803784162578837e-06,
127
+ "loss": 0.6803,
128
+ "step": 190
129
+ },
130
+ {
131
+ "epoch": 0.14,
132
+ "learning_rate": 9.789768745620183e-06,
133
+ "loss": 0.6872,
134
+ "step": 200
135
+ },
136
+ {
137
+ "epoch": 0.15,
138
+ "learning_rate": 9.775753328661528e-06,
139
+ "loss": 0.6846,
140
+ "step": 210
141
+ },
142
+ {
143
+ "epoch": 0.15,
144
+ "learning_rate": 9.761737911702874e-06,
145
+ "loss": 0.6828,
146
+ "step": 220
147
+ },
148
+ {
149
+ "epoch": 0.16,
150
+ "learning_rate": 9.74772249474422e-06,
151
+ "loss": 0.6699,
152
+ "step": 230
153
+ },
154
+ {
155
+ "epoch": 0.17,
156
+ "learning_rate": 9.733707077785565e-06,
157
+ "loss": 0.673,
158
+ "step": 240
159
+ },
160
+ {
161
+ "epoch": 0.17,
162
+ "learning_rate": 9.71969166082691e-06,
163
+ "loss": 0.6733,
164
+ "step": 250
165
+ },
166
+ {
167
+ "epoch": 0.18,
168
+ "learning_rate": 9.705676243868257e-06,
169
+ "loss": 0.6674,
170
+ "step": 260
171
+ },
172
+ {
173
+ "epoch": 0.19,
174
+ "learning_rate": 9.6916608269096e-06,
175
+ "loss": 0.6708,
176
+ "step": 270
177
+ },
178
+ {
179
+ "epoch": 0.19,
180
+ "learning_rate": 9.677645409950946e-06,
181
+ "loss": 0.6661,
182
+ "step": 280
183
+ },
184
+ {
185
+ "epoch": 0.2,
186
+ "learning_rate": 9.663629992992292e-06,
187
+ "loss": 0.677,
188
+ "step": 290
189
+ },
190
+ {
191
+ "epoch": 0.21,
192
+ "learning_rate": 9.649614576033638e-06,
193
+ "loss": 0.6657,
194
+ "step": 300
195
+ },
196
+ {
197
+ "epoch": 0.21,
198
+ "eval_custom_hf_pref_accuracy": 0.5485,
199
+ "eval_custom_hf_pref_kendalltau": 0.097,
200
+ "eval_custom_hf_pref_loss": 0.6715624928474426,
201
+ "eval_custom_hf_pref_neg_score": 0.46836376190185547,
202
+ "eval_custom_hf_pref_pos_score": 0.522513747215271,
203
+ "eval_custom_hf_pref_runtime": 3.1872,
204
+ "eval_custom_hf_pref_samples_per_second": 627.506,
205
+ "eval_custom_hf_pref_score_diff": 0.054149970412254333,
206
+ "eval_custom_hf_pref_steps_per_second": 78.438,
207
+ "step": 300
208
+ },
209
+ {
210
+ "epoch": 0.22,
211
+ "learning_rate": 9.635599159074983e-06,
212
+ "loss": 0.6747,
213
+ "step": 310
214
+ },
215
+ {
216
+ "epoch": 0.22,
217
+ "learning_rate": 9.621583742116329e-06,
218
+ "loss": 0.6668,
219
+ "step": 320
220
+ },
221
+ {
222
+ "epoch": 0.23,
223
+ "learning_rate": 9.607568325157675e-06,
224
+ "loss": 0.6599,
225
+ "step": 330
226
+ },
227
+ {
228
+ "epoch": 0.24,
229
+ "learning_rate": 9.59355290819902e-06,
230
+ "loss": 0.663,
231
+ "step": 340
232
+ },
233
+ {
234
+ "epoch": 0.24,
235
+ "learning_rate": 9.579537491240364e-06,
236
+ "loss": 0.6643,
237
+ "step": 350
238
+ },
239
+ {
240
+ "epoch": 0.25,
241
+ "learning_rate": 9.565522074281712e-06,
242
+ "loss": 0.6697,
243
+ "step": 360
244
+ },
245
+ {
246
+ "epoch": 0.26,
247
+ "learning_rate": 9.551506657323056e-06,
248
+ "loss": 0.6497,
249
+ "step": 370
250
+ },
251
+ {
252
+ "epoch": 0.26,
253
+ "learning_rate": 9.537491240364401e-06,
254
+ "loss": 0.6544,
255
+ "step": 380
256
+ },
257
+ {
258
+ "epoch": 0.27,
259
+ "learning_rate": 9.523475823405747e-06,
260
+ "loss": 0.6736,
261
+ "step": 390
262
+ },
263
+ {
264
+ "epoch": 0.28,
265
+ "learning_rate": 9.509460406447093e-06,
266
+ "loss": 0.6572,
267
+ "step": 400
268
+ },
269
+ {
270
+ "epoch": 0.29,
271
+ "learning_rate": 9.495444989488438e-06,
272
+ "loss": 0.6616,
273
+ "step": 410
274
+ },
275
+ {
276
+ "epoch": 0.29,
277
+ "learning_rate": 9.481429572529784e-06,
278
+ "loss": 0.6703,
279
+ "step": 420
280
+ },
281
+ {
282
+ "epoch": 0.3,
283
+ "learning_rate": 9.467414155571128e-06,
284
+ "loss": 0.6621,
285
+ "step": 430
286
+ },
287
+ {
288
+ "epoch": 0.31,
289
+ "learning_rate": 9.453398738612475e-06,
290
+ "loss": 0.6755,
291
+ "step": 440
292
+ },
293
+ {
294
+ "epoch": 0.31,
295
+ "learning_rate": 9.43938332165382e-06,
296
+ "loss": 0.6693,
297
+ "step": 450
298
+ },
299
+ {
300
+ "epoch": 0.32,
301
+ "learning_rate": 9.425367904695165e-06,
302
+ "loss": 0.6629,
303
+ "step": 460
304
+ },
305
+ {
306
+ "epoch": 0.33,
307
+ "learning_rate": 9.41135248773651e-06,
308
+ "loss": 0.672,
309
+ "step": 470
310
+ },
311
+ {
312
+ "epoch": 0.33,
313
+ "learning_rate": 9.397337070777856e-06,
314
+ "loss": 0.6682,
315
+ "step": 480
316
+ },
317
+ {
318
+ "epoch": 0.34,
319
+ "learning_rate": 9.383321653819202e-06,
320
+ "loss": 0.6667,
321
+ "step": 490
322
+ },
323
+ {
324
+ "epoch": 0.35,
325
+ "learning_rate": 9.369306236860548e-06,
326
+ "loss": 0.6632,
327
+ "step": 500
328
+ },
329
+ {
330
+ "epoch": 0.35,
331
+ "learning_rate": 9.355290819901893e-06,
332
+ "loss": 0.6374,
333
+ "step": 510
334
+ },
335
+ {
336
+ "epoch": 0.36,
337
+ "learning_rate": 9.341275402943239e-06,
338
+ "loss": 0.6528,
339
+ "step": 520
340
+ },
341
+ {
342
+ "epoch": 0.37,
343
+ "learning_rate": 9.327259985984583e-06,
344
+ "loss": 0.6658,
345
+ "step": 530
346
+ },
347
+ {
348
+ "epoch": 0.38,
349
+ "learning_rate": 9.313244569025929e-06,
350
+ "loss": 0.6489,
351
+ "step": 540
352
+ },
353
+ {
354
+ "epoch": 0.38,
355
+ "learning_rate": 9.299229152067274e-06,
356
+ "loss": 0.6551,
357
+ "step": 550
358
+ },
359
+ {
360
+ "epoch": 0.39,
361
+ "learning_rate": 9.28521373510862e-06,
362
+ "loss": 0.6666,
363
+ "step": 560
364
+ },
365
+ {
366
+ "epoch": 0.4,
367
+ "learning_rate": 9.271198318149966e-06,
368
+ "loss": 0.6398,
369
+ "step": 570
370
+ },
371
+ {
372
+ "epoch": 0.4,
373
+ "learning_rate": 9.257182901191311e-06,
374
+ "loss": 0.6515,
375
+ "step": 580
376
+ },
377
+ {
378
+ "epoch": 0.41,
379
+ "learning_rate": 9.243167484232657e-06,
380
+ "loss": 0.6646,
381
+ "step": 590
382
+ },
383
+ {
384
+ "epoch": 0.42,
385
+ "learning_rate": 9.229152067274003e-06,
386
+ "loss": 0.6372,
387
+ "step": 600
388
+ },
389
+ {
390
+ "epoch": 0.42,
391
+ "eval_custom_hf_pref_accuracy": 0.5785,
392
+ "eval_custom_hf_pref_kendalltau": 0.157,
393
+ "eval_custom_hf_pref_loss": 0.6421093940734863,
394
+ "eval_custom_hf_pref_neg_score": 0.48188385367393494,
395
+ "eval_custom_hf_pref_pos_score": 0.6362329721450806,
396
+ "eval_custom_hf_pref_runtime": 3.0972,
397
+ "eval_custom_hf_pref_samples_per_second": 645.736,
398
+ "eval_custom_hf_pref_score_diff": 0.15434913337230682,
399
+ "eval_custom_hf_pref_steps_per_second": 80.717,
400
+ "step": 600
401
+ },
402
+ {
403
+ "epoch": 0.42,
404
+ "learning_rate": 9.215136650315347e-06,
405
+ "loss": 0.6566,
406
+ "step": 610
407
+ },
408
+ {
409
+ "epoch": 0.43,
410
+ "learning_rate": 9.201121233356694e-06,
411
+ "loss": 0.6631,
412
+ "step": 620
413
+ },
414
+ {
415
+ "epoch": 0.44,
416
+ "learning_rate": 9.187105816398038e-06,
417
+ "loss": 0.6516,
418
+ "step": 630
419
+ },
420
+ {
421
+ "epoch": 0.45,
422
+ "learning_rate": 9.173090399439384e-06,
423
+ "loss": 0.6468,
424
+ "step": 640
425
+ },
426
+ {
427
+ "epoch": 0.45,
428
+ "learning_rate": 9.15907498248073e-06,
429
+ "loss": 0.6501,
430
+ "step": 650
431
+ },
432
+ {
433
+ "epoch": 0.46,
434
+ "learning_rate": 9.145059565522075e-06,
435
+ "loss": 0.6638,
436
+ "step": 660
437
+ },
438
+ {
439
+ "epoch": 0.47,
440
+ "learning_rate": 9.131044148563421e-06,
441
+ "loss": 0.6633,
442
+ "step": 670
443
+ },
444
+ {
445
+ "epoch": 0.47,
446
+ "learning_rate": 9.117028731604765e-06,
447
+ "loss": 0.6627,
448
+ "step": 680
449
+ },
450
+ {
451
+ "epoch": 0.48,
452
+ "learning_rate": 9.103013314646112e-06,
453
+ "loss": 0.6355,
454
+ "step": 690
455
+ },
456
+ {
457
+ "epoch": 0.49,
458
+ "learning_rate": 9.088997897687458e-06,
459
+ "loss": 0.6287,
460
+ "step": 700
461
+ },
462
+ {
463
+ "epoch": 0.49,
464
+ "learning_rate": 9.074982480728802e-06,
465
+ "loss": 0.6534,
466
+ "step": 710
467
+ },
468
+ {
469
+ "epoch": 0.5,
470
+ "learning_rate": 9.060967063770148e-06,
471
+ "loss": 0.6698,
472
+ "step": 720
473
+ },
474
+ {
475
+ "epoch": 0.51,
476
+ "learning_rate": 9.046951646811493e-06,
477
+ "loss": 0.6452,
478
+ "step": 730
479
+ },
480
+ {
481
+ "epoch": 0.51,
482
+ "learning_rate": 9.032936229852839e-06,
483
+ "loss": 0.641,
484
+ "step": 740
485
+ },
486
+ {
487
+ "epoch": 0.52,
488
+ "learning_rate": 9.018920812894185e-06,
489
+ "loss": 0.6317,
490
+ "step": 750
491
+ },
492
+ {
493
+ "epoch": 0.53,
494
+ "learning_rate": 9.00490539593553e-06,
495
+ "loss": 0.6612,
496
+ "step": 760
497
+ },
498
+ {
499
+ "epoch": 0.54,
500
+ "learning_rate": 8.990889978976876e-06,
501
+ "loss": 0.6315,
502
+ "step": 770
503
+ },
504
+ {
505
+ "epoch": 0.54,
506
+ "learning_rate": 8.97687456201822e-06,
507
+ "loss": 0.6552,
508
+ "step": 780
509
+ },
510
+ {
511
+ "epoch": 0.55,
512
+ "learning_rate": 8.962859145059566e-06,
513
+ "loss": 0.6451,
514
+ "step": 790
515
+ },
516
+ {
517
+ "epoch": 0.56,
518
+ "learning_rate": 8.948843728100913e-06,
519
+ "loss": 0.64,
520
+ "step": 800
521
+ },
522
+ {
523
+ "epoch": 0.56,
524
+ "learning_rate": 8.934828311142257e-06,
525
+ "loss": 0.6313,
526
+ "step": 810
527
+ },
528
+ {
529
+ "epoch": 0.57,
530
+ "learning_rate": 8.920812894183603e-06,
531
+ "loss": 0.6466,
532
+ "step": 820
533
+ },
534
+ {
535
+ "epoch": 0.58,
536
+ "learning_rate": 8.906797477224948e-06,
537
+ "loss": 0.6406,
538
+ "step": 830
539
+ },
540
+ {
541
+ "epoch": 0.58,
542
+ "learning_rate": 8.892782060266294e-06,
543
+ "loss": 0.6479,
544
+ "step": 840
545
+ },
546
+ {
547
+ "epoch": 0.59,
548
+ "learning_rate": 8.87876664330764e-06,
549
+ "loss": 0.6602,
550
+ "step": 850
551
+ },
552
+ {
553
+ "epoch": 0.6,
554
+ "learning_rate": 8.864751226348984e-06,
555
+ "loss": 0.6279,
556
+ "step": 860
557
+ },
558
+ {
559
+ "epoch": 0.61,
560
+ "learning_rate": 8.850735809390331e-06,
561
+ "loss": 0.6439,
562
+ "step": 870
563
+ },
564
+ {
565
+ "epoch": 0.61,
566
+ "learning_rate": 8.836720392431675e-06,
567
+ "loss": 0.6192,
568
+ "step": 880
569
+ },
570
+ {
571
+ "epoch": 0.62,
572
+ "learning_rate": 8.82270497547302e-06,
573
+ "loss": 0.6529,
574
+ "step": 890
575
+ },
576
+ {
577
+ "epoch": 0.63,
578
+ "learning_rate": 8.808689558514366e-06,
579
+ "loss": 0.6246,
580
+ "step": 900
581
+ },
582
+ {
583
+ "epoch": 0.63,
584
+ "eval_custom_hf_pref_accuracy": 0.5795,
585
+ "eval_custom_hf_pref_kendalltau": 0.159,
586
+ "eval_custom_hf_pref_loss": 0.6517187356948853,
587
+ "eval_custom_hf_pref_neg_score": -0.18239864706993103,
588
+ "eval_custom_hf_pref_pos_score": -0.047199830412864685,
589
+ "eval_custom_hf_pref_runtime": 3.0666,
590
+ "eval_custom_hf_pref_samples_per_second": 652.191,
591
+ "eval_custom_hf_pref_score_diff": 0.13519881665706635,
592
+ "eval_custom_hf_pref_steps_per_second": 81.524,
593
+ "step": 900
594
+ },
595
+ {
596
+ "epoch": 0.63,
597
+ "learning_rate": 8.794674141555712e-06,
598
+ "loss": 0.6453,
599
+ "step": 910
600
+ },
601
+ {
602
+ "epoch": 0.64,
603
+ "learning_rate": 8.780658724597058e-06,
604
+ "loss": 0.6557,
605
+ "step": 920
606
+ },
607
+ {
608
+ "epoch": 0.65,
609
+ "learning_rate": 8.766643307638403e-06,
610
+ "loss": 0.6271,
611
+ "step": 930
612
+ },
613
+ {
614
+ "epoch": 0.65,
615
+ "learning_rate": 8.752627890679747e-06,
616
+ "loss": 0.6472,
617
+ "step": 940
618
+ },
619
+ {
620
+ "epoch": 0.66,
621
+ "learning_rate": 8.738612473721095e-06,
622
+ "loss": 0.6299,
623
+ "step": 950
624
+ },
625
+ {
626
+ "epoch": 0.67,
627
+ "learning_rate": 8.724597056762439e-06,
628
+ "loss": 0.6403,
629
+ "step": 960
630
+ },
631
+ {
632
+ "epoch": 0.67,
633
+ "learning_rate": 8.710581639803784e-06,
634
+ "loss": 0.6343,
635
+ "step": 970
636
+ },
637
+ {
638
+ "epoch": 0.68,
639
+ "learning_rate": 8.69656622284513e-06,
640
+ "loss": 0.6274,
641
+ "step": 980
642
+ },
643
+ {
644
+ "epoch": 0.69,
645
+ "learning_rate": 8.682550805886476e-06,
646
+ "loss": 0.6534,
647
+ "step": 990
648
+ },
649
+ {
650
+ "epoch": 0.7,
651
+ "learning_rate": 8.668535388927821e-06,
652
+ "loss": 0.6244,
653
+ "step": 1000
654
+ },
655
+ {
656
+ "epoch": 0.7,
657
+ "learning_rate": 8.654519971969167e-06,
658
+ "loss": 0.6237,
659
+ "step": 1010
660
+ },
661
+ {
662
+ "epoch": 0.71,
663
+ "learning_rate": 8.640504555010513e-06,
664
+ "loss": 0.6531,
665
+ "step": 1020
666
+ },
667
+ {
668
+ "epoch": 0.72,
669
+ "learning_rate": 8.626489138051858e-06,
670
+ "loss": 0.6347,
671
+ "step": 1030
672
+ },
673
+ {
674
+ "epoch": 0.72,
675
+ "learning_rate": 8.612473721093202e-06,
676
+ "loss": 0.6513,
677
+ "step": 1040
678
+ },
679
+ {
680
+ "epoch": 0.73,
681
+ "learning_rate": 8.598458304134548e-06,
682
+ "loss": 0.648,
683
+ "step": 1050
684
+ },
685
+ {
686
+ "epoch": 0.74,
687
+ "learning_rate": 8.584442887175894e-06,
688
+ "loss": 0.6472,
689
+ "step": 1060
690
+ },
691
+ {
692
+ "epoch": 0.74,
693
+ "learning_rate": 8.57042747021724e-06,
694
+ "loss": 0.6305,
695
+ "step": 1070
696
+ },
697
+ {
698
+ "epoch": 0.75,
699
+ "learning_rate": 8.556412053258585e-06,
700
+ "loss": 0.6241,
701
+ "step": 1080
702
+ },
703
+ {
704
+ "epoch": 0.76,
705
+ "learning_rate": 8.54239663629993e-06,
706
+ "loss": 0.6763,
707
+ "step": 1090
708
+ },
709
+ {
710
+ "epoch": 0.77,
711
+ "learning_rate": 8.528381219341276e-06,
712
+ "loss": 0.6219,
713
+ "step": 1100
714
+ },
715
+ {
716
+ "epoch": 0.77,
717
+ "learning_rate": 8.514365802382622e-06,
718
+ "loss": 0.6384,
719
+ "step": 1110
720
+ },
721
+ {
722
+ "epoch": 0.78,
723
+ "learning_rate": 8.500350385423966e-06,
724
+ "loss": 0.6188,
725
+ "step": 1120
726
+ },
727
+ {
728
+ "epoch": 0.79,
729
+ "learning_rate": 8.486334968465313e-06,
730
+ "loss": 0.6295,
731
+ "step": 1130
732
+ },
733
+ {
734
+ "epoch": 0.79,
735
+ "learning_rate": 8.472319551506657e-06,
736
+ "loss": 0.6276,
737
+ "step": 1140
738
+ },
739
+ {
740
+ "epoch": 0.8,
741
+ "learning_rate": 8.458304134548003e-06,
742
+ "loss": 0.6501,
743
+ "step": 1150
744
+ },
745
+ {
746
+ "epoch": 0.81,
747
+ "learning_rate": 8.444288717589349e-06,
748
+ "loss": 0.6282,
749
+ "step": 1160
750
+ },
751
+ {
752
+ "epoch": 0.81,
753
+ "learning_rate": 8.430273300630695e-06,
754
+ "loss": 0.628,
755
+ "step": 1170
756
+ },
757
+ {
758
+ "epoch": 0.82,
759
+ "learning_rate": 8.41625788367204e-06,
760
+ "loss": 0.6018,
761
+ "step": 1180
762
+ },
763
+ {
764
+ "epoch": 0.83,
765
+ "learning_rate": 8.402242466713386e-06,
766
+ "loss": 0.6307,
767
+ "step": 1190
768
+ },
769
+ {
770
+ "epoch": 0.83,
771
+ "learning_rate": 8.388227049754732e-06,
772
+ "loss": 0.6083,
773
+ "step": 1200
774
+ },
775
+ {
776
+ "epoch": 0.83,
777
+ "eval_custom_hf_pref_accuracy": 0.564,
778
+ "eval_custom_hf_pref_kendalltau": 0.128,
779
+ "eval_custom_hf_pref_loss": 0.6535468697547913,
780
+ "eval_custom_hf_pref_neg_score": -0.3474700450897217,
781
+ "eval_custom_hf_pref_pos_score": -0.20932382345199585,
782
+ "eval_custom_hf_pref_runtime": 3.0757,
783
+ "eval_custom_hf_pref_samples_per_second": 650.252,
784
+ "eval_custom_hf_pref_score_diff": 0.13814620673656464,
785
+ "eval_custom_hf_pref_steps_per_second": 81.282,
786
+ "step": 1200
787
+ },
788
+ {
789
+ "epoch": 0.84,
790
+ "learning_rate": 8.374211632796077e-06,
791
+ "loss": 0.629,
792
+ "step": 1210
793
+ },
794
+ {
795
+ "epoch": 0.85,
796
+ "learning_rate": 8.360196215837421e-06,
797
+ "loss": 0.6346,
798
+ "step": 1220
799
+ },
800
+ {
801
+ "epoch": 0.86,
802
+ "learning_rate": 8.346180798878767e-06,
803
+ "loss": 0.6366,
804
+ "step": 1230
805
+ },
806
+ {
807
+ "epoch": 0.86,
808
+ "learning_rate": 8.332165381920113e-06,
809
+ "loss": 0.6231,
810
+ "step": 1240
811
+ },
812
+ {
813
+ "epoch": 0.87,
814
+ "learning_rate": 8.318149964961458e-06,
815
+ "loss": 0.6154,
816
+ "step": 1250
817
+ },
818
+ {
819
+ "epoch": 0.88,
820
+ "learning_rate": 8.304134548002804e-06,
821
+ "loss": 0.6288,
822
+ "step": 1260
823
+ },
824
+ {
825
+ "epoch": 0.88,
826
+ "learning_rate": 8.29011913104415e-06,
827
+ "loss": 0.612,
828
+ "step": 1270
829
+ },
830
+ {
831
+ "epoch": 0.89,
832
+ "learning_rate": 8.276103714085495e-06,
833
+ "loss": 0.6154,
834
+ "step": 1280
835
+ },
836
+ {
837
+ "epoch": 0.9,
838
+ "learning_rate": 8.262088297126841e-06,
839
+ "loss": 0.6107,
840
+ "step": 1290
841
+ },
842
+ {
843
+ "epoch": 0.9,
844
+ "learning_rate": 8.248072880168185e-06,
845
+ "loss": 0.5914,
846
+ "step": 1300
847
+ },
848
+ {
849
+ "epoch": 0.91,
850
+ "learning_rate": 8.234057463209532e-06,
851
+ "loss": 0.6288,
852
+ "step": 1310
853
+ },
854
+ {
855
+ "epoch": 0.92,
856
+ "learning_rate": 8.220042046250876e-06,
857
+ "loss": 0.6553,
858
+ "step": 1320
859
+ },
860
+ {
861
+ "epoch": 0.93,
862
+ "learning_rate": 8.206026629292222e-06,
863
+ "loss": 0.6075,
864
+ "step": 1330
865
+ },
866
+ {
867
+ "epoch": 0.93,
868
+ "learning_rate": 8.192011212333568e-06,
869
+ "loss": 0.6223,
870
+ "step": 1340
871
+ },
872
+ {
873
+ "epoch": 0.94,
874
+ "learning_rate": 8.177995795374913e-06,
875
+ "loss": 0.5895,
876
+ "step": 1350
877
+ },
878
+ {
879
+ "epoch": 0.95,
880
+ "learning_rate": 8.163980378416259e-06,
881
+ "loss": 0.6592,
882
+ "step": 1360
883
+ },
884
+ {
885
+ "epoch": 0.95,
886
+ "learning_rate": 8.149964961457603e-06,
887
+ "loss": 0.6402,
888
+ "step": 1370
889
+ },
890
+ {
891
+ "epoch": 0.96,
892
+ "learning_rate": 8.13594954449895e-06,
893
+ "loss": 0.6228,
894
+ "step": 1380
895
+ },
896
+ {
897
+ "epoch": 0.97,
898
+ "learning_rate": 8.121934127540296e-06,
899
+ "loss": 0.6145,
900
+ "step": 1390
901
+ },
902
+ {
903
+ "epoch": 0.97,
904
+ "learning_rate": 8.10791871058164e-06,
905
+ "loss": 0.5906,
906
+ "step": 1400
907
+ },
908
+ {
909
+ "epoch": 0.98,
910
+ "learning_rate": 8.093903293622986e-06,
911
+ "loss": 0.6153,
912
+ "step": 1410
913
+ },
914
+ {
915
+ "epoch": 0.99,
916
+ "learning_rate": 8.079887876664331e-06,
917
+ "loss": 0.6331,
918
+ "step": 1420
919
+ },
920
+ {
921
+ "epoch": 0.99,
922
+ "learning_rate": 8.065872459705677e-06,
923
+ "loss": 0.6308,
924
+ "step": 1430
925
+ },
926
+ {
927
+ "epoch": 1.0,
928
+ "learning_rate": 8.051857042747023e-06,
929
+ "loss": 0.6269,
930
+ "step": 1440
931
+ },
932
+ {
933
+ "epoch": 1.01,
934
+ "learning_rate": 8.037841625788367e-06,
935
+ "loss": 0.6243,
936
+ "step": 1450
937
+ },
938
+ {
939
+ "epoch": 1.02,
940
+ "learning_rate": 8.023826208829714e-06,
941
+ "loss": 0.6369,
942
+ "step": 1460
943
+ },
944
+ {
945
+ "epoch": 1.02,
946
+ "learning_rate": 8.009810791871058e-06,
947
+ "loss": 0.6384,
948
+ "step": 1470
949
+ },
950
+ {
951
+ "epoch": 1.03,
952
+ "learning_rate": 7.995795374912404e-06,
953
+ "loss": 0.6497,
954
+ "step": 1480
955
+ },
956
+ {
957
+ "epoch": 1.04,
958
+ "learning_rate": 7.981779957953751e-06,
959
+ "loss": 0.586,
960
+ "step": 1490
961
+ },
962
+ {
963
+ "epoch": 1.04,
964
+ "learning_rate": 7.967764540995095e-06,
965
+ "loss": 0.6172,
966
+ "step": 1500
967
+ },
968
+ {
969
+ "epoch": 1.04,
970
+ "eval_custom_hf_pref_accuracy": 0.5905,
971
+ "eval_custom_hf_pref_kendalltau": 0.181,
972
+ "eval_custom_hf_pref_loss": 0.6357187628746033,
973
+ "eval_custom_hf_pref_neg_score": -0.4265330135822296,
974
+ "eval_custom_hf_pref_pos_score": -0.18024785816669464,
975
+ "eval_custom_hf_pref_runtime": 3.0869,
976
+ "eval_custom_hf_pref_samples_per_second": 647.904,
977
+ "eval_custom_hf_pref_score_diff": 0.24628514051437378,
978
+ "eval_custom_hf_pref_steps_per_second": 80.988,
979
+ "step": 1500
980
+ },
981
+ {
982
+ "epoch": 1.05,
983
+ "learning_rate": 7.95374912403644e-06,
984
+ "loss": 0.6429,
985
+ "step": 1510
986
+ },
987
+ {
988
+ "epoch": 1.06,
989
+ "learning_rate": 7.939733707077786e-06,
990
+ "loss": 0.6088,
991
+ "step": 1520
992
+ },
993
+ {
994
+ "epoch": 1.06,
995
+ "learning_rate": 7.925718290119132e-06,
996
+ "loss": 0.6179,
997
+ "step": 1530
998
+ },
999
+ {
1000
+ "epoch": 1.07,
1001
+ "learning_rate": 7.911702873160478e-06,
1002
+ "loss": 0.606,
1003
+ "step": 1540
1004
+ },
1005
+ {
1006
+ "epoch": 1.08,
1007
+ "learning_rate": 7.897687456201822e-06,
1008
+ "loss": 0.6117,
1009
+ "step": 1550
1010
+ },
1011
+ {
1012
+ "epoch": 1.09,
1013
+ "learning_rate": 7.883672039243167e-06,
1014
+ "loss": 0.6182,
1015
+ "step": 1560
1016
+ },
1017
+ {
1018
+ "epoch": 1.09,
1019
+ "learning_rate": 7.869656622284513e-06,
1020
+ "loss": 0.6339,
1021
+ "step": 1570
1022
+ },
1023
+ {
1024
+ "epoch": 1.1,
1025
+ "learning_rate": 7.855641205325859e-06,
1026
+ "loss": 0.6183,
1027
+ "step": 1580
1028
+ },
1029
+ {
1030
+ "epoch": 1.11,
1031
+ "learning_rate": 7.841625788367204e-06,
1032
+ "loss": 0.5998,
1033
+ "step": 1590
1034
+ },
1035
+ {
1036
+ "epoch": 1.11,
1037
+ "learning_rate": 7.82761037140855e-06,
1038
+ "loss": 0.6292,
1039
+ "step": 1600
1040
+ },
1041
+ {
1042
+ "epoch": 1.12,
1043
+ "learning_rate": 7.813594954449896e-06,
1044
+ "loss": 0.6168,
1045
+ "step": 1610
1046
+ },
1047
+ {
1048
+ "epoch": 1.13,
1049
+ "learning_rate": 7.799579537491241e-06,
1050
+ "loss": 0.6152,
1051
+ "step": 1620
1052
+ },
1053
+ {
1054
+ "epoch": 1.13,
1055
+ "learning_rate": 7.785564120532585e-06,
1056
+ "loss": 0.6316,
1057
+ "step": 1630
1058
+ },
1059
+ {
1060
+ "epoch": 1.14,
1061
+ "learning_rate": 7.771548703573933e-06,
1062
+ "loss": 0.615,
1063
+ "step": 1640
1064
+ },
1065
+ {
1066
+ "epoch": 1.15,
1067
+ "learning_rate": 7.757533286615277e-06,
1068
+ "loss": 0.6316,
1069
+ "step": 1650
1070
+ },
1071
+ {
1072
+ "epoch": 1.15,
1073
+ "learning_rate": 7.743517869656622e-06,
1074
+ "loss": 0.6387,
1075
+ "step": 1660
1076
+ },
1077
+ {
1078
+ "epoch": 1.16,
1079
+ "learning_rate": 7.729502452697968e-06,
1080
+ "loss": 0.6166,
1081
+ "step": 1670
1082
+ },
1083
+ {
1084
+ "epoch": 1.17,
1085
+ "learning_rate": 7.715487035739314e-06,
1086
+ "loss": 0.6316,
1087
+ "step": 1680
1088
+ },
1089
+ {
1090
+ "epoch": 1.18,
1091
+ "learning_rate": 7.70147161878066e-06,
1092
+ "loss": 0.6261,
1093
+ "step": 1690
1094
+ },
1095
+ {
1096
+ "epoch": 1.18,
1097
+ "learning_rate": 7.687456201822005e-06,
1098
+ "loss": 0.5966,
1099
+ "step": 1700
1100
+ },
1101
+ {
1102
+ "epoch": 1.19,
1103
+ "learning_rate": 7.67344078486335e-06,
1104
+ "loss": 0.6077,
1105
+ "step": 1710
1106
+ },
1107
+ {
1108
+ "epoch": 1.2,
1109
+ "learning_rate": 7.659425367904697e-06,
1110
+ "loss": 0.6325,
1111
+ "step": 1720
1112
+ },
1113
+ {
1114
+ "epoch": 1.2,
1115
+ "learning_rate": 7.64540995094604e-06,
1116
+ "loss": 0.6158,
1117
+ "step": 1730
1118
+ },
1119
+ {
1120
+ "epoch": 1.21,
1121
+ "learning_rate": 7.631394533987386e-06,
1122
+ "loss": 0.6196,
1123
+ "step": 1740
1124
+ },
1125
+ {
1126
+ "epoch": 1.22,
1127
+ "learning_rate": 7.617379117028733e-06,
1128
+ "loss": 0.6411,
1129
+ "step": 1750
1130
+ },
1131
+ {
1132
+ "epoch": 1.22,
1133
+ "learning_rate": 7.6033637000700775e-06,
1134
+ "loss": 0.616,
1135
+ "step": 1760
1136
+ },
1137
+ {
1138
+ "epoch": 1.23,
1139
+ "learning_rate": 7.589348283111423e-06,
1140
+ "loss": 0.6252,
1141
+ "step": 1770
1142
+ },
1143
+ {
1144
+ "epoch": 1.24,
1145
+ "learning_rate": 7.575332866152769e-06,
1146
+ "loss": 0.6171,
1147
+ "step": 1780
1148
+ },
1149
+ {
1150
+ "epoch": 1.25,
1151
+ "learning_rate": 7.5613174491941146e-06,
1152
+ "loss": 0.6135,
1153
+ "step": 1790
1154
+ },
1155
+ {
1156
+ "epoch": 1.25,
1157
+ "learning_rate": 7.547302032235459e-06,
1158
+ "loss": 0.6322,
1159
+ "step": 1800
1160
+ },
1161
+ {
1162
+ "epoch": 1.25,
1163
+ "eval_custom_hf_pref_accuracy": 0.6035,
1164
+ "eval_custom_hf_pref_kendalltau": 0.207,
1165
+ "eval_custom_hf_pref_loss": 0.6257030963897705,
1166
+ "eval_custom_hf_pref_neg_score": -0.47859320044517517,
1167
+ "eval_custom_hf_pref_pos_score": -0.20773868262767792,
1168
+ "eval_custom_hf_pref_runtime": 3.1023,
1169
+ "eval_custom_hf_pref_samples_per_second": 644.681,
1170
+ "eval_custom_hf_pref_score_diff": 0.27085456252098083,
1171
+ "eval_custom_hf_pref_steps_per_second": 80.585,
1172
+ "step": 1800
1173
+ },
1174
+ {
1175
+ "epoch": 1.26,
1176
+ "learning_rate": 7.533286615276805e-06,
1177
+ "loss": 0.6259,
1178
+ "step": 1810
1179
+ },
1180
+ {
1181
+ "epoch": 1.27,
1182
+ "learning_rate": 7.519271198318151e-06,
1183
+ "loss": 0.5938,
1184
+ "step": 1820
1185
+ },
1186
+ {
1187
+ "epoch": 1.27,
1188
+ "learning_rate": 7.505255781359496e-06,
1189
+ "loss": 0.5876,
1190
+ "step": 1830
1191
+ },
1192
+ {
1193
+ "epoch": 1.28,
1194
+ "learning_rate": 7.491240364400841e-06,
1195
+ "loss": 0.6386,
1196
+ "step": 1840
1197
+ },
1198
+ {
1199
+ "epoch": 1.29,
1200
+ "learning_rate": 7.477224947442187e-06,
1201
+ "loss": 0.5974,
1202
+ "step": 1850
1203
+ },
1204
+ {
1205
+ "epoch": 1.29,
1206
+ "learning_rate": 7.463209530483533e-06,
1207
+ "loss": 0.6019,
1208
+ "step": 1860
1209
+ },
1210
+ {
1211
+ "epoch": 1.3,
1212
+ "learning_rate": 7.449194113524878e-06,
1213
+ "loss": 0.5902,
1214
+ "step": 1870
1215
+ },
1216
+ {
1217
+ "epoch": 1.31,
1218
+ "learning_rate": 7.435178696566223e-06,
1219
+ "loss": 0.5528,
1220
+ "step": 1880
1221
+ },
1222
+ {
1223
+ "epoch": 1.31,
1224
+ "learning_rate": 7.42116327960757e-06,
1225
+ "loss": 0.6031,
1226
+ "step": 1890
1227
+ },
1228
+ {
1229
+ "epoch": 1.32,
1230
+ "learning_rate": 7.4071478626489144e-06,
1231
+ "loss": 0.5838,
1232
+ "step": 1900
1233
+ },
1234
+ {
1235
+ "epoch": 1.33,
1236
+ "learning_rate": 7.39313244569026e-06,
1237
+ "loss": 0.6493,
1238
+ "step": 1910
1239
+ },
1240
+ {
1241
+ "epoch": 1.34,
1242
+ "learning_rate": 7.379117028731605e-06,
1243
+ "loss": 0.6307,
1244
+ "step": 1920
1245
+ },
1246
+ {
1247
+ "epoch": 1.34,
1248
+ "learning_rate": 7.3651016117729515e-06,
1249
+ "loss": 0.5951,
1250
+ "step": 1930
1251
+ },
1252
+ {
1253
+ "epoch": 1.35,
1254
+ "learning_rate": 7.351086194814296e-06,
1255
+ "loss": 0.5965,
1256
+ "step": 1940
1257
+ },
1258
+ {
1259
+ "epoch": 1.36,
1260
+ "learning_rate": 7.337070777855642e-06,
1261
+ "loss": 0.5882,
1262
+ "step": 1950
1263
+ },
1264
+ {
1265
+ "epoch": 1.36,
1266
+ "learning_rate": 7.323055360896987e-06,
1267
+ "loss": 0.5943,
1268
+ "step": 1960
1269
+ },
1270
+ {
1271
+ "epoch": 1.37,
1272
+ "learning_rate": 7.309039943938333e-06,
1273
+ "loss": 0.586,
1274
+ "step": 1970
1275
+ },
1276
+ {
1277
+ "epoch": 1.38,
1278
+ "learning_rate": 7.295024526979678e-06,
1279
+ "loss": 0.6234,
1280
+ "step": 1980
1281
+ },
1282
+ {
1283
+ "epoch": 1.38,
1284
+ "learning_rate": 7.281009110021023e-06,
1285
+ "loss": 0.634,
1286
+ "step": 1990
1287
+ },
1288
+ {
1289
+ "epoch": 1.39,
1290
+ "learning_rate": 7.2669936930623695e-06,
1291
+ "loss": 0.6235,
1292
+ "step": 2000
1293
+ },
1294
+ {
1295
+ "epoch": 1.4,
1296
+ "learning_rate": 7.252978276103715e-06,
1297
+ "loss": 0.5714,
1298
+ "step": 2010
1299
+ },
1300
+ {
1301
+ "epoch": 1.41,
1302
+ "learning_rate": 7.23896285914506e-06,
1303
+ "loss": 0.6068,
1304
+ "step": 2020
1305
+ },
1306
+ {
1307
+ "epoch": 1.41,
1308
+ "learning_rate": 7.224947442186405e-06,
1309
+ "loss": 0.6222,
1310
+ "step": 2030
1311
+ },
1312
+ {
1313
+ "epoch": 1.42,
1314
+ "learning_rate": 7.210932025227751e-06,
1315
+ "loss": 0.6029,
1316
+ "step": 2040
1317
+ },
1318
+ {
1319
+ "epoch": 1.43,
1320
+ "learning_rate": 7.196916608269097e-06,
1321
+ "loss": 0.6093,
1322
+ "step": 2050
1323
+ },
1324
+ {
1325
+ "epoch": 1.43,
1326
+ "learning_rate": 7.182901191310442e-06,
1327
+ "loss": 0.6151,
1328
+ "step": 2060
1329
+ },
1330
+ {
1331
+ "epoch": 1.44,
1332
+ "learning_rate": 7.168885774351787e-06,
1333
+ "loss": 0.617,
1334
+ "step": 2070
1335
+ },
1336
+ {
1337
+ "epoch": 1.45,
1338
+ "learning_rate": 7.154870357393133e-06,
1339
+ "loss": 0.6077,
1340
+ "step": 2080
1341
+ },
1342
+ {
1343
+ "epoch": 1.45,
1344
+ "learning_rate": 7.140854940434478e-06,
1345
+ "loss": 0.6247,
1346
+ "step": 2090
1347
+ },
1348
+ {
1349
+ "epoch": 1.46,
1350
+ "learning_rate": 7.126839523475824e-06,
1351
+ "loss": 0.5979,
1352
+ "step": 2100
1353
+ },
1354
+ {
1355
+ "epoch": 1.46,
1356
+ "eval_custom_hf_pref_accuracy": 0.6175,
1357
+ "eval_custom_hf_pref_kendalltau": 0.235,
1358
+ "eval_custom_hf_pref_loss": 0.6143593788146973,
1359
+ "eval_custom_hf_pref_neg_score": -0.34517455101013184,
1360
+ "eval_custom_hf_pref_pos_score": 0.005482231266796589,
1361
+ "eval_custom_hf_pref_runtime": 3.0949,
1362
+ "eval_custom_hf_pref_samples_per_second": 646.224,
1363
+ "eval_custom_hf_pref_score_diff": 0.35065677762031555,
1364
+ "eval_custom_hf_pref_steps_per_second": 80.778,
1365
+ "step": 2100
1366
+ },
1367
+ {
1368
+ "epoch": 1.47,
1369
+ "learning_rate": 7.11282410651717e-06,
1370
+ "loss": 0.5953,
1371
+ "step": 2110
1372
+ },
1373
+ {
1374
+ "epoch": 1.47,
1375
+ "learning_rate": 7.098808689558515e-06,
1376
+ "loss": 0.5933,
1377
+ "step": 2120
1378
+ },
1379
+ {
1380
+ "epoch": 1.48,
1381
+ "learning_rate": 7.08479327259986e-06,
1382
+ "loss": 0.6466,
1383
+ "step": 2130
1384
+ },
1385
+ {
1386
+ "epoch": 1.49,
1387
+ "learning_rate": 7.070777855641206e-06,
1388
+ "loss": 0.6169,
1389
+ "step": 2140
1390
+ },
1391
+ {
1392
+ "epoch": 1.5,
1393
+ "learning_rate": 7.056762438682552e-06,
1394
+ "loss": 0.6024,
1395
+ "step": 2150
1396
+ },
1397
+ {
1398
+ "epoch": 1.5,
1399
+ "learning_rate": 7.042747021723897e-06,
1400
+ "loss": 0.6208,
1401
+ "step": 2160
1402
+ },
1403
+ {
1404
+ "epoch": 1.51,
1405
+ "learning_rate": 7.028731604765242e-06,
1406
+ "loss": 0.6015,
1407
+ "step": 2170
1408
+ },
1409
+ {
1410
+ "epoch": 1.52,
1411
+ "learning_rate": 7.014716187806588e-06,
1412
+ "loss": 0.6428,
1413
+ "step": 2180
1414
+ },
1415
+ {
1416
+ "epoch": 1.52,
1417
+ "learning_rate": 7.000700770847933e-06,
1418
+ "loss": 0.6282,
1419
+ "step": 2190
1420
+ },
1421
+ {
1422
+ "epoch": 1.53,
1423
+ "learning_rate": 6.986685353889279e-06,
1424
+ "loss": 0.5968,
1425
+ "step": 2200
1426
+ },
1427
+ {
1428
+ "epoch": 1.54,
1429
+ "learning_rate": 6.972669936930624e-06,
1430
+ "loss": 0.6262,
1431
+ "step": 2210
1432
+ },
1433
+ {
1434
+ "epoch": 1.54,
1435
+ "learning_rate": 6.95865451997197e-06,
1436
+ "loss": 0.6052,
1437
+ "step": 2220
1438
+ },
1439
+ {
1440
+ "epoch": 1.55,
1441
+ "learning_rate": 6.944639103013315e-06,
1442
+ "loss": 0.618,
1443
+ "step": 2230
1444
+ },
1445
+ {
1446
+ "epoch": 1.56,
1447
+ "learning_rate": 6.930623686054661e-06,
1448
+ "loss": 0.606,
1449
+ "step": 2240
1450
+ },
1451
+ {
1452
+ "epoch": 1.57,
1453
+ "learning_rate": 6.9166082690960055e-06,
1454
+ "loss": 0.6208,
1455
+ "step": 2250
1456
+ },
1457
+ {
1458
+ "epoch": 1.57,
1459
+ "learning_rate": 6.902592852137352e-06,
1460
+ "loss": 0.6052,
1461
+ "step": 2260
1462
+ },
1463
+ {
1464
+ "epoch": 1.58,
1465
+ "learning_rate": 6.888577435178697e-06,
1466
+ "loss": 0.6348,
1467
+ "step": 2270
1468
+ },
1469
+ {
1470
+ "epoch": 1.59,
1471
+ "learning_rate": 6.8745620182200425e-06,
1472
+ "loss": 0.6177,
1473
+ "step": 2280
1474
+ },
1475
+ {
1476
+ "epoch": 1.59,
1477
+ "learning_rate": 6.860546601261388e-06,
1478
+ "loss": 0.6076,
1479
+ "step": 2290
1480
+ },
1481
+ {
1482
+ "epoch": 1.6,
1483
+ "learning_rate": 6.846531184302734e-06,
1484
+ "loss": 0.5978,
1485
+ "step": 2300
1486
+ },
1487
+ {
1488
+ "epoch": 1.61,
1489
+ "learning_rate": 6.832515767344079e-06,
1490
+ "loss": 0.5999,
1491
+ "step": 2310
1492
+ },
1493
+ {
1494
+ "epoch": 1.61,
1495
+ "learning_rate": 6.818500350385424e-06,
1496
+ "loss": 0.5955,
1497
+ "step": 2320
1498
+ },
1499
+ {
1500
+ "epoch": 1.62,
1501
+ "learning_rate": 6.80448493342677e-06,
1502
+ "loss": 0.6491,
1503
+ "step": 2330
1504
+ },
1505
+ {
1506
+ "epoch": 1.63,
1507
+ "learning_rate": 6.790469516468116e-06,
1508
+ "loss": 0.6059,
1509
+ "step": 2340
1510
+ },
1511
+ {
1512
+ "epoch": 1.63,
1513
+ "learning_rate": 6.7764540995094605e-06,
1514
+ "loss": 0.6086,
1515
+ "step": 2350
1516
+ },
1517
+ {
1518
+ "epoch": 1.64,
1519
+ "learning_rate": 6.762438682550806e-06,
1520
+ "loss": 0.5981,
1521
+ "step": 2360
1522
+ },
1523
+ {
1524
+ "epoch": 1.65,
1525
+ "learning_rate": 6.748423265592152e-06,
1526
+ "loss": 0.6151,
1527
+ "step": 2370
1528
+ },
1529
+ {
1530
+ "epoch": 1.66,
1531
+ "learning_rate": 6.7344078486334976e-06,
1532
+ "loss": 0.633,
1533
+ "step": 2380
1534
+ },
1535
+ {
1536
+ "epoch": 1.66,
1537
+ "learning_rate": 6.720392431674842e-06,
1538
+ "loss": 0.6074,
1539
+ "step": 2390
1540
+ },
1541
+ {
1542
+ "epoch": 1.67,
1543
+ "learning_rate": 6.706377014716189e-06,
1544
+ "loss": 0.592,
1545
+ "step": 2400
1546
+ },
1547
+ {
1548
+ "epoch": 1.67,
1549
+ "eval_custom_hf_pref_accuracy": 0.62,
1550
+ "eval_custom_hf_pref_kendalltau": 0.24,
1551
+ "eval_custom_hf_pref_loss": 0.6024765372276306,
1552
+ "eval_custom_hf_pref_neg_score": -0.1819182187318802,
1553
+ "eval_custom_hf_pref_pos_score": 0.2181234061717987,
1554
+ "eval_custom_hf_pref_runtime": 3.1567,
1555
+ "eval_custom_hf_pref_samples_per_second": 633.564,
1556
+ "eval_custom_hf_pref_score_diff": 0.4000416398048401,
1557
+ "eval_custom_hf_pref_steps_per_second": 79.195,
1558
+ "step": 2400
1559
+ },
1560
+ {
1561
+ "epoch": 1.68,
1562
+ "learning_rate": 6.692361597757534e-06,
1563
+ "loss": 0.569,
1564
+ "step": 2410
1565
+ },
1566
+ {
1567
+ "epoch": 1.68,
1568
+ "learning_rate": 6.6783461807988794e-06,
1569
+ "loss": 0.6234,
1570
+ "step": 2420
1571
+ },
1572
+ {
1573
+ "epoch": 1.69,
1574
+ "learning_rate": 6.664330763840224e-06,
1575
+ "loss": 0.6026,
1576
+ "step": 2430
1577
+ },
1578
+ {
1579
+ "epoch": 1.7,
1580
+ "learning_rate": 6.650315346881571e-06,
1581
+ "loss": 0.604,
1582
+ "step": 2440
1583
+ },
1584
+ {
1585
+ "epoch": 1.7,
1586
+ "learning_rate": 6.636299929922916e-06,
1587
+ "loss": 0.6344,
1588
+ "step": 2450
1589
+ },
1590
+ {
1591
+ "epoch": 1.71,
1592
+ "learning_rate": 6.622284512964261e-06,
1593
+ "loss": 0.6428,
1594
+ "step": 2460
1595
+ },
1596
+ {
1597
+ "epoch": 1.72,
1598
+ "learning_rate": 6.608269096005606e-06,
1599
+ "loss": 0.6277,
1600
+ "step": 2470
1601
+ },
1602
+ {
1603
+ "epoch": 1.73,
1604
+ "learning_rate": 6.594253679046953e-06,
1605
+ "loss": 0.6206,
1606
+ "step": 2480
1607
+ },
1608
+ {
1609
+ "epoch": 1.73,
1610
+ "learning_rate": 6.5802382620882975e-06,
1611
+ "loss": 0.6334,
1612
+ "step": 2490
1613
+ },
1614
+ {
1615
+ "epoch": 1.74,
1616
+ "learning_rate": 6.566222845129643e-06,
1617
+ "loss": 0.6269,
1618
+ "step": 2500
1619
+ },
1620
+ {
1621
+ "epoch": 1.75,
1622
+ "learning_rate": 6.552207428170989e-06,
1623
+ "loss": 0.6252,
1624
+ "step": 2510
1625
+ },
1626
+ {
1627
+ "epoch": 1.75,
1628
+ "learning_rate": 6.5381920112123345e-06,
1629
+ "loss": 0.5913,
1630
+ "step": 2520
1631
+ },
1632
+ {
1633
+ "epoch": 1.76,
1634
+ "learning_rate": 6.524176594253679e-06,
1635
+ "loss": 0.5826,
1636
+ "step": 2530
1637
+ },
1638
+ {
1639
+ "epoch": 1.77,
1640
+ "learning_rate": 6.510161177295025e-06,
1641
+ "loss": 0.5981,
1642
+ "step": 2540
1643
+ },
1644
+ {
1645
+ "epoch": 1.77,
1646
+ "learning_rate": 6.496145760336371e-06,
1647
+ "loss": 0.6052,
1648
+ "step": 2550
1649
+ },
1650
+ {
1651
+ "epoch": 1.78,
1652
+ "learning_rate": 6.482130343377716e-06,
1653
+ "loss": 0.5933,
1654
+ "step": 2560
1655
+ },
1656
+ {
1657
+ "epoch": 1.79,
1658
+ "learning_rate": 6.468114926419061e-06,
1659
+ "loss": 0.5952,
1660
+ "step": 2570
1661
+ },
1662
+ {
1663
+ "epoch": 1.79,
1664
+ "learning_rate": 6.454099509460406e-06,
1665
+ "loss": 0.6018,
1666
+ "step": 2580
1667
+ },
1668
+ {
1669
+ "epoch": 1.8,
1670
+ "learning_rate": 6.4400840925017525e-06,
1671
+ "loss": 0.5824,
1672
+ "step": 2590
1673
+ },
1674
+ {
1675
+ "epoch": 1.81,
1676
+ "learning_rate": 6.426068675543098e-06,
1677
+ "loss": 0.5687,
1678
+ "step": 2600
1679
+ },
1680
+ {
1681
+ "epoch": 1.82,
1682
+ "learning_rate": 6.412053258584443e-06,
1683
+ "loss": 0.6095,
1684
+ "step": 2610
1685
+ },
1686
+ {
1687
+ "epoch": 1.82,
1688
+ "learning_rate": 6.3980378416257896e-06,
1689
+ "loss": 0.5991,
1690
+ "step": 2620
1691
+ },
1692
+ {
1693
+ "epoch": 1.83,
1694
+ "learning_rate": 6.384022424667134e-06,
1695
+ "loss": 0.5811,
1696
+ "step": 2630
1697
+ },
1698
+ {
1699
+ "epoch": 1.84,
1700
+ "learning_rate": 6.37000700770848e-06,
1701
+ "loss": 0.606,
1702
+ "step": 2640
1703
+ },
1704
+ {
1705
+ "epoch": 1.84,
1706
+ "learning_rate": 6.355991590749825e-06,
1707
+ "loss": 0.609,
1708
+ "step": 2650
1709
+ },
1710
+ {
1711
+ "epoch": 1.85,
1712
+ "learning_rate": 6.341976173791171e-06,
1713
+ "loss": 0.6127,
1714
+ "step": 2660
1715
+ },
1716
+ {
1717
+ "epoch": 1.86,
1718
+ "learning_rate": 6.327960756832516e-06,
1719
+ "loss": 0.614,
1720
+ "step": 2670
1721
+ },
1722
+ {
1723
+ "epoch": 1.86,
1724
+ "learning_rate": 6.313945339873861e-06,
1725
+ "loss": 0.6075,
1726
+ "step": 2680
1727
+ },
1728
+ {
1729
+ "epoch": 1.87,
1730
+ "learning_rate": 6.299929922915208e-06,
1731
+ "loss": 0.6046,
1732
+ "step": 2690
1733
+ },
1734
+ {
1735
+ "epoch": 1.88,
1736
+ "learning_rate": 6.285914505956553e-06,
1737
+ "loss": 0.6014,
1738
+ "step": 2700
1739
+ },
1740
+ {
1741
+ "epoch": 1.88,
1742
+ "eval_custom_hf_pref_accuracy": 0.614,
1743
+ "eval_custom_hf_pref_kendalltau": 0.228,
1744
+ "eval_custom_hf_pref_loss": 0.6012265682220459,
1745
+ "eval_custom_hf_pref_neg_score": -0.45326927304267883,
1746
+ "eval_custom_hf_pref_pos_score": -0.06561196595430374,
1747
+ "eval_custom_hf_pref_runtime": 3.1381,
1748
+ "eval_custom_hf_pref_samples_per_second": 637.337,
1749
+ "eval_custom_hf_pref_score_diff": 0.3876572847366333,
1750
+ "eval_custom_hf_pref_steps_per_second": 79.667,
1751
+ "step": 2700
1752
+ },
1753
+ {
1754
+ "epoch": 1.89,
1755
+ "learning_rate": 6.271899088997898e-06,
1756
+ "loss": 0.6322,
1757
+ "step": 2710
1758
+ },
1759
+ {
1760
+ "epoch": 1.89,
1761
+ "learning_rate": 6.257883672039243e-06,
1762
+ "loss": 0.5794,
1763
+ "step": 2720
1764
+ },
1765
+ {
1766
+ "epoch": 1.9,
1767
+ "learning_rate": 6.2438682550805895e-06,
1768
+ "loss": 0.6534,
1769
+ "step": 2730
1770
+ },
1771
+ {
1772
+ "epoch": 1.91,
1773
+ "learning_rate": 6.229852838121935e-06,
1774
+ "loss": 0.6054,
1775
+ "step": 2740
1776
+ },
1777
+ {
1778
+ "epoch": 1.91,
1779
+ "learning_rate": 6.21583742116328e-06,
1780
+ "loss": 0.5961,
1781
+ "step": 2750
1782
+ },
1783
+ {
1784
+ "epoch": 1.92,
1785
+ "learning_rate": 6.201822004204625e-06,
1786
+ "loss": 0.6372,
1787
+ "step": 2760
1788
+ },
1789
+ {
1790
+ "epoch": 1.93,
1791
+ "learning_rate": 6.187806587245971e-06,
1792
+ "loss": 0.6136,
1793
+ "step": 2770
1794
+ },
1795
+ {
1796
+ "epoch": 1.93,
1797
+ "learning_rate": 6.173791170287316e-06,
1798
+ "loss": 0.6435,
1799
+ "step": 2780
1800
+ },
1801
+ {
1802
+ "epoch": 1.94,
1803
+ "learning_rate": 6.159775753328662e-06,
1804
+ "loss": 0.599,
1805
+ "step": 2790
1806
+ },
1807
+ {
1808
+ "epoch": 1.95,
1809
+ "learning_rate": 6.145760336370008e-06,
1810
+ "loss": 0.6341,
1811
+ "step": 2800
1812
+ },
1813
+ {
1814
+ "epoch": 1.95,
1815
+ "learning_rate": 6.131744919411353e-06,
1816
+ "loss": 0.5967,
1817
+ "step": 2810
1818
+ },
1819
+ {
1820
+ "epoch": 1.96,
1821
+ "learning_rate": 6.117729502452698e-06,
1822
+ "loss": 0.6156,
1823
+ "step": 2820
1824
+ },
1825
+ {
1826
+ "epoch": 1.97,
1827
+ "learning_rate": 6.103714085494044e-06,
1828
+ "loss": 0.6235,
1829
+ "step": 2830
1830
+ },
1831
+ {
1832
+ "epoch": 1.98,
1833
+ "learning_rate": 6.08969866853539e-06,
1834
+ "loss": 0.6027,
1835
+ "step": 2840
1836
+ },
1837
+ {
1838
+ "epoch": 1.98,
1839
+ "learning_rate": 6.075683251576735e-06,
1840
+ "loss": 0.6196,
1841
+ "step": 2850
1842
+ },
1843
+ {
1844
+ "epoch": 1.99,
1845
+ "learning_rate": 6.06166783461808e-06,
1846
+ "loss": 0.586,
1847
+ "step": 2860
1848
+ },
1849
+ {
1850
+ "epoch": 2.0,
1851
+ "learning_rate": 6.0476524176594255e-06,
1852
+ "loss": 0.586,
1853
+ "step": 2870
1854
+ },
1855
+ {
1856
+ "epoch": 2.0,
1857
+ "learning_rate": 6.033637000700771e-06,
1858
+ "loss": 0.6173,
1859
+ "step": 2880
1860
+ },
1861
+ {
1862
+ "epoch": 2.01,
1863
+ "learning_rate": 6.019621583742117e-06,
1864
+ "loss": 0.6127,
1865
+ "step": 2890
1866
+ },
1867
+ {
1868
+ "epoch": 2.02,
1869
+ "learning_rate": 6.005606166783462e-06,
1870
+ "loss": 0.6308,
1871
+ "step": 2900
1872
+ },
1873
+ {
1874
+ "epoch": 2.02,
1875
+ "learning_rate": 5.991590749824808e-06,
1876
+ "loss": 0.6055,
1877
+ "step": 2910
1878
+ },
1879
+ {
1880
+ "epoch": 2.03,
1881
+ "learning_rate": 5.977575332866153e-06,
1882
+ "loss": 0.6339,
1883
+ "step": 2920
1884
+ },
1885
+ {
1886
+ "epoch": 2.04,
1887
+ "learning_rate": 5.963559915907499e-06,
1888
+ "loss": 0.6525,
1889
+ "step": 2930
1890
+ },
1891
+ {
1892
+ "epoch": 2.05,
1893
+ "learning_rate": 5.9495444989488436e-06,
1894
+ "loss": 0.6512,
1895
+ "step": 2940
1896
+ },
1897
+ {
1898
+ "epoch": 2.05,
1899
+ "learning_rate": 5.93552908199019e-06,
1900
+ "loss": 0.6068,
1901
+ "step": 2950
1902
+ },
1903
+ {
1904
+ "epoch": 2.06,
1905
+ "learning_rate": 5.921513665031535e-06,
1906
+ "loss": 0.6137,
1907
+ "step": 2960
1908
+ },
1909
+ {
1910
+ "epoch": 2.07,
1911
+ "learning_rate": 5.907498248072881e-06,
1912
+ "loss": 0.6017,
1913
+ "step": 2970
1914
+ },
1915
+ {
1916
+ "epoch": 2.07,
1917
+ "learning_rate": 5.8934828311142254e-06,
1918
+ "loss": 0.6071,
1919
+ "step": 2980
1920
+ },
1921
+ {
1922
+ "epoch": 2.08,
1923
+ "learning_rate": 5.879467414155572e-06,
1924
+ "loss": 0.585,
1925
+ "step": 2990
1926
+ },
1927
+ {
1928
+ "epoch": 2.09,
1929
+ "learning_rate": 5.865451997196917e-06,
1930
+ "loss": 0.6075,
1931
+ "step": 3000
1932
+ },
1933
+ {
1934
+ "epoch": 2.09,
1935
+ "eval_custom_hf_pref_accuracy": 0.613,
1936
+ "eval_custom_hf_pref_kendalltau": 0.226,
1937
+ "eval_custom_hf_pref_loss": 0.6039531230926514,
1938
+ "eval_custom_hf_pref_neg_score": -0.5177181363105774,
1939
+ "eval_custom_hf_pref_pos_score": -0.16565346717834473,
1940
+ "eval_custom_hf_pref_runtime": 3.1422,
1941
+ "eval_custom_hf_pref_samples_per_second": 636.497,
1942
+ "eval_custom_hf_pref_score_diff": 0.3520646393299103,
1943
+ "eval_custom_hf_pref_steps_per_second": 79.562,
1944
+ "step": 3000
1945
+ },
1946
+ {
1947
+ "epoch": 2.09,
1948
+ "learning_rate": 5.8514365802382625e-06,
1949
+ "loss": 0.6178,
1950
+ "step": 3010
1951
+ },
1952
+ {
1953
+ "epoch": 2.1,
1954
+ "learning_rate": 5.837421163279608e-06,
1955
+ "loss": 0.6208,
1956
+ "step": 3020
1957
+ },
1958
+ {
1959
+ "epoch": 2.11,
1960
+ "learning_rate": 5.823405746320954e-06,
1961
+ "loss": 0.6236,
1962
+ "step": 3030
1963
+ },
1964
+ {
1965
+ "epoch": 2.11,
1966
+ "learning_rate": 5.809390329362299e-06,
1967
+ "loss": 0.6193,
1968
+ "step": 3040
1969
+ },
1970
+ {
1971
+ "epoch": 2.12,
1972
+ "learning_rate": 5.795374912403644e-06,
1973
+ "loss": 0.5971,
1974
+ "step": 3050
1975
+ },
1976
+ {
1977
+ "epoch": 2.13,
1978
+ "learning_rate": 5.78135949544499e-06,
1979
+ "loss": 0.6193,
1980
+ "step": 3060
1981
+ },
1982
+ {
1983
+ "epoch": 2.14,
1984
+ "learning_rate": 5.767344078486336e-06,
1985
+ "loss": 0.6173,
1986
+ "step": 3070
1987
+ },
1988
+ {
1989
+ "epoch": 2.14,
1990
+ "learning_rate": 5.7533286615276805e-06,
1991
+ "loss": 0.5993,
1992
+ "step": 3080
1993
+ },
1994
+ {
1995
+ "epoch": 2.15,
1996
+ "learning_rate": 5.739313244569027e-06,
1997
+ "loss": 0.6155,
1998
+ "step": 3090
1999
+ },
2000
+ {
2001
+ "epoch": 2.16,
2002
+ "learning_rate": 5.725297827610372e-06,
2003
+ "loss": 0.6124,
2004
+ "step": 3100
2005
+ },
2006
+ {
2007
+ "epoch": 2.16,
2008
+ "learning_rate": 5.7112824106517175e-06,
2009
+ "loss": 0.5955,
2010
+ "step": 3110
2011
+ },
2012
+ {
2013
+ "epoch": 2.17,
2014
+ "learning_rate": 5.697266993693062e-06,
2015
+ "loss": 0.5802,
2016
+ "step": 3120
2017
+ },
2018
+ {
2019
+ "epoch": 2.18,
2020
+ "learning_rate": 5.683251576734409e-06,
2021
+ "loss": 0.6156,
2022
+ "step": 3130
2023
+ },
2024
+ {
2025
+ "epoch": 2.18,
2026
+ "learning_rate": 5.669236159775754e-06,
2027
+ "loss": 0.6344,
2028
+ "step": 3140
2029
+ },
2030
+ {
2031
+ "epoch": 2.19,
2032
+ "learning_rate": 5.655220742817099e-06,
2033
+ "loss": 0.5634,
2034
+ "step": 3150
2035
+ },
2036
+ {
2037
+ "epoch": 2.2,
2038
+ "learning_rate": 5.641205325858444e-06,
2039
+ "loss": 0.5955,
2040
+ "step": 3160
2041
+ },
2042
+ {
2043
+ "epoch": 2.21,
2044
+ "learning_rate": 5.627189908899791e-06,
2045
+ "loss": 0.6102,
2046
+ "step": 3170
2047
+ },
2048
+ {
2049
+ "epoch": 2.21,
2050
+ "learning_rate": 5.6131744919411356e-06,
2051
+ "loss": 0.6409,
2052
+ "step": 3180
2053
+ },
2054
+ {
2055
+ "epoch": 2.22,
2056
+ "learning_rate": 5.599159074982481e-06,
2057
+ "loss": 0.6575,
2058
+ "step": 3190
2059
+ },
2060
+ {
2061
+ "epoch": 2.23,
2062
+ "learning_rate": 5.585143658023827e-06,
2063
+ "loss": 0.6236,
2064
+ "step": 3200
2065
+ },
2066
+ {
2067
+ "epoch": 2.23,
2068
+ "learning_rate": 5.571128241065173e-06,
2069
+ "loss": 0.6078,
2070
+ "step": 3210
2071
+ },
2072
+ {
2073
+ "epoch": 2.24,
2074
+ "learning_rate": 5.557112824106517e-06,
2075
+ "loss": 0.6161,
2076
+ "step": 3220
2077
+ },
2078
+ {
2079
+ "epoch": 2.25,
2080
+ "learning_rate": 5.543097407147863e-06,
2081
+ "loss": 0.5838,
2082
+ "step": 3230
2083
+ },
2084
+ {
2085
+ "epoch": 2.25,
2086
+ "learning_rate": 5.529081990189209e-06,
2087
+ "loss": 0.6179,
2088
+ "step": 3240
2089
+ },
2090
+ {
2091
+ "epoch": 2.26,
2092
+ "learning_rate": 5.5150665732305544e-06,
2093
+ "loss": 0.5729,
2094
+ "step": 3250
2095
+ },
2096
+ {
2097
+ "epoch": 2.27,
2098
+ "learning_rate": 5.501051156271899e-06,
2099
+ "loss": 0.6353,
2100
+ "step": 3260
2101
+ },
2102
+ {
2103
+ "epoch": 2.27,
2104
+ "learning_rate": 5.487035739313244e-06,
2105
+ "loss": 0.5723,
2106
+ "step": 3270
2107
+ },
2108
+ {
2109
+ "epoch": 2.28,
2110
+ "learning_rate": 5.473020322354591e-06,
2111
+ "loss": 0.5983,
2112
+ "step": 3280
2113
+ },
2114
+ {
2115
+ "epoch": 2.29,
2116
+ "learning_rate": 5.459004905395936e-06,
2117
+ "loss": 0.5695,
2118
+ "step": 3290
2119
+ },
2120
+ {
2121
+ "epoch": 2.3,
2122
+ "learning_rate": 5.444989488437281e-06,
2123
+ "loss": 0.6189,
2124
+ "step": 3300
2125
+ },
2126
+ {
2127
+ "epoch": 2.3,
2128
+ "eval_custom_hf_pref_accuracy": 0.605,
2129
+ "eval_custom_hf_pref_kendalltau": 0.21,
2130
+ "eval_custom_hf_pref_loss": 0.6013281345367432,
2131
+ "eval_custom_hf_pref_neg_score": -0.3515034317970276,
2132
+ "eval_custom_hf_pref_pos_score": 0.02793554961681366,
2133
+ "eval_custom_hf_pref_runtime": 3.0849,
2134
+ "eval_custom_hf_pref_samples_per_second": 648.326,
2135
+ "eval_custom_hf_pref_score_diff": 0.37943896651268005,
2136
+ "eval_custom_hf_pref_steps_per_second": 81.041,
2137
+ "step": 3300
2138
+ },
2139
+ {
2140
+ "epoch": 2.3,
2141
+ "learning_rate": 5.430974071478628e-06,
2142
+ "loss": 0.6177,
2143
+ "step": 3310
2144
+ },
2145
+ {
2146
+ "epoch": 2.31,
2147
+ "learning_rate": 5.4169586545199725e-06,
2148
+ "loss": 0.616,
2149
+ "step": 3320
2150
+ },
2151
+ {
2152
+ "epoch": 2.32,
2153
+ "learning_rate": 5.402943237561318e-06,
2154
+ "loss": 0.6038,
2155
+ "step": 3330
2156
+ },
2157
+ {
2158
+ "epoch": 2.32,
2159
+ "learning_rate": 5.388927820602663e-06,
2160
+ "loss": 0.6017,
2161
+ "step": 3340
2162
+ },
2163
+ {
2164
+ "epoch": 2.33,
2165
+ "learning_rate": 5.3749124036440095e-06,
2166
+ "loss": 0.5796,
2167
+ "step": 3350
2168
+ },
2169
+ {
2170
+ "epoch": 2.34,
2171
+ "learning_rate": 5.360896986685354e-06,
2172
+ "loss": 0.6032,
2173
+ "step": 3360
2174
+ },
2175
+ {
2176
+ "epoch": 2.34,
2177
+ "learning_rate": 5.346881569726699e-06,
2178
+ "loss": 0.626,
2179
+ "step": 3370
2180
+ },
2181
+ {
2182
+ "epoch": 2.35,
2183
+ "learning_rate": 5.332866152768045e-06,
2184
+ "loss": 0.5943,
2185
+ "step": 3380
2186
+ },
2187
+ {
2188
+ "epoch": 2.36,
2189
+ "learning_rate": 5.318850735809391e-06,
2190
+ "loss": 0.5708,
2191
+ "step": 3390
2192
+ },
2193
+ {
2194
+ "epoch": 2.37,
2195
+ "learning_rate": 5.304835318850736e-06,
2196
+ "loss": 0.6458,
2197
+ "step": 3400
2198
+ },
2199
+ {
2200
+ "epoch": 2.37,
2201
+ "learning_rate": 5.290819901892081e-06,
2202
+ "loss": 0.6138,
2203
+ "step": 3410
2204
+ },
2205
+ {
2206
+ "epoch": 2.38,
2207
+ "learning_rate": 5.2768044849334275e-06,
2208
+ "loss": 0.6278,
2209
+ "step": 3420
2210
+ },
2211
+ {
2212
+ "epoch": 2.39,
2213
+ "learning_rate": 5.262789067974773e-06,
2214
+ "loss": 0.6138,
2215
+ "step": 3430
2216
+ },
2217
+ {
2218
+ "epoch": 2.39,
2219
+ "learning_rate": 5.248773651016118e-06,
2220
+ "loss": 0.6036,
2221
+ "step": 3440
2222
+ },
2223
+ {
2224
+ "epoch": 2.4,
2225
+ "learning_rate": 5.234758234057463e-06,
2226
+ "loss": 0.5893,
2227
+ "step": 3450
2228
+ },
2229
+ {
2230
+ "epoch": 2.41,
2231
+ "learning_rate": 5.220742817098809e-06,
2232
+ "loss": 0.618,
2233
+ "step": 3460
2234
+ },
2235
+ {
2236
+ "epoch": 2.41,
2237
+ "learning_rate": 5.206727400140154e-06,
2238
+ "loss": 0.5952,
2239
+ "step": 3470
2240
+ },
2241
+ {
2242
+ "epoch": 2.42,
2243
+ "learning_rate": 5.1927119831815e-06,
2244
+ "loss": 0.6331,
2245
+ "step": 3480
2246
+ },
2247
+ {
2248
+ "epoch": 2.43,
2249
+ "learning_rate": 5.178696566222845e-06,
2250
+ "loss": 0.6091,
2251
+ "step": 3490
2252
+ },
2253
+ {
2254
+ "epoch": 2.43,
2255
+ "learning_rate": 5.164681149264191e-06,
2256
+ "loss": 0.6245,
2257
+ "step": 3500
2258
+ },
2259
+ {
2260
+ "epoch": 2.44,
2261
+ "learning_rate": 5.150665732305536e-06,
2262
+ "loss": 0.5887,
2263
+ "step": 3510
2264
+ },
2265
+ {
2266
+ "epoch": 2.45,
2267
+ "learning_rate": 5.136650315346882e-06,
2268
+ "loss": 0.608,
2269
+ "step": 3520
2270
+ },
2271
+ {
2272
+ "epoch": 2.46,
2273
+ "learning_rate": 5.122634898388228e-06,
2274
+ "loss": 0.5959,
2275
+ "step": 3530
2276
+ },
2277
+ {
2278
+ "epoch": 2.46,
2279
+ "learning_rate": 5.108619481429573e-06,
2280
+ "loss": 0.6515,
2281
+ "step": 3540
2282
+ },
2283
+ {
2284
+ "epoch": 2.47,
2285
+ "learning_rate": 5.094604064470918e-06,
2286
+ "loss": 0.629,
2287
+ "step": 3550
2288
+ },
2289
+ {
2290
+ "epoch": 2.48,
2291
+ "learning_rate": 5.080588647512264e-06,
2292
+ "loss": 0.5898,
2293
+ "step": 3560
2294
+ },
2295
+ {
2296
+ "epoch": 2.48,
2297
+ "learning_rate": 5.06657323055361e-06,
2298
+ "loss": 0.5561,
2299
+ "step": 3570
2300
+ },
2301
+ {
2302
+ "epoch": 2.49,
2303
+ "learning_rate": 5.052557813594955e-06,
2304
+ "loss": 0.6031,
2305
+ "step": 3580
2306
+ },
2307
+ {
2308
+ "epoch": 2.5,
2309
+ "learning_rate": 5.0385423966363e-06,
2310
+ "loss": 0.5934,
2311
+ "step": 3590
2312
+ },
2313
+ {
2314
+ "epoch": 2.5,
2315
+ "learning_rate": 5.024526979677646e-06,
2316
+ "loss": 0.6393,
2317
+ "step": 3600
2318
+ },
2319
+ {
2320
+ "epoch": 2.5,
2321
+ "eval_custom_hf_pref_accuracy": 0.6085,
2322
+ "eval_custom_hf_pref_kendalltau": 0.217,
2323
+ "eval_custom_hf_pref_loss": 0.6061797142028809,
2324
+ "eval_custom_hf_pref_neg_score": -0.4991738796234131,
2325
+ "eval_custom_hf_pref_pos_score": -0.14666078984737396,
2326
+ "eval_custom_hf_pref_runtime": 3.0873,
2327
+ "eval_custom_hf_pref_samples_per_second": 647.819,
2328
+ "eval_custom_hf_pref_score_diff": 0.3525131344795227,
2329
+ "eval_custom_hf_pref_steps_per_second": 80.977,
2330
+ "step": 3600
2331
+ },
2332
+ {
2333
+ "epoch": 2.51,
2334
+ "learning_rate": 5.010511562718991e-06,
2335
+ "loss": 0.5903,
2336
+ "step": 3610
2337
+ },
2338
+ {
2339
+ "epoch": 2.52,
2340
+ "learning_rate": 4.996496145760337e-06,
2341
+ "loss": 0.5731,
2342
+ "step": 3620
2343
+ },
2344
+ {
2345
+ "epoch": 2.53,
2346
+ "learning_rate": 4.9824807288016825e-06,
2347
+ "loss": 0.6383,
2348
+ "step": 3630
2349
+ },
2350
+ {
2351
+ "epoch": 2.53,
2352
+ "learning_rate": 4.968465311843027e-06,
2353
+ "loss": 0.6008,
2354
+ "step": 3640
2355
+ },
2356
+ {
2357
+ "epoch": 2.54,
2358
+ "learning_rate": 4.954449894884373e-06,
2359
+ "loss": 0.6005,
2360
+ "step": 3650
2361
+ },
2362
+ {
2363
+ "epoch": 2.55,
2364
+ "learning_rate": 4.940434477925719e-06,
2365
+ "loss": 0.6348,
2366
+ "step": 3660
2367
+ },
2368
+ {
2369
+ "epoch": 2.55,
2370
+ "learning_rate": 4.926419060967064e-06,
2371
+ "loss": 0.6109,
2372
+ "step": 3670
2373
+ },
2374
+ {
2375
+ "epoch": 2.56,
2376
+ "learning_rate": 4.912403644008409e-06,
2377
+ "loss": 0.6219,
2378
+ "step": 3680
2379
+ },
2380
+ {
2381
+ "epoch": 2.57,
2382
+ "learning_rate": 4.898388227049755e-06,
2383
+ "loss": 0.589,
2384
+ "step": 3690
2385
+ },
2386
+ {
2387
+ "epoch": 2.57,
2388
+ "learning_rate": 4.8843728100911005e-06,
2389
+ "loss": 0.6278,
2390
+ "step": 3700
2391
+ },
2392
+ {
2393
+ "epoch": 2.58,
2394
+ "learning_rate": 4.870357393132446e-06,
2395
+ "loss": 0.6119,
2396
+ "step": 3710
2397
+ },
2398
+ {
2399
+ "epoch": 2.59,
2400
+ "learning_rate": 4.856341976173792e-06,
2401
+ "loss": 0.5835,
2402
+ "step": 3720
2403
+ },
2404
+ {
2405
+ "epoch": 2.59,
2406
+ "learning_rate": 4.842326559215137e-06,
2407
+ "loss": 0.6209,
2408
+ "step": 3730
2409
+ },
2410
+ {
2411
+ "epoch": 2.6,
2412
+ "learning_rate": 4.828311142256482e-06,
2413
+ "loss": 0.5998,
2414
+ "step": 3740
2415
+ },
2416
+ {
2417
+ "epoch": 2.61,
2418
+ "learning_rate": 4.814295725297828e-06,
2419
+ "loss": 0.5851,
2420
+ "step": 3750
2421
+ },
2422
+ {
2423
+ "epoch": 2.62,
2424
+ "learning_rate": 4.800280308339174e-06,
2425
+ "loss": 0.6362,
2426
+ "step": 3760
2427
+ },
2428
+ {
2429
+ "epoch": 2.62,
2430
+ "learning_rate": 4.786264891380519e-06,
2431
+ "loss": 0.6101,
2432
+ "step": 3770
2433
+ },
2434
+ {
2435
+ "epoch": 2.63,
2436
+ "learning_rate": 4.772249474421864e-06,
2437
+ "loss": 0.634,
2438
+ "step": 3780
2439
+ },
2440
+ {
2441
+ "epoch": 2.64,
2442
+ "learning_rate": 4.75823405746321e-06,
2443
+ "loss": 0.6183,
2444
+ "step": 3790
2445
+ },
2446
+ {
2447
+ "epoch": 2.64,
2448
+ "learning_rate": 4.744218640504556e-06,
2449
+ "loss": 0.6076,
2450
+ "step": 3800
2451
+ },
2452
+ {
2453
+ "epoch": 2.65,
2454
+ "learning_rate": 4.730203223545901e-06,
2455
+ "loss": 0.6406,
2456
+ "step": 3810
2457
+ },
2458
+ {
2459
+ "epoch": 2.66,
2460
+ "learning_rate": 4.716187806587246e-06,
2461
+ "loss": 0.5459,
2462
+ "step": 3820
2463
+ },
2464
+ {
2465
+ "epoch": 2.66,
2466
+ "learning_rate": 4.702172389628592e-06,
2467
+ "loss": 0.6272,
2468
+ "step": 3830
2469
+ },
2470
+ {
2471
+ "epoch": 2.67,
2472
+ "learning_rate": 4.6881569726699375e-06,
2473
+ "loss": 0.6035,
2474
+ "step": 3840
2475
+ },
2476
+ {
2477
+ "epoch": 2.68,
2478
+ "learning_rate": 4.674141555711283e-06,
2479
+ "loss": 0.6483,
2480
+ "step": 3850
2481
+ },
2482
+ {
2483
+ "epoch": 2.69,
2484
+ "learning_rate": 4.660126138752628e-06,
2485
+ "loss": 0.6028,
2486
+ "step": 3860
2487
+ },
2488
+ {
2489
+ "epoch": 2.69,
2490
+ "learning_rate": 4.646110721793974e-06,
2491
+ "loss": 0.6274,
2492
+ "step": 3870
2493
+ },
2494
+ {
2495
+ "epoch": 2.7,
2496
+ "learning_rate": 4.632095304835319e-06,
2497
+ "loss": 0.5906,
2498
+ "step": 3880
2499
+ },
2500
+ {
2501
+ "epoch": 2.71,
2502
+ "learning_rate": 4.618079887876665e-06,
2503
+ "loss": 0.6186,
2504
+ "step": 3890
2505
+ },
2506
+ {
2507
+ "epoch": 2.71,
2508
+ "learning_rate": 4.604064470918011e-06,
2509
+ "loss": 0.619,
2510
+ "step": 3900
2511
+ },
2512
+ {
2513
+ "epoch": 2.71,
2514
+ "eval_custom_hf_pref_accuracy": 0.613,
2515
+ "eval_custom_hf_pref_kendalltau": 0.226,
2516
+ "eval_custom_hf_pref_loss": 0.6025937795639038,
2517
+ "eval_custom_hf_pref_neg_score": -0.5224867463111877,
2518
+ "eval_custom_hf_pref_pos_score": -0.15825501084327698,
2519
+ "eval_custom_hf_pref_runtime": 3.0937,
2520
+ "eval_custom_hf_pref_samples_per_second": 646.485,
2521
+ "eval_custom_hf_pref_score_diff": 0.36423179507255554,
2522
+ "eval_custom_hf_pref_steps_per_second": 80.811,
2523
+ "step": 3900
2524
+ },
2525
+ {
2526
+ "epoch": 2.72,
2527
+ "learning_rate": 4.5900490539593555e-06,
2528
+ "loss": 0.6094,
2529
+ "step": 3910
2530
+ },
2531
+ {
2532
+ "epoch": 2.73,
2533
+ "learning_rate": 4.576033637000701e-06,
2534
+ "loss": 0.5902,
2535
+ "step": 3920
2536
+ },
2537
+ {
2538
+ "epoch": 2.73,
2539
+ "learning_rate": 4.562018220042047e-06,
2540
+ "loss": 0.5853,
2541
+ "step": 3930
2542
+ },
2543
+ {
2544
+ "epoch": 2.74,
2545
+ "learning_rate": 4.5480028030833925e-06,
2546
+ "loss": 0.6198,
2547
+ "step": 3940
2548
+ },
2549
+ {
2550
+ "epoch": 2.75,
2551
+ "learning_rate": 4.533987386124737e-06,
2552
+ "loss": 0.6498,
2553
+ "step": 3950
2554
+ },
2555
+ {
2556
+ "epoch": 2.75,
2557
+ "learning_rate": 4.519971969166083e-06,
2558
+ "loss": 0.6078,
2559
+ "step": 3960
2560
+ },
2561
+ {
2562
+ "epoch": 2.76,
2563
+ "learning_rate": 4.505956552207429e-06,
2564
+ "loss": 0.6154,
2565
+ "step": 3970
2566
+ },
2567
+ {
2568
+ "epoch": 2.77,
2569
+ "learning_rate": 4.491941135248774e-06,
2570
+ "loss": 0.6019,
2571
+ "step": 3980
2572
+ },
2573
+ {
2574
+ "epoch": 2.78,
2575
+ "learning_rate": 4.477925718290119e-06,
2576
+ "loss": 0.6166,
2577
+ "step": 3990
2578
+ },
2579
+ {
2580
+ "epoch": 2.78,
2581
+ "learning_rate": 4.463910301331465e-06,
2582
+ "loss": 0.6226,
2583
+ "step": 4000
2584
+ }
2585
+ ],
2586
+ "max_steps": 7185,
2587
+ "num_train_epochs": 5,
2588
+ "total_flos": 0.0,
2589
+ "trial_name": null,
2590
+ "trial_params": null
2591
+ }
checkpoint-4000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15d17db566d266013af03a48da12b993aa2b485f122344645b19641417f7840d
3
+ size 4881
checkpoint-5000/config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "tlc4418/pythia_70m_sft",
3
+ "architectures": [
4
+ "GPTNeoXRewardModel"
5
+ ],
6
+ "bos_token_id": 0,
7
+ "classifier_dropout": 0.1,
8
+ "eos_token_id": 0,
9
+ "hidden_act": "gelu",
10
+ "hidden_size": 512,
11
+ "initializer_range": 0.02,
12
+ "intermediate_size": 2048,
13
+ "layer_norm_eps": 1e-05,
14
+ "max_position_embeddings": 2048,
15
+ "model_type": "gpt_neox_reward_model",
16
+ "num_attention_heads": 8,
17
+ "num_hidden_layers": 6,
18
+ "pooling": "mean",
19
+ "reward_head_activation": "gelu",
20
+ "reward_head_dropout": 0.0,
21
+ "reward_head_hidden_dim": null,
22
+ "reward_head_type": "layernorm_linear",
23
+ "rotary_emb_base": 10000,
24
+ "rotary_pct": 0.25,
25
+ "tie_word_embeddings": false,
26
+ "torch_dtype": "bfloat16",
27
+ "transformers_version": "4.29.2",
28
+ "use_cache": true,
29
+ "use_parallel_residual": true,
30
+ "vocab_size": 50288
31
+ }
checkpoint-5000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5f97f1bbf3d71f7c9327588fc0a283a9480b2b3af93d9b7a408dd44dd503920f
3
+ size 178704331
checkpoint-5000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1e4b8390998307d7f76ad40137359d8ccd2bddcf8900c768fdfe9cc94ed6c453
3
+ size 114527658
checkpoint-5000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0a89d1514c0481cde885739c0642a9c433b3c0cf124a2cba4d70dd7aa4005559
3
+ size 14645
checkpoint-5000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad5809b1125a8b8d59bbfd55f25693fa27f204ad97152b395aac8149077ddd68
3
+ size 1465
checkpoint-5000/special_tokens_map.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<|prefix_end|>",
4
+ "<|assistant|>",
5
+ "<|prefix_begin|>",
6
+ "<|prompter|>",
7
+ "<|system|>"
8
+ ],
9
+ "bos_token": "<|endoftext|>",
10
+ "eos_token": "<|endoftext|>",
11
+ "pad_token": "<|padding|>",
12
+ "sep_token": "<|endoftext|>",
13
+ "unk_token": "<|endoftext|>"
14
+ }
checkpoint-5000/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-5000/tokenizer_config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "bos_token": "<|endoftext|>",
4
+ "clean_up_tokenization_spaces": true,
5
+ "eos_token": "<|endoftext|>",
6
+ "model_max_length": 1000000000000000019884624838656,
7
+ "tokenizer_class": "GPTNeoXTokenizer",
8
+ "unk_token": "<|endoftext|>"
9
+ }
checkpoint-5000/trainer_state.json ADDED
@@ -0,0 +1,3230 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.4782608695652173,
5
+ "global_step": 5000,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.0,
12
+ "learning_rate": 2.0000000000000002e-07,
13
+ "loss": 0.6842,
14
+ "step": 1
15
+ },
16
+ {
17
+ "epoch": 0.01,
18
+ "learning_rate": 2.0000000000000003e-06,
19
+ "loss": 0.6871,
20
+ "step": 10
21
+ },
22
+ {
23
+ "epoch": 0.01,
24
+ "learning_rate": 4.000000000000001e-06,
25
+ "loss": 0.691,
26
+ "step": 20
27
+ },
28
+ {
29
+ "epoch": 0.02,
30
+ "learning_rate": 6e-06,
31
+ "loss": 0.6848,
32
+ "step": 30
33
+ },
34
+ {
35
+ "epoch": 0.03,
36
+ "learning_rate": 8.000000000000001e-06,
37
+ "loss": 0.6873,
38
+ "step": 40
39
+ },
40
+ {
41
+ "epoch": 0.03,
42
+ "learning_rate": 1e-05,
43
+ "loss": 0.6902,
44
+ "step": 50
45
+ },
46
+ {
47
+ "epoch": 0.04,
48
+ "learning_rate": 9.985984583041346e-06,
49
+ "loss": 0.6842,
50
+ "step": 60
51
+ },
52
+ {
53
+ "epoch": 0.05,
54
+ "learning_rate": 9.971969166082692e-06,
55
+ "loss": 0.6908,
56
+ "step": 70
57
+ },
58
+ {
59
+ "epoch": 0.06,
60
+ "learning_rate": 9.957953749124038e-06,
61
+ "loss": 0.6865,
62
+ "step": 80
63
+ },
64
+ {
65
+ "epoch": 0.06,
66
+ "learning_rate": 9.943938332165382e-06,
67
+ "loss": 0.6834,
68
+ "step": 90
69
+ },
70
+ {
71
+ "epoch": 0.07,
72
+ "learning_rate": 9.929922915206728e-06,
73
+ "loss": 0.6884,
74
+ "step": 100
75
+ },
76
+ {
77
+ "epoch": 0.08,
78
+ "learning_rate": 9.915907498248075e-06,
79
+ "loss": 0.6837,
80
+ "step": 110
81
+ },
82
+ {
83
+ "epoch": 0.08,
84
+ "learning_rate": 9.901892081289419e-06,
85
+ "loss": 0.6883,
86
+ "step": 120
87
+ },
88
+ {
89
+ "epoch": 0.09,
90
+ "learning_rate": 9.887876664330765e-06,
91
+ "loss": 0.6816,
92
+ "step": 130
93
+ },
94
+ {
95
+ "epoch": 0.1,
96
+ "learning_rate": 9.87386124737211e-06,
97
+ "loss": 0.6906,
98
+ "step": 140
99
+ },
100
+ {
101
+ "epoch": 0.1,
102
+ "learning_rate": 9.859845830413456e-06,
103
+ "loss": 0.6875,
104
+ "step": 150
105
+ },
106
+ {
107
+ "epoch": 0.11,
108
+ "learning_rate": 9.845830413454802e-06,
109
+ "loss": 0.6805,
110
+ "step": 160
111
+ },
112
+ {
113
+ "epoch": 0.12,
114
+ "learning_rate": 9.831814996496146e-06,
115
+ "loss": 0.6857,
116
+ "step": 170
117
+ },
118
+ {
119
+ "epoch": 0.13,
120
+ "learning_rate": 9.817799579537493e-06,
121
+ "loss": 0.6804,
122
+ "step": 180
123
+ },
124
+ {
125
+ "epoch": 0.13,
126
+ "learning_rate": 9.803784162578837e-06,
127
+ "loss": 0.6803,
128
+ "step": 190
129
+ },
130
+ {
131
+ "epoch": 0.14,
132
+ "learning_rate": 9.789768745620183e-06,
133
+ "loss": 0.6872,
134
+ "step": 200
135
+ },
136
+ {
137
+ "epoch": 0.15,
138
+ "learning_rate": 9.775753328661528e-06,
139
+ "loss": 0.6846,
140
+ "step": 210
141
+ },
142
+ {
143
+ "epoch": 0.15,
144
+ "learning_rate": 9.761737911702874e-06,
145
+ "loss": 0.6828,
146
+ "step": 220
147
+ },
148
+ {
149
+ "epoch": 0.16,
150
+ "learning_rate": 9.74772249474422e-06,
151
+ "loss": 0.6699,
152
+ "step": 230
153
+ },
154
+ {
155
+ "epoch": 0.17,
156
+ "learning_rate": 9.733707077785565e-06,
157
+ "loss": 0.673,
158
+ "step": 240
159
+ },
160
+ {
161
+ "epoch": 0.17,
162
+ "learning_rate": 9.71969166082691e-06,
163
+ "loss": 0.6733,
164
+ "step": 250
165
+ },
166
+ {
167
+ "epoch": 0.18,
168
+ "learning_rate": 9.705676243868257e-06,
169
+ "loss": 0.6674,
170
+ "step": 260
171
+ },
172
+ {
173
+ "epoch": 0.19,
174
+ "learning_rate": 9.6916608269096e-06,
175
+ "loss": 0.6708,
176
+ "step": 270
177
+ },
178
+ {
179
+ "epoch": 0.19,
180
+ "learning_rate": 9.677645409950946e-06,
181
+ "loss": 0.6661,
182
+ "step": 280
183
+ },
184
+ {
185
+ "epoch": 0.2,
186
+ "learning_rate": 9.663629992992292e-06,
187
+ "loss": 0.677,
188
+ "step": 290
189
+ },
190
+ {
191
+ "epoch": 0.21,
192
+ "learning_rate": 9.649614576033638e-06,
193
+ "loss": 0.6657,
194
+ "step": 300
195
+ },
196
+ {
197
+ "epoch": 0.21,
198
+ "eval_custom_hf_pref_accuracy": 0.5485,
199
+ "eval_custom_hf_pref_kendalltau": 0.097,
200
+ "eval_custom_hf_pref_loss": 0.6715624928474426,
201
+ "eval_custom_hf_pref_neg_score": 0.46836376190185547,
202
+ "eval_custom_hf_pref_pos_score": 0.522513747215271,
203
+ "eval_custom_hf_pref_runtime": 3.1872,
204
+ "eval_custom_hf_pref_samples_per_second": 627.506,
205
+ "eval_custom_hf_pref_score_diff": 0.054149970412254333,
206
+ "eval_custom_hf_pref_steps_per_second": 78.438,
207
+ "step": 300
208
+ },
209
+ {
210
+ "epoch": 0.22,
211
+ "learning_rate": 9.635599159074983e-06,
212
+ "loss": 0.6747,
213
+ "step": 310
214
+ },
215
+ {
216
+ "epoch": 0.22,
217
+ "learning_rate": 9.621583742116329e-06,
218
+ "loss": 0.6668,
219
+ "step": 320
220
+ },
221
+ {
222
+ "epoch": 0.23,
223
+ "learning_rate": 9.607568325157675e-06,
224
+ "loss": 0.6599,
225
+ "step": 330
226
+ },
227
+ {
228
+ "epoch": 0.24,
229
+ "learning_rate": 9.59355290819902e-06,
230
+ "loss": 0.663,
231
+ "step": 340
232
+ },
233
+ {
234
+ "epoch": 0.24,
235
+ "learning_rate": 9.579537491240364e-06,
236
+ "loss": 0.6643,
237
+ "step": 350
238
+ },
239
+ {
240
+ "epoch": 0.25,
241
+ "learning_rate": 9.565522074281712e-06,
242
+ "loss": 0.6697,
243
+ "step": 360
244
+ },
245
+ {
246
+ "epoch": 0.26,
247
+ "learning_rate": 9.551506657323056e-06,
248
+ "loss": 0.6497,
249
+ "step": 370
250
+ },
251
+ {
252
+ "epoch": 0.26,
253
+ "learning_rate": 9.537491240364401e-06,
254
+ "loss": 0.6544,
255
+ "step": 380
256
+ },
257
+ {
258
+ "epoch": 0.27,
259
+ "learning_rate": 9.523475823405747e-06,
260
+ "loss": 0.6736,
261
+ "step": 390
262
+ },
263
+ {
264
+ "epoch": 0.28,
265
+ "learning_rate": 9.509460406447093e-06,
266
+ "loss": 0.6572,
267
+ "step": 400
268
+ },
269
+ {
270
+ "epoch": 0.29,
271
+ "learning_rate": 9.495444989488438e-06,
272
+ "loss": 0.6616,
273
+ "step": 410
274
+ },
275
+ {
276
+ "epoch": 0.29,
277
+ "learning_rate": 9.481429572529784e-06,
278
+ "loss": 0.6703,
279
+ "step": 420
280
+ },
281
+ {
282
+ "epoch": 0.3,
283
+ "learning_rate": 9.467414155571128e-06,
284
+ "loss": 0.6621,
285
+ "step": 430
286
+ },
287
+ {
288
+ "epoch": 0.31,
289
+ "learning_rate": 9.453398738612475e-06,
290
+ "loss": 0.6755,
291
+ "step": 440
292
+ },
293
+ {
294
+ "epoch": 0.31,
295
+ "learning_rate": 9.43938332165382e-06,
296
+ "loss": 0.6693,
297
+ "step": 450
298
+ },
299
+ {
300
+ "epoch": 0.32,
301
+ "learning_rate": 9.425367904695165e-06,
302
+ "loss": 0.6629,
303
+ "step": 460
304
+ },
305
+ {
306
+ "epoch": 0.33,
307
+ "learning_rate": 9.41135248773651e-06,
308
+ "loss": 0.672,
309
+ "step": 470
310
+ },
311
+ {
312
+ "epoch": 0.33,
313
+ "learning_rate": 9.397337070777856e-06,
314
+ "loss": 0.6682,
315
+ "step": 480
316
+ },
317
+ {
318
+ "epoch": 0.34,
319
+ "learning_rate": 9.383321653819202e-06,
320
+ "loss": 0.6667,
321
+ "step": 490
322
+ },
323
+ {
324
+ "epoch": 0.35,
325
+ "learning_rate": 9.369306236860548e-06,
326
+ "loss": 0.6632,
327
+ "step": 500
328
+ },
329
+ {
330
+ "epoch": 0.35,
331
+ "learning_rate": 9.355290819901893e-06,
332
+ "loss": 0.6374,
333
+ "step": 510
334
+ },
335
+ {
336
+ "epoch": 0.36,
337
+ "learning_rate": 9.341275402943239e-06,
338
+ "loss": 0.6528,
339
+ "step": 520
340
+ },
341
+ {
342
+ "epoch": 0.37,
343
+ "learning_rate": 9.327259985984583e-06,
344
+ "loss": 0.6658,
345
+ "step": 530
346
+ },
347
+ {
348
+ "epoch": 0.38,
349
+ "learning_rate": 9.313244569025929e-06,
350
+ "loss": 0.6489,
351
+ "step": 540
352
+ },
353
+ {
354
+ "epoch": 0.38,
355
+ "learning_rate": 9.299229152067274e-06,
356
+ "loss": 0.6551,
357
+ "step": 550
358
+ },
359
+ {
360
+ "epoch": 0.39,
361
+ "learning_rate": 9.28521373510862e-06,
362
+ "loss": 0.6666,
363
+ "step": 560
364
+ },
365
+ {
366
+ "epoch": 0.4,
367
+ "learning_rate": 9.271198318149966e-06,
368
+ "loss": 0.6398,
369
+ "step": 570
370
+ },
371
+ {
372
+ "epoch": 0.4,
373
+ "learning_rate": 9.257182901191311e-06,
374
+ "loss": 0.6515,
375
+ "step": 580
376
+ },
377
+ {
378
+ "epoch": 0.41,
379
+ "learning_rate": 9.243167484232657e-06,
380
+ "loss": 0.6646,
381
+ "step": 590
382
+ },
383
+ {
384
+ "epoch": 0.42,
385
+ "learning_rate": 9.229152067274003e-06,
386
+ "loss": 0.6372,
387
+ "step": 600
388
+ },
389
+ {
390
+ "epoch": 0.42,
391
+ "eval_custom_hf_pref_accuracy": 0.5785,
392
+ "eval_custom_hf_pref_kendalltau": 0.157,
393
+ "eval_custom_hf_pref_loss": 0.6421093940734863,
394
+ "eval_custom_hf_pref_neg_score": 0.48188385367393494,
395
+ "eval_custom_hf_pref_pos_score": 0.6362329721450806,
396
+ "eval_custom_hf_pref_runtime": 3.0972,
397
+ "eval_custom_hf_pref_samples_per_second": 645.736,
398
+ "eval_custom_hf_pref_score_diff": 0.15434913337230682,
399
+ "eval_custom_hf_pref_steps_per_second": 80.717,
400
+ "step": 600
401
+ },
402
+ {
403
+ "epoch": 0.42,
404
+ "learning_rate": 9.215136650315347e-06,
405
+ "loss": 0.6566,
406
+ "step": 610
407
+ },
408
+ {
409
+ "epoch": 0.43,
410
+ "learning_rate": 9.201121233356694e-06,
411
+ "loss": 0.6631,
412
+ "step": 620
413
+ },
414
+ {
415
+ "epoch": 0.44,
416
+ "learning_rate": 9.187105816398038e-06,
417
+ "loss": 0.6516,
418
+ "step": 630
419
+ },
420
+ {
421
+ "epoch": 0.45,
422
+ "learning_rate": 9.173090399439384e-06,
423
+ "loss": 0.6468,
424
+ "step": 640
425
+ },
426
+ {
427
+ "epoch": 0.45,
428
+ "learning_rate": 9.15907498248073e-06,
429
+ "loss": 0.6501,
430
+ "step": 650
431
+ },
432
+ {
433
+ "epoch": 0.46,
434
+ "learning_rate": 9.145059565522075e-06,
435
+ "loss": 0.6638,
436
+ "step": 660
437
+ },
438
+ {
439
+ "epoch": 0.47,
440
+ "learning_rate": 9.131044148563421e-06,
441
+ "loss": 0.6633,
442
+ "step": 670
443
+ },
444
+ {
445
+ "epoch": 0.47,
446
+ "learning_rate": 9.117028731604765e-06,
447
+ "loss": 0.6627,
448
+ "step": 680
449
+ },
450
+ {
451
+ "epoch": 0.48,
452
+ "learning_rate": 9.103013314646112e-06,
453
+ "loss": 0.6355,
454
+ "step": 690
455
+ },
456
+ {
457
+ "epoch": 0.49,
458
+ "learning_rate": 9.088997897687458e-06,
459
+ "loss": 0.6287,
460
+ "step": 700
461
+ },
462
+ {
463
+ "epoch": 0.49,
464
+ "learning_rate": 9.074982480728802e-06,
465
+ "loss": 0.6534,
466
+ "step": 710
467
+ },
468
+ {
469
+ "epoch": 0.5,
470
+ "learning_rate": 9.060967063770148e-06,
471
+ "loss": 0.6698,
472
+ "step": 720
473
+ },
474
+ {
475
+ "epoch": 0.51,
476
+ "learning_rate": 9.046951646811493e-06,
477
+ "loss": 0.6452,
478
+ "step": 730
479
+ },
480
+ {
481
+ "epoch": 0.51,
482
+ "learning_rate": 9.032936229852839e-06,
483
+ "loss": 0.641,
484
+ "step": 740
485
+ },
486
+ {
487
+ "epoch": 0.52,
488
+ "learning_rate": 9.018920812894185e-06,
489
+ "loss": 0.6317,
490
+ "step": 750
491
+ },
492
+ {
493
+ "epoch": 0.53,
494
+ "learning_rate": 9.00490539593553e-06,
495
+ "loss": 0.6612,
496
+ "step": 760
497
+ },
498
+ {
499
+ "epoch": 0.54,
500
+ "learning_rate": 8.990889978976876e-06,
501
+ "loss": 0.6315,
502
+ "step": 770
503
+ },
504
+ {
505
+ "epoch": 0.54,
506
+ "learning_rate": 8.97687456201822e-06,
507
+ "loss": 0.6552,
508
+ "step": 780
509
+ },
510
+ {
511
+ "epoch": 0.55,
512
+ "learning_rate": 8.962859145059566e-06,
513
+ "loss": 0.6451,
514
+ "step": 790
515
+ },
516
+ {
517
+ "epoch": 0.56,
518
+ "learning_rate": 8.948843728100913e-06,
519
+ "loss": 0.64,
520
+ "step": 800
521
+ },
522
+ {
523
+ "epoch": 0.56,
524
+ "learning_rate": 8.934828311142257e-06,
525
+ "loss": 0.6313,
526
+ "step": 810
527
+ },
528
+ {
529
+ "epoch": 0.57,
530
+ "learning_rate": 8.920812894183603e-06,
531
+ "loss": 0.6466,
532
+ "step": 820
533
+ },
534
+ {
535
+ "epoch": 0.58,
536
+ "learning_rate": 8.906797477224948e-06,
537
+ "loss": 0.6406,
538
+ "step": 830
539
+ },
540
+ {
541
+ "epoch": 0.58,
542
+ "learning_rate": 8.892782060266294e-06,
543
+ "loss": 0.6479,
544
+ "step": 840
545
+ },
546
+ {
547
+ "epoch": 0.59,
548
+ "learning_rate": 8.87876664330764e-06,
549
+ "loss": 0.6602,
550
+ "step": 850
551
+ },
552
+ {
553
+ "epoch": 0.6,
554
+ "learning_rate": 8.864751226348984e-06,
555
+ "loss": 0.6279,
556
+ "step": 860
557
+ },
558
+ {
559
+ "epoch": 0.61,
560
+ "learning_rate": 8.850735809390331e-06,
561
+ "loss": 0.6439,
562
+ "step": 870
563
+ },
564
+ {
565
+ "epoch": 0.61,
566
+ "learning_rate": 8.836720392431675e-06,
567
+ "loss": 0.6192,
568
+ "step": 880
569
+ },
570
+ {
571
+ "epoch": 0.62,
572
+ "learning_rate": 8.82270497547302e-06,
573
+ "loss": 0.6529,
574
+ "step": 890
575
+ },
576
+ {
577
+ "epoch": 0.63,
578
+ "learning_rate": 8.808689558514366e-06,
579
+ "loss": 0.6246,
580
+ "step": 900
581
+ },
582
+ {
583
+ "epoch": 0.63,
584
+ "eval_custom_hf_pref_accuracy": 0.5795,
585
+ "eval_custom_hf_pref_kendalltau": 0.159,
586
+ "eval_custom_hf_pref_loss": 0.6517187356948853,
587
+ "eval_custom_hf_pref_neg_score": -0.18239864706993103,
588
+ "eval_custom_hf_pref_pos_score": -0.047199830412864685,
589
+ "eval_custom_hf_pref_runtime": 3.0666,
590
+ "eval_custom_hf_pref_samples_per_second": 652.191,
591
+ "eval_custom_hf_pref_score_diff": 0.13519881665706635,
592
+ "eval_custom_hf_pref_steps_per_second": 81.524,
593
+ "step": 900
594
+ },
595
+ {
596
+ "epoch": 0.63,
597
+ "learning_rate": 8.794674141555712e-06,
598
+ "loss": 0.6453,
599
+ "step": 910
600
+ },
601
+ {
602
+ "epoch": 0.64,
603
+ "learning_rate": 8.780658724597058e-06,
604
+ "loss": 0.6557,
605
+ "step": 920
606
+ },
607
+ {
608
+ "epoch": 0.65,
609
+ "learning_rate": 8.766643307638403e-06,
610
+ "loss": 0.6271,
611
+ "step": 930
612
+ },
613
+ {
614
+ "epoch": 0.65,
615
+ "learning_rate": 8.752627890679747e-06,
616
+ "loss": 0.6472,
617
+ "step": 940
618
+ },
619
+ {
620
+ "epoch": 0.66,
621
+ "learning_rate": 8.738612473721095e-06,
622
+ "loss": 0.6299,
623
+ "step": 950
624
+ },
625
+ {
626
+ "epoch": 0.67,
627
+ "learning_rate": 8.724597056762439e-06,
628
+ "loss": 0.6403,
629
+ "step": 960
630
+ },
631
+ {
632
+ "epoch": 0.67,
633
+ "learning_rate": 8.710581639803784e-06,
634
+ "loss": 0.6343,
635
+ "step": 970
636
+ },
637
+ {
638
+ "epoch": 0.68,
639
+ "learning_rate": 8.69656622284513e-06,
640
+ "loss": 0.6274,
641
+ "step": 980
642
+ },
643
+ {
644
+ "epoch": 0.69,
645
+ "learning_rate": 8.682550805886476e-06,
646
+ "loss": 0.6534,
647
+ "step": 990
648
+ },
649
+ {
650
+ "epoch": 0.7,
651
+ "learning_rate": 8.668535388927821e-06,
652
+ "loss": 0.6244,
653
+ "step": 1000
654
+ },
655
+ {
656
+ "epoch": 0.7,
657
+ "learning_rate": 8.654519971969167e-06,
658
+ "loss": 0.6237,
659
+ "step": 1010
660
+ },
661
+ {
662
+ "epoch": 0.71,
663
+ "learning_rate": 8.640504555010513e-06,
664
+ "loss": 0.6531,
665
+ "step": 1020
666
+ },
667
+ {
668
+ "epoch": 0.72,
669
+ "learning_rate": 8.626489138051858e-06,
670
+ "loss": 0.6347,
671
+ "step": 1030
672
+ },
673
+ {
674
+ "epoch": 0.72,
675
+ "learning_rate": 8.612473721093202e-06,
676
+ "loss": 0.6513,
677
+ "step": 1040
678
+ },
679
+ {
680
+ "epoch": 0.73,
681
+ "learning_rate": 8.598458304134548e-06,
682
+ "loss": 0.648,
683
+ "step": 1050
684
+ },
685
+ {
686
+ "epoch": 0.74,
687
+ "learning_rate": 8.584442887175894e-06,
688
+ "loss": 0.6472,
689
+ "step": 1060
690
+ },
691
+ {
692
+ "epoch": 0.74,
693
+ "learning_rate": 8.57042747021724e-06,
694
+ "loss": 0.6305,
695
+ "step": 1070
696
+ },
697
+ {
698
+ "epoch": 0.75,
699
+ "learning_rate": 8.556412053258585e-06,
700
+ "loss": 0.6241,
701
+ "step": 1080
702
+ },
703
+ {
704
+ "epoch": 0.76,
705
+ "learning_rate": 8.54239663629993e-06,
706
+ "loss": 0.6763,
707
+ "step": 1090
708
+ },
709
+ {
710
+ "epoch": 0.77,
711
+ "learning_rate": 8.528381219341276e-06,
712
+ "loss": 0.6219,
713
+ "step": 1100
714
+ },
715
+ {
716
+ "epoch": 0.77,
717
+ "learning_rate": 8.514365802382622e-06,
718
+ "loss": 0.6384,
719
+ "step": 1110
720
+ },
721
+ {
722
+ "epoch": 0.78,
723
+ "learning_rate": 8.500350385423966e-06,
724
+ "loss": 0.6188,
725
+ "step": 1120
726
+ },
727
+ {
728
+ "epoch": 0.79,
729
+ "learning_rate": 8.486334968465313e-06,
730
+ "loss": 0.6295,
731
+ "step": 1130
732
+ },
733
+ {
734
+ "epoch": 0.79,
735
+ "learning_rate": 8.472319551506657e-06,
736
+ "loss": 0.6276,
737
+ "step": 1140
738
+ },
739
+ {
740
+ "epoch": 0.8,
741
+ "learning_rate": 8.458304134548003e-06,
742
+ "loss": 0.6501,
743
+ "step": 1150
744
+ },
745
+ {
746
+ "epoch": 0.81,
747
+ "learning_rate": 8.444288717589349e-06,
748
+ "loss": 0.6282,
749
+ "step": 1160
750
+ },
751
+ {
752
+ "epoch": 0.81,
753
+ "learning_rate": 8.430273300630695e-06,
754
+ "loss": 0.628,
755
+ "step": 1170
756
+ },
757
+ {
758
+ "epoch": 0.82,
759
+ "learning_rate": 8.41625788367204e-06,
760
+ "loss": 0.6018,
761
+ "step": 1180
762
+ },
763
+ {
764
+ "epoch": 0.83,
765
+ "learning_rate": 8.402242466713386e-06,
766
+ "loss": 0.6307,
767
+ "step": 1190
768
+ },
769
+ {
770
+ "epoch": 0.83,
771
+ "learning_rate": 8.388227049754732e-06,
772
+ "loss": 0.6083,
773
+ "step": 1200
774
+ },
775
+ {
776
+ "epoch": 0.83,
777
+ "eval_custom_hf_pref_accuracy": 0.564,
778
+ "eval_custom_hf_pref_kendalltau": 0.128,
779
+ "eval_custom_hf_pref_loss": 0.6535468697547913,
780
+ "eval_custom_hf_pref_neg_score": -0.3474700450897217,
781
+ "eval_custom_hf_pref_pos_score": -0.20932382345199585,
782
+ "eval_custom_hf_pref_runtime": 3.0757,
783
+ "eval_custom_hf_pref_samples_per_second": 650.252,
784
+ "eval_custom_hf_pref_score_diff": 0.13814620673656464,
785
+ "eval_custom_hf_pref_steps_per_second": 81.282,
786
+ "step": 1200
787
+ },
788
+ {
789
+ "epoch": 0.84,
790
+ "learning_rate": 8.374211632796077e-06,
791
+ "loss": 0.629,
792
+ "step": 1210
793
+ },
794
+ {
795
+ "epoch": 0.85,
796
+ "learning_rate": 8.360196215837421e-06,
797
+ "loss": 0.6346,
798
+ "step": 1220
799
+ },
800
+ {
801
+ "epoch": 0.86,
802
+ "learning_rate": 8.346180798878767e-06,
803
+ "loss": 0.6366,
804
+ "step": 1230
805
+ },
806
+ {
807
+ "epoch": 0.86,
808
+ "learning_rate": 8.332165381920113e-06,
809
+ "loss": 0.6231,
810
+ "step": 1240
811
+ },
812
+ {
813
+ "epoch": 0.87,
814
+ "learning_rate": 8.318149964961458e-06,
815
+ "loss": 0.6154,
816
+ "step": 1250
817
+ },
818
+ {
819
+ "epoch": 0.88,
820
+ "learning_rate": 8.304134548002804e-06,
821
+ "loss": 0.6288,
822
+ "step": 1260
823
+ },
824
+ {
825
+ "epoch": 0.88,
826
+ "learning_rate": 8.29011913104415e-06,
827
+ "loss": 0.612,
828
+ "step": 1270
829
+ },
830
+ {
831
+ "epoch": 0.89,
832
+ "learning_rate": 8.276103714085495e-06,
833
+ "loss": 0.6154,
834
+ "step": 1280
835
+ },
836
+ {
837
+ "epoch": 0.9,
838
+ "learning_rate": 8.262088297126841e-06,
839
+ "loss": 0.6107,
840
+ "step": 1290
841
+ },
842
+ {
843
+ "epoch": 0.9,
844
+ "learning_rate": 8.248072880168185e-06,
845
+ "loss": 0.5914,
846
+ "step": 1300
847
+ },
848
+ {
849
+ "epoch": 0.91,
850
+ "learning_rate": 8.234057463209532e-06,
851
+ "loss": 0.6288,
852
+ "step": 1310
853
+ },
854
+ {
855
+ "epoch": 0.92,
856
+ "learning_rate": 8.220042046250876e-06,
857
+ "loss": 0.6553,
858
+ "step": 1320
859
+ },
860
+ {
861
+ "epoch": 0.93,
862
+ "learning_rate": 8.206026629292222e-06,
863
+ "loss": 0.6075,
864
+ "step": 1330
865
+ },
866
+ {
867
+ "epoch": 0.93,
868
+ "learning_rate": 8.192011212333568e-06,
869
+ "loss": 0.6223,
870
+ "step": 1340
871
+ },
872
+ {
873
+ "epoch": 0.94,
874
+ "learning_rate": 8.177995795374913e-06,
875
+ "loss": 0.5895,
876
+ "step": 1350
877
+ },
878
+ {
879
+ "epoch": 0.95,
880
+ "learning_rate": 8.163980378416259e-06,
881
+ "loss": 0.6592,
882
+ "step": 1360
883
+ },
884
+ {
885
+ "epoch": 0.95,
886
+ "learning_rate": 8.149964961457603e-06,
887
+ "loss": 0.6402,
888
+ "step": 1370
889
+ },
890
+ {
891
+ "epoch": 0.96,
892
+ "learning_rate": 8.13594954449895e-06,
893
+ "loss": 0.6228,
894
+ "step": 1380
895
+ },
896
+ {
897
+ "epoch": 0.97,
898
+ "learning_rate": 8.121934127540296e-06,
899
+ "loss": 0.6145,
900
+ "step": 1390
901
+ },
902
+ {
903
+ "epoch": 0.97,
904
+ "learning_rate": 8.10791871058164e-06,
905
+ "loss": 0.5906,
906
+ "step": 1400
907
+ },
908
+ {
909
+ "epoch": 0.98,
910
+ "learning_rate": 8.093903293622986e-06,
911
+ "loss": 0.6153,
912
+ "step": 1410
913
+ },
914
+ {
915
+ "epoch": 0.99,
916
+ "learning_rate": 8.079887876664331e-06,
917
+ "loss": 0.6331,
918
+ "step": 1420
919
+ },
920
+ {
921
+ "epoch": 0.99,
922
+ "learning_rate": 8.065872459705677e-06,
923
+ "loss": 0.6308,
924
+ "step": 1430
925
+ },
926
+ {
927
+ "epoch": 1.0,
928
+ "learning_rate": 8.051857042747023e-06,
929
+ "loss": 0.6269,
930
+ "step": 1440
931
+ },
932
+ {
933
+ "epoch": 1.01,
934
+ "learning_rate": 8.037841625788367e-06,
935
+ "loss": 0.6243,
936
+ "step": 1450
937
+ },
938
+ {
939
+ "epoch": 1.02,
940
+ "learning_rate": 8.023826208829714e-06,
941
+ "loss": 0.6369,
942
+ "step": 1460
943
+ },
944
+ {
945
+ "epoch": 1.02,
946
+ "learning_rate": 8.009810791871058e-06,
947
+ "loss": 0.6384,
948
+ "step": 1470
949
+ },
950
+ {
951
+ "epoch": 1.03,
952
+ "learning_rate": 7.995795374912404e-06,
953
+ "loss": 0.6497,
954
+ "step": 1480
955
+ },
956
+ {
957
+ "epoch": 1.04,
958
+ "learning_rate": 7.981779957953751e-06,
959
+ "loss": 0.586,
960
+ "step": 1490
961
+ },
962
+ {
963
+ "epoch": 1.04,
964
+ "learning_rate": 7.967764540995095e-06,
965
+ "loss": 0.6172,
966
+ "step": 1500
967
+ },
968
+ {
969
+ "epoch": 1.04,
970
+ "eval_custom_hf_pref_accuracy": 0.5905,
971
+ "eval_custom_hf_pref_kendalltau": 0.181,
972
+ "eval_custom_hf_pref_loss": 0.6357187628746033,
973
+ "eval_custom_hf_pref_neg_score": -0.4265330135822296,
974
+ "eval_custom_hf_pref_pos_score": -0.18024785816669464,
975
+ "eval_custom_hf_pref_runtime": 3.0869,
976
+ "eval_custom_hf_pref_samples_per_second": 647.904,
977
+ "eval_custom_hf_pref_score_diff": 0.24628514051437378,
978
+ "eval_custom_hf_pref_steps_per_second": 80.988,
979
+ "step": 1500
980
+ },
981
+ {
982
+ "epoch": 1.05,
983
+ "learning_rate": 7.95374912403644e-06,
984
+ "loss": 0.6429,
985
+ "step": 1510
986
+ },
987
+ {
988
+ "epoch": 1.06,
989
+ "learning_rate": 7.939733707077786e-06,
990
+ "loss": 0.6088,
991
+ "step": 1520
992
+ },
993
+ {
994
+ "epoch": 1.06,
995
+ "learning_rate": 7.925718290119132e-06,
996
+ "loss": 0.6179,
997
+ "step": 1530
998
+ },
999
+ {
1000
+ "epoch": 1.07,
1001
+ "learning_rate": 7.911702873160478e-06,
1002
+ "loss": 0.606,
1003
+ "step": 1540
1004
+ },
1005
+ {
1006
+ "epoch": 1.08,
1007
+ "learning_rate": 7.897687456201822e-06,
1008
+ "loss": 0.6117,
1009
+ "step": 1550
1010
+ },
1011
+ {
1012
+ "epoch": 1.09,
1013
+ "learning_rate": 7.883672039243167e-06,
1014
+ "loss": 0.6182,
1015
+ "step": 1560
1016
+ },
1017
+ {
1018
+ "epoch": 1.09,
1019
+ "learning_rate": 7.869656622284513e-06,
1020
+ "loss": 0.6339,
1021
+ "step": 1570
1022
+ },
1023
+ {
1024
+ "epoch": 1.1,
1025
+ "learning_rate": 7.855641205325859e-06,
1026
+ "loss": 0.6183,
1027
+ "step": 1580
1028
+ },
1029
+ {
1030
+ "epoch": 1.11,
1031
+ "learning_rate": 7.841625788367204e-06,
1032
+ "loss": 0.5998,
1033
+ "step": 1590
1034
+ },
1035
+ {
1036
+ "epoch": 1.11,
1037
+ "learning_rate": 7.82761037140855e-06,
1038
+ "loss": 0.6292,
1039
+ "step": 1600
1040
+ },
1041
+ {
1042
+ "epoch": 1.12,
1043
+ "learning_rate": 7.813594954449896e-06,
1044
+ "loss": 0.6168,
1045
+ "step": 1610
1046
+ },
1047
+ {
1048
+ "epoch": 1.13,
1049
+ "learning_rate": 7.799579537491241e-06,
1050
+ "loss": 0.6152,
1051
+ "step": 1620
1052
+ },
1053
+ {
1054
+ "epoch": 1.13,
1055
+ "learning_rate": 7.785564120532585e-06,
1056
+ "loss": 0.6316,
1057
+ "step": 1630
1058
+ },
1059
+ {
1060
+ "epoch": 1.14,
1061
+ "learning_rate": 7.771548703573933e-06,
1062
+ "loss": 0.615,
1063
+ "step": 1640
1064
+ },
1065
+ {
1066
+ "epoch": 1.15,
1067
+ "learning_rate": 7.757533286615277e-06,
1068
+ "loss": 0.6316,
1069
+ "step": 1650
1070
+ },
1071
+ {
1072
+ "epoch": 1.15,
1073
+ "learning_rate": 7.743517869656622e-06,
1074
+ "loss": 0.6387,
1075
+ "step": 1660
1076
+ },
1077
+ {
1078
+ "epoch": 1.16,
1079
+ "learning_rate": 7.729502452697968e-06,
1080
+ "loss": 0.6166,
1081
+ "step": 1670
1082
+ },
1083
+ {
1084
+ "epoch": 1.17,
1085
+ "learning_rate": 7.715487035739314e-06,
1086
+ "loss": 0.6316,
1087
+ "step": 1680
1088
+ },
1089
+ {
1090
+ "epoch": 1.18,
1091
+ "learning_rate": 7.70147161878066e-06,
1092
+ "loss": 0.6261,
1093
+ "step": 1690
1094
+ },
1095
+ {
1096
+ "epoch": 1.18,
1097
+ "learning_rate": 7.687456201822005e-06,
1098
+ "loss": 0.5966,
1099
+ "step": 1700
1100
+ },
1101
+ {
1102
+ "epoch": 1.19,
1103
+ "learning_rate": 7.67344078486335e-06,
1104
+ "loss": 0.6077,
1105
+ "step": 1710
1106
+ },
1107
+ {
1108
+ "epoch": 1.2,
1109
+ "learning_rate": 7.659425367904697e-06,
1110
+ "loss": 0.6325,
1111
+ "step": 1720
1112
+ },
1113
+ {
1114
+ "epoch": 1.2,
1115
+ "learning_rate": 7.64540995094604e-06,
1116
+ "loss": 0.6158,
1117
+ "step": 1730
1118
+ },
1119
+ {
1120
+ "epoch": 1.21,
1121
+ "learning_rate": 7.631394533987386e-06,
1122
+ "loss": 0.6196,
1123
+ "step": 1740
1124
+ },
1125
+ {
1126
+ "epoch": 1.22,
1127
+ "learning_rate": 7.617379117028733e-06,
1128
+ "loss": 0.6411,
1129
+ "step": 1750
1130
+ },
1131
+ {
1132
+ "epoch": 1.22,
1133
+ "learning_rate": 7.6033637000700775e-06,
1134
+ "loss": 0.616,
1135
+ "step": 1760
1136
+ },
1137
+ {
1138
+ "epoch": 1.23,
1139
+ "learning_rate": 7.589348283111423e-06,
1140
+ "loss": 0.6252,
1141
+ "step": 1770
1142
+ },
1143
+ {
1144
+ "epoch": 1.24,
1145
+ "learning_rate": 7.575332866152769e-06,
1146
+ "loss": 0.6171,
1147
+ "step": 1780
1148
+ },
1149
+ {
1150
+ "epoch": 1.25,
1151
+ "learning_rate": 7.5613174491941146e-06,
1152
+ "loss": 0.6135,
1153
+ "step": 1790
1154
+ },
1155
+ {
1156
+ "epoch": 1.25,
1157
+ "learning_rate": 7.547302032235459e-06,
1158
+ "loss": 0.6322,
1159
+ "step": 1800
1160
+ },
1161
+ {
1162
+ "epoch": 1.25,
1163
+ "eval_custom_hf_pref_accuracy": 0.6035,
1164
+ "eval_custom_hf_pref_kendalltau": 0.207,
1165
+ "eval_custom_hf_pref_loss": 0.6257030963897705,
1166
+ "eval_custom_hf_pref_neg_score": -0.47859320044517517,
1167
+ "eval_custom_hf_pref_pos_score": -0.20773868262767792,
1168
+ "eval_custom_hf_pref_runtime": 3.1023,
1169
+ "eval_custom_hf_pref_samples_per_second": 644.681,
1170
+ "eval_custom_hf_pref_score_diff": 0.27085456252098083,
1171
+ "eval_custom_hf_pref_steps_per_second": 80.585,
1172
+ "step": 1800
1173
+ },
1174
+ {
1175
+ "epoch": 1.26,
1176
+ "learning_rate": 7.533286615276805e-06,
1177
+ "loss": 0.6259,
1178
+ "step": 1810
1179
+ },
1180
+ {
1181
+ "epoch": 1.27,
1182
+ "learning_rate": 7.519271198318151e-06,
1183
+ "loss": 0.5938,
1184
+ "step": 1820
1185
+ },
1186
+ {
1187
+ "epoch": 1.27,
1188
+ "learning_rate": 7.505255781359496e-06,
1189
+ "loss": 0.5876,
1190
+ "step": 1830
1191
+ },
1192
+ {
1193
+ "epoch": 1.28,
1194
+ "learning_rate": 7.491240364400841e-06,
1195
+ "loss": 0.6386,
1196
+ "step": 1840
1197
+ },
1198
+ {
1199
+ "epoch": 1.29,
1200
+ "learning_rate": 7.477224947442187e-06,
1201
+ "loss": 0.5974,
1202
+ "step": 1850
1203
+ },
1204
+ {
1205
+ "epoch": 1.29,
1206
+ "learning_rate": 7.463209530483533e-06,
1207
+ "loss": 0.6019,
1208
+ "step": 1860
1209
+ },
1210
+ {
1211
+ "epoch": 1.3,
1212
+ "learning_rate": 7.449194113524878e-06,
1213
+ "loss": 0.5902,
1214
+ "step": 1870
1215
+ },
1216
+ {
1217
+ "epoch": 1.31,
1218
+ "learning_rate": 7.435178696566223e-06,
1219
+ "loss": 0.5528,
1220
+ "step": 1880
1221
+ },
1222
+ {
1223
+ "epoch": 1.31,
1224
+ "learning_rate": 7.42116327960757e-06,
1225
+ "loss": 0.6031,
1226
+ "step": 1890
1227
+ },
1228
+ {
1229
+ "epoch": 1.32,
1230
+ "learning_rate": 7.4071478626489144e-06,
1231
+ "loss": 0.5838,
1232
+ "step": 1900
1233
+ },
1234
+ {
1235
+ "epoch": 1.33,
1236
+ "learning_rate": 7.39313244569026e-06,
1237
+ "loss": 0.6493,
1238
+ "step": 1910
1239
+ },
1240
+ {
1241
+ "epoch": 1.34,
1242
+ "learning_rate": 7.379117028731605e-06,
1243
+ "loss": 0.6307,
1244
+ "step": 1920
1245
+ },
1246
+ {
1247
+ "epoch": 1.34,
1248
+ "learning_rate": 7.3651016117729515e-06,
1249
+ "loss": 0.5951,
1250
+ "step": 1930
1251
+ },
1252
+ {
1253
+ "epoch": 1.35,
1254
+ "learning_rate": 7.351086194814296e-06,
1255
+ "loss": 0.5965,
1256
+ "step": 1940
1257
+ },
1258
+ {
1259
+ "epoch": 1.36,
1260
+ "learning_rate": 7.337070777855642e-06,
1261
+ "loss": 0.5882,
1262
+ "step": 1950
1263
+ },
1264
+ {
1265
+ "epoch": 1.36,
1266
+ "learning_rate": 7.323055360896987e-06,
1267
+ "loss": 0.5943,
1268
+ "step": 1960
1269
+ },
1270
+ {
1271
+ "epoch": 1.37,
1272
+ "learning_rate": 7.309039943938333e-06,
1273
+ "loss": 0.586,
1274
+ "step": 1970
1275
+ },
1276
+ {
1277
+ "epoch": 1.38,
1278
+ "learning_rate": 7.295024526979678e-06,
1279
+ "loss": 0.6234,
1280
+ "step": 1980
1281
+ },
1282
+ {
1283
+ "epoch": 1.38,
1284
+ "learning_rate": 7.281009110021023e-06,
1285
+ "loss": 0.634,
1286
+ "step": 1990
1287
+ },
1288
+ {
1289
+ "epoch": 1.39,
1290
+ "learning_rate": 7.2669936930623695e-06,
1291
+ "loss": 0.6235,
1292
+ "step": 2000
1293
+ },
1294
+ {
1295
+ "epoch": 1.4,
1296
+ "learning_rate": 7.252978276103715e-06,
1297
+ "loss": 0.5714,
1298
+ "step": 2010
1299
+ },
1300
+ {
1301
+ "epoch": 1.41,
1302
+ "learning_rate": 7.23896285914506e-06,
1303
+ "loss": 0.6068,
1304
+ "step": 2020
1305
+ },
1306
+ {
1307
+ "epoch": 1.41,
1308
+ "learning_rate": 7.224947442186405e-06,
1309
+ "loss": 0.6222,
1310
+ "step": 2030
1311
+ },
1312
+ {
1313
+ "epoch": 1.42,
1314
+ "learning_rate": 7.210932025227751e-06,
1315
+ "loss": 0.6029,
1316
+ "step": 2040
1317
+ },
1318
+ {
1319
+ "epoch": 1.43,
1320
+ "learning_rate": 7.196916608269097e-06,
1321
+ "loss": 0.6093,
1322
+ "step": 2050
1323
+ },
1324
+ {
1325
+ "epoch": 1.43,
1326
+ "learning_rate": 7.182901191310442e-06,
1327
+ "loss": 0.6151,
1328
+ "step": 2060
1329
+ },
1330
+ {
1331
+ "epoch": 1.44,
1332
+ "learning_rate": 7.168885774351787e-06,
1333
+ "loss": 0.617,
1334
+ "step": 2070
1335
+ },
1336
+ {
1337
+ "epoch": 1.45,
1338
+ "learning_rate": 7.154870357393133e-06,
1339
+ "loss": 0.6077,
1340
+ "step": 2080
1341
+ },
1342
+ {
1343
+ "epoch": 1.45,
1344
+ "learning_rate": 7.140854940434478e-06,
1345
+ "loss": 0.6247,
1346
+ "step": 2090
1347
+ },
1348
+ {
1349
+ "epoch": 1.46,
1350
+ "learning_rate": 7.126839523475824e-06,
1351
+ "loss": 0.5979,
1352
+ "step": 2100
1353
+ },
1354
+ {
1355
+ "epoch": 1.46,
1356
+ "eval_custom_hf_pref_accuracy": 0.6175,
1357
+ "eval_custom_hf_pref_kendalltau": 0.235,
1358
+ "eval_custom_hf_pref_loss": 0.6143593788146973,
1359
+ "eval_custom_hf_pref_neg_score": -0.34517455101013184,
1360
+ "eval_custom_hf_pref_pos_score": 0.005482231266796589,
1361
+ "eval_custom_hf_pref_runtime": 3.0949,
1362
+ "eval_custom_hf_pref_samples_per_second": 646.224,
1363
+ "eval_custom_hf_pref_score_diff": 0.35065677762031555,
1364
+ "eval_custom_hf_pref_steps_per_second": 80.778,
1365
+ "step": 2100
1366
+ },
1367
+ {
1368
+ "epoch": 1.47,
1369
+ "learning_rate": 7.11282410651717e-06,
1370
+ "loss": 0.5953,
1371
+ "step": 2110
1372
+ },
1373
+ {
1374
+ "epoch": 1.47,
1375
+ "learning_rate": 7.098808689558515e-06,
1376
+ "loss": 0.5933,
1377
+ "step": 2120
1378
+ },
1379
+ {
1380
+ "epoch": 1.48,
1381
+ "learning_rate": 7.08479327259986e-06,
1382
+ "loss": 0.6466,
1383
+ "step": 2130
1384
+ },
1385
+ {
1386
+ "epoch": 1.49,
1387
+ "learning_rate": 7.070777855641206e-06,
1388
+ "loss": 0.6169,
1389
+ "step": 2140
1390
+ },
1391
+ {
1392
+ "epoch": 1.5,
1393
+ "learning_rate": 7.056762438682552e-06,
1394
+ "loss": 0.6024,
1395
+ "step": 2150
1396
+ },
1397
+ {
1398
+ "epoch": 1.5,
1399
+ "learning_rate": 7.042747021723897e-06,
1400
+ "loss": 0.6208,
1401
+ "step": 2160
1402
+ },
1403
+ {
1404
+ "epoch": 1.51,
1405
+ "learning_rate": 7.028731604765242e-06,
1406
+ "loss": 0.6015,
1407
+ "step": 2170
1408
+ },
1409
+ {
1410
+ "epoch": 1.52,
1411
+ "learning_rate": 7.014716187806588e-06,
1412
+ "loss": 0.6428,
1413
+ "step": 2180
1414
+ },
1415
+ {
1416
+ "epoch": 1.52,
1417
+ "learning_rate": 7.000700770847933e-06,
1418
+ "loss": 0.6282,
1419
+ "step": 2190
1420
+ },
1421
+ {
1422
+ "epoch": 1.53,
1423
+ "learning_rate": 6.986685353889279e-06,
1424
+ "loss": 0.5968,
1425
+ "step": 2200
1426
+ },
1427
+ {
1428
+ "epoch": 1.54,
1429
+ "learning_rate": 6.972669936930624e-06,
1430
+ "loss": 0.6262,
1431
+ "step": 2210
1432
+ },
1433
+ {
1434
+ "epoch": 1.54,
1435
+ "learning_rate": 6.95865451997197e-06,
1436
+ "loss": 0.6052,
1437
+ "step": 2220
1438
+ },
1439
+ {
1440
+ "epoch": 1.55,
1441
+ "learning_rate": 6.944639103013315e-06,
1442
+ "loss": 0.618,
1443
+ "step": 2230
1444
+ },
1445
+ {
1446
+ "epoch": 1.56,
1447
+ "learning_rate": 6.930623686054661e-06,
1448
+ "loss": 0.606,
1449
+ "step": 2240
1450
+ },
1451
+ {
1452
+ "epoch": 1.57,
1453
+ "learning_rate": 6.9166082690960055e-06,
1454
+ "loss": 0.6208,
1455
+ "step": 2250
1456
+ },
1457
+ {
1458
+ "epoch": 1.57,
1459
+ "learning_rate": 6.902592852137352e-06,
1460
+ "loss": 0.6052,
1461
+ "step": 2260
1462
+ },
1463
+ {
1464
+ "epoch": 1.58,
1465
+ "learning_rate": 6.888577435178697e-06,
1466
+ "loss": 0.6348,
1467
+ "step": 2270
1468
+ },
1469
+ {
1470
+ "epoch": 1.59,
1471
+ "learning_rate": 6.8745620182200425e-06,
1472
+ "loss": 0.6177,
1473
+ "step": 2280
1474
+ },
1475
+ {
1476
+ "epoch": 1.59,
1477
+ "learning_rate": 6.860546601261388e-06,
1478
+ "loss": 0.6076,
1479
+ "step": 2290
1480
+ },
1481
+ {
1482
+ "epoch": 1.6,
1483
+ "learning_rate": 6.846531184302734e-06,
1484
+ "loss": 0.5978,
1485
+ "step": 2300
1486
+ },
1487
+ {
1488
+ "epoch": 1.61,
1489
+ "learning_rate": 6.832515767344079e-06,
1490
+ "loss": 0.5999,
1491
+ "step": 2310
1492
+ },
1493
+ {
1494
+ "epoch": 1.61,
1495
+ "learning_rate": 6.818500350385424e-06,
1496
+ "loss": 0.5955,
1497
+ "step": 2320
1498
+ },
1499
+ {
1500
+ "epoch": 1.62,
1501
+ "learning_rate": 6.80448493342677e-06,
1502
+ "loss": 0.6491,
1503
+ "step": 2330
1504
+ },
1505
+ {
1506
+ "epoch": 1.63,
1507
+ "learning_rate": 6.790469516468116e-06,
1508
+ "loss": 0.6059,
1509
+ "step": 2340
1510
+ },
1511
+ {
1512
+ "epoch": 1.63,
1513
+ "learning_rate": 6.7764540995094605e-06,
1514
+ "loss": 0.6086,
1515
+ "step": 2350
1516
+ },
1517
+ {
1518
+ "epoch": 1.64,
1519
+ "learning_rate": 6.762438682550806e-06,
1520
+ "loss": 0.5981,
1521
+ "step": 2360
1522
+ },
1523
+ {
1524
+ "epoch": 1.65,
1525
+ "learning_rate": 6.748423265592152e-06,
1526
+ "loss": 0.6151,
1527
+ "step": 2370
1528
+ },
1529
+ {
1530
+ "epoch": 1.66,
1531
+ "learning_rate": 6.7344078486334976e-06,
1532
+ "loss": 0.633,
1533
+ "step": 2380
1534
+ },
1535
+ {
1536
+ "epoch": 1.66,
1537
+ "learning_rate": 6.720392431674842e-06,
1538
+ "loss": 0.6074,
1539
+ "step": 2390
1540
+ },
1541
+ {
1542
+ "epoch": 1.67,
1543
+ "learning_rate": 6.706377014716189e-06,
1544
+ "loss": 0.592,
1545
+ "step": 2400
1546
+ },
1547
+ {
1548
+ "epoch": 1.67,
1549
+ "eval_custom_hf_pref_accuracy": 0.62,
1550
+ "eval_custom_hf_pref_kendalltau": 0.24,
1551
+ "eval_custom_hf_pref_loss": 0.6024765372276306,
1552
+ "eval_custom_hf_pref_neg_score": -0.1819182187318802,
1553
+ "eval_custom_hf_pref_pos_score": 0.2181234061717987,
1554
+ "eval_custom_hf_pref_runtime": 3.1567,
1555
+ "eval_custom_hf_pref_samples_per_second": 633.564,
1556
+ "eval_custom_hf_pref_score_diff": 0.4000416398048401,
1557
+ "eval_custom_hf_pref_steps_per_second": 79.195,
1558
+ "step": 2400
1559
+ },
1560
+ {
1561
+ "epoch": 1.68,
1562
+ "learning_rate": 6.692361597757534e-06,
1563
+ "loss": 0.569,
1564
+ "step": 2410
1565
+ },
1566
+ {
1567
+ "epoch": 1.68,
1568
+ "learning_rate": 6.6783461807988794e-06,
1569
+ "loss": 0.6234,
1570
+ "step": 2420
1571
+ },
1572
+ {
1573
+ "epoch": 1.69,
1574
+ "learning_rate": 6.664330763840224e-06,
1575
+ "loss": 0.6026,
1576
+ "step": 2430
1577
+ },
1578
+ {
1579
+ "epoch": 1.7,
1580
+ "learning_rate": 6.650315346881571e-06,
1581
+ "loss": 0.604,
1582
+ "step": 2440
1583
+ },
1584
+ {
1585
+ "epoch": 1.7,
1586
+ "learning_rate": 6.636299929922916e-06,
1587
+ "loss": 0.6344,
1588
+ "step": 2450
1589
+ },
1590
+ {
1591
+ "epoch": 1.71,
1592
+ "learning_rate": 6.622284512964261e-06,
1593
+ "loss": 0.6428,
1594
+ "step": 2460
1595
+ },
1596
+ {
1597
+ "epoch": 1.72,
1598
+ "learning_rate": 6.608269096005606e-06,
1599
+ "loss": 0.6277,
1600
+ "step": 2470
1601
+ },
1602
+ {
1603
+ "epoch": 1.73,
1604
+ "learning_rate": 6.594253679046953e-06,
1605
+ "loss": 0.6206,
1606
+ "step": 2480
1607
+ },
1608
+ {
1609
+ "epoch": 1.73,
1610
+ "learning_rate": 6.5802382620882975e-06,
1611
+ "loss": 0.6334,
1612
+ "step": 2490
1613
+ },
1614
+ {
1615
+ "epoch": 1.74,
1616
+ "learning_rate": 6.566222845129643e-06,
1617
+ "loss": 0.6269,
1618
+ "step": 2500
1619
+ },
1620
+ {
1621
+ "epoch": 1.75,
1622
+ "learning_rate": 6.552207428170989e-06,
1623
+ "loss": 0.6252,
1624
+ "step": 2510
1625
+ },
1626
+ {
1627
+ "epoch": 1.75,
1628
+ "learning_rate": 6.5381920112123345e-06,
1629
+ "loss": 0.5913,
1630
+ "step": 2520
1631
+ },
1632
+ {
1633
+ "epoch": 1.76,
1634
+ "learning_rate": 6.524176594253679e-06,
1635
+ "loss": 0.5826,
1636
+ "step": 2530
1637
+ },
1638
+ {
1639
+ "epoch": 1.77,
1640
+ "learning_rate": 6.510161177295025e-06,
1641
+ "loss": 0.5981,
1642
+ "step": 2540
1643
+ },
1644
+ {
1645
+ "epoch": 1.77,
1646
+ "learning_rate": 6.496145760336371e-06,
1647
+ "loss": 0.6052,
1648
+ "step": 2550
1649
+ },
1650
+ {
1651
+ "epoch": 1.78,
1652
+ "learning_rate": 6.482130343377716e-06,
1653
+ "loss": 0.5933,
1654
+ "step": 2560
1655
+ },
1656
+ {
1657
+ "epoch": 1.79,
1658
+ "learning_rate": 6.468114926419061e-06,
1659
+ "loss": 0.5952,
1660
+ "step": 2570
1661
+ },
1662
+ {
1663
+ "epoch": 1.79,
1664
+ "learning_rate": 6.454099509460406e-06,
1665
+ "loss": 0.6018,
1666
+ "step": 2580
1667
+ },
1668
+ {
1669
+ "epoch": 1.8,
1670
+ "learning_rate": 6.4400840925017525e-06,
1671
+ "loss": 0.5824,
1672
+ "step": 2590
1673
+ },
1674
+ {
1675
+ "epoch": 1.81,
1676
+ "learning_rate": 6.426068675543098e-06,
1677
+ "loss": 0.5687,
1678
+ "step": 2600
1679
+ },
1680
+ {
1681
+ "epoch": 1.82,
1682
+ "learning_rate": 6.412053258584443e-06,
1683
+ "loss": 0.6095,
1684
+ "step": 2610
1685
+ },
1686
+ {
1687
+ "epoch": 1.82,
1688
+ "learning_rate": 6.3980378416257896e-06,
1689
+ "loss": 0.5991,
1690
+ "step": 2620
1691
+ },
1692
+ {
1693
+ "epoch": 1.83,
1694
+ "learning_rate": 6.384022424667134e-06,
1695
+ "loss": 0.5811,
1696
+ "step": 2630
1697
+ },
1698
+ {
1699
+ "epoch": 1.84,
1700
+ "learning_rate": 6.37000700770848e-06,
1701
+ "loss": 0.606,
1702
+ "step": 2640
1703
+ },
1704
+ {
1705
+ "epoch": 1.84,
1706
+ "learning_rate": 6.355991590749825e-06,
1707
+ "loss": 0.609,
1708
+ "step": 2650
1709
+ },
1710
+ {
1711
+ "epoch": 1.85,
1712
+ "learning_rate": 6.341976173791171e-06,
1713
+ "loss": 0.6127,
1714
+ "step": 2660
1715
+ },
1716
+ {
1717
+ "epoch": 1.86,
1718
+ "learning_rate": 6.327960756832516e-06,
1719
+ "loss": 0.614,
1720
+ "step": 2670
1721
+ },
1722
+ {
1723
+ "epoch": 1.86,
1724
+ "learning_rate": 6.313945339873861e-06,
1725
+ "loss": 0.6075,
1726
+ "step": 2680
1727
+ },
1728
+ {
1729
+ "epoch": 1.87,
1730
+ "learning_rate": 6.299929922915208e-06,
1731
+ "loss": 0.6046,
1732
+ "step": 2690
1733
+ },
1734
+ {
1735
+ "epoch": 1.88,
1736
+ "learning_rate": 6.285914505956553e-06,
1737
+ "loss": 0.6014,
1738
+ "step": 2700
1739
+ },
1740
+ {
1741
+ "epoch": 1.88,
1742
+ "eval_custom_hf_pref_accuracy": 0.614,
1743
+ "eval_custom_hf_pref_kendalltau": 0.228,
1744
+ "eval_custom_hf_pref_loss": 0.6012265682220459,
1745
+ "eval_custom_hf_pref_neg_score": -0.45326927304267883,
1746
+ "eval_custom_hf_pref_pos_score": -0.06561196595430374,
1747
+ "eval_custom_hf_pref_runtime": 3.1381,
1748
+ "eval_custom_hf_pref_samples_per_second": 637.337,
1749
+ "eval_custom_hf_pref_score_diff": 0.3876572847366333,
1750
+ "eval_custom_hf_pref_steps_per_second": 79.667,
1751
+ "step": 2700
1752
+ },
1753
+ {
1754
+ "epoch": 1.89,
1755
+ "learning_rate": 6.271899088997898e-06,
1756
+ "loss": 0.6322,
1757
+ "step": 2710
1758
+ },
1759
+ {
1760
+ "epoch": 1.89,
1761
+ "learning_rate": 6.257883672039243e-06,
1762
+ "loss": 0.5794,
1763
+ "step": 2720
1764
+ },
1765
+ {
1766
+ "epoch": 1.9,
1767
+ "learning_rate": 6.2438682550805895e-06,
1768
+ "loss": 0.6534,
1769
+ "step": 2730
1770
+ },
1771
+ {
1772
+ "epoch": 1.91,
1773
+ "learning_rate": 6.229852838121935e-06,
1774
+ "loss": 0.6054,
1775
+ "step": 2740
1776
+ },
1777
+ {
1778
+ "epoch": 1.91,
1779
+ "learning_rate": 6.21583742116328e-06,
1780
+ "loss": 0.5961,
1781
+ "step": 2750
1782
+ },
1783
+ {
1784
+ "epoch": 1.92,
1785
+ "learning_rate": 6.201822004204625e-06,
1786
+ "loss": 0.6372,
1787
+ "step": 2760
1788
+ },
1789
+ {
1790
+ "epoch": 1.93,
1791
+ "learning_rate": 6.187806587245971e-06,
1792
+ "loss": 0.6136,
1793
+ "step": 2770
1794
+ },
1795
+ {
1796
+ "epoch": 1.93,
1797
+ "learning_rate": 6.173791170287316e-06,
1798
+ "loss": 0.6435,
1799
+ "step": 2780
1800
+ },
1801
+ {
1802
+ "epoch": 1.94,
1803
+ "learning_rate": 6.159775753328662e-06,
1804
+ "loss": 0.599,
1805
+ "step": 2790
1806
+ },
1807
+ {
1808
+ "epoch": 1.95,
1809
+ "learning_rate": 6.145760336370008e-06,
1810
+ "loss": 0.6341,
1811
+ "step": 2800
1812
+ },
1813
+ {
1814
+ "epoch": 1.95,
1815
+ "learning_rate": 6.131744919411353e-06,
1816
+ "loss": 0.5967,
1817
+ "step": 2810
1818
+ },
1819
+ {
1820
+ "epoch": 1.96,
1821
+ "learning_rate": 6.117729502452698e-06,
1822
+ "loss": 0.6156,
1823
+ "step": 2820
1824
+ },
1825
+ {
1826
+ "epoch": 1.97,
1827
+ "learning_rate": 6.103714085494044e-06,
1828
+ "loss": 0.6235,
1829
+ "step": 2830
1830
+ },
1831
+ {
1832
+ "epoch": 1.98,
1833
+ "learning_rate": 6.08969866853539e-06,
1834
+ "loss": 0.6027,
1835
+ "step": 2840
1836
+ },
1837
+ {
1838
+ "epoch": 1.98,
1839
+ "learning_rate": 6.075683251576735e-06,
1840
+ "loss": 0.6196,
1841
+ "step": 2850
1842
+ },
1843
+ {
1844
+ "epoch": 1.99,
1845
+ "learning_rate": 6.06166783461808e-06,
1846
+ "loss": 0.586,
1847
+ "step": 2860
1848
+ },
1849
+ {
1850
+ "epoch": 2.0,
1851
+ "learning_rate": 6.0476524176594255e-06,
1852
+ "loss": 0.586,
1853
+ "step": 2870
1854
+ },
1855
+ {
1856
+ "epoch": 2.0,
1857
+ "learning_rate": 6.033637000700771e-06,
1858
+ "loss": 0.6173,
1859
+ "step": 2880
1860
+ },
1861
+ {
1862
+ "epoch": 2.01,
1863
+ "learning_rate": 6.019621583742117e-06,
1864
+ "loss": 0.6127,
1865
+ "step": 2890
1866
+ },
1867
+ {
1868
+ "epoch": 2.02,
1869
+ "learning_rate": 6.005606166783462e-06,
1870
+ "loss": 0.6308,
1871
+ "step": 2900
1872
+ },
1873
+ {
1874
+ "epoch": 2.02,
1875
+ "learning_rate": 5.991590749824808e-06,
1876
+ "loss": 0.6055,
1877
+ "step": 2910
1878
+ },
1879
+ {
1880
+ "epoch": 2.03,
1881
+ "learning_rate": 5.977575332866153e-06,
1882
+ "loss": 0.6339,
1883
+ "step": 2920
1884
+ },
1885
+ {
1886
+ "epoch": 2.04,
1887
+ "learning_rate": 5.963559915907499e-06,
1888
+ "loss": 0.6525,
1889
+ "step": 2930
1890
+ },
1891
+ {
1892
+ "epoch": 2.05,
1893
+ "learning_rate": 5.9495444989488436e-06,
1894
+ "loss": 0.6512,
1895
+ "step": 2940
1896
+ },
1897
+ {
1898
+ "epoch": 2.05,
1899
+ "learning_rate": 5.93552908199019e-06,
1900
+ "loss": 0.6068,
1901
+ "step": 2950
1902
+ },
1903
+ {
1904
+ "epoch": 2.06,
1905
+ "learning_rate": 5.921513665031535e-06,
1906
+ "loss": 0.6137,
1907
+ "step": 2960
1908
+ },
1909
+ {
1910
+ "epoch": 2.07,
1911
+ "learning_rate": 5.907498248072881e-06,
1912
+ "loss": 0.6017,
1913
+ "step": 2970
1914
+ },
1915
+ {
1916
+ "epoch": 2.07,
1917
+ "learning_rate": 5.8934828311142254e-06,
1918
+ "loss": 0.6071,
1919
+ "step": 2980
1920
+ },
1921
+ {
1922
+ "epoch": 2.08,
1923
+ "learning_rate": 5.879467414155572e-06,
1924
+ "loss": 0.585,
1925
+ "step": 2990
1926
+ },
1927
+ {
1928
+ "epoch": 2.09,
1929
+ "learning_rate": 5.865451997196917e-06,
1930
+ "loss": 0.6075,
1931
+ "step": 3000
1932
+ },
1933
+ {
1934
+ "epoch": 2.09,
1935
+ "eval_custom_hf_pref_accuracy": 0.613,
1936
+ "eval_custom_hf_pref_kendalltau": 0.226,
1937
+ "eval_custom_hf_pref_loss": 0.6039531230926514,
1938
+ "eval_custom_hf_pref_neg_score": -0.5177181363105774,
1939
+ "eval_custom_hf_pref_pos_score": -0.16565346717834473,
1940
+ "eval_custom_hf_pref_runtime": 3.1422,
1941
+ "eval_custom_hf_pref_samples_per_second": 636.497,
1942
+ "eval_custom_hf_pref_score_diff": 0.3520646393299103,
1943
+ "eval_custom_hf_pref_steps_per_second": 79.562,
1944
+ "step": 3000
1945
+ },
1946
+ {
1947
+ "epoch": 2.09,
1948
+ "learning_rate": 5.8514365802382625e-06,
1949
+ "loss": 0.6178,
1950
+ "step": 3010
1951
+ },
1952
+ {
1953
+ "epoch": 2.1,
1954
+ "learning_rate": 5.837421163279608e-06,
1955
+ "loss": 0.6208,
1956
+ "step": 3020
1957
+ },
1958
+ {
1959
+ "epoch": 2.11,
1960
+ "learning_rate": 5.823405746320954e-06,
1961
+ "loss": 0.6236,
1962
+ "step": 3030
1963
+ },
1964
+ {
1965
+ "epoch": 2.11,
1966
+ "learning_rate": 5.809390329362299e-06,
1967
+ "loss": 0.6193,
1968
+ "step": 3040
1969
+ },
1970
+ {
1971
+ "epoch": 2.12,
1972
+ "learning_rate": 5.795374912403644e-06,
1973
+ "loss": 0.5971,
1974
+ "step": 3050
1975
+ },
1976
+ {
1977
+ "epoch": 2.13,
1978
+ "learning_rate": 5.78135949544499e-06,
1979
+ "loss": 0.6193,
1980
+ "step": 3060
1981
+ },
1982
+ {
1983
+ "epoch": 2.14,
1984
+ "learning_rate": 5.767344078486336e-06,
1985
+ "loss": 0.6173,
1986
+ "step": 3070
1987
+ },
1988
+ {
1989
+ "epoch": 2.14,
1990
+ "learning_rate": 5.7533286615276805e-06,
1991
+ "loss": 0.5993,
1992
+ "step": 3080
1993
+ },
1994
+ {
1995
+ "epoch": 2.15,
1996
+ "learning_rate": 5.739313244569027e-06,
1997
+ "loss": 0.6155,
1998
+ "step": 3090
1999
+ },
2000
+ {
2001
+ "epoch": 2.16,
2002
+ "learning_rate": 5.725297827610372e-06,
2003
+ "loss": 0.6124,
2004
+ "step": 3100
2005
+ },
2006
+ {
2007
+ "epoch": 2.16,
2008
+ "learning_rate": 5.7112824106517175e-06,
2009
+ "loss": 0.5955,
2010
+ "step": 3110
2011
+ },
2012
+ {
2013
+ "epoch": 2.17,
2014
+ "learning_rate": 5.697266993693062e-06,
2015
+ "loss": 0.5802,
2016
+ "step": 3120
2017
+ },
2018
+ {
2019
+ "epoch": 2.18,
2020
+ "learning_rate": 5.683251576734409e-06,
2021
+ "loss": 0.6156,
2022
+ "step": 3130
2023
+ },
2024
+ {
2025
+ "epoch": 2.18,
2026
+ "learning_rate": 5.669236159775754e-06,
2027
+ "loss": 0.6344,
2028
+ "step": 3140
2029
+ },
2030
+ {
2031
+ "epoch": 2.19,
2032
+ "learning_rate": 5.655220742817099e-06,
2033
+ "loss": 0.5634,
2034
+ "step": 3150
2035
+ },
2036
+ {
2037
+ "epoch": 2.2,
2038
+ "learning_rate": 5.641205325858444e-06,
2039
+ "loss": 0.5955,
2040
+ "step": 3160
2041
+ },
2042
+ {
2043
+ "epoch": 2.21,
2044
+ "learning_rate": 5.627189908899791e-06,
2045
+ "loss": 0.6102,
2046
+ "step": 3170
2047
+ },
2048
+ {
2049
+ "epoch": 2.21,
2050
+ "learning_rate": 5.6131744919411356e-06,
2051
+ "loss": 0.6409,
2052
+ "step": 3180
2053
+ },
2054
+ {
2055
+ "epoch": 2.22,
2056
+ "learning_rate": 5.599159074982481e-06,
2057
+ "loss": 0.6575,
2058
+ "step": 3190
2059
+ },
2060
+ {
2061
+ "epoch": 2.23,
2062
+ "learning_rate": 5.585143658023827e-06,
2063
+ "loss": 0.6236,
2064
+ "step": 3200
2065
+ },
2066
+ {
2067
+ "epoch": 2.23,
2068
+ "learning_rate": 5.571128241065173e-06,
2069
+ "loss": 0.6078,
2070
+ "step": 3210
2071
+ },
2072
+ {
2073
+ "epoch": 2.24,
2074
+ "learning_rate": 5.557112824106517e-06,
2075
+ "loss": 0.6161,
2076
+ "step": 3220
2077
+ },
2078
+ {
2079
+ "epoch": 2.25,
2080
+ "learning_rate": 5.543097407147863e-06,
2081
+ "loss": 0.5838,
2082
+ "step": 3230
2083
+ },
2084
+ {
2085
+ "epoch": 2.25,
2086
+ "learning_rate": 5.529081990189209e-06,
2087
+ "loss": 0.6179,
2088
+ "step": 3240
2089
+ },
2090
+ {
2091
+ "epoch": 2.26,
2092
+ "learning_rate": 5.5150665732305544e-06,
2093
+ "loss": 0.5729,
2094
+ "step": 3250
2095
+ },
2096
+ {
2097
+ "epoch": 2.27,
2098
+ "learning_rate": 5.501051156271899e-06,
2099
+ "loss": 0.6353,
2100
+ "step": 3260
2101
+ },
2102
+ {
2103
+ "epoch": 2.27,
2104
+ "learning_rate": 5.487035739313244e-06,
2105
+ "loss": 0.5723,
2106
+ "step": 3270
2107
+ },
2108
+ {
2109
+ "epoch": 2.28,
2110
+ "learning_rate": 5.473020322354591e-06,
2111
+ "loss": 0.5983,
2112
+ "step": 3280
2113
+ },
2114
+ {
2115
+ "epoch": 2.29,
2116
+ "learning_rate": 5.459004905395936e-06,
2117
+ "loss": 0.5695,
2118
+ "step": 3290
2119
+ },
2120
+ {
2121
+ "epoch": 2.3,
2122
+ "learning_rate": 5.444989488437281e-06,
2123
+ "loss": 0.6189,
2124
+ "step": 3300
2125
+ },
2126
+ {
2127
+ "epoch": 2.3,
2128
+ "eval_custom_hf_pref_accuracy": 0.605,
2129
+ "eval_custom_hf_pref_kendalltau": 0.21,
2130
+ "eval_custom_hf_pref_loss": 0.6013281345367432,
2131
+ "eval_custom_hf_pref_neg_score": -0.3515034317970276,
2132
+ "eval_custom_hf_pref_pos_score": 0.02793554961681366,
2133
+ "eval_custom_hf_pref_runtime": 3.0849,
2134
+ "eval_custom_hf_pref_samples_per_second": 648.326,
2135
+ "eval_custom_hf_pref_score_diff": 0.37943896651268005,
2136
+ "eval_custom_hf_pref_steps_per_second": 81.041,
2137
+ "step": 3300
2138
+ },
2139
+ {
2140
+ "epoch": 2.3,
2141
+ "learning_rate": 5.430974071478628e-06,
2142
+ "loss": 0.6177,
2143
+ "step": 3310
2144
+ },
2145
+ {
2146
+ "epoch": 2.31,
2147
+ "learning_rate": 5.4169586545199725e-06,
2148
+ "loss": 0.616,
2149
+ "step": 3320
2150
+ },
2151
+ {
2152
+ "epoch": 2.32,
2153
+ "learning_rate": 5.402943237561318e-06,
2154
+ "loss": 0.6038,
2155
+ "step": 3330
2156
+ },
2157
+ {
2158
+ "epoch": 2.32,
2159
+ "learning_rate": 5.388927820602663e-06,
2160
+ "loss": 0.6017,
2161
+ "step": 3340
2162
+ },
2163
+ {
2164
+ "epoch": 2.33,
2165
+ "learning_rate": 5.3749124036440095e-06,
2166
+ "loss": 0.5796,
2167
+ "step": 3350
2168
+ },
2169
+ {
2170
+ "epoch": 2.34,
2171
+ "learning_rate": 5.360896986685354e-06,
2172
+ "loss": 0.6032,
2173
+ "step": 3360
2174
+ },
2175
+ {
2176
+ "epoch": 2.34,
2177
+ "learning_rate": 5.346881569726699e-06,
2178
+ "loss": 0.626,
2179
+ "step": 3370
2180
+ },
2181
+ {
2182
+ "epoch": 2.35,
2183
+ "learning_rate": 5.332866152768045e-06,
2184
+ "loss": 0.5943,
2185
+ "step": 3380
2186
+ },
2187
+ {
2188
+ "epoch": 2.36,
2189
+ "learning_rate": 5.318850735809391e-06,
2190
+ "loss": 0.5708,
2191
+ "step": 3390
2192
+ },
2193
+ {
2194
+ "epoch": 2.37,
2195
+ "learning_rate": 5.304835318850736e-06,
2196
+ "loss": 0.6458,
2197
+ "step": 3400
2198
+ },
2199
+ {
2200
+ "epoch": 2.37,
2201
+ "learning_rate": 5.290819901892081e-06,
2202
+ "loss": 0.6138,
2203
+ "step": 3410
2204
+ },
2205
+ {
2206
+ "epoch": 2.38,
2207
+ "learning_rate": 5.2768044849334275e-06,
2208
+ "loss": 0.6278,
2209
+ "step": 3420
2210
+ },
2211
+ {
2212
+ "epoch": 2.39,
2213
+ "learning_rate": 5.262789067974773e-06,
2214
+ "loss": 0.6138,
2215
+ "step": 3430
2216
+ },
2217
+ {
2218
+ "epoch": 2.39,
2219
+ "learning_rate": 5.248773651016118e-06,
2220
+ "loss": 0.6036,
2221
+ "step": 3440
2222
+ },
2223
+ {
2224
+ "epoch": 2.4,
2225
+ "learning_rate": 5.234758234057463e-06,
2226
+ "loss": 0.5893,
2227
+ "step": 3450
2228
+ },
2229
+ {
2230
+ "epoch": 2.41,
2231
+ "learning_rate": 5.220742817098809e-06,
2232
+ "loss": 0.618,
2233
+ "step": 3460
2234
+ },
2235
+ {
2236
+ "epoch": 2.41,
2237
+ "learning_rate": 5.206727400140154e-06,
2238
+ "loss": 0.5952,
2239
+ "step": 3470
2240
+ },
2241
+ {
2242
+ "epoch": 2.42,
2243
+ "learning_rate": 5.1927119831815e-06,
2244
+ "loss": 0.6331,
2245
+ "step": 3480
2246
+ },
2247
+ {
2248
+ "epoch": 2.43,
2249
+ "learning_rate": 5.178696566222845e-06,
2250
+ "loss": 0.6091,
2251
+ "step": 3490
2252
+ },
2253
+ {
2254
+ "epoch": 2.43,
2255
+ "learning_rate": 5.164681149264191e-06,
2256
+ "loss": 0.6245,
2257
+ "step": 3500
2258
+ },
2259
+ {
2260
+ "epoch": 2.44,
2261
+ "learning_rate": 5.150665732305536e-06,
2262
+ "loss": 0.5887,
2263
+ "step": 3510
2264
+ },
2265
+ {
2266
+ "epoch": 2.45,
2267
+ "learning_rate": 5.136650315346882e-06,
2268
+ "loss": 0.608,
2269
+ "step": 3520
2270
+ },
2271
+ {
2272
+ "epoch": 2.46,
2273
+ "learning_rate": 5.122634898388228e-06,
2274
+ "loss": 0.5959,
2275
+ "step": 3530
2276
+ },
2277
+ {
2278
+ "epoch": 2.46,
2279
+ "learning_rate": 5.108619481429573e-06,
2280
+ "loss": 0.6515,
2281
+ "step": 3540
2282
+ },
2283
+ {
2284
+ "epoch": 2.47,
2285
+ "learning_rate": 5.094604064470918e-06,
2286
+ "loss": 0.629,
2287
+ "step": 3550
2288
+ },
2289
+ {
2290
+ "epoch": 2.48,
2291
+ "learning_rate": 5.080588647512264e-06,
2292
+ "loss": 0.5898,
2293
+ "step": 3560
2294
+ },
2295
+ {
2296
+ "epoch": 2.48,
2297
+ "learning_rate": 5.06657323055361e-06,
2298
+ "loss": 0.5561,
2299
+ "step": 3570
2300
+ },
2301
+ {
2302
+ "epoch": 2.49,
2303
+ "learning_rate": 5.052557813594955e-06,
2304
+ "loss": 0.6031,
2305
+ "step": 3580
2306
+ },
2307
+ {
2308
+ "epoch": 2.5,
2309
+ "learning_rate": 5.0385423966363e-06,
2310
+ "loss": 0.5934,
2311
+ "step": 3590
2312
+ },
2313
+ {
2314
+ "epoch": 2.5,
2315
+ "learning_rate": 5.024526979677646e-06,
2316
+ "loss": 0.6393,
2317
+ "step": 3600
2318
+ },
2319
+ {
2320
+ "epoch": 2.5,
2321
+ "eval_custom_hf_pref_accuracy": 0.6085,
2322
+ "eval_custom_hf_pref_kendalltau": 0.217,
2323
+ "eval_custom_hf_pref_loss": 0.6061797142028809,
2324
+ "eval_custom_hf_pref_neg_score": -0.4991738796234131,
2325
+ "eval_custom_hf_pref_pos_score": -0.14666078984737396,
2326
+ "eval_custom_hf_pref_runtime": 3.0873,
2327
+ "eval_custom_hf_pref_samples_per_second": 647.819,
2328
+ "eval_custom_hf_pref_score_diff": 0.3525131344795227,
2329
+ "eval_custom_hf_pref_steps_per_second": 80.977,
2330
+ "step": 3600
2331
+ },
2332
+ {
2333
+ "epoch": 2.51,
2334
+ "learning_rate": 5.010511562718991e-06,
2335
+ "loss": 0.5903,
2336
+ "step": 3610
2337
+ },
2338
+ {
2339
+ "epoch": 2.52,
2340
+ "learning_rate": 4.996496145760337e-06,
2341
+ "loss": 0.5731,
2342
+ "step": 3620
2343
+ },
2344
+ {
2345
+ "epoch": 2.53,
2346
+ "learning_rate": 4.9824807288016825e-06,
2347
+ "loss": 0.6383,
2348
+ "step": 3630
2349
+ },
2350
+ {
2351
+ "epoch": 2.53,
2352
+ "learning_rate": 4.968465311843027e-06,
2353
+ "loss": 0.6008,
2354
+ "step": 3640
2355
+ },
2356
+ {
2357
+ "epoch": 2.54,
2358
+ "learning_rate": 4.954449894884373e-06,
2359
+ "loss": 0.6005,
2360
+ "step": 3650
2361
+ },
2362
+ {
2363
+ "epoch": 2.55,
2364
+ "learning_rate": 4.940434477925719e-06,
2365
+ "loss": 0.6348,
2366
+ "step": 3660
2367
+ },
2368
+ {
2369
+ "epoch": 2.55,
2370
+ "learning_rate": 4.926419060967064e-06,
2371
+ "loss": 0.6109,
2372
+ "step": 3670
2373
+ },
2374
+ {
2375
+ "epoch": 2.56,
2376
+ "learning_rate": 4.912403644008409e-06,
2377
+ "loss": 0.6219,
2378
+ "step": 3680
2379
+ },
2380
+ {
2381
+ "epoch": 2.57,
2382
+ "learning_rate": 4.898388227049755e-06,
2383
+ "loss": 0.589,
2384
+ "step": 3690
2385
+ },
2386
+ {
2387
+ "epoch": 2.57,
2388
+ "learning_rate": 4.8843728100911005e-06,
2389
+ "loss": 0.6278,
2390
+ "step": 3700
2391
+ },
2392
+ {
2393
+ "epoch": 2.58,
2394
+ "learning_rate": 4.870357393132446e-06,
2395
+ "loss": 0.6119,
2396
+ "step": 3710
2397
+ },
2398
+ {
2399
+ "epoch": 2.59,
2400
+ "learning_rate": 4.856341976173792e-06,
2401
+ "loss": 0.5835,
2402
+ "step": 3720
2403
+ },
2404
+ {
2405
+ "epoch": 2.59,
2406
+ "learning_rate": 4.842326559215137e-06,
2407
+ "loss": 0.6209,
2408
+ "step": 3730
2409
+ },
2410
+ {
2411
+ "epoch": 2.6,
2412
+ "learning_rate": 4.828311142256482e-06,
2413
+ "loss": 0.5998,
2414
+ "step": 3740
2415
+ },
2416
+ {
2417
+ "epoch": 2.61,
2418
+ "learning_rate": 4.814295725297828e-06,
2419
+ "loss": 0.5851,
2420
+ "step": 3750
2421
+ },
2422
+ {
2423
+ "epoch": 2.62,
2424
+ "learning_rate": 4.800280308339174e-06,
2425
+ "loss": 0.6362,
2426
+ "step": 3760
2427
+ },
2428
+ {
2429
+ "epoch": 2.62,
2430
+ "learning_rate": 4.786264891380519e-06,
2431
+ "loss": 0.6101,
2432
+ "step": 3770
2433
+ },
2434
+ {
2435
+ "epoch": 2.63,
2436
+ "learning_rate": 4.772249474421864e-06,
2437
+ "loss": 0.634,
2438
+ "step": 3780
2439
+ },
2440
+ {
2441
+ "epoch": 2.64,
2442
+ "learning_rate": 4.75823405746321e-06,
2443
+ "loss": 0.6183,
2444
+ "step": 3790
2445
+ },
2446
+ {
2447
+ "epoch": 2.64,
2448
+ "learning_rate": 4.744218640504556e-06,
2449
+ "loss": 0.6076,
2450
+ "step": 3800
2451
+ },
2452
+ {
2453
+ "epoch": 2.65,
2454
+ "learning_rate": 4.730203223545901e-06,
2455
+ "loss": 0.6406,
2456
+ "step": 3810
2457
+ },
2458
+ {
2459
+ "epoch": 2.66,
2460
+ "learning_rate": 4.716187806587246e-06,
2461
+ "loss": 0.5459,
2462
+ "step": 3820
2463
+ },
2464
+ {
2465
+ "epoch": 2.66,
2466
+ "learning_rate": 4.702172389628592e-06,
2467
+ "loss": 0.6272,
2468
+ "step": 3830
2469
+ },
2470
+ {
2471
+ "epoch": 2.67,
2472
+ "learning_rate": 4.6881569726699375e-06,
2473
+ "loss": 0.6035,
2474
+ "step": 3840
2475
+ },
2476
+ {
2477
+ "epoch": 2.68,
2478
+ "learning_rate": 4.674141555711283e-06,
2479
+ "loss": 0.6483,
2480
+ "step": 3850
2481
+ },
2482
+ {
2483
+ "epoch": 2.69,
2484
+ "learning_rate": 4.660126138752628e-06,
2485
+ "loss": 0.6028,
2486
+ "step": 3860
2487
+ },
2488
+ {
2489
+ "epoch": 2.69,
2490
+ "learning_rate": 4.646110721793974e-06,
2491
+ "loss": 0.6274,
2492
+ "step": 3870
2493
+ },
2494
+ {
2495
+ "epoch": 2.7,
2496
+ "learning_rate": 4.632095304835319e-06,
2497
+ "loss": 0.5906,
2498
+ "step": 3880
2499
+ },
2500
+ {
2501
+ "epoch": 2.71,
2502
+ "learning_rate": 4.618079887876665e-06,
2503
+ "loss": 0.6186,
2504
+ "step": 3890
2505
+ },
2506
+ {
2507
+ "epoch": 2.71,
2508
+ "learning_rate": 4.604064470918011e-06,
2509
+ "loss": 0.619,
2510
+ "step": 3900
2511
+ },
2512
+ {
2513
+ "epoch": 2.71,
2514
+ "eval_custom_hf_pref_accuracy": 0.613,
2515
+ "eval_custom_hf_pref_kendalltau": 0.226,
2516
+ "eval_custom_hf_pref_loss": 0.6025937795639038,
2517
+ "eval_custom_hf_pref_neg_score": -0.5224867463111877,
2518
+ "eval_custom_hf_pref_pos_score": -0.15825501084327698,
2519
+ "eval_custom_hf_pref_runtime": 3.0937,
2520
+ "eval_custom_hf_pref_samples_per_second": 646.485,
2521
+ "eval_custom_hf_pref_score_diff": 0.36423179507255554,
2522
+ "eval_custom_hf_pref_steps_per_second": 80.811,
2523
+ "step": 3900
2524
+ },
2525
+ {
2526
+ "epoch": 2.72,
2527
+ "learning_rate": 4.5900490539593555e-06,
2528
+ "loss": 0.6094,
2529
+ "step": 3910
2530
+ },
2531
+ {
2532
+ "epoch": 2.73,
2533
+ "learning_rate": 4.576033637000701e-06,
2534
+ "loss": 0.5902,
2535
+ "step": 3920
2536
+ },
2537
+ {
2538
+ "epoch": 2.73,
2539
+ "learning_rate": 4.562018220042047e-06,
2540
+ "loss": 0.5853,
2541
+ "step": 3930
2542
+ },
2543
+ {
2544
+ "epoch": 2.74,
2545
+ "learning_rate": 4.5480028030833925e-06,
2546
+ "loss": 0.6198,
2547
+ "step": 3940
2548
+ },
2549
+ {
2550
+ "epoch": 2.75,
2551
+ "learning_rate": 4.533987386124737e-06,
2552
+ "loss": 0.6498,
2553
+ "step": 3950
2554
+ },
2555
+ {
2556
+ "epoch": 2.75,
2557
+ "learning_rate": 4.519971969166083e-06,
2558
+ "loss": 0.6078,
2559
+ "step": 3960
2560
+ },
2561
+ {
2562
+ "epoch": 2.76,
2563
+ "learning_rate": 4.505956552207429e-06,
2564
+ "loss": 0.6154,
2565
+ "step": 3970
2566
+ },
2567
+ {
2568
+ "epoch": 2.77,
2569
+ "learning_rate": 4.491941135248774e-06,
2570
+ "loss": 0.6019,
2571
+ "step": 3980
2572
+ },
2573
+ {
2574
+ "epoch": 2.78,
2575
+ "learning_rate": 4.477925718290119e-06,
2576
+ "loss": 0.6166,
2577
+ "step": 3990
2578
+ },
2579
+ {
2580
+ "epoch": 2.78,
2581
+ "learning_rate": 4.463910301331465e-06,
2582
+ "loss": 0.6226,
2583
+ "step": 4000
2584
+ },
2585
+ {
2586
+ "epoch": 2.79,
2587
+ "learning_rate": 4.4498948843728106e-06,
2588
+ "loss": 0.6139,
2589
+ "step": 4010
2590
+ },
2591
+ {
2592
+ "epoch": 2.8,
2593
+ "learning_rate": 4.435879467414156e-06,
2594
+ "loss": 0.5833,
2595
+ "step": 4020
2596
+ },
2597
+ {
2598
+ "epoch": 2.8,
2599
+ "learning_rate": 4.421864050455502e-06,
2600
+ "loss": 0.6128,
2601
+ "step": 4030
2602
+ },
2603
+ {
2604
+ "epoch": 2.81,
2605
+ "learning_rate": 4.407848633496847e-06,
2606
+ "loss": 0.5927,
2607
+ "step": 4040
2608
+ },
2609
+ {
2610
+ "epoch": 2.82,
2611
+ "learning_rate": 4.393833216538192e-06,
2612
+ "loss": 0.6445,
2613
+ "step": 4050
2614
+ },
2615
+ {
2616
+ "epoch": 2.82,
2617
+ "learning_rate": 4.379817799579538e-06,
2618
+ "loss": 0.5848,
2619
+ "step": 4060
2620
+ },
2621
+ {
2622
+ "epoch": 2.83,
2623
+ "learning_rate": 4.365802382620884e-06,
2624
+ "loss": 0.618,
2625
+ "step": 4070
2626
+ },
2627
+ {
2628
+ "epoch": 2.84,
2629
+ "learning_rate": 4.351786965662229e-06,
2630
+ "loss": 0.5996,
2631
+ "step": 4080
2632
+ },
2633
+ {
2634
+ "epoch": 2.85,
2635
+ "learning_rate": 4.337771548703574e-06,
2636
+ "loss": 0.5851,
2637
+ "step": 4090
2638
+ },
2639
+ {
2640
+ "epoch": 2.85,
2641
+ "learning_rate": 4.32375613174492e-06,
2642
+ "loss": 0.6112,
2643
+ "step": 4100
2644
+ },
2645
+ {
2646
+ "epoch": 2.86,
2647
+ "learning_rate": 4.309740714786266e-06,
2648
+ "loss": 0.6492,
2649
+ "step": 4110
2650
+ },
2651
+ {
2652
+ "epoch": 2.87,
2653
+ "learning_rate": 4.295725297827611e-06,
2654
+ "loss": 0.6016,
2655
+ "step": 4120
2656
+ },
2657
+ {
2658
+ "epoch": 2.87,
2659
+ "learning_rate": 4.281709880868956e-06,
2660
+ "loss": 0.6518,
2661
+ "step": 4130
2662
+ },
2663
+ {
2664
+ "epoch": 2.88,
2665
+ "learning_rate": 4.267694463910302e-06,
2666
+ "loss": 0.5722,
2667
+ "step": 4140
2668
+ },
2669
+ {
2670
+ "epoch": 2.89,
2671
+ "learning_rate": 4.253679046951647e-06,
2672
+ "loss": 0.579,
2673
+ "step": 4150
2674
+ },
2675
+ {
2676
+ "epoch": 2.89,
2677
+ "learning_rate": 4.239663629992993e-06,
2678
+ "loss": 0.5974,
2679
+ "step": 4160
2680
+ },
2681
+ {
2682
+ "epoch": 2.9,
2683
+ "learning_rate": 4.225648213034338e-06,
2684
+ "loss": 0.6183,
2685
+ "step": 4170
2686
+ },
2687
+ {
2688
+ "epoch": 2.91,
2689
+ "learning_rate": 4.211632796075684e-06,
2690
+ "loss": 0.6302,
2691
+ "step": 4180
2692
+ },
2693
+ {
2694
+ "epoch": 2.91,
2695
+ "learning_rate": 4.1976173791170285e-06,
2696
+ "loss": 0.6088,
2697
+ "step": 4190
2698
+ },
2699
+ {
2700
+ "epoch": 2.92,
2701
+ "learning_rate": 4.183601962158374e-06,
2702
+ "loss": 0.6204,
2703
+ "step": 4200
2704
+ },
2705
+ {
2706
+ "epoch": 2.92,
2707
+ "eval_custom_hf_pref_accuracy": 0.6205,
2708
+ "eval_custom_hf_pref_kendalltau": 0.241,
2709
+ "eval_custom_hf_pref_loss": 0.5986875295639038,
2710
+ "eval_custom_hf_pref_neg_score": -0.47119298577308655,
2711
+ "eval_custom_hf_pref_pos_score": -0.10675738751888275,
2712
+ "eval_custom_hf_pref_runtime": 3.1679,
2713
+ "eval_custom_hf_pref_samples_per_second": 631.326,
2714
+ "eval_custom_hf_pref_score_diff": 0.3644356429576874,
2715
+ "eval_custom_hf_pref_steps_per_second": 78.916,
2716
+ "step": 4200
2717
+ },
2718
+ {
2719
+ "epoch": 2.93,
2720
+ "learning_rate": 4.169586545199721e-06,
2721
+ "loss": 0.5967,
2722
+ "step": 4210
2723
+ },
2724
+ {
2725
+ "epoch": 2.94,
2726
+ "learning_rate": 4.1555711282410655e-06,
2727
+ "loss": 0.5978,
2728
+ "step": 4220
2729
+ },
2730
+ {
2731
+ "epoch": 2.94,
2732
+ "learning_rate": 4.141555711282411e-06,
2733
+ "loss": 0.6013,
2734
+ "step": 4230
2735
+ },
2736
+ {
2737
+ "epoch": 2.95,
2738
+ "learning_rate": 4.127540294323756e-06,
2739
+ "loss": 0.6363,
2740
+ "step": 4240
2741
+ },
2742
+ {
2743
+ "epoch": 2.96,
2744
+ "learning_rate": 4.113524877365102e-06,
2745
+ "loss": 0.597,
2746
+ "step": 4250
2747
+ },
2748
+ {
2749
+ "epoch": 2.96,
2750
+ "learning_rate": 4.099509460406447e-06,
2751
+ "loss": 0.5893,
2752
+ "step": 4260
2753
+ },
2754
+ {
2755
+ "epoch": 2.97,
2756
+ "learning_rate": 4.085494043447793e-06,
2757
+ "loss": 0.643,
2758
+ "step": 4270
2759
+ },
2760
+ {
2761
+ "epoch": 2.98,
2762
+ "learning_rate": 4.071478626489138e-06,
2763
+ "loss": 0.5555,
2764
+ "step": 4280
2765
+ },
2766
+ {
2767
+ "epoch": 2.98,
2768
+ "learning_rate": 4.0574632095304836e-06,
2769
+ "loss": 0.6336,
2770
+ "step": 4290
2771
+ },
2772
+ {
2773
+ "epoch": 2.99,
2774
+ "learning_rate": 4.043447792571829e-06,
2775
+ "loss": 0.5994,
2776
+ "step": 4300
2777
+ },
2778
+ {
2779
+ "epoch": 3.0,
2780
+ "learning_rate": 4.029432375613175e-06,
2781
+ "loss": 0.5717,
2782
+ "step": 4310
2783
+ },
2784
+ {
2785
+ "epoch": 3.01,
2786
+ "learning_rate": 4.015416958654521e-06,
2787
+ "loss": 0.6058,
2788
+ "step": 4320
2789
+ },
2790
+ {
2791
+ "epoch": 3.01,
2792
+ "learning_rate": 4.001401541695865e-06,
2793
+ "loss": 0.6151,
2794
+ "step": 4330
2795
+ },
2796
+ {
2797
+ "epoch": 3.02,
2798
+ "learning_rate": 3.987386124737211e-06,
2799
+ "loss": 0.5971,
2800
+ "step": 4340
2801
+ },
2802
+ {
2803
+ "epoch": 3.03,
2804
+ "learning_rate": 3.973370707778557e-06,
2805
+ "loss": 0.6147,
2806
+ "step": 4350
2807
+ },
2808
+ {
2809
+ "epoch": 3.03,
2810
+ "learning_rate": 3.9593552908199024e-06,
2811
+ "loss": 0.5664,
2812
+ "step": 4360
2813
+ },
2814
+ {
2815
+ "epoch": 3.04,
2816
+ "learning_rate": 3.945339873861247e-06,
2817
+ "loss": 0.609,
2818
+ "step": 4370
2819
+ },
2820
+ {
2821
+ "epoch": 3.05,
2822
+ "learning_rate": 3.931324456902593e-06,
2823
+ "loss": 0.6032,
2824
+ "step": 4380
2825
+ },
2826
+ {
2827
+ "epoch": 3.05,
2828
+ "learning_rate": 3.917309039943939e-06,
2829
+ "loss": 0.6202,
2830
+ "step": 4390
2831
+ },
2832
+ {
2833
+ "epoch": 3.06,
2834
+ "learning_rate": 3.903293622985284e-06,
2835
+ "loss": 0.5965,
2836
+ "step": 4400
2837
+ },
2838
+ {
2839
+ "epoch": 3.07,
2840
+ "learning_rate": 3.88927820602663e-06,
2841
+ "loss": 0.6127,
2842
+ "step": 4410
2843
+ },
2844
+ {
2845
+ "epoch": 3.07,
2846
+ "learning_rate": 3.875262789067975e-06,
2847
+ "loss": 0.6345,
2848
+ "step": 4420
2849
+ },
2850
+ {
2851
+ "epoch": 3.08,
2852
+ "learning_rate": 3.8612473721093205e-06,
2853
+ "loss": 0.629,
2854
+ "step": 4430
2855
+ },
2856
+ {
2857
+ "epoch": 3.09,
2858
+ "learning_rate": 3.847231955150666e-06,
2859
+ "loss": 0.6408,
2860
+ "step": 4440
2861
+ },
2862
+ {
2863
+ "epoch": 3.1,
2864
+ "learning_rate": 3.833216538192012e-06,
2865
+ "loss": 0.584,
2866
+ "step": 4450
2867
+ },
2868
+ {
2869
+ "epoch": 3.1,
2870
+ "learning_rate": 3.819201121233357e-06,
2871
+ "loss": 0.604,
2872
+ "step": 4460
2873
+ },
2874
+ {
2875
+ "epoch": 3.11,
2876
+ "learning_rate": 3.8051857042747028e-06,
2877
+ "loss": 0.5901,
2878
+ "step": 4470
2879
+ },
2880
+ {
2881
+ "epoch": 3.12,
2882
+ "learning_rate": 3.7911702873160476e-06,
2883
+ "loss": 0.5901,
2884
+ "step": 4480
2885
+ },
2886
+ {
2887
+ "epoch": 3.12,
2888
+ "learning_rate": 3.7771548703573933e-06,
2889
+ "loss": 0.5902,
2890
+ "step": 4490
2891
+ },
2892
+ {
2893
+ "epoch": 3.13,
2894
+ "learning_rate": 3.7631394533987385e-06,
2895
+ "loss": 0.6193,
2896
+ "step": 4500
2897
+ },
2898
+ {
2899
+ "epoch": 3.13,
2900
+ "eval_custom_hf_pref_accuracy": 0.6265,
2901
+ "eval_custom_hf_pref_kendalltau": 0.253,
2902
+ "eval_custom_hf_pref_loss": 0.5968906283378601,
2903
+ "eval_custom_hf_pref_neg_score": -0.27363714575767517,
2904
+ "eval_custom_hf_pref_pos_score": 0.12795554101467133,
2905
+ "eval_custom_hf_pref_runtime": 3.1482,
2906
+ "eval_custom_hf_pref_samples_per_second": 635.277,
2907
+ "eval_custom_hf_pref_score_diff": 0.4015926420688629,
2908
+ "eval_custom_hf_pref_steps_per_second": 79.41,
2909
+ "step": 4500
2910
+ },
2911
+ {
2912
+ "epoch": 3.14,
2913
+ "learning_rate": 3.749124036440084e-06,
2914
+ "loss": 0.6134,
2915
+ "step": 4510
2916
+ },
2917
+ {
2918
+ "epoch": 3.14,
2919
+ "learning_rate": 3.7351086194814303e-06,
2920
+ "loss": 0.6028,
2921
+ "step": 4520
2922
+ },
2923
+ {
2924
+ "epoch": 3.15,
2925
+ "learning_rate": 3.721093202522775e-06,
2926
+ "loss": 0.6098,
2927
+ "step": 4530
2928
+ },
2929
+ {
2930
+ "epoch": 3.16,
2931
+ "learning_rate": 3.707077785564121e-06,
2932
+ "loss": 0.6323,
2933
+ "step": 4540
2934
+ },
2935
+ {
2936
+ "epoch": 3.17,
2937
+ "learning_rate": 3.693062368605466e-06,
2938
+ "loss": 0.5975,
2939
+ "step": 4550
2940
+ },
2941
+ {
2942
+ "epoch": 3.17,
2943
+ "learning_rate": 3.6790469516468117e-06,
2944
+ "loss": 0.59,
2945
+ "step": 4560
2946
+ },
2947
+ {
2948
+ "epoch": 3.18,
2949
+ "learning_rate": 3.665031534688157e-06,
2950
+ "loss": 0.5724,
2951
+ "step": 4570
2952
+ },
2953
+ {
2954
+ "epoch": 3.19,
2955
+ "learning_rate": 3.6510161177295027e-06,
2956
+ "loss": 0.6176,
2957
+ "step": 4580
2958
+ },
2959
+ {
2960
+ "epoch": 3.19,
2961
+ "learning_rate": 3.637000700770848e-06,
2962
+ "loss": 0.6124,
2963
+ "step": 4590
2964
+ },
2965
+ {
2966
+ "epoch": 3.2,
2967
+ "learning_rate": 3.6229852838121936e-06,
2968
+ "loss": 0.5846,
2969
+ "step": 4600
2970
+ },
2971
+ {
2972
+ "epoch": 3.21,
2973
+ "learning_rate": 3.6089698668535393e-06,
2974
+ "loss": 0.596,
2975
+ "step": 4610
2976
+ },
2977
+ {
2978
+ "epoch": 3.21,
2979
+ "learning_rate": 3.5949544498948845e-06,
2980
+ "loss": 0.6126,
2981
+ "step": 4620
2982
+ },
2983
+ {
2984
+ "epoch": 3.22,
2985
+ "learning_rate": 3.58093903293623e-06,
2986
+ "loss": 0.5881,
2987
+ "step": 4630
2988
+ },
2989
+ {
2990
+ "epoch": 3.23,
2991
+ "learning_rate": 3.5669236159775754e-06,
2992
+ "loss": 0.6052,
2993
+ "step": 4640
2994
+ },
2995
+ {
2996
+ "epoch": 3.23,
2997
+ "learning_rate": 3.552908199018921e-06,
2998
+ "loss": 0.6158,
2999
+ "step": 4650
3000
+ },
3001
+ {
3002
+ "epoch": 3.24,
3003
+ "learning_rate": 3.5388927820602664e-06,
3004
+ "loss": 0.5856,
3005
+ "step": 4660
3006
+ },
3007
+ {
3008
+ "epoch": 3.25,
3009
+ "learning_rate": 3.524877365101612e-06,
3010
+ "loss": 0.6291,
3011
+ "step": 4670
3012
+ },
3013
+ {
3014
+ "epoch": 3.26,
3015
+ "learning_rate": 3.5108619481429573e-06,
3016
+ "loss": 0.6234,
3017
+ "step": 4680
3018
+ },
3019
+ {
3020
+ "epoch": 3.26,
3021
+ "learning_rate": 3.496846531184303e-06,
3022
+ "loss": 0.5928,
3023
+ "step": 4690
3024
+ },
3025
+ {
3026
+ "epoch": 3.27,
3027
+ "learning_rate": 3.4828311142256482e-06,
3028
+ "loss": 0.5998,
3029
+ "step": 4700
3030
+ },
3031
+ {
3032
+ "epoch": 3.28,
3033
+ "learning_rate": 3.468815697266994e-06,
3034
+ "loss": 0.601,
3035
+ "step": 4710
3036
+ },
3037
+ {
3038
+ "epoch": 3.28,
3039
+ "learning_rate": 3.4548002803083396e-06,
3040
+ "loss": 0.6353,
3041
+ "step": 4720
3042
+ },
3043
+ {
3044
+ "epoch": 3.29,
3045
+ "learning_rate": 3.440784863349685e-06,
3046
+ "loss": 0.6226,
3047
+ "step": 4730
3048
+ },
3049
+ {
3050
+ "epoch": 3.3,
3051
+ "learning_rate": 3.4267694463910305e-06,
3052
+ "loss": 0.5706,
3053
+ "step": 4740
3054
+ },
3055
+ {
3056
+ "epoch": 3.3,
3057
+ "learning_rate": 3.4127540294323758e-06,
3058
+ "loss": 0.5953,
3059
+ "step": 4750
3060
+ },
3061
+ {
3062
+ "epoch": 3.31,
3063
+ "learning_rate": 3.3987386124737214e-06,
3064
+ "loss": 0.5982,
3065
+ "step": 4760
3066
+ },
3067
+ {
3068
+ "epoch": 3.32,
3069
+ "learning_rate": 3.3847231955150667e-06,
3070
+ "loss": 0.5873,
3071
+ "step": 4770
3072
+ },
3073
+ {
3074
+ "epoch": 3.33,
3075
+ "learning_rate": 3.3707077785564124e-06,
3076
+ "loss": 0.609,
3077
+ "step": 4780
3078
+ },
3079
+ {
3080
+ "epoch": 3.33,
3081
+ "learning_rate": 3.3566923615977576e-06,
3082
+ "loss": 0.6384,
3083
+ "step": 4790
3084
+ },
3085
+ {
3086
+ "epoch": 3.34,
3087
+ "learning_rate": 3.3426769446391033e-06,
3088
+ "loss": 0.5835,
3089
+ "step": 4800
3090
+ },
3091
+ {
3092
+ "epoch": 3.34,
3093
+ "eval_custom_hf_pref_accuracy": 0.614,
3094
+ "eval_custom_hf_pref_kendalltau": 0.228,
3095
+ "eval_custom_hf_pref_loss": 0.6002968549728394,
3096
+ "eval_custom_hf_pref_neg_score": -0.48366308212280273,
3097
+ "eval_custom_hf_pref_pos_score": -0.10953208804130554,
3098
+ "eval_custom_hf_pref_runtime": 3.1498,
3099
+ "eval_custom_hf_pref_samples_per_second": 634.964,
3100
+ "eval_custom_hf_pref_score_diff": 0.3741309940814972,
3101
+ "eval_custom_hf_pref_steps_per_second": 79.371,
3102
+ "step": 4800
3103
+ },
3104
+ {
3105
+ "epoch": 3.35,
3106
+ "learning_rate": 3.328661527680449e-06,
3107
+ "loss": 0.6107,
3108
+ "step": 4810
3109
+ },
3110
+ {
3111
+ "epoch": 3.35,
3112
+ "learning_rate": 3.3146461107217942e-06,
3113
+ "loss": 0.6198,
3114
+ "step": 4820
3115
+ },
3116
+ {
3117
+ "epoch": 3.36,
3118
+ "learning_rate": 3.30063069376314e-06,
3119
+ "loss": 0.6077,
3120
+ "step": 4830
3121
+ },
3122
+ {
3123
+ "epoch": 3.37,
3124
+ "learning_rate": 3.286615276804485e-06,
3125
+ "loss": 0.6009,
3126
+ "step": 4840
3127
+ },
3128
+ {
3129
+ "epoch": 3.37,
3130
+ "learning_rate": 3.272599859845831e-06,
3131
+ "loss": 0.627,
3132
+ "step": 4850
3133
+ },
3134
+ {
3135
+ "epoch": 3.38,
3136
+ "learning_rate": 3.258584442887176e-06,
3137
+ "loss": 0.5839,
3138
+ "step": 4860
3139
+ },
3140
+ {
3141
+ "epoch": 3.39,
3142
+ "learning_rate": 3.2445690259285218e-06,
3143
+ "loss": 0.6205,
3144
+ "step": 4870
3145
+ },
3146
+ {
3147
+ "epoch": 3.39,
3148
+ "learning_rate": 3.230553608969867e-06,
3149
+ "loss": 0.6206,
3150
+ "step": 4880
3151
+ },
3152
+ {
3153
+ "epoch": 3.4,
3154
+ "learning_rate": 3.2165381920112127e-06,
3155
+ "loss": 0.6411,
3156
+ "step": 4890
3157
+ },
3158
+ {
3159
+ "epoch": 3.41,
3160
+ "learning_rate": 3.202522775052558e-06,
3161
+ "loss": 0.6231,
3162
+ "step": 4900
3163
+ },
3164
+ {
3165
+ "epoch": 3.42,
3166
+ "learning_rate": 3.1885073580939036e-06,
3167
+ "loss": 0.6324,
3168
+ "step": 4910
3169
+ },
3170
+ {
3171
+ "epoch": 3.42,
3172
+ "learning_rate": 3.1744919411352493e-06,
3173
+ "loss": 0.6094,
3174
+ "step": 4920
3175
+ },
3176
+ {
3177
+ "epoch": 3.43,
3178
+ "learning_rate": 3.1604765241765945e-06,
3179
+ "loss": 0.5963,
3180
+ "step": 4930
3181
+ },
3182
+ {
3183
+ "epoch": 3.44,
3184
+ "learning_rate": 3.1464611072179402e-06,
3185
+ "loss": 0.6245,
3186
+ "step": 4940
3187
+ },
3188
+ {
3189
+ "epoch": 3.44,
3190
+ "learning_rate": 3.1324456902592855e-06,
3191
+ "loss": 0.602,
3192
+ "step": 4950
3193
+ },
3194
+ {
3195
+ "epoch": 3.45,
3196
+ "learning_rate": 3.118430273300631e-06,
3197
+ "loss": 0.5815,
3198
+ "step": 4960
3199
+ },
3200
+ {
3201
+ "epoch": 3.46,
3202
+ "learning_rate": 3.1044148563419764e-06,
3203
+ "loss": 0.5761,
3204
+ "step": 4970
3205
+ },
3206
+ {
3207
+ "epoch": 3.46,
3208
+ "learning_rate": 3.090399439383322e-06,
3209
+ "loss": 0.6007,
3210
+ "step": 4980
3211
+ },
3212
+ {
3213
+ "epoch": 3.47,
3214
+ "learning_rate": 3.0763840224246673e-06,
3215
+ "loss": 0.6386,
3216
+ "step": 4990
3217
+ },
3218
+ {
3219
+ "epoch": 3.48,
3220
+ "learning_rate": 3.062368605466013e-06,
3221
+ "loss": 0.6073,
3222
+ "step": 5000
3223
+ }
3224
+ ],
3225
+ "max_steps": 7185,
3226
+ "num_train_epochs": 5,
3227
+ "total_flos": 0.0,
3228
+ "trial_name": null,
3229
+ "trial_params": null
3230
+ }
checkpoint-5000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15d17db566d266013af03a48da12b993aa2b485f122344645b19641417f7840d
3
+ size 4881
checkpoint-6000/config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "tlc4418/pythia_70m_sft",
3
+ "architectures": [
4
+ "GPTNeoXRewardModel"
5
+ ],
6
+ "bos_token_id": 0,
7
+ "classifier_dropout": 0.1,
8
+ "eos_token_id": 0,
9
+ "hidden_act": "gelu",
10
+ "hidden_size": 512,
11
+ "initializer_range": 0.02,
12
+ "intermediate_size": 2048,
13
+ "layer_norm_eps": 1e-05,
14
+ "max_position_embeddings": 2048,
15
+ "model_type": "gpt_neox_reward_model",
16
+ "num_attention_heads": 8,
17
+ "num_hidden_layers": 6,
18
+ "pooling": "mean",
19
+ "reward_head_activation": "gelu",
20
+ "reward_head_dropout": 0.0,
21
+ "reward_head_hidden_dim": null,
22
+ "reward_head_type": "layernorm_linear",
23
+ "rotary_emb_base": 10000,
24
+ "rotary_pct": 0.25,
25
+ "tie_word_embeddings": false,
26
+ "torch_dtype": "bfloat16",
27
+ "transformers_version": "4.29.2",
28
+ "use_cache": true,
29
+ "use_parallel_residual": true,
30
+ "vocab_size": 50288
31
+ }
checkpoint-6000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b8175bd8e1bd977fae23d588b043f61b3e00ef8469f42c9fee011e0f4097e09b
3
+ size 178704331
checkpoint-6000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fdc9b65d6e190a1082e329bdf2ca402d8e54fa7c209a9955069fe8be24ada1fe
3
+ size 114527658
checkpoint-6000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d24f414bdae146a6f812a1fc61264695a89b0f20e3adfa70b4026df57d962359
3
+ size 14709
checkpoint-6000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b015afe1f47c2fa70d51c369a2e065a9d41ad9859b1a16609013886acb0c52fa
3
+ size 1465
checkpoint-6000/special_tokens_map.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<|prefix_end|>",
4
+ "<|assistant|>",
5
+ "<|prefix_begin|>",
6
+ "<|prompter|>",
7
+ "<|system|>"
8
+ ],
9
+ "bos_token": "<|endoftext|>",
10
+ "eos_token": "<|endoftext|>",
11
+ "pad_token": "<|padding|>",
12
+ "sep_token": "<|endoftext|>",
13
+ "unk_token": "<|endoftext|>"
14
+ }
checkpoint-6000/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-6000/tokenizer_config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "bos_token": "<|endoftext|>",
4
+ "clean_up_tokenization_spaces": true,
5
+ "eos_token": "<|endoftext|>",
6
+ "model_max_length": 1000000000000000019884624838656,
7
+ "tokenizer_class": "GPTNeoXTokenizer",
8
+ "unk_token": "<|endoftext|>"
9
+ }
checkpoint-6000/trainer_state.json ADDED
@@ -0,0 +1,3882 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 4.173913043478261,
5
+ "global_step": 6000,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.0,
12
+ "learning_rate": 2.0000000000000002e-07,
13
+ "loss": 0.6842,
14
+ "step": 1
15
+ },
16
+ {
17
+ "epoch": 0.01,
18
+ "learning_rate": 2.0000000000000003e-06,
19
+ "loss": 0.6871,
20
+ "step": 10
21
+ },
22
+ {
23
+ "epoch": 0.01,
24
+ "learning_rate": 4.000000000000001e-06,
25
+ "loss": 0.691,
26
+ "step": 20
27
+ },
28
+ {
29
+ "epoch": 0.02,
30
+ "learning_rate": 6e-06,
31
+ "loss": 0.6848,
32
+ "step": 30
33
+ },
34
+ {
35
+ "epoch": 0.03,
36
+ "learning_rate": 8.000000000000001e-06,
37
+ "loss": 0.6873,
38
+ "step": 40
39
+ },
40
+ {
41
+ "epoch": 0.03,
42
+ "learning_rate": 1e-05,
43
+ "loss": 0.6902,
44
+ "step": 50
45
+ },
46
+ {
47
+ "epoch": 0.04,
48
+ "learning_rate": 9.985984583041346e-06,
49
+ "loss": 0.6842,
50
+ "step": 60
51
+ },
52
+ {
53
+ "epoch": 0.05,
54
+ "learning_rate": 9.971969166082692e-06,
55
+ "loss": 0.6908,
56
+ "step": 70
57
+ },
58
+ {
59
+ "epoch": 0.06,
60
+ "learning_rate": 9.957953749124038e-06,
61
+ "loss": 0.6865,
62
+ "step": 80
63
+ },
64
+ {
65
+ "epoch": 0.06,
66
+ "learning_rate": 9.943938332165382e-06,
67
+ "loss": 0.6834,
68
+ "step": 90
69
+ },
70
+ {
71
+ "epoch": 0.07,
72
+ "learning_rate": 9.929922915206728e-06,
73
+ "loss": 0.6884,
74
+ "step": 100
75
+ },
76
+ {
77
+ "epoch": 0.08,
78
+ "learning_rate": 9.915907498248075e-06,
79
+ "loss": 0.6837,
80
+ "step": 110
81
+ },
82
+ {
83
+ "epoch": 0.08,
84
+ "learning_rate": 9.901892081289419e-06,
85
+ "loss": 0.6883,
86
+ "step": 120
87
+ },
88
+ {
89
+ "epoch": 0.09,
90
+ "learning_rate": 9.887876664330765e-06,
91
+ "loss": 0.6816,
92
+ "step": 130
93
+ },
94
+ {
95
+ "epoch": 0.1,
96
+ "learning_rate": 9.87386124737211e-06,
97
+ "loss": 0.6906,
98
+ "step": 140
99
+ },
100
+ {
101
+ "epoch": 0.1,
102
+ "learning_rate": 9.859845830413456e-06,
103
+ "loss": 0.6875,
104
+ "step": 150
105
+ },
106
+ {
107
+ "epoch": 0.11,
108
+ "learning_rate": 9.845830413454802e-06,
109
+ "loss": 0.6805,
110
+ "step": 160
111
+ },
112
+ {
113
+ "epoch": 0.12,
114
+ "learning_rate": 9.831814996496146e-06,
115
+ "loss": 0.6857,
116
+ "step": 170
117
+ },
118
+ {
119
+ "epoch": 0.13,
120
+ "learning_rate": 9.817799579537493e-06,
121
+ "loss": 0.6804,
122
+ "step": 180
123
+ },
124
+ {
125
+ "epoch": 0.13,
126
+ "learning_rate": 9.803784162578837e-06,
127
+ "loss": 0.6803,
128
+ "step": 190
129
+ },
130
+ {
131
+ "epoch": 0.14,
132
+ "learning_rate": 9.789768745620183e-06,
133
+ "loss": 0.6872,
134
+ "step": 200
135
+ },
136
+ {
137
+ "epoch": 0.15,
138
+ "learning_rate": 9.775753328661528e-06,
139
+ "loss": 0.6846,
140
+ "step": 210
141
+ },
142
+ {
143
+ "epoch": 0.15,
144
+ "learning_rate": 9.761737911702874e-06,
145
+ "loss": 0.6828,
146
+ "step": 220
147
+ },
148
+ {
149
+ "epoch": 0.16,
150
+ "learning_rate": 9.74772249474422e-06,
151
+ "loss": 0.6699,
152
+ "step": 230
153
+ },
154
+ {
155
+ "epoch": 0.17,
156
+ "learning_rate": 9.733707077785565e-06,
157
+ "loss": 0.673,
158
+ "step": 240
159
+ },
160
+ {
161
+ "epoch": 0.17,
162
+ "learning_rate": 9.71969166082691e-06,
163
+ "loss": 0.6733,
164
+ "step": 250
165
+ },
166
+ {
167
+ "epoch": 0.18,
168
+ "learning_rate": 9.705676243868257e-06,
169
+ "loss": 0.6674,
170
+ "step": 260
171
+ },
172
+ {
173
+ "epoch": 0.19,
174
+ "learning_rate": 9.6916608269096e-06,
175
+ "loss": 0.6708,
176
+ "step": 270
177
+ },
178
+ {
179
+ "epoch": 0.19,
180
+ "learning_rate": 9.677645409950946e-06,
181
+ "loss": 0.6661,
182
+ "step": 280
183
+ },
184
+ {
185
+ "epoch": 0.2,
186
+ "learning_rate": 9.663629992992292e-06,
187
+ "loss": 0.677,
188
+ "step": 290
189
+ },
190
+ {
191
+ "epoch": 0.21,
192
+ "learning_rate": 9.649614576033638e-06,
193
+ "loss": 0.6657,
194
+ "step": 300
195
+ },
196
+ {
197
+ "epoch": 0.21,
198
+ "eval_custom_hf_pref_accuracy": 0.5485,
199
+ "eval_custom_hf_pref_kendalltau": 0.097,
200
+ "eval_custom_hf_pref_loss": 0.6715624928474426,
201
+ "eval_custom_hf_pref_neg_score": 0.46836376190185547,
202
+ "eval_custom_hf_pref_pos_score": 0.522513747215271,
203
+ "eval_custom_hf_pref_runtime": 3.1872,
204
+ "eval_custom_hf_pref_samples_per_second": 627.506,
205
+ "eval_custom_hf_pref_score_diff": 0.054149970412254333,
206
+ "eval_custom_hf_pref_steps_per_second": 78.438,
207
+ "step": 300
208
+ },
209
+ {
210
+ "epoch": 0.22,
211
+ "learning_rate": 9.635599159074983e-06,
212
+ "loss": 0.6747,
213
+ "step": 310
214
+ },
215
+ {
216
+ "epoch": 0.22,
217
+ "learning_rate": 9.621583742116329e-06,
218
+ "loss": 0.6668,
219
+ "step": 320
220
+ },
221
+ {
222
+ "epoch": 0.23,
223
+ "learning_rate": 9.607568325157675e-06,
224
+ "loss": 0.6599,
225
+ "step": 330
226
+ },
227
+ {
228
+ "epoch": 0.24,
229
+ "learning_rate": 9.59355290819902e-06,
230
+ "loss": 0.663,
231
+ "step": 340
232
+ },
233
+ {
234
+ "epoch": 0.24,
235
+ "learning_rate": 9.579537491240364e-06,
236
+ "loss": 0.6643,
237
+ "step": 350
238
+ },
239
+ {
240
+ "epoch": 0.25,
241
+ "learning_rate": 9.565522074281712e-06,
242
+ "loss": 0.6697,
243
+ "step": 360
244
+ },
245
+ {
246
+ "epoch": 0.26,
247
+ "learning_rate": 9.551506657323056e-06,
248
+ "loss": 0.6497,
249
+ "step": 370
250
+ },
251
+ {
252
+ "epoch": 0.26,
253
+ "learning_rate": 9.537491240364401e-06,
254
+ "loss": 0.6544,
255
+ "step": 380
256
+ },
257
+ {
258
+ "epoch": 0.27,
259
+ "learning_rate": 9.523475823405747e-06,
260
+ "loss": 0.6736,
261
+ "step": 390
262
+ },
263
+ {
264
+ "epoch": 0.28,
265
+ "learning_rate": 9.509460406447093e-06,
266
+ "loss": 0.6572,
267
+ "step": 400
268
+ },
269
+ {
270
+ "epoch": 0.29,
271
+ "learning_rate": 9.495444989488438e-06,
272
+ "loss": 0.6616,
273
+ "step": 410
274
+ },
275
+ {
276
+ "epoch": 0.29,
277
+ "learning_rate": 9.481429572529784e-06,
278
+ "loss": 0.6703,
279
+ "step": 420
280
+ },
281
+ {
282
+ "epoch": 0.3,
283
+ "learning_rate": 9.467414155571128e-06,
284
+ "loss": 0.6621,
285
+ "step": 430
286
+ },
287
+ {
288
+ "epoch": 0.31,
289
+ "learning_rate": 9.453398738612475e-06,
290
+ "loss": 0.6755,
291
+ "step": 440
292
+ },
293
+ {
294
+ "epoch": 0.31,
295
+ "learning_rate": 9.43938332165382e-06,
296
+ "loss": 0.6693,
297
+ "step": 450
298
+ },
299
+ {
300
+ "epoch": 0.32,
301
+ "learning_rate": 9.425367904695165e-06,
302
+ "loss": 0.6629,
303
+ "step": 460
304
+ },
305
+ {
306
+ "epoch": 0.33,
307
+ "learning_rate": 9.41135248773651e-06,
308
+ "loss": 0.672,
309
+ "step": 470
310
+ },
311
+ {
312
+ "epoch": 0.33,
313
+ "learning_rate": 9.397337070777856e-06,
314
+ "loss": 0.6682,
315
+ "step": 480
316
+ },
317
+ {
318
+ "epoch": 0.34,
319
+ "learning_rate": 9.383321653819202e-06,
320
+ "loss": 0.6667,
321
+ "step": 490
322
+ },
323
+ {
324
+ "epoch": 0.35,
325
+ "learning_rate": 9.369306236860548e-06,
326
+ "loss": 0.6632,
327
+ "step": 500
328
+ },
329
+ {
330
+ "epoch": 0.35,
331
+ "learning_rate": 9.355290819901893e-06,
332
+ "loss": 0.6374,
333
+ "step": 510
334
+ },
335
+ {
336
+ "epoch": 0.36,
337
+ "learning_rate": 9.341275402943239e-06,
338
+ "loss": 0.6528,
339
+ "step": 520
340
+ },
341
+ {
342
+ "epoch": 0.37,
343
+ "learning_rate": 9.327259985984583e-06,
344
+ "loss": 0.6658,
345
+ "step": 530
346
+ },
347
+ {
348
+ "epoch": 0.38,
349
+ "learning_rate": 9.313244569025929e-06,
350
+ "loss": 0.6489,
351
+ "step": 540
352
+ },
353
+ {
354
+ "epoch": 0.38,
355
+ "learning_rate": 9.299229152067274e-06,
356
+ "loss": 0.6551,
357
+ "step": 550
358
+ },
359
+ {
360
+ "epoch": 0.39,
361
+ "learning_rate": 9.28521373510862e-06,
362
+ "loss": 0.6666,
363
+ "step": 560
364
+ },
365
+ {
366
+ "epoch": 0.4,
367
+ "learning_rate": 9.271198318149966e-06,
368
+ "loss": 0.6398,
369
+ "step": 570
370
+ },
371
+ {
372
+ "epoch": 0.4,
373
+ "learning_rate": 9.257182901191311e-06,
374
+ "loss": 0.6515,
375
+ "step": 580
376
+ },
377
+ {
378
+ "epoch": 0.41,
379
+ "learning_rate": 9.243167484232657e-06,
380
+ "loss": 0.6646,
381
+ "step": 590
382
+ },
383
+ {
384
+ "epoch": 0.42,
385
+ "learning_rate": 9.229152067274003e-06,
386
+ "loss": 0.6372,
387
+ "step": 600
388
+ },
389
+ {
390
+ "epoch": 0.42,
391
+ "eval_custom_hf_pref_accuracy": 0.5785,
392
+ "eval_custom_hf_pref_kendalltau": 0.157,
393
+ "eval_custom_hf_pref_loss": 0.6421093940734863,
394
+ "eval_custom_hf_pref_neg_score": 0.48188385367393494,
395
+ "eval_custom_hf_pref_pos_score": 0.6362329721450806,
396
+ "eval_custom_hf_pref_runtime": 3.0972,
397
+ "eval_custom_hf_pref_samples_per_second": 645.736,
398
+ "eval_custom_hf_pref_score_diff": 0.15434913337230682,
399
+ "eval_custom_hf_pref_steps_per_second": 80.717,
400
+ "step": 600
401
+ },
402
+ {
403
+ "epoch": 0.42,
404
+ "learning_rate": 9.215136650315347e-06,
405
+ "loss": 0.6566,
406
+ "step": 610
407
+ },
408
+ {
409
+ "epoch": 0.43,
410
+ "learning_rate": 9.201121233356694e-06,
411
+ "loss": 0.6631,
412
+ "step": 620
413
+ },
414
+ {
415
+ "epoch": 0.44,
416
+ "learning_rate": 9.187105816398038e-06,
417
+ "loss": 0.6516,
418
+ "step": 630
419
+ },
420
+ {
421
+ "epoch": 0.45,
422
+ "learning_rate": 9.173090399439384e-06,
423
+ "loss": 0.6468,
424
+ "step": 640
425
+ },
426
+ {
427
+ "epoch": 0.45,
428
+ "learning_rate": 9.15907498248073e-06,
429
+ "loss": 0.6501,
430
+ "step": 650
431
+ },
432
+ {
433
+ "epoch": 0.46,
434
+ "learning_rate": 9.145059565522075e-06,
435
+ "loss": 0.6638,
436
+ "step": 660
437
+ },
438
+ {
439
+ "epoch": 0.47,
440
+ "learning_rate": 9.131044148563421e-06,
441
+ "loss": 0.6633,
442
+ "step": 670
443
+ },
444
+ {
445
+ "epoch": 0.47,
446
+ "learning_rate": 9.117028731604765e-06,
447
+ "loss": 0.6627,
448
+ "step": 680
449
+ },
450
+ {
451
+ "epoch": 0.48,
452
+ "learning_rate": 9.103013314646112e-06,
453
+ "loss": 0.6355,
454
+ "step": 690
455
+ },
456
+ {
457
+ "epoch": 0.49,
458
+ "learning_rate": 9.088997897687458e-06,
459
+ "loss": 0.6287,
460
+ "step": 700
461
+ },
462
+ {
463
+ "epoch": 0.49,
464
+ "learning_rate": 9.074982480728802e-06,
465
+ "loss": 0.6534,
466
+ "step": 710
467
+ },
468
+ {
469
+ "epoch": 0.5,
470
+ "learning_rate": 9.060967063770148e-06,
471
+ "loss": 0.6698,
472
+ "step": 720
473
+ },
474
+ {
475
+ "epoch": 0.51,
476
+ "learning_rate": 9.046951646811493e-06,
477
+ "loss": 0.6452,
478
+ "step": 730
479
+ },
480
+ {
481
+ "epoch": 0.51,
482
+ "learning_rate": 9.032936229852839e-06,
483
+ "loss": 0.641,
484
+ "step": 740
485
+ },
486
+ {
487
+ "epoch": 0.52,
488
+ "learning_rate": 9.018920812894185e-06,
489
+ "loss": 0.6317,
490
+ "step": 750
491
+ },
492
+ {
493
+ "epoch": 0.53,
494
+ "learning_rate": 9.00490539593553e-06,
495
+ "loss": 0.6612,
496
+ "step": 760
497
+ },
498
+ {
499
+ "epoch": 0.54,
500
+ "learning_rate": 8.990889978976876e-06,
501
+ "loss": 0.6315,
502
+ "step": 770
503
+ },
504
+ {
505
+ "epoch": 0.54,
506
+ "learning_rate": 8.97687456201822e-06,
507
+ "loss": 0.6552,
508
+ "step": 780
509
+ },
510
+ {
511
+ "epoch": 0.55,
512
+ "learning_rate": 8.962859145059566e-06,
513
+ "loss": 0.6451,
514
+ "step": 790
515
+ },
516
+ {
517
+ "epoch": 0.56,
518
+ "learning_rate": 8.948843728100913e-06,
519
+ "loss": 0.64,
520
+ "step": 800
521
+ },
522
+ {
523
+ "epoch": 0.56,
524
+ "learning_rate": 8.934828311142257e-06,
525
+ "loss": 0.6313,
526
+ "step": 810
527
+ },
528
+ {
529
+ "epoch": 0.57,
530
+ "learning_rate": 8.920812894183603e-06,
531
+ "loss": 0.6466,
532
+ "step": 820
533
+ },
534
+ {
535
+ "epoch": 0.58,
536
+ "learning_rate": 8.906797477224948e-06,
537
+ "loss": 0.6406,
538
+ "step": 830
539
+ },
540
+ {
541
+ "epoch": 0.58,
542
+ "learning_rate": 8.892782060266294e-06,
543
+ "loss": 0.6479,
544
+ "step": 840
545
+ },
546
+ {
547
+ "epoch": 0.59,
548
+ "learning_rate": 8.87876664330764e-06,
549
+ "loss": 0.6602,
550
+ "step": 850
551
+ },
552
+ {
553
+ "epoch": 0.6,
554
+ "learning_rate": 8.864751226348984e-06,
555
+ "loss": 0.6279,
556
+ "step": 860
557
+ },
558
+ {
559
+ "epoch": 0.61,
560
+ "learning_rate": 8.850735809390331e-06,
561
+ "loss": 0.6439,
562
+ "step": 870
563
+ },
564
+ {
565
+ "epoch": 0.61,
566
+ "learning_rate": 8.836720392431675e-06,
567
+ "loss": 0.6192,
568
+ "step": 880
569
+ },
570
+ {
571
+ "epoch": 0.62,
572
+ "learning_rate": 8.82270497547302e-06,
573
+ "loss": 0.6529,
574
+ "step": 890
575
+ },
576
+ {
577
+ "epoch": 0.63,
578
+ "learning_rate": 8.808689558514366e-06,
579
+ "loss": 0.6246,
580
+ "step": 900
581
+ },
582
+ {
583
+ "epoch": 0.63,
584
+ "eval_custom_hf_pref_accuracy": 0.5795,
585
+ "eval_custom_hf_pref_kendalltau": 0.159,
586
+ "eval_custom_hf_pref_loss": 0.6517187356948853,
587
+ "eval_custom_hf_pref_neg_score": -0.18239864706993103,
588
+ "eval_custom_hf_pref_pos_score": -0.047199830412864685,
589
+ "eval_custom_hf_pref_runtime": 3.0666,
590
+ "eval_custom_hf_pref_samples_per_second": 652.191,
591
+ "eval_custom_hf_pref_score_diff": 0.13519881665706635,
592
+ "eval_custom_hf_pref_steps_per_second": 81.524,
593
+ "step": 900
594
+ },
595
+ {
596
+ "epoch": 0.63,
597
+ "learning_rate": 8.794674141555712e-06,
598
+ "loss": 0.6453,
599
+ "step": 910
600
+ },
601
+ {
602
+ "epoch": 0.64,
603
+ "learning_rate": 8.780658724597058e-06,
604
+ "loss": 0.6557,
605
+ "step": 920
606
+ },
607
+ {
608
+ "epoch": 0.65,
609
+ "learning_rate": 8.766643307638403e-06,
610
+ "loss": 0.6271,
611
+ "step": 930
612
+ },
613
+ {
614
+ "epoch": 0.65,
615
+ "learning_rate": 8.752627890679747e-06,
616
+ "loss": 0.6472,
617
+ "step": 940
618
+ },
619
+ {
620
+ "epoch": 0.66,
621
+ "learning_rate": 8.738612473721095e-06,
622
+ "loss": 0.6299,
623
+ "step": 950
624
+ },
625
+ {
626
+ "epoch": 0.67,
627
+ "learning_rate": 8.724597056762439e-06,
628
+ "loss": 0.6403,
629
+ "step": 960
630
+ },
631
+ {
632
+ "epoch": 0.67,
633
+ "learning_rate": 8.710581639803784e-06,
634
+ "loss": 0.6343,
635
+ "step": 970
636
+ },
637
+ {
638
+ "epoch": 0.68,
639
+ "learning_rate": 8.69656622284513e-06,
640
+ "loss": 0.6274,
641
+ "step": 980
642
+ },
643
+ {
644
+ "epoch": 0.69,
645
+ "learning_rate": 8.682550805886476e-06,
646
+ "loss": 0.6534,
647
+ "step": 990
648
+ },
649
+ {
650
+ "epoch": 0.7,
651
+ "learning_rate": 8.668535388927821e-06,
652
+ "loss": 0.6244,
653
+ "step": 1000
654
+ },
655
+ {
656
+ "epoch": 0.7,
657
+ "learning_rate": 8.654519971969167e-06,
658
+ "loss": 0.6237,
659
+ "step": 1010
660
+ },
661
+ {
662
+ "epoch": 0.71,
663
+ "learning_rate": 8.640504555010513e-06,
664
+ "loss": 0.6531,
665
+ "step": 1020
666
+ },
667
+ {
668
+ "epoch": 0.72,
669
+ "learning_rate": 8.626489138051858e-06,
670
+ "loss": 0.6347,
671
+ "step": 1030
672
+ },
673
+ {
674
+ "epoch": 0.72,
675
+ "learning_rate": 8.612473721093202e-06,
676
+ "loss": 0.6513,
677
+ "step": 1040
678
+ },
679
+ {
680
+ "epoch": 0.73,
681
+ "learning_rate": 8.598458304134548e-06,
682
+ "loss": 0.648,
683
+ "step": 1050
684
+ },
685
+ {
686
+ "epoch": 0.74,
687
+ "learning_rate": 8.584442887175894e-06,
688
+ "loss": 0.6472,
689
+ "step": 1060
690
+ },
691
+ {
692
+ "epoch": 0.74,
693
+ "learning_rate": 8.57042747021724e-06,
694
+ "loss": 0.6305,
695
+ "step": 1070
696
+ },
697
+ {
698
+ "epoch": 0.75,
699
+ "learning_rate": 8.556412053258585e-06,
700
+ "loss": 0.6241,
701
+ "step": 1080
702
+ },
703
+ {
704
+ "epoch": 0.76,
705
+ "learning_rate": 8.54239663629993e-06,
706
+ "loss": 0.6763,
707
+ "step": 1090
708
+ },
709
+ {
710
+ "epoch": 0.77,
711
+ "learning_rate": 8.528381219341276e-06,
712
+ "loss": 0.6219,
713
+ "step": 1100
714
+ },
715
+ {
716
+ "epoch": 0.77,
717
+ "learning_rate": 8.514365802382622e-06,
718
+ "loss": 0.6384,
719
+ "step": 1110
720
+ },
721
+ {
722
+ "epoch": 0.78,
723
+ "learning_rate": 8.500350385423966e-06,
724
+ "loss": 0.6188,
725
+ "step": 1120
726
+ },
727
+ {
728
+ "epoch": 0.79,
729
+ "learning_rate": 8.486334968465313e-06,
730
+ "loss": 0.6295,
731
+ "step": 1130
732
+ },
733
+ {
734
+ "epoch": 0.79,
735
+ "learning_rate": 8.472319551506657e-06,
736
+ "loss": 0.6276,
737
+ "step": 1140
738
+ },
739
+ {
740
+ "epoch": 0.8,
741
+ "learning_rate": 8.458304134548003e-06,
742
+ "loss": 0.6501,
743
+ "step": 1150
744
+ },
745
+ {
746
+ "epoch": 0.81,
747
+ "learning_rate": 8.444288717589349e-06,
748
+ "loss": 0.6282,
749
+ "step": 1160
750
+ },
751
+ {
752
+ "epoch": 0.81,
753
+ "learning_rate": 8.430273300630695e-06,
754
+ "loss": 0.628,
755
+ "step": 1170
756
+ },
757
+ {
758
+ "epoch": 0.82,
759
+ "learning_rate": 8.41625788367204e-06,
760
+ "loss": 0.6018,
761
+ "step": 1180
762
+ },
763
+ {
764
+ "epoch": 0.83,
765
+ "learning_rate": 8.402242466713386e-06,
766
+ "loss": 0.6307,
767
+ "step": 1190
768
+ },
769
+ {
770
+ "epoch": 0.83,
771
+ "learning_rate": 8.388227049754732e-06,
772
+ "loss": 0.6083,
773
+ "step": 1200
774
+ },
775
+ {
776
+ "epoch": 0.83,
777
+ "eval_custom_hf_pref_accuracy": 0.564,
778
+ "eval_custom_hf_pref_kendalltau": 0.128,
779
+ "eval_custom_hf_pref_loss": 0.6535468697547913,
780
+ "eval_custom_hf_pref_neg_score": -0.3474700450897217,
781
+ "eval_custom_hf_pref_pos_score": -0.20932382345199585,
782
+ "eval_custom_hf_pref_runtime": 3.0757,
783
+ "eval_custom_hf_pref_samples_per_second": 650.252,
784
+ "eval_custom_hf_pref_score_diff": 0.13814620673656464,
785
+ "eval_custom_hf_pref_steps_per_second": 81.282,
786
+ "step": 1200
787
+ },
788
+ {
789
+ "epoch": 0.84,
790
+ "learning_rate": 8.374211632796077e-06,
791
+ "loss": 0.629,
792
+ "step": 1210
793
+ },
794
+ {
795
+ "epoch": 0.85,
796
+ "learning_rate": 8.360196215837421e-06,
797
+ "loss": 0.6346,
798
+ "step": 1220
799
+ },
800
+ {
801
+ "epoch": 0.86,
802
+ "learning_rate": 8.346180798878767e-06,
803
+ "loss": 0.6366,
804
+ "step": 1230
805
+ },
806
+ {
807
+ "epoch": 0.86,
808
+ "learning_rate": 8.332165381920113e-06,
809
+ "loss": 0.6231,
810
+ "step": 1240
811
+ },
812
+ {
813
+ "epoch": 0.87,
814
+ "learning_rate": 8.318149964961458e-06,
815
+ "loss": 0.6154,
816
+ "step": 1250
817
+ },
818
+ {
819
+ "epoch": 0.88,
820
+ "learning_rate": 8.304134548002804e-06,
821
+ "loss": 0.6288,
822
+ "step": 1260
823
+ },
824
+ {
825
+ "epoch": 0.88,
826
+ "learning_rate": 8.29011913104415e-06,
827
+ "loss": 0.612,
828
+ "step": 1270
829
+ },
830
+ {
831
+ "epoch": 0.89,
832
+ "learning_rate": 8.276103714085495e-06,
833
+ "loss": 0.6154,
834
+ "step": 1280
835
+ },
836
+ {
837
+ "epoch": 0.9,
838
+ "learning_rate": 8.262088297126841e-06,
839
+ "loss": 0.6107,
840
+ "step": 1290
841
+ },
842
+ {
843
+ "epoch": 0.9,
844
+ "learning_rate": 8.248072880168185e-06,
845
+ "loss": 0.5914,
846
+ "step": 1300
847
+ },
848
+ {
849
+ "epoch": 0.91,
850
+ "learning_rate": 8.234057463209532e-06,
851
+ "loss": 0.6288,
852
+ "step": 1310
853
+ },
854
+ {
855
+ "epoch": 0.92,
856
+ "learning_rate": 8.220042046250876e-06,
857
+ "loss": 0.6553,
858
+ "step": 1320
859
+ },
860
+ {
861
+ "epoch": 0.93,
862
+ "learning_rate": 8.206026629292222e-06,
863
+ "loss": 0.6075,
864
+ "step": 1330
865
+ },
866
+ {
867
+ "epoch": 0.93,
868
+ "learning_rate": 8.192011212333568e-06,
869
+ "loss": 0.6223,
870
+ "step": 1340
871
+ },
872
+ {
873
+ "epoch": 0.94,
874
+ "learning_rate": 8.177995795374913e-06,
875
+ "loss": 0.5895,
876
+ "step": 1350
877
+ },
878
+ {
879
+ "epoch": 0.95,
880
+ "learning_rate": 8.163980378416259e-06,
881
+ "loss": 0.6592,
882
+ "step": 1360
883
+ },
884
+ {
885
+ "epoch": 0.95,
886
+ "learning_rate": 8.149964961457603e-06,
887
+ "loss": 0.6402,
888
+ "step": 1370
889
+ },
890
+ {
891
+ "epoch": 0.96,
892
+ "learning_rate": 8.13594954449895e-06,
893
+ "loss": 0.6228,
894
+ "step": 1380
895
+ },
896
+ {
897
+ "epoch": 0.97,
898
+ "learning_rate": 8.121934127540296e-06,
899
+ "loss": 0.6145,
900
+ "step": 1390
901
+ },
902
+ {
903
+ "epoch": 0.97,
904
+ "learning_rate": 8.10791871058164e-06,
905
+ "loss": 0.5906,
906
+ "step": 1400
907
+ },
908
+ {
909
+ "epoch": 0.98,
910
+ "learning_rate": 8.093903293622986e-06,
911
+ "loss": 0.6153,
912
+ "step": 1410
913
+ },
914
+ {
915
+ "epoch": 0.99,
916
+ "learning_rate": 8.079887876664331e-06,
917
+ "loss": 0.6331,
918
+ "step": 1420
919
+ },
920
+ {
921
+ "epoch": 0.99,
922
+ "learning_rate": 8.065872459705677e-06,
923
+ "loss": 0.6308,
924
+ "step": 1430
925
+ },
926
+ {
927
+ "epoch": 1.0,
928
+ "learning_rate": 8.051857042747023e-06,
929
+ "loss": 0.6269,
930
+ "step": 1440
931
+ },
932
+ {
933
+ "epoch": 1.01,
934
+ "learning_rate": 8.037841625788367e-06,
935
+ "loss": 0.6243,
936
+ "step": 1450
937
+ },
938
+ {
939
+ "epoch": 1.02,
940
+ "learning_rate": 8.023826208829714e-06,
941
+ "loss": 0.6369,
942
+ "step": 1460
943
+ },
944
+ {
945
+ "epoch": 1.02,
946
+ "learning_rate": 8.009810791871058e-06,
947
+ "loss": 0.6384,
948
+ "step": 1470
949
+ },
950
+ {
951
+ "epoch": 1.03,
952
+ "learning_rate": 7.995795374912404e-06,
953
+ "loss": 0.6497,
954
+ "step": 1480
955
+ },
956
+ {
957
+ "epoch": 1.04,
958
+ "learning_rate": 7.981779957953751e-06,
959
+ "loss": 0.586,
960
+ "step": 1490
961
+ },
962
+ {
963
+ "epoch": 1.04,
964
+ "learning_rate": 7.967764540995095e-06,
965
+ "loss": 0.6172,
966
+ "step": 1500
967
+ },
968
+ {
969
+ "epoch": 1.04,
970
+ "eval_custom_hf_pref_accuracy": 0.5905,
971
+ "eval_custom_hf_pref_kendalltau": 0.181,
972
+ "eval_custom_hf_pref_loss": 0.6357187628746033,
973
+ "eval_custom_hf_pref_neg_score": -0.4265330135822296,
974
+ "eval_custom_hf_pref_pos_score": -0.18024785816669464,
975
+ "eval_custom_hf_pref_runtime": 3.0869,
976
+ "eval_custom_hf_pref_samples_per_second": 647.904,
977
+ "eval_custom_hf_pref_score_diff": 0.24628514051437378,
978
+ "eval_custom_hf_pref_steps_per_second": 80.988,
979
+ "step": 1500
980
+ },
981
+ {
982
+ "epoch": 1.05,
983
+ "learning_rate": 7.95374912403644e-06,
984
+ "loss": 0.6429,
985
+ "step": 1510
986
+ },
987
+ {
988
+ "epoch": 1.06,
989
+ "learning_rate": 7.939733707077786e-06,
990
+ "loss": 0.6088,
991
+ "step": 1520
992
+ },
993
+ {
994
+ "epoch": 1.06,
995
+ "learning_rate": 7.925718290119132e-06,
996
+ "loss": 0.6179,
997
+ "step": 1530
998
+ },
999
+ {
1000
+ "epoch": 1.07,
1001
+ "learning_rate": 7.911702873160478e-06,
1002
+ "loss": 0.606,
1003
+ "step": 1540
1004
+ },
1005
+ {
1006
+ "epoch": 1.08,
1007
+ "learning_rate": 7.897687456201822e-06,
1008
+ "loss": 0.6117,
1009
+ "step": 1550
1010
+ },
1011
+ {
1012
+ "epoch": 1.09,
1013
+ "learning_rate": 7.883672039243167e-06,
1014
+ "loss": 0.6182,
1015
+ "step": 1560
1016
+ },
1017
+ {
1018
+ "epoch": 1.09,
1019
+ "learning_rate": 7.869656622284513e-06,
1020
+ "loss": 0.6339,
1021
+ "step": 1570
1022
+ },
1023
+ {
1024
+ "epoch": 1.1,
1025
+ "learning_rate": 7.855641205325859e-06,
1026
+ "loss": 0.6183,
1027
+ "step": 1580
1028
+ },
1029
+ {
1030
+ "epoch": 1.11,
1031
+ "learning_rate": 7.841625788367204e-06,
1032
+ "loss": 0.5998,
1033
+ "step": 1590
1034
+ },
1035
+ {
1036
+ "epoch": 1.11,
1037
+ "learning_rate": 7.82761037140855e-06,
1038
+ "loss": 0.6292,
1039
+ "step": 1600
1040
+ },
1041
+ {
1042
+ "epoch": 1.12,
1043
+ "learning_rate": 7.813594954449896e-06,
1044
+ "loss": 0.6168,
1045
+ "step": 1610
1046
+ },
1047
+ {
1048
+ "epoch": 1.13,
1049
+ "learning_rate": 7.799579537491241e-06,
1050
+ "loss": 0.6152,
1051
+ "step": 1620
1052
+ },
1053
+ {
1054
+ "epoch": 1.13,
1055
+ "learning_rate": 7.785564120532585e-06,
1056
+ "loss": 0.6316,
1057
+ "step": 1630
1058
+ },
1059
+ {
1060
+ "epoch": 1.14,
1061
+ "learning_rate": 7.771548703573933e-06,
1062
+ "loss": 0.615,
1063
+ "step": 1640
1064
+ },
1065
+ {
1066
+ "epoch": 1.15,
1067
+ "learning_rate": 7.757533286615277e-06,
1068
+ "loss": 0.6316,
1069
+ "step": 1650
1070
+ },
1071
+ {
1072
+ "epoch": 1.15,
1073
+ "learning_rate": 7.743517869656622e-06,
1074
+ "loss": 0.6387,
1075
+ "step": 1660
1076
+ },
1077
+ {
1078
+ "epoch": 1.16,
1079
+ "learning_rate": 7.729502452697968e-06,
1080
+ "loss": 0.6166,
1081
+ "step": 1670
1082
+ },
1083
+ {
1084
+ "epoch": 1.17,
1085
+ "learning_rate": 7.715487035739314e-06,
1086
+ "loss": 0.6316,
1087
+ "step": 1680
1088
+ },
1089
+ {
1090
+ "epoch": 1.18,
1091
+ "learning_rate": 7.70147161878066e-06,
1092
+ "loss": 0.6261,
1093
+ "step": 1690
1094
+ },
1095
+ {
1096
+ "epoch": 1.18,
1097
+ "learning_rate": 7.687456201822005e-06,
1098
+ "loss": 0.5966,
1099
+ "step": 1700
1100
+ },
1101
+ {
1102
+ "epoch": 1.19,
1103
+ "learning_rate": 7.67344078486335e-06,
1104
+ "loss": 0.6077,
1105
+ "step": 1710
1106
+ },
1107
+ {
1108
+ "epoch": 1.2,
1109
+ "learning_rate": 7.659425367904697e-06,
1110
+ "loss": 0.6325,
1111
+ "step": 1720
1112
+ },
1113
+ {
1114
+ "epoch": 1.2,
1115
+ "learning_rate": 7.64540995094604e-06,
1116
+ "loss": 0.6158,
1117
+ "step": 1730
1118
+ },
1119
+ {
1120
+ "epoch": 1.21,
1121
+ "learning_rate": 7.631394533987386e-06,
1122
+ "loss": 0.6196,
1123
+ "step": 1740
1124
+ },
1125
+ {
1126
+ "epoch": 1.22,
1127
+ "learning_rate": 7.617379117028733e-06,
1128
+ "loss": 0.6411,
1129
+ "step": 1750
1130
+ },
1131
+ {
1132
+ "epoch": 1.22,
1133
+ "learning_rate": 7.6033637000700775e-06,
1134
+ "loss": 0.616,
1135
+ "step": 1760
1136
+ },
1137
+ {
1138
+ "epoch": 1.23,
1139
+ "learning_rate": 7.589348283111423e-06,
1140
+ "loss": 0.6252,
1141
+ "step": 1770
1142
+ },
1143
+ {
1144
+ "epoch": 1.24,
1145
+ "learning_rate": 7.575332866152769e-06,
1146
+ "loss": 0.6171,
1147
+ "step": 1780
1148
+ },
1149
+ {
1150
+ "epoch": 1.25,
1151
+ "learning_rate": 7.5613174491941146e-06,
1152
+ "loss": 0.6135,
1153
+ "step": 1790
1154
+ },
1155
+ {
1156
+ "epoch": 1.25,
1157
+ "learning_rate": 7.547302032235459e-06,
1158
+ "loss": 0.6322,
1159
+ "step": 1800
1160
+ },
1161
+ {
1162
+ "epoch": 1.25,
1163
+ "eval_custom_hf_pref_accuracy": 0.6035,
1164
+ "eval_custom_hf_pref_kendalltau": 0.207,
1165
+ "eval_custom_hf_pref_loss": 0.6257030963897705,
1166
+ "eval_custom_hf_pref_neg_score": -0.47859320044517517,
1167
+ "eval_custom_hf_pref_pos_score": -0.20773868262767792,
1168
+ "eval_custom_hf_pref_runtime": 3.1023,
1169
+ "eval_custom_hf_pref_samples_per_second": 644.681,
1170
+ "eval_custom_hf_pref_score_diff": 0.27085456252098083,
1171
+ "eval_custom_hf_pref_steps_per_second": 80.585,
1172
+ "step": 1800
1173
+ },
1174
+ {
1175
+ "epoch": 1.26,
1176
+ "learning_rate": 7.533286615276805e-06,
1177
+ "loss": 0.6259,
1178
+ "step": 1810
1179
+ },
1180
+ {
1181
+ "epoch": 1.27,
1182
+ "learning_rate": 7.519271198318151e-06,
1183
+ "loss": 0.5938,
1184
+ "step": 1820
1185
+ },
1186
+ {
1187
+ "epoch": 1.27,
1188
+ "learning_rate": 7.505255781359496e-06,
1189
+ "loss": 0.5876,
1190
+ "step": 1830
1191
+ },
1192
+ {
1193
+ "epoch": 1.28,
1194
+ "learning_rate": 7.491240364400841e-06,
1195
+ "loss": 0.6386,
1196
+ "step": 1840
1197
+ },
1198
+ {
1199
+ "epoch": 1.29,
1200
+ "learning_rate": 7.477224947442187e-06,
1201
+ "loss": 0.5974,
1202
+ "step": 1850
1203
+ },
1204
+ {
1205
+ "epoch": 1.29,
1206
+ "learning_rate": 7.463209530483533e-06,
1207
+ "loss": 0.6019,
1208
+ "step": 1860
1209
+ },
1210
+ {
1211
+ "epoch": 1.3,
1212
+ "learning_rate": 7.449194113524878e-06,
1213
+ "loss": 0.5902,
1214
+ "step": 1870
1215
+ },
1216
+ {
1217
+ "epoch": 1.31,
1218
+ "learning_rate": 7.435178696566223e-06,
1219
+ "loss": 0.5528,
1220
+ "step": 1880
1221
+ },
1222
+ {
1223
+ "epoch": 1.31,
1224
+ "learning_rate": 7.42116327960757e-06,
1225
+ "loss": 0.6031,
1226
+ "step": 1890
1227
+ },
1228
+ {
1229
+ "epoch": 1.32,
1230
+ "learning_rate": 7.4071478626489144e-06,
1231
+ "loss": 0.5838,
1232
+ "step": 1900
1233
+ },
1234
+ {
1235
+ "epoch": 1.33,
1236
+ "learning_rate": 7.39313244569026e-06,
1237
+ "loss": 0.6493,
1238
+ "step": 1910
1239
+ },
1240
+ {
1241
+ "epoch": 1.34,
1242
+ "learning_rate": 7.379117028731605e-06,
1243
+ "loss": 0.6307,
1244
+ "step": 1920
1245
+ },
1246
+ {
1247
+ "epoch": 1.34,
1248
+ "learning_rate": 7.3651016117729515e-06,
1249
+ "loss": 0.5951,
1250
+ "step": 1930
1251
+ },
1252
+ {
1253
+ "epoch": 1.35,
1254
+ "learning_rate": 7.351086194814296e-06,
1255
+ "loss": 0.5965,
1256
+ "step": 1940
1257
+ },
1258
+ {
1259
+ "epoch": 1.36,
1260
+ "learning_rate": 7.337070777855642e-06,
1261
+ "loss": 0.5882,
1262
+ "step": 1950
1263
+ },
1264
+ {
1265
+ "epoch": 1.36,
1266
+ "learning_rate": 7.323055360896987e-06,
1267
+ "loss": 0.5943,
1268
+ "step": 1960
1269
+ },
1270
+ {
1271
+ "epoch": 1.37,
1272
+ "learning_rate": 7.309039943938333e-06,
1273
+ "loss": 0.586,
1274
+ "step": 1970
1275
+ },
1276
+ {
1277
+ "epoch": 1.38,
1278
+ "learning_rate": 7.295024526979678e-06,
1279
+ "loss": 0.6234,
1280
+ "step": 1980
1281
+ },
1282
+ {
1283
+ "epoch": 1.38,
1284
+ "learning_rate": 7.281009110021023e-06,
1285
+ "loss": 0.634,
1286
+ "step": 1990
1287
+ },
1288
+ {
1289
+ "epoch": 1.39,
1290
+ "learning_rate": 7.2669936930623695e-06,
1291
+ "loss": 0.6235,
1292
+ "step": 2000
1293
+ },
1294
+ {
1295
+ "epoch": 1.4,
1296
+ "learning_rate": 7.252978276103715e-06,
1297
+ "loss": 0.5714,
1298
+ "step": 2010
1299
+ },
1300
+ {
1301
+ "epoch": 1.41,
1302
+ "learning_rate": 7.23896285914506e-06,
1303
+ "loss": 0.6068,
1304
+ "step": 2020
1305
+ },
1306
+ {
1307
+ "epoch": 1.41,
1308
+ "learning_rate": 7.224947442186405e-06,
1309
+ "loss": 0.6222,
1310
+ "step": 2030
1311
+ },
1312
+ {
1313
+ "epoch": 1.42,
1314
+ "learning_rate": 7.210932025227751e-06,
1315
+ "loss": 0.6029,
1316
+ "step": 2040
1317
+ },
1318
+ {
1319
+ "epoch": 1.43,
1320
+ "learning_rate": 7.196916608269097e-06,
1321
+ "loss": 0.6093,
1322
+ "step": 2050
1323
+ },
1324
+ {
1325
+ "epoch": 1.43,
1326
+ "learning_rate": 7.182901191310442e-06,
1327
+ "loss": 0.6151,
1328
+ "step": 2060
1329
+ },
1330
+ {
1331
+ "epoch": 1.44,
1332
+ "learning_rate": 7.168885774351787e-06,
1333
+ "loss": 0.617,
1334
+ "step": 2070
1335
+ },
1336
+ {
1337
+ "epoch": 1.45,
1338
+ "learning_rate": 7.154870357393133e-06,
1339
+ "loss": 0.6077,
1340
+ "step": 2080
1341
+ },
1342
+ {
1343
+ "epoch": 1.45,
1344
+ "learning_rate": 7.140854940434478e-06,
1345
+ "loss": 0.6247,
1346
+ "step": 2090
1347
+ },
1348
+ {
1349
+ "epoch": 1.46,
1350
+ "learning_rate": 7.126839523475824e-06,
1351
+ "loss": 0.5979,
1352
+ "step": 2100
1353
+ },
1354
+ {
1355
+ "epoch": 1.46,
1356
+ "eval_custom_hf_pref_accuracy": 0.6175,
1357
+ "eval_custom_hf_pref_kendalltau": 0.235,
1358
+ "eval_custom_hf_pref_loss": 0.6143593788146973,
1359
+ "eval_custom_hf_pref_neg_score": -0.34517455101013184,
1360
+ "eval_custom_hf_pref_pos_score": 0.005482231266796589,
1361
+ "eval_custom_hf_pref_runtime": 3.0949,
1362
+ "eval_custom_hf_pref_samples_per_second": 646.224,
1363
+ "eval_custom_hf_pref_score_diff": 0.35065677762031555,
1364
+ "eval_custom_hf_pref_steps_per_second": 80.778,
1365
+ "step": 2100
1366
+ },
1367
+ {
1368
+ "epoch": 1.47,
1369
+ "learning_rate": 7.11282410651717e-06,
1370
+ "loss": 0.5953,
1371
+ "step": 2110
1372
+ },
1373
+ {
1374
+ "epoch": 1.47,
1375
+ "learning_rate": 7.098808689558515e-06,
1376
+ "loss": 0.5933,
1377
+ "step": 2120
1378
+ },
1379
+ {
1380
+ "epoch": 1.48,
1381
+ "learning_rate": 7.08479327259986e-06,
1382
+ "loss": 0.6466,
1383
+ "step": 2130
1384
+ },
1385
+ {
1386
+ "epoch": 1.49,
1387
+ "learning_rate": 7.070777855641206e-06,
1388
+ "loss": 0.6169,
1389
+ "step": 2140
1390
+ },
1391
+ {
1392
+ "epoch": 1.5,
1393
+ "learning_rate": 7.056762438682552e-06,
1394
+ "loss": 0.6024,
1395
+ "step": 2150
1396
+ },
1397
+ {
1398
+ "epoch": 1.5,
1399
+ "learning_rate": 7.042747021723897e-06,
1400
+ "loss": 0.6208,
1401
+ "step": 2160
1402
+ },
1403
+ {
1404
+ "epoch": 1.51,
1405
+ "learning_rate": 7.028731604765242e-06,
1406
+ "loss": 0.6015,
1407
+ "step": 2170
1408
+ },
1409
+ {
1410
+ "epoch": 1.52,
1411
+ "learning_rate": 7.014716187806588e-06,
1412
+ "loss": 0.6428,
1413
+ "step": 2180
1414
+ },
1415
+ {
1416
+ "epoch": 1.52,
1417
+ "learning_rate": 7.000700770847933e-06,
1418
+ "loss": 0.6282,
1419
+ "step": 2190
1420
+ },
1421
+ {
1422
+ "epoch": 1.53,
1423
+ "learning_rate": 6.986685353889279e-06,
1424
+ "loss": 0.5968,
1425
+ "step": 2200
1426
+ },
1427
+ {
1428
+ "epoch": 1.54,
1429
+ "learning_rate": 6.972669936930624e-06,
1430
+ "loss": 0.6262,
1431
+ "step": 2210
1432
+ },
1433
+ {
1434
+ "epoch": 1.54,
1435
+ "learning_rate": 6.95865451997197e-06,
1436
+ "loss": 0.6052,
1437
+ "step": 2220
1438
+ },
1439
+ {
1440
+ "epoch": 1.55,
1441
+ "learning_rate": 6.944639103013315e-06,
1442
+ "loss": 0.618,
1443
+ "step": 2230
1444
+ },
1445
+ {
1446
+ "epoch": 1.56,
1447
+ "learning_rate": 6.930623686054661e-06,
1448
+ "loss": 0.606,
1449
+ "step": 2240
1450
+ },
1451
+ {
1452
+ "epoch": 1.57,
1453
+ "learning_rate": 6.9166082690960055e-06,
1454
+ "loss": 0.6208,
1455
+ "step": 2250
1456
+ },
1457
+ {
1458
+ "epoch": 1.57,
1459
+ "learning_rate": 6.902592852137352e-06,
1460
+ "loss": 0.6052,
1461
+ "step": 2260
1462
+ },
1463
+ {
1464
+ "epoch": 1.58,
1465
+ "learning_rate": 6.888577435178697e-06,
1466
+ "loss": 0.6348,
1467
+ "step": 2270
1468
+ },
1469
+ {
1470
+ "epoch": 1.59,
1471
+ "learning_rate": 6.8745620182200425e-06,
1472
+ "loss": 0.6177,
1473
+ "step": 2280
1474
+ },
1475
+ {
1476
+ "epoch": 1.59,
1477
+ "learning_rate": 6.860546601261388e-06,
1478
+ "loss": 0.6076,
1479
+ "step": 2290
1480
+ },
1481
+ {
1482
+ "epoch": 1.6,
1483
+ "learning_rate": 6.846531184302734e-06,
1484
+ "loss": 0.5978,
1485
+ "step": 2300
1486
+ },
1487
+ {
1488
+ "epoch": 1.61,
1489
+ "learning_rate": 6.832515767344079e-06,
1490
+ "loss": 0.5999,
1491
+ "step": 2310
1492
+ },
1493
+ {
1494
+ "epoch": 1.61,
1495
+ "learning_rate": 6.818500350385424e-06,
1496
+ "loss": 0.5955,
1497
+ "step": 2320
1498
+ },
1499
+ {
1500
+ "epoch": 1.62,
1501
+ "learning_rate": 6.80448493342677e-06,
1502
+ "loss": 0.6491,
1503
+ "step": 2330
1504
+ },
1505
+ {
1506
+ "epoch": 1.63,
1507
+ "learning_rate": 6.790469516468116e-06,
1508
+ "loss": 0.6059,
1509
+ "step": 2340
1510
+ },
1511
+ {
1512
+ "epoch": 1.63,
1513
+ "learning_rate": 6.7764540995094605e-06,
1514
+ "loss": 0.6086,
1515
+ "step": 2350
1516
+ },
1517
+ {
1518
+ "epoch": 1.64,
1519
+ "learning_rate": 6.762438682550806e-06,
1520
+ "loss": 0.5981,
1521
+ "step": 2360
1522
+ },
1523
+ {
1524
+ "epoch": 1.65,
1525
+ "learning_rate": 6.748423265592152e-06,
1526
+ "loss": 0.6151,
1527
+ "step": 2370
1528
+ },
1529
+ {
1530
+ "epoch": 1.66,
1531
+ "learning_rate": 6.7344078486334976e-06,
1532
+ "loss": 0.633,
1533
+ "step": 2380
1534
+ },
1535
+ {
1536
+ "epoch": 1.66,
1537
+ "learning_rate": 6.720392431674842e-06,
1538
+ "loss": 0.6074,
1539
+ "step": 2390
1540
+ },
1541
+ {
1542
+ "epoch": 1.67,
1543
+ "learning_rate": 6.706377014716189e-06,
1544
+ "loss": 0.592,
1545
+ "step": 2400
1546
+ },
1547
+ {
1548
+ "epoch": 1.67,
1549
+ "eval_custom_hf_pref_accuracy": 0.62,
1550
+ "eval_custom_hf_pref_kendalltau": 0.24,
1551
+ "eval_custom_hf_pref_loss": 0.6024765372276306,
1552
+ "eval_custom_hf_pref_neg_score": -0.1819182187318802,
1553
+ "eval_custom_hf_pref_pos_score": 0.2181234061717987,
1554
+ "eval_custom_hf_pref_runtime": 3.1567,
1555
+ "eval_custom_hf_pref_samples_per_second": 633.564,
1556
+ "eval_custom_hf_pref_score_diff": 0.4000416398048401,
1557
+ "eval_custom_hf_pref_steps_per_second": 79.195,
1558
+ "step": 2400
1559
+ },
1560
+ {
1561
+ "epoch": 1.68,
1562
+ "learning_rate": 6.692361597757534e-06,
1563
+ "loss": 0.569,
1564
+ "step": 2410
1565
+ },
1566
+ {
1567
+ "epoch": 1.68,
1568
+ "learning_rate": 6.6783461807988794e-06,
1569
+ "loss": 0.6234,
1570
+ "step": 2420
1571
+ },
1572
+ {
1573
+ "epoch": 1.69,
1574
+ "learning_rate": 6.664330763840224e-06,
1575
+ "loss": 0.6026,
1576
+ "step": 2430
1577
+ },
1578
+ {
1579
+ "epoch": 1.7,
1580
+ "learning_rate": 6.650315346881571e-06,
1581
+ "loss": 0.604,
1582
+ "step": 2440
1583
+ },
1584
+ {
1585
+ "epoch": 1.7,
1586
+ "learning_rate": 6.636299929922916e-06,
1587
+ "loss": 0.6344,
1588
+ "step": 2450
1589
+ },
1590
+ {
1591
+ "epoch": 1.71,
1592
+ "learning_rate": 6.622284512964261e-06,
1593
+ "loss": 0.6428,
1594
+ "step": 2460
1595
+ },
1596
+ {
1597
+ "epoch": 1.72,
1598
+ "learning_rate": 6.608269096005606e-06,
1599
+ "loss": 0.6277,
1600
+ "step": 2470
1601
+ },
1602
+ {
1603
+ "epoch": 1.73,
1604
+ "learning_rate": 6.594253679046953e-06,
1605
+ "loss": 0.6206,
1606
+ "step": 2480
1607
+ },
1608
+ {
1609
+ "epoch": 1.73,
1610
+ "learning_rate": 6.5802382620882975e-06,
1611
+ "loss": 0.6334,
1612
+ "step": 2490
1613
+ },
1614
+ {
1615
+ "epoch": 1.74,
1616
+ "learning_rate": 6.566222845129643e-06,
1617
+ "loss": 0.6269,
1618
+ "step": 2500
1619
+ },
1620
+ {
1621
+ "epoch": 1.75,
1622
+ "learning_rate": 6.552207428170989e-06,
1623
+ "loss": 0.6252,
1624
+ "step": 2510
1625
+ },
1626
+ {
1627
+ "epoch": 1.75,
1628
+ "learning_rate": 6.5381920112123345e-06,
1629
+ "loss": 0.5913,
1630
+ "step": 2520
1631
+ },
1632
+ {
1633
+ "epoch": 1.76,
1634
+ "learning_rate": 6.524176594253679e-06,
1635
+ "loss": 0.5826,
1636
+ "step": 2530
1637
+ },
1638
+ {
1639
+ "epoch": 1.77,
1640
+ "learning_rate": 6.510161177295025e-06,
1641
+ "loss": 0.5981,
1642
+ "step": 2540
1643
+ },
1644
+ {
1645
+ "epoch": 1.77,
1646
+ "learning_rate": 6.496145760336371e-06,
1647
+ "loss": 0.6052,
1648
+ "step": 2550
1649
+ },
1650
+ {
1651
+ "epoch": 1.78,
1652
+ "learning_rate": 6.482130343377716e-06,
1653
+ "loss": 0.5933,
1654
+ "step": 2560
1655
+ },
1656
+ {
1657
+ "epoch": 1.79,
1658
+ "learning_rate": 6.468114926419061e-06,
1659
+ "loss": 0.5952,
1660
+ "step": 2570
1661
+ },
1662
+ {
1663
+ "epoch": 1.79,
1664
+ "learning_rate": 6.454099509460406e-06,
1665
+ "loss": 0.6018,
1666
+ "step": 2580
1667
+ },
1668
+ {
1669
+ "epoch": 1.8,
1670
+ "learning_rate": 6.4400840925017525e-06,
1671
+ "loss": 0.5824,
1672
+ "step": 2590
1673
+ },
1674
+ {
1675
+ "epoch": 1.81,
1676
+ "learning_rate": 6.426068675543098e-06,
1677
+ "loss": 0.5687,
1678
+ "step": 2600
1679
+ },
1680
+ {
1681
+ "epoch": 1.82,
1682
+ "learning_rate": 6.412053258584443e-06,
1683
+ "loss": 0.6095,
1684
+ "step": 2610
1685
+ },
1686
+ {
1687
+ "epoch": 1.82,
1688
+ "learning_rate": 6.3980378416257896e-06,
1689
+ "loss": 0.5991,
1690
+ "step": 2620
1691
+ },
1692
+ {
1693
+ "epoch": 1.83,
1694
+ "learning_rate": 6.384022424667134e-06,
1695
+ "loss": 0.5811,
1696
+ "step": 2630
1697
+ },
1698
+ {
1699
+ "epoch": 1.84,
1700
+ "learning_rate": 6.37000700770848e-06,
1701
+ "loss": 0.606,
1702
+ "step": 2640
1703
+ },
1704
+ {
1705
+ "epoch": 1.84,
1706
+ "learning_rate": 6.355991590749825e-06,
1707
+ "loss": 0.609,
1708
+ "step": 2650
1709
+ },
1710
+ {
1711
+ "epoch": 1.85,
1712
+ "learning_rate": 6.341976173791171e-06,
1713
+ "loss": 0.6127,
1714
+ "step": 2660
1715
+ },
1716
+ {
1717
+ "epoch": 1.86,
1718
+ "learning_rate": 6.327960756832516e-06,
1719
+ "loss": 0.614,
1720
+ "step": 2670
1721
+ },
1722
+ {
1723
+ "epoch": 1.86,
1724
+ "learning_rate": 6.313945339873861e-06,
1725
+ "loss": 0.6075,
1726
+ "step": 2680
1727
+ },
1728
+ {
1729
+ "epoch": 1.87,
1730
+ "learning_rate": 6.299929922915208e-06,
1731
+ "loss": 0.6046,
1732
+ "step": 2690
1733
+ },
1734
+ {
1735
+ "epoch": 1.88,
1736
+ "learning_rate": 6.285914505956553e-06,
1737
+ "loss": 0.6014,
1738
+ "step": 2700
1739
+ },
1740
+ {
1741
+ "epoch": 1.88,
1742
+ "eval_custom_hf_pref_accuracy": 0.614,
1743
+ "eval_custom_hf_pref_kendalltau": 0.228,
1744
+ "eval_custom_hf_pref_loss": 0.6012265682220459,
1745
+ "eval_custom_hf_pref_neg_score": -0.45326927304267883,
1746
+ "eval_custom_hf_pref_pos_score": -0.06561196595430374,
1747
+ "eval_custom_hf_pref_runtime": 3.1381,
1748
+ "eval_custom_hf_pref_samples_per_second": 637.337,
1749
+ "eval_custom_hf_pref_score_diff": 0.3876572847366333,
1750
+ "eval_custom_hf_pref_steps_per_second": 79.667,
1751
+ "step": 2700
1752
+ },
1753
+ {
1754
+ "epoch": 1.89,
1755
+ "learning_rate": 6.271899088997898e-06,
1756
+ "loss": 0.6322,
1757
+ "step": 2710
1758
+ },
1759
+ {
1760
+ "epoch": 1.89,
1761
+ "learning_rate": 6.257883672039243e-06,
1762
+ "loss": 0.5794,
1763
+ "step": 2720
1764
+ },
1765
+ {
1766
+ "epoch": 1.9,
1767
+ "learning_rate": 6.2438682550805895e-06,
1768
+ "loss": 0.6534,
1769
+ "step": 2730
1770
+ },
1771
+ {
1772
+ "epoch": 1.91,
1773
+ "learning_rate": 6.229852838121935e-06,
1774
+ "loss": 0.6054,
1775
+ "step": 2740
1776
+ },
1777
+ {
1778
+ "epoch": 1.91,
1779
+ "learning_rate": 6.21583742116328e-06,
1780
+ "loss": 0.5961,
1781
+ "step": 2750
1782
+ },
1783
+ {
1784
+ "epoch": 1.92,
1785
+ "learning_rate": 6.201822004204625e-06,
1786
+ "loss": 0.6372,
1787
+ "step": 2760
1788
+ },
1789
+ {
1790
+ "epoch": 1.93,
1791
+ "learning_rate": 6.187806587245971e-06,
1792
+ "loss": 0.6136,
1793
+ "step": 2770
1794
+ },
1795
+ {
1796
+ "epoch": 1.93,
1797
+ "learning_rate": 6.173791170287316e-06,
1798
+ "loss": 0.6435,
1799
+ "step": 2780
1800
+ },
1801
+ {
1802
+ "epoch": 1.94,
1803
+ "learning_rate": 6.159775753328662e-06,
1804
+ "loss": 0.599,
1805
+ "step": 2790
1806
+ },
1807
+ {
1808
+ "epoch": 1.95,
1809
+ "learning_rate": 6.145760336370008e-06,
1810
+ "loss": 0.6341,
1811
+ "step": 2800
1812
+ },
1813
+ {
1814
+ "epoch": 1.95,
1815
+ "learning_rate": 6.131744919411353e-06,
1816
+ "loss": 0.5967,
1817
+ "step": 2810
1818
+ },
1819
+ {
1820
+ "epoch": 1.96,
1821
+ "learning_rate": 6.117729502452698e-06,
1822
+ "loss": 0.6156,
1823
+ "step": 2820
1824
+ },
1825
+ {
1826
+ "epoch": 1.97,
1827
+ "learning_rate": 6.103714085494044e-06,
1828
+ "loss": 0.6235,
1829
+ "step": 2830
1830
+ },
1831
+ {
1832
+ "epoch": 1.98,
1833
+ "learning_rate": 6.08969866853539e-06,
1834
+ "loss": 0.6027,
1835
+ "step": 2840
1836
+ },
1837
+ {
1838
+ "epoch": 1.98,
1839
+ "learning_rate": 6.075683251576735e-06,
1840
+ "loss": 0.6196,
1841
+ "step": 2850
1842
+ },
1843
+ {
1844
+ "epoch": 1.99,
1845
+ "learning_rate": 6.06166783461808e-06,
1846
+ "loss": 0.586,
1847
+ "step": 2860
1848
+ },
1849
+ {
1850
+ "epoch": 2.0,
1851
+ "learning_rate": 6.0476524176594255e-06,
1852
+ "loss": 0.586,
1853
+ "step": 2870
1854
+ },
1855
+ {
1856
+ "epoch": 2.0,
1857
+ "learning_rate": 6.033637000700771e-06,
1858
+ "loss": 0.6173,
1859
+ "step": 2880
1860
+ },
1861
+ {
1862
+ "epoch": 2.01,
1863
+ "learning_rate": 6.019621583742117e-06,
1864
+ "loss": 0.6127,
1865
+ "step": 2890
1866
+ },
1867
+ {
1868
+ "epoch": 2.02,
1869
+ "learning_rate": 6.005606166783462e-06,
1870
+ "loss": 0.6308,
1871
+ "step": 2900
1872
+ },
1873
+ {
1874
+ "epoch": 2.02,
1875
+ "learning_rate": 5.991590749824808e-06,
1876
+ "loss": 0.6055,
1877
+ "step": 2910
1878
+ },
1879
+ {
1880
+ "epoch": 2.03,
1881
+ "learning_rate": 5.977575332866153e-06,
1882
+ "loss": 0.6339,
1883
+ "step": 2920
1884
+ },
1885
+ {
1886
+ "epoch": 2.04,
1887
+ "learning_rate": 5.963559915907499e-06,
1888
+ "loss": 0.6525,
1889
+ "step": 2930
1890
+ },
1891
+ {
1892
+ "epoch": 2.05,
1893
+ "learning_rate": 5.9495444989488436e-06,
1894
+ "loss": 0.6512,
1895
+ "step": 2940
1896
+ },
1897
+ {
1898
+ "epoch": 2.05,
1899
+ "learning_rate": 5.93552908199019e-06,
1900
+ "loss": 0.6068,
1901
+ "step": 2950
1902
+ },
1903
+ {
1904
+ "epoch": 2.06,
1905
+ "learning_rate": 5.921513665031535e-06,
1906
+ "loss": 0.6137,
1907
+ "step": 2960
1908
+ },
1909
+ {
1910
+ "epoch": 2.07,
1911
+ "learning_rate": 5.907498248072881e-06,
1912
+ "loss": 0.6017,
1913
+ "step": 2970
1914
+ },
1915
+ {
1916
+ "epoch": 2.07,
1917
+ "learning_rate": 5.8934828311142254e-06,
1918
+ "loss": 0.6071,
1919
+ "step": 2980
1920
+ },
1921
+ {
1922
+ "epoch": 2.08,
1923
+ "learning_rate": 5.879467414155572e-06,
1924
+ "loss": 0.585,
1925
+ "step": 2990
1926
+ },
1927
+ {
1928
+ "epoch": 2.09,
1929
+ "learning_rate": 5.865451997196917e-06,
1930
+ "loss": 0.6075,
1931
+ "step": 3000
1932
+ },
1933
+ {
1934
+ "epoch": 2.09,
1935
+ "eval_custom_hf_pref_accuracy": 0.613,
1936
+ "eval_custom_hf_pref_kendalltau": 0.226,
1937
+ "eval_custom_hf_pref_loss": 0.6039531230926514,
1938
+ "eval_custom_hf_pref_neg_score": -0.5177181363105774,
1939
+ "eval_custom_hf_pref_pos_score": -0.16565346717834473,
1940
+ "eval_custom_hf_pref_runtime": 3.1422,
1941
+ "eval_custom_hf_pref_samples_per_second": 636.497,
1942
+ "eval_custom_hf_pref_score_diff": 0.3520646393299103,
1943
+ "eval_custom_hf_pref_steps_per_second": 79.562,
1944
+ "step": 3000
1945
+ },
1946
+ {
1947
+ "epoch": 2.09,
1948
+ "learning_rate": 5.8514365802382625e-06,
1949
+ "loss": 0.6178,
1950
+ "step": 3010
1951
+ },
1952
+ {
1953
+ "epoch": 2.1,
1954
+ "learning_rate": 5.837421163279608e-06,
1955
+ "loss": 0.6208,
1956
+ "step": 3020
1957
+ },
1958
+ {
1959
+ "epoch": 2.11,
1960
+ "learning_rate": 5.823405746320954e-06,
1961
+ "loss": 0.6236,
1962
+ "step": 3030
1963
+ },
1964
+ {
1965
+ "epoch": 2.11,
1966
+ "learning_rate": 5.809390329362299e-06,
1967
+ "loss": 0.6193,
1968
+ "step": 3040
1969
+ },
1970
+ {
1971
+ "epoch": 2.12,
1972
+ "learning_rate": 5.795374912403644e-06,
1973
+ "loss": 0.5971,
1974
+ "step": 3050
1975
+ },
1976
+ {
1977
+ "epoch": 2.13,
1978
+ "learning_rate": 5.78135949544499e-06,
1979
+ "loss": 0.6193,
1980
+ "step": 3060
1981
+ },
1982
+ {
1983
+ "epoch": 2.14,
1984
+ "learning_rate": 5.767344078486336e-06,
1985
+ "loss": 0.6173,
1986
+ "step": 3070
1987
+ },
1988
+ {
1989
+ "epoch": 2.14,
1990
+ "learning_rate": 5.7533286615276805e-06,
1991
+ "loss": 0.5993,
1992
+ "step": 3080
1993
+ },
1994
+ {
1995
+ "epoch": 2.15,
1996
+ "learning_rate": 5.739313244569027e-06,
1997
+ "loss": 0.6155,
1998
+ "step": 3090
1999
+ },
2000
+ {
2001
+ "epoch": 2.16,
2002
+ "learning_rate": 5.725297827610372e-06,
2003
+ "loss": 0.6124,
2004
+ "step": 3100
2005
+ },
2006
+ {
2007
+ "epoch": 2.16,
2008
+ "learning_rate": 5.7112824106517175e-06,
2009
+ "loss": 0.5955,
2010
+ "step": 3110
2011
+ },
2012
+ {
2013
+ "epoch": 2.17,
2014
+ "learning_rate": 5.697266993693062e-06,
2015
+ "loss": 0.5802,
2016
+ "step": 3120
2017
+ },
2018
+ {
2019
+ "epoch": 2.18,
2020
+ "learning_rate": 5.683251576734409e-06,
2021
+ "loss": 0.6156,
2022
+ "step": 3130
2023
+ },
2024
+ {
2025
+ "epoch": 2.18,
2026
+ "learning_rate": 5.669236159775754e-06,
2027
+ "loss": 0.6344,
2028
+ "step": 3140
2029
+ },
2030
+ {
2031
+ "epoch": 2.19,
2032
+ "learning_rate": 5.655220742817099e-06,
2033
+ "loss": 0.5634,
2034
+ "step": 3150
2035
+ },
2036
+ {
2037
+ "epoch": 2.2,
2038
+ "learning_rate": 5.641205325858444e-06,
2039
+ "loss": 0.5955,
2040
+ "step": 3160
2041
+ },
2042
+ {
2043
+ "epoch": 2.21,
2044
+ "learning_rate": 5.627189908899791e-06,
2045
+ "loss": 0.6102,
2046
+ "step": 3170
2047
+ },
2048
+ {
2049
+ "epoch": 2.21,
2050
+ "learning_rate": 5.6131744919411356e-06,
2051
+ "loss": 0.6409,
2052
+ "step": 3180
2053
+ },
2054
+ {
2055
+ "epoch": 2.22,
2056
+ "learning_rate": 5.599159074982481e-06,
2057
+ "loss": 0.6575,
2058
+ "step": 3190
2059
+ },
2060
+ {
2061
+ "epoch": 2.23,
2062
+ "learning_rate": 5.585143658023827e-06,
2063
+ "loss": 0.6236,
2064
+ "step": 3200
2065
+ },
2066
+ {
2067
+ "epoch": 2.23,
2068
+ "learning_rate": 5.571128241065173e-06,
2069
+ "loss": 0.6078,
2070
+ "step": 3210
2071
+ },
2072
+ {
2073
+ "epoch": 2.24,
2074
+ "learning_rate": 5.557112824106517e-06,
2075
+ "loss": 0.6161,
2076
+ "step": 3220
2077
+ },
2078
+ {
2079
+ "epoch": 2.25,
2080
+ "learning_rate": 5.543097407147863e-06,
2081
+ "loss": 0.5838,
2082
+ "step": 3230
2083
+ },
2084
+ {
2085
+ "epoch": 2.25,
2086
+ "learning_rate": 5.529081990189209e-06,
2087
+ "loss": 0.6179,
2088
+ "step": 3240
2089
+ },
2090
+ {
2091
+ "epoch": 2.26,
2092
+ "learning_rate": 5.5150665732305544e-06,
2093
+ "loss": 0.5729,
2094
+ "step": 3250
2095
+ },
2096
+ {
2097
+ "epoch": 2.27,
2098
+ "learning_rate": 5.501051156271899e-06,
2099
+ "loss": 0.6353,
2100
+ "step": 3260
2101
+ },
2102
+ {
2103
+ "epoch": 2.27,
2104
+ "learning_rate": 5.487035739313244e-06,
2105
+ "loss": 0.5723,
2106
+ "step": 3270
2107
+ },
2108
+ {
2109
+ "epoch": 2.28,
2110
+ "learning_rate": 5.473020322354591e-06,
2111
+ "loss": 0.5983,
2112
+ "step": 3280
2113
+ },
2114
+ {
2115
+ "epoch": 2.29,
2116
+ "learning_rate": 5.459004905395936e-06,
2117
+ "loss": 0.5695,
2118
+ "step": 3290
2119
+ },
2120
+ {
2121
+ "epoch": 2.3,
2122
+ "learning_rate": 5.444989488437281e-06,
2123
+ "loss": 0.6189,
2124
+ "step": 3300
2125
+ },
2126
+ {
2127
+ "epoch": 2.3,
2128
+ "eval_custom_hf_pref_accuracy": 0.605,
2129
+ "eval_custom_hf_pref_kendalltau": 0.21,
2130
+ "eval_custom_hf_pref_loss": 0.6013281345367432,
2131
+ "eval_custom_hf_pref_neg_score": -0.3515034317970276,
2132
+ "eval_custom_hf_pref_pos_score": 0.02793554961681366,
2133
+ "eval_custom_hf_pref_runtime": 3.0849,
2134
+ "eval_custom_hf_pref_samples_per_second": 648.326,
2135
+ "eval_custom_hf_pref_score_diff": 0.37943896651268005,
2136
+ "eval_custom_hf_pref_steps_per_second": 81.041,
2137
+ "step": 3300
2138
+ },
2139
+ {
2140
+ "epoch": 2.3,
2141
+ "learning_rate": 5.430974071478628e-06,
2142
+ "loss": 0.6177,
2143
+ "step": 3310
2144
+ },
2145
+ {
2146
+ "epoch": 2.31,
2147
+ "learning_rate": 5.4169586545199725e-06,
2148
+ "loss": 0.616,
2149
+ "step": 3320
2150
+ },
2151
+ {
2152
+ "epoch": 2.32,
2153
+ "learning_rate": 5.402943237561318e-06,
2154
+ "loss": 0.6038,
2155
+ "step": 3330
2156
+ },
2157
+ {
2158
+ "epoch": 2.32,
2159
+ "learning_rate": 5.388927820602663e-06,
2160
+ "loss": 0.6017,
2161
+ "step": 3340
2162
+ },
2163
+ {
2164
+ "epoch": 2.33,
2165
+ "learning_rate": 5.3749124036440095e-06,
2166
+ "loss": 0.5796,
2167
+ "step": 3350
2168
+ },
2169
+ {
2170
+ "epoch": 2.34,
2171
+ "learning_rate": 5.360896986685354e-06,
2172
+ "loss": 0.6032,
2173
+ "step": 3360
2174
+ },
2175
+ {
2176
+ "epoch": 2.34,
2177
+ "learning_rate": 5.346881569726699e-06,
2178
+ "loss": 0.626,
2179
+ "step": 3370
2180
+ },
2181
+ {
2182
+ "epoch": 2.35,
2183
+ "learning_rate": 5.332866152768045e-06,
2184
+ "loss": 0.5943,
2185
+ "step": 3380
2186
+ },
2187
+ {
2188
+ "epoch": 2.36,
2189
+ "learning_rate": 5.318850735809391e-06,
2190
+ "loss": 0.5708,
2191
+ "step": 3390
2192
+ },
2193
+ {
2194
+ "epoch": 2.37,
2195
+ "learning_rate": 5.304835318850736e-06,
2196
+ "loss": 0.6458,
2197
+ "step": 3400
2198
+ },
2199
+ {
2200
+ "epoch": 2.37,
2201
+ "learning_rate": 5.290819901892081e-06,
2202
+ "loss": 0.6138,
2203
+ "step": 3410
2204
+ },
2205
+ {
2206
+ "epoch": 2.38,
2207
+ "learning_rate": 5.2768044849334275e-06,
2208
+ "loss": 0.6278,
2209
+ "step": 3420
2210
+ },
2211
+ {
2212
+ "epoch": 2.39,
2213
+ "learning_rate": 5.262789067974773e-06,
2214
+ "loss": 0.6138,
2215
+ "step": 3430
2216
+ },
2217
+ {
2218
+ "epoch": 2.39,
2219
+ "learning_rate": 5.248773651016118e-06,
2220
+ "loss": 0.6036,
2221
+ "step": 3440
2222
+ },
2223
+ {
2224
+ "epoch": 2.4,
2225
+ "learning_rate": 5.234758234057463e-06,
2226
+ "loss": 0.5893,
2227
+ "step": 3450
2228
+ },
2229
+ {
2230
+ "epoch": 2.41,
2231
+ "learning_rate": 5.220742817098809e-06,
2232
+ "loss": 0.618,
2233
+ "step": 3460
2234
+ },
2235
+ {
2236
+ "epoch": 2.41,
2237
+ "learning_rate": 5.206727400140154e-06,
2238
+ "loss": 0.5952,
2239
+ "step": 3470
2240
+ },
2241
+ {
2242
+ "epoch": 2.42,
2243
+ "learning_rate": 5.1927119831815e-06,
2244
+ "loss": 0.6331,
2245
+ "step": 3480
2246
+ },
2247
+ {
2248
+ "epoch": 2.43,
2249
+ "learning_rate": 5.178696566222845e-06,
2250
+ "loss": 0.6091,
2251
+ "step": 3490
2252
+ },
2253
+ {
2254
+ "epoch": 2.43,
2255
+ "learning_rate": 5.164681149264191e-06,
2256
+ "loss": 0.6245,
2257
+ "step": 3500
2258
+ },
2259
+ {
2260
+ "epoch": 2.44,
2261
+ "learning_rate": 5.150665732305536e-06,
2262
+ "loss": 0.5887,
2263
+ "step": 3510
2264
+ },
2265
+ {
2266
+ "epoch": 2.45,
2267
+ "learning_rate": 5.136650315346882e-06,
2268
+ "loss": 0.608,
2269
+ "step": 3520
2270
+ },
2271
+ {
2272
+ "epoch": 2.46,
2273
+ "learning_rate": 5.122634898388228e-06,
2274
+ "loss": 0.5959,
2275
+ "step": 3530
2276
+ },
2277
+ {
2278
+ "epoch": 2.46,
2279
+ "learning_rate": 5.108619481429573e-06,
2280
+ "loss": 0.6515,
2281
+ "step": 3540
2282
+ },
2283
+ {
2284
+ "epoch": 2.47,
2285
+ "learning_rate": 5.094604064470918e-06,
2286
+ "loss": 0.629,
2287
+ "step": 3550
2288
+ },
2289
+ {
2290
+ "epoch": 2.48,
2291
+ "learning_rate": 5.080588647512264e-06,
2292
+ "loss": 0.5898,
2293
+ "step": 3560
2294
+ },
2295
+ {
2296
+ "epoch": 2.48,
2297
+ "learning_rate": 5.06657323055361e-06,
2298
+ "loss": 0.5561,
2299
+ "step": 3570
2300
+ },
2301
+ {
2302
+ "epoch": 2.49,
2303
+ "learning_rate": 5.052557813594955e-06,
2304
+ "loss": 0.6031,
2305
+ "step": 3580
2306
+ },
2307
+ {
2308
+ "epoch": 2.5,
2309
+ "learning_rate": 5.0385423966363e-06,
2310
+ "loss": 0.5934,
2311
+ "step": 3590
2312
+ },
2313
+ {
2314
+ "epoch": 2.5,
2315
+ "learning_rate": 5.024526979677646e-06,
2316
+ "loss": 0.6393,
2317
+ "step": 3600
2318
+ },
2319
+ {
2320
+ "epoch": 2.5,
2321
+ "eval_custom_hf_pref_accuracy": 0.6085,
2322
+ "eval_custom_hf_pref_kendalltau": 0.217,
2323
+ "eval_custom_hf_pref_loss": 0.6061797142028809,
2324
+ "eval_custom_hf_pref_neg_score": -0.4991738796234131,
2325
+ "eval_custom_hf_pref_pos_score": -0.14666078984737396,
2326
+ "eval_custom_hf_pref_runtime": 3.0873,
2327
+ "eval_custom_hf_pref_samples_per_second": 647.819,
2328
+ "eval_custom_hf_pref_score_diff": 0.3525131344795227,
2329
+ "eval_custom_hf_pref_steps_per_second": 80.977,
2330
+ "step": 3600
2331
+ },
2332
+ {
2333
+ "epoch": 2.51,
2334
+ "learning_rate": 5.010511562718991e-06,
2335
+ "loss": 0.5903,
2336
+ "step": 3610
2337
+ },
2338
+ {
2339
+ "epoch": 2.52,
2340
+ "learning_rate": 4.996496145760337e-06,
2341
+ "loss": 0.5731,
2342
+ "step": 3620
2343
+ },
2344
+ {
2345
+ "epoch": 2.53,
2346
+ "learning_rate": 4.9824807288016825e-06,
2347
+ "loss": 0.6383,
2348
+ "step": 3630
2349
+ },
2350
+ {
2351
+ "epoch": 2.53,
2352
+ "learning_rate": 4.968465311843027e-06,
2353
+ "loss": 0.6008,
2354
+ "step": 3640
2355
+ },
2356
+ {
2357
+ "epoch": 2.54,
2358
+ "learning_rate": 4.954449894884373e-06,
2359
+ "loss": 0.6005,
2360
+ "step": 3650
2361
+ },
2362
+ {
2363
+ "epoch": 2.55,
2364
+ "learning_rate": 4.940434477925719e-06,
2365
+ "loss": 0.6348,
2366
+ "step": 3660
2367
+ },
2368
+ {
2369
+ "epoch": 2.55,
2370
+ "learning_rate": 4.926419060967064e-06,
2371
+ "loss": 0.6109,
2372
+ "step": 3670
2373
+ },
2374
+ {
2375
+ "epoch": 2.56,
2376
+ "learning_rate": 4.912403644008409e-06,
2377
+ "loss": 0.6219,
2378
+ "step": 3680
2379
+ },
2380
+ {
2381
+ "epoch": 2.57,
2382
+ "learning_rate": 4.898388227049755e-06,
2383
+ "loss": 0.589,
2384
+ "step": 3690
2385
+ },
2386
+ {
2387
+ "epoch": 2.57,
2388
+ "learning_rate": 4.8843728100911005e-06,
2389
+ "loss": 0.6278,
2390
+ "step": 3700
2391
+ },
2392
+ {
2393
+ "epoch": 2.58,
2394
+ "learning_rate": 4.870357393132446e-06,
2395
+ "loss": 0.6119,
2396
+ "step": 3710
2397
+ },
2398
+ {
2399
+ "epoch": 2.59,
2400
+ "learning_rate": 4.856341976173792e-06,
2401
+ "loss": 0.5835,
2402
+ "step": 3720
2403
+ },
2404
+ {
2405
+ "epoch": 2.59,
2406
+ "learning_rate": 4.842326559215137e-06,
2407
+ "loss": 0.6209,
2408
+ "step": 3730
2409
+ },
2410
+ {
2411
+ "epoch": 2.6,
2412
+ "learning_rate": 4.828311142256482e-06,
2413
+ "loss": 0.5998,
2414
+ "step": 3740
2415
+ },
2416
+ {
2417
+ "epoch": 2.61,
2418
+ "learning_rate": 4.814295725297828e-06,
2419
+ "loss": 0.5851,
2420
+ "step": 3750
2421
+ },
2422
+ {
2423
+ "epoch": 2.62,
2424
+ "learning_rate": 4.800280308339174e-06,
2425
+ "loss": 0.6362,
2426
+ "step": 3760
2427
+ },
2428
+ {
2429
+ "epoch": 2.62,
2430
+ "learning_rate": 4.786264891380519e-06,
2431
+ "loss": 0.6101,
2432
+ "step": 3770
2433
+ },
2434
+ {
2435
+ "epoch": 2.63,
2436
+ "learning_rate": 4.772249474421864e-06,
2437
+ "loss": 0.634,
2438
+ "step": 3780
2439
+ },
2440
+ {
2441
+ "epoch": 2.64,
2442
+ "learning_rate": 4.75823405746321e-06,
2443
+ "loss": 0.6183,
2444
+ "step": 3790
2445
+ },
2446
+ {
2447
+ "epoch": 2.64,
2448
+ "learning_rate": 4.744218640504556e-06,
2449
+ "loss": 0.6076,
2450
+ "step": 3800
2451
+ },
2452
+ {
2453
+ "epoch": 2.65,
2454
+ "learning_rate": 4.730203223545901e-06,
2455
+ "loss": 0.6406,
2456
+ "step": 3810
2457
+ },
2458
+ {
2459
+ "epoch": 2.66,
2460
+ "learning_rate": 4.716187806587246e-06,
2461
+ "loss": 0.5459,
2462
+ "step": 3820
2463
+ },
2464
+ {
2465
+ "epoch": 2.66,
2466
+ "learning_rate": 4.702172389628592e-06,
2467
+ "loss": 0.6272,
2468
+ "step": 3830
2469
+ },
2470
+ {
2471
+ "epoch": 2.67,
2472
+ "learning_rate": 4.6881569726699375e-06,
2473
+ "loss": 0.6035,
2474
+ "step": 3840
2475
+ },
2476
+ {
2477
+ "epoch": 2.68,
2478
+ "learning_rate": 4.674141555711283e-06,
2479
+ "loss": 0.6483,
2480
+ "step": 3850
2481
+ },
2482
+ {
2483
+ "epoch": 2.69,
2484
+ "learning_rate": 4.660126138752628e-06,
2485
+ "loss": 0.6028,
2486
+ "step": 3860
2487
+ },
2488
+ {
2489
+ "epoch": 2.69,
2490
+ "learning_rate": 4.646110721793974e-06,
2491
+ "loss": 0.6274,
2492
+ "step": 3870
2493
+ },
2494
+ {
2495
+ "epoch": 2.7,
2496
+ "learning_rate": 4.632095304835319e-06,
2497
+ "loss": 0.5906,
2498
+ "step": 3880
2499
+ },
2500
+ {
2501
+ "epoch": 2.71,
2502
+ "learning_rate": 4.618079887876665e-06,
2503
+ "loss": 0.6186,
2504
+ "step": 3890
2505
+ },
2506
+ {
2507
+ "epoch": 2.71,
2508
+ "learning_rate": 4.604064470918011e-06,
2509
+ "loss": 0.619,
2510
+ "step": 3900
2511
+ },
2512
+ {
2513
+ "epoch": 2.71,
2514
+ "eval_custom_hf_pref_accuracy": 0.613,
2515
+ "eval_custom_hf_pref_kendalltau": 0.226,
2516
+ "eval_custom_hf_pref_loss": 0.6025937795639038,
2517
+ "eval_custom_hf_pref_neg_score": -0.5224867463111877,
2518
+ "eval_custom_hf_pref_pos_score": -0.15825501084327698,
2519
+ "eval_custom_hf_pref_runtime": 3.0937,
2520
+ "eval_custom_hf_pref_samples_per_second": 646.485,
2521
+ "eval_custom_hf_pref_score_diff": 0.36423179507255554,
2522
+ "eval_custom_hf_pref_steps_per_second": 80.811,
2523
+ "step": 3900
2524
+ },
2525
+ {
2526
+ "epoch": 2.72,
2527
+ "learning_rate": 4.5900490539593555e-06,
2528
+ "loss": 0.6094,
2529
+ "step": 3910
2530
+ },
2531
+ {
2532
+ "epoch": 2.73,
2533
+ "learning_rate": 4.576033637000701e-06,
2534
+ "loss": 0.5902,
2535
+ "step": 3920
2536
+ },
2537
+ {
2538
+ "epoch": 2.73,
2539
+ "learning_rate": 4.562018220042047e-06,
2540
+ "loss": 0.5853,
2541
+ "step": 3930
2542
+ },
2543
+ {
2544
+ "epoch": 2.74,
2545
+ "learning_rate": 4.5480028030833925e-06,
2546
+ "loss": 0.6198,
2547
+ "step": 3940
2548
+ },
2549
+ {
2550
+ "epoch": 2.75,
2551
+ "learning_rate": 4.533987386124737e-06,
2552
+ "loss": 0.6498,
2553
+ "step": 3950
2554
+ },
2555
+ {
2556
+ "epoch": 2.75,
2557
+ "learning_rate": 4.519971969166083e-06,
2558
+ "loss": 0.6078,
2559
+ "step": 3960
2560
+ },
2561
+ {
2562
+ "epoch": 2.76,
2563
+ "learning_rate": 4.505956552207429e-06,
2564
+ "loss": 0.6154,
2565
+ "step": 3970
2566
+ },
2567
+ {
2568
+ "epoch": 2.77,
2569
+ "learning_rate": 4.491941135248774e-06,
2570
+ "loss": 0.6019,
2571
+ "step": 3980
2572
+ },
2573
+ {
2574
+ "epoch": 2.78,
2575
+ "learning_rate": 4.477925718290119e-06,
2576
+ "loss": 0.6166,
2577
+ "step": 3990
2578
+ },
2579
+ {
2580
+ "epoch": 2.78,
2581
+ "learning_rate": 4.463910301331465e-06,
2582
+ "loss": 0.6226,
2583
+ "step": 4000
2584
+ },
2585
+ {
2586
+ "epoch": 2.79,
2587
+ "learning_rate": 4.4498948843728106e-06,
2588
+ "loss": 0.6139,
2589
+ "step": 4010
2590
+ },
2591
+ {
2592
+ "epoch": 2.8,
2593
+ "learning_rate": 4.435879467414156e-06,
2594
+ "loss": 0.5833,
2595
+ "step": 4020
2596
+ },
2597
+ {
2598
+ "epoch": 2.8,
2599
+ "learning_rate": 4.421864050455502e-06,
2600
+ "loss": 0.6128,
2601
+ "step": 4030
2602
+ },
2603
+ {
2604
+ "epoch": 2.81,
2605
+ "learning_rate": 4.407848633496847e-06,
2606
+ "loss": 0.5927,
2607
+ "step": 4040
2608
+ },
2609
+ {
2610
+ "epoch": 2.82,
2611
+ "learning_rate": 4.393833216538192e-06,
2612
+ "loss": 0.6445,
2613
+ "step": 4050
2614
+ },
2615
+ {
2616
+ "epoch": 2.82,
2617
+ "learning_rate": 4.379817799579538e-06,
2618
+ "loss": 0.5848,
2619
+ "step": 4060
2620
+ },
2621
+ {
2622
+ "epoch": 2.83,
2623
+ "learning_rate": 4.365802382620884e-06,
2624
+ "loss": 0.618,
2625
+ "step": 4070
2626
+ },
2627
+ {
2628
+ "epoch": 2.84,
2629
+ "learning_rate": 4.351786965662229e-06,
2630
+ "loss": 0.5996,
2631
+ "step": 4080
2632
+ },
2633
+ {
2634
+ "epoch": 2.85,
2635
+ "learning_rate": 4.337771548703574e-06,
2636
+ "loss": 0.5851,
2637
+ "step": 4090
2638
+ },
2639
+ {
2640
+ "epoch": 2.85,
2641
+ "learning_rate": 4.32375613174492e-06,
2642
+ "loss": 0.6112,
2643
+ "step": 4100
2644
+ },
2645
+ {
2646
+ "epoch": 2.86,
2647
+ "learning_rate": 4.309740714786266e-06,
2648
+ "loss": 0.6492,
2649
+ "step": 4110
2650
+ },
2651
+ {
2652
+ "epoch": 2.87,
2653
+ "learning_rate": 4.295725297827611e-06,
2654
+ "loss": 0.6016,
2655
+ "step": 4120
2656
+ },
2657
+ {
2658
+ "epoch": 2.87,
2659
+ "learning_rate": 4.281709880868956e-06,
2660
+ "loss": 0.6518,
2661
+ "step": 4130
2662
+ },
2663
+ {
2664
+ "epoch": 2.88,
2665
+ "learning_rate": 4.267694463910302e-06,
2666
+ "loss": 0.5722,
2667
+ "step": 4140
2668
+ },
2669
+ {
2670
+ "epoch": 2.89,
2671
+ "learning_rate": 4.253679046951647e-06,
2672
+ "loss": 0.579,
2673
+ "step": 4150
2674
+ },
2675
+ {
2676
+ "epoch": 2.89,
2677
+ "learning_rate": 4.239663629992993e-06,
2678
+ "loss": 0.5974,
2679
+ "step": 4160
2680
+ },
2681
+ {
2682
+ "epoch": 2.9,
2683
+ "learning_rate": 4.225648213034338e-06,
2684
+ "loss": 0.6183,
2685
+ "step": 4170
2686
+ },
2687
+ {
2688
+ "epoch": 2.91,
2689
+ "learning_rate": 4.211632796075684e-06,
2690
+ "loss": 0.6302,
2691
+ "step": 4180
2692
+ },
2693
+ {
2694
+ "epoch": 2.91,
2695
+ "learning_rate": 4.1976173791170285e-06,
2696
+ "loss": 0.6088,
2697
+ "step": 4190
2698
+ },
2699
+ {
2700
+ "epoch": 2.92,
2701
+ "learning_rate": 4.183601962158374e-06,
2702
+ "loss": 0.6204,
2703
+ "step": 4200
2704
+ },
2705
+ {
2706
+ "epoch": 2.92,
2707
+ "eval_custom_hf_pref_accuracy": 0.6205,
2708
+ "eval_custom_hf_pref_kendalltau": 0.241,
2709
+ "eval_custom_hf_pref_loss": 0.5986875295639038,
2710
+ "eval_custom_hf_pref_neg_score": -0.47119298577308655,
2711
+ "eval_custom_hf_pref_pos_score": -0.10675738751888275,
2712
+ "eval_custom_hf_pref_runtime": 3.1679,
2713
+ "eval_custom_hf_pref_samples_per_second": 631.326,
2714
+ "eval_custom_hf_pref_score_diff": 0.3644356429576874,
2715
+ "eval_custom_hf_pref_steps_per_second": 78.916,
2716
+ "step": 4200
2717
+ },
2718
+ {
2719
+ "epoch": 2.93,
2720
+ "learning_rate": 4.169586545199721e-06,
2721
+ "loss": 0.5967,
2722
+ "step": 4210
2723
+ },
2724
+ {
2725
+ "epoch": 2.94,
2726
+ "learning_rate": 4.1555711282410655e-06,
2727
+ "loss": 0.5978,
2728
+ "step": 4220
2729
+ },
2730
+ {
2731
+ "epoch": 2.94,
2732
+ "learning_rate": 4.141555711282411e-06,
2733
+ "loss": 0.6013,
2734
+ "step": 4230
2735
+ },
2736
+ {
2737
+ "epoch": 2.95,
2738
+ "learning_rate": 4.127540294323756e-06,
2739
+ "loss": 0.6363,
2740
+ "step": 4240
2741
+ },
2742
+ {
2743
+ "epoch": 2.96,
2744
+ "learning_rate": 4.113524877365102e-06,
2745
+ "loss": 0.597,
2746
+ "step": 4250
2747
+ },
2748
+ {
2749
+ "epoch": 2.96,
2750
+ "learning_rate": 4.099509460406447e-06,
2751
+ "loss": 0.5893,
2752
+ "step": 4260
2753
+ },
2754
+ {
2755
+ "epoch": 2.97,
2756
+ "learning_rate": 4.085494043447793e-06,
2757
+ "loss": 0.643,
2758
+ "step": 4270
2759
+ },
2760
+ {
2761
+ "epoch": 2.98,
2762
+ "learning_rate": 4.071478626489138e-06,
2763
+ "loss": 0.5555,
2764
+ "step": 4280
2765
+ },
2766
+ {
2767
+ "epoch": 2.98,
2768
+ "learning_rate": 4.0574632095304836e-06,
2769
+ "loss": 0.6336,
2770
+ "step": 4290
2771
+ },
2772
+ {
2773
+ "epoch": 2.99,
2774
+ "learning_rate": 4.043447792571829e-06,
2775
+ "loss": 0.5994,
2776
+ "step": 4300
2777
+ },
2778
+ {
2779
+ "epoch": 3.0,
2780
+ "learning_rate": 4.029432375613175e-06,
2781
+ "loss": 0.5717,
2782
+ "step": 4310
2783
+ },
2784
+ {
2785
+ "epoch": 3.01,
2786
+ "learning_rate": 4.015416958654521e-06,
2787
+ "loss": 0.6058,
2788
+ "step": 4320
2789
+ },
2790
+ {
2791
+ "epoch": 3.01,
2792
+ "learning_rate": 4.001401541695865e-06,
2793
+ "loss": 0.6151,
2794
+ "step": 4330
2795
+ },
2796
+ {
2797
+ "epoch": 3.02,
2798
+ "learning_rate": 3.987386124737211e-06,
2799
+ "loss": 0.5971,
2800
+ "step": 4340
2801
+ },
2802
+ {
2803
+ "epoch": 3.03,
2804
+ "learning_rate": 3.973370707778557e-06,
2805
+ "loss": 0.6147,
2806
+ "step": 4350
2807
+ },
2808
+ {
2809
+ "epoch": 3.03,
2810
+ "learning_rate": 3.9593552908199024e-06,
2811
+ "loss": 0.5664,
2812
+ "step": 4360
2813
+ },
2814
+ {
2815
+ "epoch": 3.04,
2816
+ "learning_rate": 3.945339873861247e-06,
2817
+ "loss": 0.609,
2818
+ "step": 4370
2819
+ },
2820
+ {
2821
+ "epoch": 3.05,
2822
+ "learning_rate": 3.931324456902593e-06,
2823
+ "loss": 0.6032,
2824
+ "step": 4380
2825
+ },
2826
+ {
2827
+ "epoch": 3.05,
2828
+ "learning_rate": 3.917309039943939e-06,
2829
+ "loss": 0.6202,
2830
+ "step": 4390
2831
+ },
2832
+ {
2833
+ "epoch": 3.06,
2834
+ "learning_rate": 3.903293622985284e-06,
2835
+ "loss": 0.5965,
2836
+ "step": 4400
2837
+ },
2838
+ {
2839
+ "epoch": 3.07,
2840
+ "learning_rate": 3.88927820602663e-06,
2841
+ "loss": 0.6127,
2842
+ "step": 4410
2843
+ },
2844
+ {
2845
+ "epoch": 3.07,
2846
+ "learning_rate": 3.875262789067975e-06,
2847
+ "loss": 0.6345,
2848
+ "step": 4420
2849
+ },
2850
+ {
2851
+ "epoch": 3.08,
2852
+ "learning_rate": 3.8612473721093205e-06,
2853
+ "loss": 0.629,
2854
+ "step": 4430
2855
+ },
2856
+ {
2857
+ "epoch": 3.09,
2858
+ "learning_rate": 3.847231955150666e-06,
2859
+ "loss": 0.6408,
2860
+ "step": 4440
2861
+ },
2862
+ {
2863
+ "epoch": 3.1,
2864
+ "learning_rate": 3.833216538192012e-06,
2865
+ "loss": 0.584,
2866
+ "step": 4450
2867
+ },
2868
+ {
2869
+ "epoch": 3.1,
2870
+ "learning_rate": 3.819201121233357e-06,
2871
+ "loss": 0.604,
2872
+ "step": 4460
2873
+ },
2874
+ {
2875
+ "epoch": 3.11,
2876
+ "learning_rate": 3.8051857042747028e-06,
2877
+ "loss": 0.5901,
2878
+ "step": 4470
2879
+ },
2880
+ {
2881
+ "epoch": 3.12,
2882
+ "learning_rate": 3.7911702873160476e-06,
2883
+ "loss": 0.5901,
2884
+ "step": 4480
2885
+ },
2886
+ {
2887
+ "epoch": 3.12,
2888
+ "learning_rate": 3.7771548703573933e-06,
2889
+ "loss": 0.5902,
2890
+ "step": 4490
2891
+ },
2892
+ {
2893
+ "epoch": 3.13,
2894
+ "learning_rate": 3.7631394533987385e-06,
2895
+ "loss": 0.6193,
2896
+ "step": 4500
2897
+ },
2898
+ {
2899
+ "epoch": 3.13,
2900
+ "eval_custom_hf_pref_accuracy": 0.6265,
2901
+ "eval_custom_hf_pref_kendalltau": 0.253,
2902
+ "eval_custom_hf_pref_loss": 0.5968906283378601,
2903
+ "eval_custom_hf_pref_neg_score": -0.27363714575767517,
2904
+ "eval_custom_hf_pref_pos_score": 0.12795554101467133,
2905
+ "eval_custom_hf_pref_runtime": 3.1482,
2906
+ "eval_custom_hf_pref_samples_per_second": 635.277,
2907
+ "eval_custom_hf_pref_score_diff": 0.4015926420688629,
2908
+ "eval_custom_hf_pref_steps_per_second": 79.41,
2909
+ "step": 4500
2910
+ },
2911
+ {
2912
+ "epoch": 3.14,
2913
+ "learning_rate": 3.749124036440084e-06,
2914
+ "loss": 0.6134,
2915
+ "step": 4510
2916
+ },
2917
+ {
2918
+ "epoch": 3.14,
2919
+ "learning_rate": 3.7351086194814303e-06,
2920
+ "loss": 0.6028,
2921
+ "step": 4520
2922
+ },
2923
+ {
2924
+ "epoch": 3.15,
2925
+ "learning_rate": 3.721093202522775e-06,
2926
+ "loss": 0.6098,
2927
+ "step": 4530
2928
+ },
2929
+ {
2930
+ "epoch": 3.16,
2931
+ "learning_rate": 3.707077785564121e-06,
2932
+ "loss": 0.6323,
2933
+ "step": 4540
2934
+ },
2935
+ {
2936
+ "epoch": 3.17,
2937
+ "learning_rate": 3.693062368605466e-06,
2938
+ "loss": 0.5975,
2939
+ "step": 4550
2940
+ },
2941
+ {
2942
+ "epoch": 3.17,
2943
+ "learning_rate": 3.6790469516468117e-06,
2944
+ "loss": 0.59,
2945
+ "step": 4560
2946
+ },
2947
+ {
2948
+ "epoch": 3.18,
2949
+ "learning_rate": 3.665031534688157e-06,
2950
+ "loss": 0.5724,
2951
+ "step": 4570
2952
+ },
2953
+ {
2954
+ "epoch": 3.19,
2955
+ "learning_rate": 3.6510161177295027e-06,
2956
+ "loss": 0.6176,
2957
+ "step": 4580
2958
+ },
2959
+ {
2960
+ "epoch": 3.19,
2961
+ "learning_rate": 3.637000700770848e-06,
2962
+ "loss": 0.6124,
2963
+ "step": 4590
2964
+ },
2965
+ {
2966
+ "epoch": 3.2,
2967
+ "learning_rate": 3.6229852838121936e-06,
2968
+ "loss": 0.5846,
2969
+ "step": 4600
2970
+ },
2971
+ {
2972
+ "epoch": 3.21,
2973
+ "learning_rate": 3.6089698668535393e-06,
2974
+ "loss": 0.596,
2975
+ "step": 4610
2976
+ },
2977
+ {
2978
+ "epoch": 3.21,
2979
+ "learning_rate": 3.5949544498948845e-06,
2980
+ "loss": 0.6126,
2981
+ "step": 4620
2982
+ },
2983
+ {
2984
+ "epoch": 3.22,
2985
+ "learning_rate": 3.58093903293623e-06,
2986
+ "loss": 0.5881,
2987
+ "step": 4630
2988
+ },
2989
+ {
2990
+ "epoch": 3.23,
2991
+ "learning_rate": 3.5669236159775754e-06,
2992
+ "loss": 0.6052,
2993
+ "step": 4640
2994
+ },
2995
+ {
2996
+ "epoch": 3.23,
2997
+ "learning_rate": 3.552908199018921e-06,
2998
+ "loss": 0.6158,
2999
+ "step": 4650
3000
+ },
3001
+ {
3002
+ "epoch": 3.24,
3003
+ "learning_rate": 3.5388927820602664e-06,
3004
+ "loss": 0.5856,
3005
+ "step": 4660
3006
+ },
3007
+ {
3008
+ "epoch": 3.25,
3009
+ "learning_rate": 3.524877365101612e-06,
3010
+ "loss": 0.6291,
3011
+ "step": 4670
3012
+ },
3013
+ {
3014
+ "epoch": 3.26,
3015
+ "learning_rate": 3.5108619481429573e-06,
3016
+ "loss": 0.6234,
3017
+ "step": 4680
3018
+ },
3019
+ {
3020
+ "epoch": 3.26,
3021
+ "learning_rate": 3.496846531184303e-06,
3022
+ "loss": 0.5928,
3023
+ "step": 4690
3024
+ },
3025
+ {
3026
+ "epoch": 3.27,
3027
+ "learning_rate": 3.4828311142256482e-06,
3028
+ "loss": 0.5998,
3029
+ "step": 4700
3030
+ },
3031
+ {
3032
+ "epoch": 3.28,
3033
+ "learning_rate": 3.468815697266994e-06,
3034
+ "loss": 0.601,
3035
+ "step": 4710
3036
+ },
3037
+ {
3038
+ "epoch": 3.28,
3039
+ "learning_rate": 3.4548002803083396e-06,
3040
+ "loss": 0.6353,
3041
+ "step": 4720
3042
+ },
3043
+ {
3044
+ "epoch": 3.29,
3045
+ "learning_rate": 3.440784863349685e-06,
3046
+ "loss": 0.6226,
3047
+ "step": 4730
3048
+ },
3049
+ {
3050
+ "epoch": 3.3,
3051
+ "learning_rate": 3.4267694463910305e-06,
3052
+ "loss": 0.5706,
3053
+ "step": 4740
3054
+ },
3055
+ {
3056
+ "epoch": 3.3,
3057
+ "learning_rate": 3.4127540294323758e-06,
3058
+ "loss": 0.5953,
3059
+ "step": 4750
3060
+ },
3061
+ {
3062
+ "epoch": 3.31,
3063
+ "learning_rate": 3.3987386124737214e-06,
3064
+ "loss": 0.5982,
3065
+ "step": 4760
3066
+ },
3067
+ {
3068
+ "epoch": 3.32,
3069
+ "learning_rate": 3.3847231955150667e-06,
3070
+ "loss": 0.5873,
3071
+ "step": 4770
3072
+ },
3073
+ {
3074
+ "epoch": 3.33,
3075
+ "learning_rate": 3.3707077785564124e-06,
3076
+ "loss": 0.609,
3077
+ "step": 4780
3078
+ },
3079
+ {
3080
+ "epoch": 3.33,
3081
+ "learning_rate": 3.3566923615977576e-06,
3082
+ "loss": 0.6384,
3083
+ "step": 4790
3084
+ },
3085
+ {
3086
+ "epoch": 3.34,
3087
+ "learning_rate": 3.3426769446391033e-06,
3088
+ "loss": 0.5835,
3089
+ "step": 4800
3090
+ },
3091
+ {
3092
+ "epoch": 3.34,
3093
+ "eval_custom_hf_pref_accuracy": 0.614,
3094
+ "eval_custom_hf_pref_kendalltau": 0.228,
3095
+ "eval_custom_hf_pref_loss": 0.6002968549728394,
3096
+ "eval_custom_hf_pref_neg_score": -0.48366308212280273,
3097
+ "eval_custom_hf_pref_pos_score": -0.10953208804130554,
3098
+ "eval_custom_hf_pref_runtime": 3.1498,
3099
+ "eval_custom_hf_pref_samples_per_second": 634.964,
3100
+ "eval_custom_hf_pref_score_diff": 0.3741309940814972,
3101
+ "eval_custom_hf_pref_steps_per_second": 79.371,
3102
+ "step": 4800
3103
+ },
3104
+ {
3105
+ "epoch": 3.35,
3106
+ "learning_rate": 3.328661527680449e-06,
3107
+ "loss": 0.6107,
3108
+ "step": 4810
3109
+ },
3110
+ {
3111
+ "epoch": 3.35,
3112
+ "learning_rate": 3.3146461107217942e-06,
3113
+ "loss": 0.6198,
3114
+ "step": 4820
3115
+ },
3116
+ {
3117
+ "epoch": 3.36,
3118
+ "learning_rate": 3.30063069376314e-06,
3119
+ "loss": 0.6077,
3120
+ "step": 4830
3121
+ },
3122
+ {
3123
+ "epoch": 3.37,
3124
+ "learning_rate": 3.286615276804485e-06,
3125
+ "loss": 0.6009,
3126
+ "step": 4840
3127
+ },
3128
+ {
3129
+ "epoch": 3.37,
3130
+ "learning_rate": 3.272599859845831e-06,
3131
+ "loss": 0.627,
3132
+ "step": 4850
3133
+ },
3134
+ {
3135
+ "epoch": 3.38,
3136
+ "learning_rate": 3.258584442887176e-06,
3137
+ "loss": 0.5839,
3138
+ "step": 4860
3139
+ },
3140
+ {
3141
+ "epoch": 3.39,
3142
+ "learning_rate": 3.2445690259285218e-06,
3143
+ "loss": 0.6205,
3144
+ "step": 4870
3145
+ },
3146
+ {
3147
+ "epoch": 3.39,
3148
+ "learning_rate": 3.230553608969867e-06,
3149
+ "loss": 0.6206,
3150
+ "step": 4880
3151
+ },
3152
+ {
3153
+ "epoch": 3.4,
3154
+ "learning_rate": 3.2165381920112127e-06,
3155
+ "loss": 0.6411,
3156
+ "step": 4890
3157
+ },
3158
+ {
3159
+ "epoch": 3.41,
3160
+ "learning_rate": 3.202522775052558e-06,
3161
+ "loss": 0.6231,
3162
+ "step": 4900
3163
+ },
3164
+ {
3165
+ "epoch": 3.42,
3166
+ "learning_rate": 3.1885073580939036e-06,
3167
+ "loss": 0.6324,
3168
+ "step": 4910
3169
+ },
3170
+ {
3171
+ "epoch": 3.42,
3172
+ "learning_rate": 3.1744919411352493e-06,
3173
+ "loss": 0.6094,
3174
+ "step": 4920
3175
+ },
3176
+ {
3177
+ "epoch": 3.43,
3178
+ "learning_rate": 3.1604765241765945e-06,
3179
+ "loss": 0.5963,
3180
+ "step": 4930
3181
+ },
3182
+ {
3183
+ "epoch": 3.44,
3184
+ "learning_rate": 3.1464611072179402e-06,
3185
+ "loss": 0.6245,
3186
+ "step": 4940
3187
+ },
3188
+ {
3189
+ "epoch": 3.44,
3190
+ "learning_rate": 3.1324456902592855e-06,
3191
+ "loss": 0.602,
3192
+ "step": 4950
3193
+ },
3194
+ {
3195
+ "epoch": 3.45,
3196
+ "learning_rate": 3.118430273300631e-06,
3197
+ "loss": 0.5815,
3198
+ "step": 4960
3199
+ },
3200
+ {
3201
+ "epoch": 3.46,
3202
+ "learning_rate": 3.1044148563419764e-06,
3203
+ "loss": 0.5761,
3204
+ "step": 4970
3205
+ },
3206
+ {
3207
+ "epoch": 3.46,
3208
+ "learning_rate": 3.090399439383322e-06,
3209
+ "loss": 0.6007,
3210
+ "step": 4980
3211
+ },
3212
+ {
3213
+ "epoch": 3.47,
3214
+ "learning_rate": 3.0763840224246673e-06,
3215
+ "loss": 0.6386,
3216
+ "step": 4990
3217
+ },
3218
+ {
3219
+ "epoch": 3.48,
3220
+ "learning_rate": 3.062368605466013e-06,
3221
+ "loss": 0.6073,
3222
+ "step": 5000
3223
+ },
3224
+ {
3225
+ "epoch": 3.49,
3226
+ "learning_rate": 3.0483531885073587e-06,
3227
+ "loss": 0.6297,
3228
+ "step": 5010
3229
+ },
3230
+ {
3231
+ "epoch": 3.49,
3232
+ "learning_rate": 3.034337771548704e-06,
3233
+ "loss": 0.6336,
3234
+ "step": 5020
3235
+ },
3236
+ {
3237
+ "epoch": 3.5,
3238
+ "learning_rate": 3.0203223545900496e-06,
3239
+ "loss": 0.5885,
3240
+ "step": 5030
3241
+ },
3242
+ {
3243
+ "epoch": 3.51,
3244
+ "learning_rate": 3.006306937631395e-06,
3245
+ "loss": 0.6201,
3246
+ "step": 5040
3247
+ },
3248
+ {
3249
+ "epoch": 3.51,
3250
+ "learning_rate": 2.9922915206727405e-06,
3251
+ "loss": 0.6233,
3252
+ "step": 5050
3253
+ },
3254
+ {
3255
+ "epoch": 3.52,
3256
+ "learning_rate": 2.978276103714086e-06,
3257
+ "loss": 0.6065,
3258
+ "step": 5060
3259
+ },
3260
+ {
3261
+ "epoch": 3.53,
3262
+ "learning_rate": 2.9642606867554315e-06,
3263
+ "loss": 0.5982,
3264
+ "step": 5070
3265
+ },
3266
+ {
3267
+ "epoch": 3.53,
3268
+ "learning_rate": 2.9502452697967763e-06,
3269
+ "loss": 0.601,
3270
+ "step": 5080
3271
+ },
3272
+ {
3273
+ "epoch": 3.54,
3274
+ "learning_rate": 2.9362298528381224e-06,
3275
+ "loss": 0.5765,
3276
+ "step": 5090
3277
+ },
3278
+ {
3279
+ "epoch": 3.55,
3280
+ "learning_rate": 2.9222144358794672e-06,
3281
+ "loss": 0.5644,
3282
+ "step": 5100
3283
+ },
3284
+ {
3285
+ "epoch": 3.55,
3286
+ "eval_custom_hf_pref_accuracy": 0.6245,
3287
+ "eval_custom_hf_pref_kendalltau": 0.249,
3288
+ "eval_custom_hf_pref_loss": 0.5992500185966492,
3289
+ "eval_custom_hf_pref_neg_score": -0.4819223880767822,
3290
+ "eval_custom_hf_pref_pos_score": -0.1118302047252655,
3291
+ "eval_custom_hf_pref_runtime": 3.1202,
3292
+ "eval_custom_hf_pref_samples_per_second": 640.985,
3293
+ "eval_custom_hf_pref_score_diff": 0.3700921833515167,
3294
+ "eval_custom_hf_pref_steps_per_second": 80.123,
3295
+ "step": 5100
3296
+ },
3297
+ {
3298
+ "epoch": 3.55,
3299
+ "learning_rate": 2.9081990189208133e-06,
3300
+ "loss": 0.6288,
3301
+ "step": 5110
3302
+ },
3303
+ {
3304
+ "epoch": 3.56,
3305
+ "learning_rate": 2.894183601962159e-06,
3306
+ "loss": 0.5764,
3307
+ "step": 5120
3308
+ },
3309
+ {
3310
+ "epoch": 3.57,
3311
+ "learning_rate": 2.880168185003504e-06,
3312
+ "loss": 0.6127,
3313
+ "step": 5130
3314
+ },
3315
+ {
3316
+ "epoch": 3.58,
3317
+ "learning_rate": 2.86615276804485e-06,
3318
+ "loss": 0.6132,
3319
+ "step": 5140
3320
+ },
3321
+ {
3322
+ "epoch": 3.58,
3323
+ "learning_rate": 2.8521373510861948e-06,
3324
+ "loss": 0.6161,
3325
+ "step": 5150
3326
+ },
3327
+ {
3328
+ "epoch": 3.59,
3329
+ "learning_rate": 2.838121934127541e-06,
3330
+ "loss": 0.5842,
3331
+ "step": 5160
3332
+ },
3333
+ {
3334
+ "epoch": 3.6,
3335
+ "learning_rate": 2.8241065171688857e-06,
3336
+ "loss": 0.5721,
3337
+ "step": 5170
3338
+ },
3339
+ {
3340
+ "epoch": 3.6,
3341
+ "learning_rate": 2.8100911002102314e-06,
3342
+ "loss": 0.5999,
3343
+ "step": 5180
3344
+ },
3345
+ {
3346
+ "epoch": 3.61,
3347
+ "learning_rate": 2.7960756832515766e-06,
3348
+ "loss": 0.6511,
3349
+ "step": 5190
3350
+ },
3351
+ {
3352
+ "epoch": 3.62,
3353
+ "learning_rate": 2.7820602662929223e-06,
3354
+ "loss": 0.5939,
3355
+ "step": 5200
3356
+ },
3357
+ {
3358
+ "epoch": 3.62,
3359
+ "learning_rate": 2.7680448493342675e-06,
3360
+ "loss": 0.6223,
3361
+ "step": 5210
3362
+ },
3363
+ {
3364
+ "epoch": 3.63,
3365
+ "learning_rate": 2.7540294323756132e-06,
3366
+ "loss": 0.6149,
3367
+ "step": 5220
3368
+ },
3369
+ {
3370
+ "epoch": 3.64,
3371
+ "learning_rate": 2.7400140154169593e-06,
3372
+ "loss": 0.5749,
3373
+ "step": 5230
3374
+ },
3375
+ {
3376
+ "epoch": 3.65,
3377
+ "learning_rate": 2.725998598458304e-06,
3378
+ "loss": 0.5945,
3379
+ "step": 5240
3380
+ },
3381
+ {
3382
+ "epoch": 3.65,
3383
+ "learning_rate": 2.71198318149965e-06,
3384
+ "loss": 0.5699,
3385
+ "step": 5250
3386
+ },
3387
+ {
3388
+ "epoch": 3.66,
3389
+ "learning_rate": 2.697967764540995e-06,
3390
+ "loss": 0.6141,
3391
+ "step": 5260
3392
+ },
3393
+ {
3394
+ "epoch": 3.67,
3395
+ "learning_rate": 2.6839523475823407e-06,
3396
+ "loss": 0.5843,
3397
+ "step": 5270
3398
+ },
3399
+ {
3400
+ "epoch": 3.67,
3401
+ "learning_rate": 2.669936930623686e-06,
3402
+ "loss": 0.6239,
3403
+ "step": 5280
3404
+ },
3405
+ {
3406
+ "epoch": 3.68,
3407
+ "learning_rate": 2.6559215136650317e-06,
3408
+ "loss": 0.6492,
3409
+ "step": 5290
3410
+ },
3411
+ {
3412
+ "epoch": 3.69,
3413
+ "learning_rate": 2.641906096706377e-06,
3414
+ "loss": 0.6047,
3415
+ "step": 5300
3416
+ },
3417
+ {
3418
+ "epoch": 3.69,
3419
+ "learning_rate": 2.6278906797477226e-06,
3420
+ "loss": 0.6219,
3421
+ "step": 5310
3422
+ },
3423
+ {
3424
+ "epoch": 3.7,
3425
+ "learning_rate": 2.6138752627890683e-06,
3426
+ "loss": 0.5969,
3427
+ "step": 5320
3428
+ },
3429
+ {
3430
+ "epoch": 3.71,
3431
+ "learning_rate": 2.5998598458304135e-06,
3432
+ "loss": 0.5605,
3433
+ "step": 5330
3434
+ },
3435
+ {
3436
+ "epoch": 3.71,
3437
+ "learning_rate": 2.585844428871759e-06,
3438
+ "loss": 0.6063,
3439
+ "step": 5340
3440
+ },
3441
+ {
3442
+ "epoch": 3.72,
3443
+ "learning_rate": 2.5718290119131045e-06,
3444
+ "loss": 0.6111,
3445
+ "step": 5350
3446
+ },
3447
+ {
3448
+ "epoch": 3.73,
3449
+ "learning_rate": 2.55781359495445e-06,
3450
+ "loss": 0.6336,
3451
+ "step": 5360
3452
+ },
3453
+ {
3454
+ "epoch": 3.74,
3455
+ "learning_rate": 2.5437981779957954e-06,
3456
+ "loss": 0.6213,
3457
+ "step": 5370
3458
+ },
3459
+ {
3460
+ "epoch": 3.74,
3461
+ "learning_rate": 2.529782761037141e-06,
3462
+ "loss": 0.5939,
3463
+ "step": 5380
3464
+ },
3465
+ {
3466
+ "epoch": 3.75,
3467
+ "learning_rate": 2.5157673440784863e-06,
3468
+ "loss": 0.5887,
3469
+ "step": 5390
3470
+ },
3471
+ {
3472
+ "epoch": 3.76,
3473
+ "learning_rate": 2.501751927119832e-06,
3474
+ "loss": 0.6451,
3475
+ "step": 5400
3476
+ },
3477
+ {
3478
+ "epoch": 3.76,
3479
+ "eval_custom_hf_pref_accuracy": 0.6225,
3480
+ "eval_custom_hf_pref_kendalltau": 0.245,
3481
+ "eval_custom_hf_pref_loss": 0.5965859293937683,
3482
+ "eval_custom_hf_pref_neg_score": -0.40171003341674805,
3483
+ "eval_custom_hf_pref_pos_score": -0.02441699430346489,
3484
+ "eval_custom_hf_pref_runtime": 3.1674,
3485
+ "eval_custom_hf_pref_samples_per_second": 631.43,
3486
+ "eval_custom_hf_pref_score_diff": 0.37729302048683167,
3487
+ "eval_custom_hf_pref_steps_per_second": 78.929,
3488
+ "step": 5400
3489
+ },
3490
+ {
3491
+ "epoch": 3.76,
3492
+ "learning_rate": 2.4877365101611777e-06,
3493
+ "loss": 0.592,
3494
+ "step": 5410
3495
+ },
3496
+ {
3497
+ "epoch": 3.77,
3498
+ "learning_rate": 2.473721093202523e-06,
3499
+ "loss": 0.5941,
3500
+ "step": 5420
3501
+ },
3502
+ {
3503
+ "epoch": 3.78,
3504
+ "learning_rate": 2.4597056762438686e-06,
3505
+ "loss": 0.5547,
3506
+ "step": 5430
3507
+ },
3508
+ {
3509
+ "epoch": 3.78,
3510
+ "learning_rate": 2.445690259285214e-06,
3511
+ "loss": 0.6006,
3512
+ "step": 5440
3513
+ },
3514
+ {
3515
+ "epoch": 3.79,
3516
+ "learning_rate": 2.431674842326559e-06,
3517
+ "loss": 0.6379,
3518
+ "step": 5450
3519
+ },
3520
+ {
3521
+ "epoch": 3.8,
3522
+ "learning_rate": 2.4176594253679048e-06,
3523
+ "loss": 0.6053,
3524
+ "step": 5460
3525
+ },
3526
+ {
3527
+ "epoch": 3.81,
3528
+ "learning_rate": 2.4036440084092505e-06,
3529
+ "loss": 0.6081,
3530
+ "step": 5470
3531
+ },
3532
+ {
3533
+ "epoch": 3.81,
3534
+ "learning_rate": 2.389628591450596e-06,
3535
+ "loss": 0.625,
3536
+ "step": 5480
3537
+ },
3538
+ {
3539
+ "epoch": 3.82,
3540
+ "learning_rate": 2.3756131744919414e-06,
3541
+ "loss": 0.5974,
3542
+ "step": 5490
3543
+ },
3544
+ {
3545
+ "epoch": 3.83,
3546
+ "learning_rate": 2.361597757533287e-06,
3547
+ "loss": 0.6095,
3548
+ "step": 5500
3549
+ },
3550
+ {
3551
+ "epoch": 3.83,
3552
+ "learning_rate": 2.3475823405746323e-06,
3553
+ "loss": 0.6189,
3554
+ "step": 5510
3555
+ },
3556
+ {
3557
+ "epoch": 3.84,
3558
+ "learning_rate": 2.3335669236159776e-06,
3559
+ "loss": 0.6212,
3560
+ "step": 5520
3561
+ },
3562
+ {
3563
+ "epoch": 3.85,
3564
+ "learning_rate": 2.3195515066573232e-06,
3565
+ "loss": 0.599,
3566
+ "step": 5530
3567
+ },
3568
+ {
3569
+ "epoch": 3.85,
3570
+ "learning_rate": 2.3055360896986685e-06,
3571
+ "loss": 0.6205,
3572
+ "step": 5540
3573
+ },
3574
+ {
3575
+ "epoch": 3.86,
3576
+ "learning_rate": 2.291520672740014e-06,
3577
+ "loss": 0.5884,
3578
+ "step": 5550
3579
+ },
3580
+ {
3581
+ "epoch": 3.87,
3582
+ "learning_rate": 2.2775052557813594e-06,
3583
+ "loss": 0.6034,
3584
+ "step": 5560
3585
+ },
3586
+ {
3587
+ "epoch": 3.87,
3588
+ "learning_rate": 2.263489838822705e-06,
3589
+ "loss": 0.5731,
3590
+ "step": 5570
3591
+ },
3592
+ {
3593
+ "epoch": 3.88,
3594
+ "learning_rate": 2.2494744218640508e-06,
3595
+ "loss": 0.5968,
3596
+ "step": 5580
3597
+ },
3598
+ {
3599
+ "epoch": 3.89,
3600
+ "learning_rate": 2.235459004905396e-06,
3601
+ "loss": 0.5856,
3602
+ "step": 5590
3603
+ },
3604
+ {
3605
+ "epoch": 3.9,
3606
+ "learning_rate": 2.2214435879467417e-06,
3607
+ "loss": 0.6373,
3608
+ "step": 5600
3609
+ },
3610
+ {
3611
+ "epoch": 3.9,
3612
+ "learning_rate": 2.207428170988087e-06,
3613
+ "loss": 0.6293,
3614
+ "step": 5610
3615
+ },
3616
+ {
3617
+ "epoch": 3.91,
3618
+ "learning_rate": 2.1934127540294326e-06,
3619
+ "loss": 0.5965,
3620
+ "step": 5620
3621
+ },
3622
+ {
3623
+ "epoch": 3.92,
3624
+ "learning_rate": 2.179397337070778e-06,
3625
+ "loss": 0.6163,
3626
+ "step": 5630
3627
+ },
3628
+ {
3629
+ "epoch": 3.92,
3630
+ "learning_rate": 2.1653819201121236e-06,
3631
+ "loss": 0.6009,
3632
+ "step": 5640
3633
+ },
3634
+ {
3635
+ "epoch": 3.93,
3636
+ "learning_rate": 2.151366503153469e-06,
3637
+ "loss": 0.5862,
3638
+ "step": 5650
3639
+ },
3640
+ {
3641
+ "epoch": 3.94,
3642
+ "learning_rate": 2.1373510861948145e-06,
3643
+ "loss": 0.5967,
3644
+ "step": 5660
3645
+ },
3646
+ {
3647
+ "epoch": 3.94,
3648
+ "learning_rate": 2.12333566923616e-06,
3649
+ "loss": 0.6222,
3650
+ "step": 5670
3651
+ },
3652
+ {
3653
+ "epoch": 3.95,
3654
+ "learning_rate": 2.1093202522775054e-06,
3655
+ "loss": 0.6039,
3656
+ "step": 5680
3657
+ },
3658
+ {
3659
+ "epoch": 3.96,
3660
+ "learning_rate": 2.095304835318851e-06,
3661
+ "loss": 0.6157,
3662
+ "step": 5690
3663
+ },
3664
+ {
3665
+ "epoch": 3.97,
3666
+ "learning_rate": 2.0812894183601963e-06,
3667
+ "loss": 0.6303,
3668
+ "step": 5700
3669
+ },
3670
+ {
3671
+ "epoch": 3.97,
3672
+ "eval_custom_hf_pref_accuracy": 0.6255,
3673
+ "eval_custom_hf_pref_kendalltau": 0.251,
3674
+ "eval_custom_hf_pref_loss": 0.5979297161102295,
3675
+ "eval_custom_hf_pref_neg_score": -0.39338263869285583,
3676
+ "eval_custom_hf_pref_pos_score": -0.022781679406762123,
3677
+ "eval_custom_hf_pref_runtime": 3.1387,
3678
+ "eval_custom_hf_pref_samples_per_second": 637.201,
3679
+ "eval_custom_hf_pref_score_diff": 0.37060093879699707,
3680
+ "eval_custom_hf_pref_steps_per_second": 79.65,
3681
+ "step": 5700
3682
+ },
3683
+ {
3684
+ "epoch": 3.97,
3685
+ "learning_rate": 2.067274001401542e-06,
3686
+ "loss": 0.6195,
3687
+ "step": 5710
3688
+ },
3689
+ {
3690
+ "epoch": 3.98,
3691
+ "learning_rate": 2.0532585844428873e-06,
3692
+ "loss": 0.612,
3693
+ "step": 5720
3694
+ },
3695
+ {
3696
+ "epoch": 3.99,
3697
+ "learning_rate": 2.039243167484233e-06,
3698
+ "loss": 0.602,
3699
+ "step": 5730
3700
+ },
3701
+ {
3702
+ "epoch": 3.99,
3703
+ "learning_rate": 2.025227750525578e-06,
3704
+ "loss": 0.6154,
3705
+ "step": 5740
3706
+ },
3707
+ {
3708
+ "epoch": 4.0,
3709
+ "learning_rate": 2.011212333566924e-06,
3710
+ "loss": 0.5905,
3711
+ "step": 5750
3712
+ },
3713
+ {
3714
+ "epoch": 4.01,
3715
+ "learning_rate": 1.997196916608269e-06,
3716
+ "loss": 0.5967,
3717
+ "step": 5760
3718
+ },
3719
+ {
3720
+ "epoch": 4.01,
3721
+ "learning_rate": 1.983181499649615e-06,
3722
+ "loss": 0.5984,
3723
+ "step": 5770
3724
+ },
3725
+ {
3726
+ "epoch": 4.02,
3727
+ "learning_rate": 1.9691660826909605e-06,
3728
+ "loss": 0.6236,
3729
+ "step": 5780
3730
+ },
3731
+ {
3732
+ "epoch": 4.03,
3733
+ "learning_rate": 1.9551506657323057e-06,
3734
+ "loss": 0.6059,
3735
+ "step": 5790
3736
+ },
3737
+ {
3738
+ "epoch": 4.03,
3739
+ "learning_rate": 1.9411352487736514e-06,
3740
+ "loss": 0.6048,
3741
+ "step": 5800
3742
+ },
3743
+ {
3744
+ "epoch": 4.04,
3745
+ "learning_rate": 1.9271198318149967e-06,
3746
+ "loss": 0.6069,
3747
+ "step": 5810
3748
+ },
3749
+ {
3750
+ "epoch": 4.05,
3751
+ "learning_rate": 1.9131044148563423e-06,
3752
+ "loss": 0.572,
3753
+ "step": 5820
3754
+ },
3755
+ {
3756
+ "epoch": 4.06,
3757
+ "learning_rate": 1.8990889978976876e-06,
3758
+ "loss": 0.6238,
3759
+ "step": 5830
3760
+ },
3761
+ {
3762
+ "epoch": 4.06,
3763
+ "learning_rate": 1.885073580939033e-06,
3764
+ "loss": 0.6091,
3765
+ "step": 5840
3766
+ },
3767
+ {
3768
+ "epoch": 4.07,
3769
+ "learning_rate": 1.8710581639803785e-06,
3770
+ "loss": 0.605,
3771
+ "step": 5850
3772
+ },
3773
+ {
3774
+ "epoch": 4.08,
3775
+ "learning_rate": 1.857042747021724e-06,
3776
+ "loss": 0.5973,
3777
+ "step": 5860
3778
+ },
3779
+ {
3780
+ "epoch": 4.08,
3781
+ "learning_rate": 1.8430273300630697e-06,
3782
+ "loss": 0.5946,
3783
+ "step": 5870
3784
+ },
3785
+ {
3786
+ "epoch": 4.09,
3787
+ "learning_rate": 1.8290119131044151e-06,
3788
+ "loss": 0.5831,
3789
+ "step": 5880
3790
+ },
3791
+ {
3792
+ "epoch": 4.1,
3793
+ "learning_rate": 1.8149964961457606e-06,
3794
+ "loss": 0.5973,
3795
+ "step": 5890
3796
+ },
3797
+ {
3798
+ "epoch": 4.1,
3799
+ "learning_rate": 1.800981079187106e-06,
3800
+ "loss": 0.6187,
3801
+ "step": 5900
3802
+ },
3803
+ {
3804
+ "epoch": 4.11,
3805
+ "learning_rate": 1.7869656622284515e-06,
3806
+ "loss": 0.5953,
3807
+ "step": 5910
3808
+ },
3809
+ {
3810
+ "epoch": 4.12,
3811
+ "learning_rate": 1.772950245269797e-06,
3812
+ "loss": 0.6321,
3813
+ "step": 5920
3814
+ },
3815
+ {
3816
+ "epoch": 4.13,
3817
+ "learning_rate": 1.7589348283111424e-06,
3818
+ "loss": 0.5647,
3819
+ "step": 5930
3820
+ },
3821
+ {
3822
+ "epoch": 4.13,
3823
+ "learning_rate": 1.744919411352488e-06,
3824
+ "loss": 0.6232,
3825
+ "step": 5940
3826
+ },
3827
+ {
3828
+ "epoch": 4.14,
3829
+ "learning_rate": 1.7309039943938334e-06,
3830
+ "loss": 0.6015,
3831
+ "step": 5950
3832
+ },
3833
+ {
3834
+ "epoch": 4.15,
3835
+ "learning_rate": 1.7168885774351786e-06,
3836
+ "loss": 0.6095,
3837
+ "step": 5960
3838
+ },
3839
+ {
3840
+ "epoch": 4.15,
3841
+ "learning_rate": 1.702873160476524e-06,
3842
+ "loss": 0.6167,
3843
+ "step": 5970
3844
+ },
3845
+ {
3846
+ "epoch": 4.16,
3847
+ "learning_rate": 1.68885774351787e-06,
3848
+ "loss": 0.6079,
3849
+ "step": 5980
3850
+ },
3851
+ {
3852
+ "epoch": 4.17,
3853
+ "learning_rate": 1.6748423265592154e-06,
3854
+ "loss": 0.605,
3855
+ "step": 5990
3856
+ },
3857
+ {
3858
+ "epoch": 4.17,
3859
+ "learning_rate": 1.660826909600561e-06,
3860
+ "loss": 0.5969,
3861
+ "step": 6000
3862
+ },
3863
+ {
3864
+ "epoch": 4.17,
3865
+ "eval_custom_hf_pref_accuracy": 0.6295,
3866
+ "eval_custom_hf_pref_kendalltau": 0.259,
3867
+ "eval_custom_hf_pref_loss": 0.5939843654632568,
3868
+ "eval_custom_hf_pref_neg_score": -0.4233732521533966,
3869
+ "eval_custom_hf_pref_pos_score": -0.03053688071668148,
3870
+ "eval_custom_hf_pref_runtime": 3.1342,
3871
+ "eval_custom_hf_pref_samples_per_second": 638.127,
3872
+ "eval_custom_hf_pref_score_diff": 0.3928363621234894,
3873
+ "eval_custom_hf_pref_steps_per_second": 79.766,
3874
+ "step": 6000
3875
+ }
3876
+ ],
3877
+ "max_steps": 7185,
3878
+ "num_train_epochs": 5,
3879
+ "total_flos": 0.0,
3880
+ "trial_name": null,
3881
+ "trial_params": null
3882
+ }
checkpoint-6000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15d17db566d266013af03a48da12b993aa2b485f122344645b19641417f7840d
3
+ size 4881
checkpoint-7000/config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "tlc4418/pythia_70m_sft",
3
+ "architectures": [
4
+ "GPTNeoXRewardModel"
5
+ ],
6
+ "bos_token_id": 0,
7
+ "classifier_dropout": 0.1,
8
+ "eos_token_id": 0,
9
+ "hidden_act": "gelu",
10
+ "hidden_size": 512,
11
+ "initializer_range": 0.02,
12
+ "intermediate_size": 2048,
13
+ "layer_norm_eps": 1e-05,
14
+ "max_position_embeddings": 2048,
15
+ "model_type": "gpt_neox_reward_model",
16
+ "num_attention_heads": 8,
17
+ "num_hidden_layers": 6,
18
+ "pooling": "mean",
19
+ "reward_head_activation": "gelu",
20
+ "reward_head_dropout": 0.0,
21
+ "reward_head_hidden_dim": null,
22
+ "reward_head_type": "layernorm_linear",
23
+ "rotary_emb_base": 10000,
24
+ "rotary_pct": 0.25,
25
+ "tie_word_embeddings": false,
26
+ "torch_dtype": "bfloat16",
27
+ "transformers_version": "4.29.2",
28
+ "use_cache": true,
29
+ "use_parallel_residual": true,
30
+ "vocab_size": 50288
31
+ }
checkpoint-7000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b75a07ce6b1abae5d6c2f3d59c1717ba18cb435a1663bcec99bd027f72a5752d
3
+ size 178704331
checkpoint-7000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d0f939c92b8a85ad10f9bdc62b83527664cb20cf08b86009fa92a558f5539b47
3
+ size 114527658
checkpoint-7000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1eb2960340f4bc2f80ae45dcc71cd5c2fe4019dace8500d0a09977f565d302a5
3
+ size 14709
checkpoint-7000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c8778a8a186ddd8df006a4c55dcec727f61f2d53fe6d71dfa7692d77e3bbf34
3
+ size 1465
checkpoint-7000/special_tokens_map.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<|prefix_end|>",
4
+ "<|assistant|>",
5
+ "<|prefix_begin|>",
6
+ "<|prompter|>",
7
+ "<|system|>"
8
+ ],
9
+ "bos_token": "<|endoftext|>",
10
+ "eos_token": "<|endoftext|>",
11
+ "pad_token": "<|padding|>",
12
+ "sep_token": "<|endoftext|>",
13
+ "unk_token": "<|endoftext|>"
14
+ }
checkpoint-7000/tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-7000/tokenizer_config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "bos_token": "<|endoftext|>",
4
+ "clean_up_tokenization_spaces": true,
5
+ "eos_token": "<|endoftext|>",
6
+ "model_max_length": 1000000000000000019884624838656,
7
+ "tokenizer_class": "GPTNeoXTokenizer",
8
+ "unk_token": "<|endoftext|>"
9
+ }
checkpoint-7000/trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-7000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15d17db566d266013af03a48da12b993aa2b485f122344645b19641417f7840d
3
+ size 4881
config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "tlc4418/pythia_70m_sft",
3
+ "architectures": [
4
+ "GPTNeoXRewardModel"
5
+ ],
6
+ "bos_token_id": 0,
7
+ "classifier_dropout": 0.1,
8
+ "eos_token_id": 0,
9
+ "hidden_act": "gelu",
10
+ "hidden_size": 512,
11
+ "initializer_range": 0.02,
12
+ "intermediate_size": 2048,
13
+ "layer_norm_eps": 1e-05,
14
+ "max_position_embeddings": 2048,
15
+ "mean": -0.1044921875,
16
+ "model_type": "gpt_neox_reward_model",
17
+ "num_attention_heads": 8,
18
+ "num_hidden_layers": 6,
19
+ "pooling": "mean",
20
+ "reward_head_activation": "gelu",
21
+ "reward_head_dropout": 0.0,
22
+ "reward_head_hidden_dim": null,
23
+ "reward_head_type": "layernorm_linear",
24
+ "rotary_emb_base": 10000,
25
+ "rotary_pct": 0.25,
26
+ "std": 1.1328125,
27
+ "tie_word_embeddings": false,
28
+ "torch_dtype": "bfloat16",
29
+ "transformers_version": "4.29.2",
30
+ "use_cache": true,
31
+ "use_parallel_residual": true,
32
+ "vocab_size": 50288,
33
+ "auto_map": {
34
+ "AutoConfig": "gpt_neox_reward_model.GPTNeoXRewardModelConfig",
35
+ "AutoModelForSequenceClassification": "gpt_neox_reward_model.GPTNeoXRewardModel"
36
+ }
37
+ }
gpt_neox_reward_model.py ADDED
@@ -0,0 +1,188 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from dataclasses import dataclass
2
+ from typing import Literal, Optional
3
+
4
+ import torch
5
+ import torch.nn as nn
6
+ from transformers import AutoConfig, AutoModelForSequenceClassification
7
+ from transformers.models.gpt_neox.modeling_gpt_neox import GPTNeoXConfig, GPTNeoXModel, GPTNeoXPreTrainedModel
8
+ from transformers.utils import ModelOutput
9
+
10
+
11
+ def _get_activation(name: str) -> nn.Module:
12
+ activations = {
13
+ "gelu": nn.GELU,
14
+ "relu": nn.ReLU,
15
+ "silu": nn.SiLU,
16
+ "tanh": nn.Tanh,
17
+ }
18
+ if name not in activations:
19
+ raise ValueError(f"Unsupported reward-head activation: {name}")
20
+ return activations[name]()
21
+
22
+
23
+ class LayerNormLinearRewardHead(nn.Module):
24
+ def __init__(self, hidden_size: int):
25
+ super().__init__()
26
+ self.norm = nn.LayerNorm(hidden_size)
27
+ self.proj = nn.Linear(hidden_size, 1)
28
+
29
+ def forward(self, pooled: torch.FloatTensor) -> torch.FloatTensor:
30
+ return self.proj(self.norm(pooled))
31
+
32
+
33
+ class MLPRewardHead(nn.Module):
34
+ def __init__(self, hidden_size: int, hidden_dim: int, dropout: float, activation: str):
35
+ super().__init__()
36
+ self.net = nn.Sequential(
37
+ nn.LayerNorm(hidden_size),
38
+ nn.Linear(hidden_size, hidden_dim),
39
+ _get_activation(activation),
40
+ nn.Dropout(dropout),
41
+ nn.Linear(hidden_dim, 1),
42
+ )
43
+
44
+ def forward(self, pooled: torch.FloatTensor) -> torch.FloatTensor:
45
+ return self.net(pooled)
46
+
47
+
48
+ class GatedMLPRewardHead(nn.Module):
49
+ def __init__(self, hidden_size: int, hidden_dim: int, dropout: float):
50
+ super().__init__()
51
+ self.norm = nn.LayerNorm(hidden_size)
52
+ self.value_proj = nn.Linear(hidden_size, hidden_dim)
53
+ self.gate_proj = nn.Linear(hidden_size, hidden_dim)
54
+ self.dropout = nn.Dropout(dropout)
55
+ self.out_proj = nn.Linear(hidden_dim, 1)
56
+
57
+ def forward(self, pooled: torch.FloatTensor) -> torch.FloatTensor:
58
+ pooled = self.norm(pooled)
59
+ hidden = self.value_proj(pooled) * torch.sigmoid(self.gate_proj(pooled))
60
+ hidden = self.dropout(hidden)
61
+ return self.out_proj(hidden)
62
+
63
+
64
+ def build_reward_head(
65
+ hidden_size: int,
66
+ head_type: str,
67
+ hidden_dim: Optional[int],
68
+ dropout: float,
69
+ activation: str,
70
+ ) -> nn.Module:
71
+ resolved_hidden_dim = hidden_dim or hidden_size
72
+
73
+ if head_type == "linear":
74
+ return nn.Linear(hidden_size, 1)
75
+ if head_type == "layernorm_linear":
76
+ return LayerNormLinearRewardHead(hidden_size)
77
+ if head_type == "mlp":
78
+ return MLPRewardHead(hidden_size, resolved_hidden_dim, dropout, activation)
79
+ if head_type == "gated_mlp":
80
+ return GatedMLPRewardHead(hidden_size, resolved_hidden_dim, dropout)
81
+ raise ValueError(f"Unsupported reward head type: {head_type}")
82
+
83
+
84
+ class GPTNeoXRewardModelConfig(GPTNeoXConfig):
85
+ model_type = "gpt_neox_reward_model"
86
+
87
+ pooling: Literal["mean", "last"]
88
+ reward_head_type: Literal["linear", "layernorm_linear", "mlp", "gated_mlp"]
89
+ reward_head_hidden_dim: Optional[int]
90
+ reward_head_dropout: float
91
+ reward_head_activation: Literal["gelu", "relu", "silu", "tanh"]
92
+
93
+ def __init__(
94
+ self,
95
+ pooling: Literal["mean", "last"] = "last",
96
+ reward_head_type: Literal["linear", "layernorm_linear", "mlp", "gated_mlp"] = "linear",
97
+ reward_head_hidden_dim: Optional[int] = None,
98
+ reward_head_dropout: float = 0.0,
99
+ reward_head_activation: Literal["gelu", "relu", "silu", "tanh"] = "gelu",
100
+ **kwargs,
101
+ ):
102
+ super().__init__(**kwargs)
103
+ self.pooling = pooling or "last"
104
+ self.reward_head_type = reward_head_type or "linear"
105
+ self.reward_head_hidden_dim = reward_head_hidden_dim
106
+ self.reward_head_dropout = reward_head_dropout
107
+ self.reward_head_activation = reward_head_activation or "gelu"
108
+
109
+
110
+ @dataclass
111
+ class GPTNeoXRewardModelOutput(ModelOutput):
112
+ logits: torch.FloatTensor = None
113
+
114
+
115
+ class GPTNeoXRewardModel(GPTNeoXPreTrainedModel):
116
+ config_class = GPTNeoXRewardModelConfig
117
+
118
+ def __init__(self, config):
119
+ if type(config) == GPTNeoXConfig:
120
+ # Preserve the upstream behavior where a base GPT-NeoX config can
121
+ # be promoted into the reward-model config during loading.
122
+ config = GPTNeoXRewardModelConfig.from_dict(config.to_dict())
123
+ super().__init__(config)
124
+
125
+ self.gpt_neox = GPTNeoXModel(config)
126
+ # Keep the reward head declared locally so it can be modified here
127
+ # without relying on the packaged `model_training` implementation.
128
+ self.out_proj = build_reward_head(
129
+ hidden_size=config.hidden_size,
130
+ head_type=getattr(config, "reward_head_type", "linear"),
131
+ hidden_dim=getattr(config, "reward_head_hidden_dim", None),
132
+ dropout=getattr(config, "reward_head_dropout", 0.0),
133
+ activation=getattr(config, "reward_head_activation", "gelu"),
134
+ )
135
+ self.pooling = config.pooling
136
+ self.post_init()
137
+
138
+ @property
139
+ def reward_head(self) -> nn.Module:
140
+ return self.out_proj
141
+
142
+ def forward(
143
+ self,
144
+ input_ids,
145
+ attention_mask: Optional[torch.FloatTensor] = None,
146
+ inputs_embeds: Optional[torch.FloatTensor] = None,
147
+ head_mask: Optional[torch.FloatTensor] = None,
148
+ use_cache: Optional[bool] = None,
149
+ return_dict: Optional[bool] = True,
150
+ ) -> GPTNeoXRewardModelOutput:
151
+ outputs = self.gpt_neox(
152
+ input_ids,
153
+ attention_mask=attention_mask,
154
+ head_mask=head_mask,
155
+ inputs_embeds=inputs_embeds,
156
+ use_cache=use_cache,
157
+ return_dict=return_dict,
158
+ )
159
+
160
+ hidden_states = outputs[0]
161
+ if self.pooling == "mean":
162
+ if attention_mask is None:
163
+ pooled = hidden_states.mean(dim=1)
164
+ else:
165
+ mask = attention_mask.to(hidden_states.dtype).unsqueeze(-1)
166
+ pooled = (hidden_states * mask).sum(dim=1) / mask.sum(dim=1).clamp_min(1.0)
167
+ elif self.pooling == "last":
168
+ if attention_mask is None:
169
+ pooled = hidden_states[:, -1]
170
+ else:
171
+ last_idx = attention_mask.long().sum(dim=1).clamp_min(1) - 1
172
+ pooled = hidden_states.gather(
173
+ 1,
174
+ last_idx.view(-1, 1, 1).expand(-1, 1, hidden_states.size(-1)),
175
+ ).squeeze(1)
176
+ else:
177
+ raise ValueError(f"Unknown pooling method: {self.pooling}")
178
+
179
+ logits = self.out_proj(pooled)
180
+
181
+ if not return_dict:
182
+ return (logits,) + outputs[1:]
183
+
184
+ return GPTNeoXRewardModelOutput(logits=logits)
185
+
186
+
187
+ AutoConfig.register("gpt_neox_reward_model", GPTNeoXRewardModelConfig)
188
+ AutoModelForSequenceClassification.register(GPTNeoXRewardModelConfig, GPTNeoXRewardModel)
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5277f93106c5f7312d2d9fe666702cc1b43a6de7f74bcfcbdfafe66eef0a8f6
3
+ size 114527658
special_tokens_map.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<|prefix_end|>",
4
+ "<|assistant|>",
5
+ "<|prefix_begin|>",
6
+ "<|prompter|>",
7
+ "<|system|>"
8
+ ],
9
+ "bos_token": "<|endoftext|>",
10
+ "eos_token": "<|endoftext|>",
11
+ "pad_token": "<|padding|>",
12
+ "sep_token": "<|endoftext|>",
13
+ "unk_token": "<|endoftext|>"
14
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "bos_token": "<|endoftext|>",
4
+ "clean_up_tokenization_spaces": true,
5
+ "eos_token": "<|endoftext|>",
6
+ "model_max_length": 1000000000000000019884624838656,
7
+ "tokenizer_class": "GPTNeoXTokenizer",
8
+ "unk_token": "<|endoftext|>"
9
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15d17db566d266013af03a48da12b993aa2b485f122344645b19641417f7840d
3
+ size 4881