Model save

README.md

@@ -0,0 +1,61 @@
+---
+license: other
+base_model: deepseek-ai/deepseek-math-7b-base
+tags:
+- trl
+- orpo
+- generated_from_trainer
+model-index:
+- name: deepseek-8b-orpo-full
+  results: []
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/thuzfz1/huggingface/runs/dtbyb3v7)
+# deepseek-8b-orpo-full
+
+This model is a fine-tuned version of [deepseek-ai/deepseek-math-7b-base](https://huggingface.co/deepseek-ai/deepseek-math-7b-base) on an unknown dataset.
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 2e-05
+- train_batch_size: 4
+- eval_batch_size: 4
+- seed: 43
+- distributed_type: multi-GPU
+- num_devices: 4
+- gradient_accumulation_steps: 2
+- total_train_batch_size: 32
+- total_eval_batch_size: 16
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: cosine
+- lr_scheduler_warmup_ratio: 0.1
+- num_epochs: 2
+
+### Training results
+
+
+
+### Framework versions
+
+- Transformers 4.42.3
+- Pytorch 2.1.2
+- Datasets 2.20.0
+- Tokenizers 0.19.1

all_results.json

@@ -0,0 +1,9 @@
+{
+    "epoch": 2.0,
+    "total_flos": 0.0,
+    "train_loss": 0.2130171504020691,
+    "train_runtime": 11716.0982,
+    "train_samples": 20000,
+    "train_samples_per_second": 3.414,
+    "train_steps_per_second": 0.107
+}

generation_config.json

@@ -0,0 +1,6 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 100000,
+  "eos_token_id": 100001,
+  "transformers_version": "4.42.3"
+}

model-00001-of-00003.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cfa72038bb07d6e7b2db06365ddf749a1f0d2e7ead871483a2bcbdf7e6d2d45a
+size 4987202208

model-00002-of-00003.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:48aa886de79efc86115d136b985a023b9e80b8940d5aadb8da0faa2e767a26cc
+size 4980945440

model-00003-of-00003.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9f1204f6ccf8c8f580baba662b33eb215db8ea21273dab74aa457a0dd671d832
+size 3852615520

model.safetensors.index.json

@@ -0,0 +1,280 @@
+{
+  "metadata": {
+    "total_size": 13820731392
+  },
+  "weight_map": {
+    "lm_head.weight": "model-00003-of-00003.safetensors",
+    "model.embed_tokens.weight": "model-00001-of-00003.safetensors",
+    "model.layers.0.input_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.0.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.0.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.1.input_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.1.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.1.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.10.input_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.10.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.10.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.10.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.10.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.10.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.11.input_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.11.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.11.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.11.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.11.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.11.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.11.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.11.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.11.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.12.input_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.12.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.12.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.12.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.12.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.12.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.12.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.12.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.12.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.13.input_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.13.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.13.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.13.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.13.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.13.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.13.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.13.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.13.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.14.input_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.14.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.14.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.14.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.14.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.14.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.14.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.14.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.14.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.15.input_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.15.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.15.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.15.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.15.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.15.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.15.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.15.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.15.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.16.input_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.16.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.16.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.16.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.16.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.16.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.16.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.16.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.16.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.17.input_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.17.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.17.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.17.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.17.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.17.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.17.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.17.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.17.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.18.input_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.18.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.18.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.18.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.18.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.18.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.18.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.18.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.18.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.19.input_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.19.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.19.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.19.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.19.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.19.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.19.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.19.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.19.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.2.input_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.2.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.2.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.20.input_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.20.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.20.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.20.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.20.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.20.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.20.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.20.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.20.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.21.input_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.21.mlp.down_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.21.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.21.mlp.up_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.21.post_attention_layernorm.weight": "model-00002-of-00003.safetensors",
+    "model.layers.21.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.21.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.21.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.21.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.22.input_layernorm.weight": "model-00003-of-00003.safetensors",
+    "model.layers.22.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.22.mlp.gate_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.22.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.22.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
+    "model.layers.22.self_attn.k_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.22.self_attn.o_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.22.self_attn.q_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.22.self_attn.v_proj.weight": "model-00002-of-00003.safetensors",
+    "model.layers.23.input_layernorm.weight": "model-00003-of-00003.safetensors",
+    "model.layers.23.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.23.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.23.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.23.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
+    "model.layers.23.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.23.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.23.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.23.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.24.input_layernorm.weight": "model-00003-of-00003.safetensors",
+    "model.layers.24.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.24.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.24.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.24.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
+    "model.layers.24.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.24.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.24.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.24.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.25.input_layernorm.weight": "model-00003-of-00003.safetensors",
+    "model.layers.25.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.25.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.25.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.25.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
+    "model.layers.25.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.25.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.25.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.25.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.26.input_layernorm.weight": "model-00003-of-00003.safetensors",
+    "model.layers.26.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.26.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.26.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.26.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
+    "model.layers.26.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.26.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.26.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.26.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.27.input_layernorm.weight": "model-00003-of-00003.safetensors",
+    "model.layers.27.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.27.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.27.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.27.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
+    "model.layers.27.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.27.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.27.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.27.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.28.input_layernorm.weight": "model-00003-of-00003.safetensors",
+    "model.layers.28.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.28.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.28.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.28.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
+    "model.layers.28.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.28.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.28.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.28.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.29.input_layernorm.weight": "model-00003-of-00003.safetensors",
+    "model.layers.29.mlp.down_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.29.mlp.gate_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.29.mlp.up_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.29.post_attention_layernorm.weight": "model-00003-of-00003.safetensors",
+    "model.layers.29.self_attn.k_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.29.self_attn.o_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.29.self_attn.q_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.29.self_attn.v_proj.weight": "model-00003-of-00003.safetensors",
+    "model.layers.3.input_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.3.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.3.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.4.input_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.4.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.4.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.5.input_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.5.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.5.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.6.input_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.6.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.6.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.7.input_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.7.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.7.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.8.input_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.8.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.8.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.8.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.8.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.9.input_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.9.mlp.down_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.9.mlp.gate_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.9.mlp.up_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.9.post_attention_layernorm.weight": "model-00001-of-00003.safetensors",
+    "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.9.self_attn.o_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00003.safetensors",
+    "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00003.safetensors",
+    "model.norm.weight": "model-00003-of-00003.safetensors"
+  }
+}

runs/Jul15_19-44-09_phe108-jieyuzhao-01/events.out.tfevents.1721098092.phe108-jieyuzhao-01.232196.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a87d5797b869c6dbb4faffaf0aff6b2c1ec244b01dec06128502e2c1a988f646
-size 112664
+oid sha256:2c5c518b0a102820dbacdc15c5838d8eaf3571d7db3e200476d0e101c28377b3
+size 113018

train_results.json

@@ -0,0 +1,9 @@
+{
+    "epoch": 2.0,
+    "total_flos": 0.0,
+    "train_loss": 0.2130171504020691,
+    "train_runtime": 11716.0982,
+    "train_samples": 20000,
+    "train_samples_per_second": 3.414,
+    "train_steps_per_second": 0.107
+}

trainer_state.json

@@ -0,0 +1,2292 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 2.0,
+  "eval_steps": 10000,
+  "global_step": 1250,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.016,
+      "grad_norm": 7.157915223499978,
+      "learning_rate": 1.6000000000000001e-06,
+      "log_odds_chosen": 0.08687938749790192,
+      "log_odds_ratio": -0.6765035390853882,
+      "logits/chosen": 42.733116149902344,
+      "logits/rejected": 42.47905731201172,
+      "logps/chosen": -1.0179941654205322,
+      "logps/rejected": -1.0789813995361328,
+      "loss": 1.6603,
+      "nll_loss": 1.6577844619750977,
+      "rewards/accuracies": 0.574999988079071,
+      "rewards/chosen": -0.10179942846298218,
+      "rewards/margins": 0.0060987225733697414,
+      "rewards/rejected": -0.10789815336465836,
+      "step": 10
+    },
+    {
+      "epoch": 0.032,
+      "grad_norm": 4.787027456951044,
+      "learning_rate": 3.2000000000000003e-06,
+      "log_odds_chosen": 0.16144487261772156,
+      "log_odds_ratio": -0.6409393548965454,
+      "logits/chosen": 39.06427764892578,
+      "logits/rejected": 37.91705322265625,
+      "logps/chosen": -0.8086442947387695,
+      "logps/rejected": -0.9057012796401978,
+      "loss": 1.4104,
+      "nll_loss": 1.320190191268921,
+      "rewards/accuracies": 0.637499988079071,
+      "rewards/chosen": -0.08086444437503815,
+      "rewards/margins": 0.009705692529678345,
+      "rewards/rejected": -0.09057013690471649,
+      "step": 20
+    },
+    {
+      "epoch": 0.048,
+      "grad_norm": 3.619155990390428,
+      "learning_rate": 4.800000000000001e-06,
+      "log_odds_chosen": -0.09232249855995178,
+      "log_odds_ratio": -0.7732639312744141,
+      "logits/chosen": 33.01856231689453,
+      "logits/rejected": 34.8216552734375,
+      "logps/chosen": -0.5950765013694763,
+      "logps/rejected": -0.5527861714363098,
+      "loss": 0.9477,
+      "nll_loss": 0.9017837643623352,
+      "rewards/accuracies": 0.4375,
+      "rewards/chosen": -0.05950764939188957,
+      "rewards/margins": -0.004229038953781128,
+      "rewards/rejected": -0.05527861788868904,
+      "step": 30
+    },
+    {
+      "epoch": 0.064,
+      "grad_norm": 0.8193426378065174,
+      "learning_rate": 6.4000000000000006e-06,
+      "log_odds_chosen": 0.13158965110778809,
+      "log_odds_ratio": -0.6801391243934631,
+      "logits/chosen": 34.19690704345703,
+      "logits/rejected": 33.757015228271484,
+      "logps/chosen": -0.38923197984695435,
+      "logps/rejected": -0.42596691846847534,
+      "loss": 0.7871,
+      "nll_loss": 0.7480510473251343,
+      "rewards/accuracies": 0.48750001192092896,
+      "rewards/chosen": -0.038923196494579315,
+      "rewards/margins": 0.003673497587442398,
+      "rewards/rejected": -0.04259669408202171,
+      "step": 40
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.837886863416422,
+      "learning_rate": 8.000000000000001e-06,
+      "log_odds_chosen": 0.010798773728311062,
+      "log_odds_ratio": -0.7395683526992798,
+      "logits/chosen": 36.046871185302734,
+      "logits/rejected": 35.39847946166992,
+      "logps/chosen": -0.3819189667701721,
+      "logps/rejected": -0.36964151263237,
+      "loss": 0.7614,
+      "nll_loss": 0.6979484558105469,
+      "rewards/accuracies": 0.5375000238418579,
+      "rewards/chosen": -0.038191892206668854,
+      "rewards/margins": -0.0012277422938495874,
+      "rewards/rejected": -0.03696415573358536,
+      "step": 50
+    },
+    {
+      "epoch": 0.096,
+      "grad_norm": 0.934326744395183,
+      "learning_rate": 9.600000000000001e-06,
+      "log_odds_chosen": 0.17773893475532532,
+      "log_odds_ratio": -0.6542357206344604,
+      "logits/chosen": 37.91405487060547,
+      "logits/rejected": 38.55655288696289,
+      "logps/chosen": -0.35373184084892273,
+      "logps/rejected": -0.40366658568382263,
+      "loss": 0.7369,
+      "nll_loss": 0.6853512525558472,
+      "rewards/accuracies": 0.699999988079071,
+      "rewards/chosen": -0.03537318482995033,
+      "rewards/margins": 0.0049934713169932365,
+      "rewards/rejected": -0.040366653352975845,
+      "step": 60
+    },
+    {
+      "epoch": 0.112,
+      "grad_norm": 0.8140121698145116,
+      "learning_rate": 1.1200000000000001e-05,
+      "log_odds_chosen": 0.27008742094039917,
+      "log_odds_ratio": -0.6275990009307861,
+      "logits/chosen": 37.89257049560547,
+      "logits/rejected": 36.801631927490234,
+      "logps/chosen": -0.3447234630584717,
+      "logps/rejected": -0.44380050897598267,
+      "loss": 0.7405,
+      "nll_loss": 0.6623181104660034,
+      "rewards/accuracies": 0.6499999761581421,
+      "rewards/chosen": -0.03447234630584717,
+      "rewards/margins": 0.009907705709338188,
+      "rewards/rejected": -0.04438004642724991,
+      "step": 70
+    },
+    {
+      "epoch": 0.128,
+      "grad_norm": 0.8866561757309486,
+      "learning_rate": 1.2800000000000001e-05,
+      "log_odds_chosen": 0.7000471949577332,
+      "log_odds_ratio": -0.46674174070358276,
+      "logits/chosen": 39.06570816040039,
+      "logits/rejected": 38.14867401123047,
+      "logps/chosen": -0.3046243488788605,
+      "logps/rejected": -0.541035532951355,
+      "loss": 0.6756,
+      "nll_loss": 0.5991424322128296,
+      "rewards/accuracies": 0.8374999761581421,
+      "rewards/chosen": -0.030462438240647316,
+      "rewards/margins": 0.023641113191843033,
+      "rewards/rejected": -0.0541035458445549,
+      "step": 80
+    },
+    {
+      "epoch": 0.144,
+      "grad_norm": 1.1050711696617728,
+      "learning_rate": 1.4400000000000001e-05,
+      "log_odds_chosen": 0.4890363812446594,
+      "log_odds_ratio": -0.5526093244552612,
+      "logits/chosen": 39.77109146118164,
+      "logits/rejected": 41.683753967285156,
+      "logps/chosen": -0.36524319648742676,
+      "logps/rejected": -0.5409058928489685,
+      "loss": 0.6761,
+      "nll_loss": 0.6159157156944275,
+      "rewards/accuracies": 0.7250000238418579,
+      "rewards/chosen": -0.036524318158626556,
+      "rewards/margins": 0.017566271126270294,
+      "rewards/rejected": -0.05409059673547745,
+      "step": 90
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 1.1283739516296225,
+      "learning_rate": 1.6000000000000003e-05,
+      "log_odds_chosen": 0.5452580451965332,
+      "log_odds_ratio": -0.5334872603416443,
+      "logits/chosen": 41.09711837768555,
+      "logits/rejected": 39.49372100830078,
+      "logps/chosen": -0.3422545790672302,
+      "logps/rejected": -0.5715359449386597,
+      "loss": 0.6441,
+      "nll_loss": 0.5890800952911377,
+      "rewards/accuracies": 0.762499988079071,
+      "rewards/chosen": -0.0342254601418972,
+      "rewards/margins": 0.022928137332201004,
+      "rewards/rejected": -0.05715359374880791,
+      "step": 100
+    },
+    {
+      "epoch": 0.176,
+      "grad_norm": 1.044328943254365,
+      "learning_rate": 1.76e-05,
+      "log_odds_chosen": 0.6057835817337036,
+      "log_odds_ratio": -0.5027212500572205,
+      "logits/chosen": 41.39733123779297,
+      "logits/rejected": 39.85953140258789,
+      "logps/chosen": -0.35752102732658386,
+      "logps/rejected": -0.590412437915802,
+      "loss": 0.6068,
+      "nll_loss": 0.5553559064865112,
+      "rewards/accuracies": 0.800000011920929,
+      "rewards/chosen": -0.035752106457948685,
+      "rewards/margins": 0.023289144039154053,
+      "rewards/rejected": -0.059041254222393036,
+      "step": 110
+    },
+    {
+      "epoch": 0.192,
+      "grad_norm": 1.0225837980847625,
+      "learning_rate": 1.9200000000000003e-05,
+      "log_odds_chosen": 0.5523650050163269,
+      "log_odds_ratio": -0.5369630455970764,
+      "logits/chosen": 43.63589859008789,
+      "logits/rejected": 41.659324645996094,
+      "logps/chosen": -0.3336974084377289,
+      "logps/rejected": -0.5324398279190063,
+      "loss": 0.5734,
+      "nll_loss": 0.5098426938056946,
+      "rewards/accuracies": 0.75,
+      "rewards/chosen": -0.03336974233388901,
+      "rewards/margins": 0.01987423747777939,
+      "rewards/rejected": -0.05324398726224899,
+      "step": 120
+    },
+    {
+      "epoch": 0.208,
+      "grad_norm": 1.1101012093036131,
+      "learning_rate": 1.9999025240093045e-05,
+      "log_odds_chosen": 0.7986687421798706,
+      "log_odds_ratio": -0.471863329410553,
+      "logits/chosen": 43.426692962646484,
+      "logits/rejected": 40.31127166748047,
+      "logps/chosen": -0.35013002157211304,
+      "logps/rejected": -0.6971026659011841,
+      "loss": 0.5401,
+      "nll_loss": 0.508377730846405,
+      "rewards/accuracies": 0.824999988079071,
+      "rewards/chosen": -0.03501300513744354,
+      "rewards/margins": 0.03469725698232651,
+      "rewards/rejected": -0.06971026211977005,
+      "step": 130
+    },
+    {
+      "epoch": 0.224,
+      "grad_norm": 1.5810963333813628,
+      "learning_rate": 1.9991228300988586e-05,
+      "log_odds_chosen": 0.9270819425582886,
+      "log_odds_ratio": -0.4264460504055023,
+      "logits/chosen": 40.22356414794922,
+      "logits/rejected": 38.093299865722656,
+      "logps/chosen": -0.3407229781150818,
+      "logps/rejected": -0.7779335975646973,
+      "loss": 0.5243,
+      "nll_loss": 0.4749884605407715,
+      "rewards/accuracies": 0.8125,
+      "rewards/chosen": -0.03407229855656624,
+      "rewards/margins": 0.04372106492519379,
+      "rewards/rejected": -0.07779336720705032,
+      "step": 140
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.9914026891453022,
+      "learning_rate": 1.9975640502598243e-05,
+      "log_odds_chosen": 0.7473500967025757,
+      "log_odds_ratio": -0.4457695484161377,
+      "logits/chosen": 38.593631744384766,
+      "logits/rejected": 40.06364059448242,
+      "logps/chosen": -0.35150039196014404,
+      "logps/rejected": -0.6378719210624695,
+      "loss": 0.507,
+      "nll_loss": 0.45233845710754395,
+      "rewards/accuracies": 0.875,
+      "rewards/chosen": -0.035150036215782166,
+      "rewards/margins": 0.02863715961575508,
+      "rewards/rejected": -0.06378719955682755,
+      "step": 150
+    },
+    {
+      "epoch": 0.256,
+      "grad_norm": 1.0114005741143277,
+      "learning_rate": 1.9952273999818312e-05,
+      "log_odds_chosen": 0.7789567708969116,
+      "log_odds_ratio": -0.47668343782424927,
+      "logits/chosen": 40.021446228027344,
+      "logits/rejected": 39.03896713256836,
+      "logps/chosen": -0.32147353887557983,
+      "logps/rejected": -0.6195149421691895,
+      "loss": 0.4762,
+      "nll_loss": 0.40015798807144165,
+      "rewards/accuracies": 0.7749999761581421,
+      "rewards/chosen": -0.0321473553776741,
+      "rewards/margins": 0.02980414591729641,
+      "rewards/rejected": -0.061951495707035065,
+      "step": 160
+    },
+    {
+      "epoch": 0.272,
+      "grad_norm": 1.1072013212789444,
+      "learning_rate": 1.9921147013144782e-05,
+      "log_odds_chosen": 0.9789760708808899,
+      "log_odds_ratio": -0.3996427059173584,
+      "logits/chosen": 38.469261169433594,
+      "logits/rejected": 39.36458206176758,
+      "logps/chosen": -0.31273895502090454,
+      "logps/rejected": -0.7282775640487671,
+      "loss": 0.4569,
+      "nll_loss": 0.4199884533882141,
+      "rewards/accuracies": 0.887499988079071,
+      "rewards/chosen": -0.031273894011974335,
+      "rewards/margins": 0.041553862392902374,
+      "rewards/rejected": -0.07282774895429611,
+      "step": 170
+    },
+    {
+      "epoch": 0.288,
+      "grad_norm": 1.0540475357463217,
+      "learning_rate": 1.988228381446553e-05,
+      "log_odds_chosen": 0.9965242147445679,
+      "log_odds_ratio": -0.4183652997016907,
+      "logits/chosen": 40.18573760986328,
+      "logits/rejected": 40.521148681640625,
+      "logps/chosen": -0.33785638213157654,
+      "logps/rejected": -0.7628054022789001,
+      "loss": 0.4311,
+      "nll_loss": 0.43037042021751404,
+      "rewards/accuracies": 0.824999988079071,
+      "rewards/chosen": -0.03378564119338989,
+      "rewards/margins": 0.04249490052461624,
+      "rewards/rejected": -0.07628054916858673,
+      "step": 180
+    },
+    {
+      "epoch": 0.304,
+      "grad_norm": 1.2807805662820098,
+      "learning_rate": 1.983571470813386e-05,
+      "log_odds_chosen": 0.9251772165298462,
+      "log_odds_ratio": -0.43279653787612915,
+      "logits/chosen": 39.66691970825195,
+      "logits/rejected": 39.22489929199219,
+      "logps/chosen": -0.3330400586128235,
+      "logps/rejected": -0.6816459894180298,
+      "loss": 0.4062,
+      "nll_loss": 0.36826083064079285,
+      "rewards/accuracies": 0.8125,
+      "rewards/chosen": -0.03330400586128235,
+      "rewards/margins": 0.03486059606075287,
+      "rewards/rejected": -0.06816460937261581,
+      "step": 190
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 1.247095232534241,
+      "learning_rate": 1.9781476007338058e-05,
+      "log_odds_chosen": 1.1870229244232178,
+      "log_odds_ratio": -0.331691175699234,
+      "logits/chosen": 36.31359100341797,
+      "logits/rejected": 38.259849548339844,
+      "logps/chosen": -0.2683125436306,
+      "logps/rejected": -0.6734791994094849,
+      "loss": 0.3637,
+      "nll_loss": 0.29906129837036133,
+      "rewards/accuracies": 0.949999988079071,
+      "rewards/chosen": -0.026831254363059998,
+      "rewards/margins": 0.04051666334271431,
+      "rewards/rejected": -0.0673479214310646,
+      "step": 200
+    },
+    {
+      "epoch": 0.336,
+      "grad_norm": 1.023611287110811,
+      "learning_rate": 1.9719610005785466e-05,
+      "log_odds_chosen": 1.10099196434021,
+      "log_odds_ratio": -0.3911464810371399,
+      "logits/chosen": 36.06108856201172,
+      "logits/rejected": 36.09987258911133,
+      "logps/chosen": -0.3143238425254822,
+      "logps/rejected": -0.7127642035484314,
+      "loss": 0.3608,
+      "nll_loss": 0.33582496643066406,
+      "rewards/accuracies": 0.862500011920929,
+      "rewards/chosen": -0.0314323827624321,
+      "rewards/margins": 0.03984404355287552,
+      "rewards/rejected": -0.07127643376588821,
+      "step": 210
+    },
+    {
+      "epoch": 0.352,
+      "grad_norm": 1.0618187057425008,
+      "learning_rate": 1.9650164944723116e-05,
+      "log_odds_chosen": 1.4383821487426758,
+      "log_odds_ratio": -0.313861608505249,
+      "logits/chosen": 34.82101821899414,
+      "logits/rejected": 38.44565200805664,
+      "logps/chosen": -0.28008711338043213,
+      "logps/rejected": -0.8806719779968262,
+      "loss": 0.3331,
+      "nll_loss": 0.2972494959831238,
+      "rewards/accuracies": 0.9375,
+      "rewards/chosen": -0.028008710592985153,
+      "rewards/margins": 0.06005847454071045,
+      "rewards/rejected": -0.0880671888589859,
+      "step": 220
+    },
+    {
+      "epoch": 0.368,
+      "grad_norm": 0.9398000108663768,
+      "learning_rate": 1.9573194975320672e-05,
+      "log_odds_chosen": 1.4377901554107666,
+      "log_odds_ratio": -0.34182658791542053,
+      "logits/chosen": 33.84839630126953,
+      "logits/rejected": 35.625667572021484,
+      "logps/chosen": -0.30297523736953735,
+      "logps/rejected": -0.9298135042190552,
+      "loss": 0.3403,
+      "nll_loss": 0.2947668433189392,
+      "rewards/accuracies": 0.8999999761581421,
+      "rewards/chosen": -0.030297517776489258,
+      "rewards/margins": 0.0626838356256485,
+      "rewards/rejected": -0.09298135340213776,
+      "step": 230
+    },
+    {
+      "epoch": 0.384,
+      "grad_norm": 1.103321212338863,
+      "learning_rate": 1.9488760116444966e-05,
+      "log_odds_chosen": 1.4992961883544922,
+      "log_odds_ratio": -0.3107326030731201,
+      "logits/chosen": 35.9039421081543,
+      "logits/rejected": 36.633705139160156,
+      "logps/chosen": -0.2847585380077362,
+      "logps/rejected": -0.9570012092590332,
+      "loss": 0.3166,
+      "nll_loss": 0.30076318979263306,
+      "rewards/accuracies": 0.925000011920929,
+      "rewards/chosen": -0.02847585640847683,
+      "rewards/margins": 0.06722426414489746,
+      "rewards/rejected": -0.09570012241601944,
+      "step": 240
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.9188155900142966,
+      "learning_rate": 1.9396926207859085e-05,
+      "log_odds_chosen": 1.232299566268921,
+      "log_odds_ratio": -0.36939191818237305,
+      "logits/chosen": 36.425289154052734,
+      "logits/rejected": 38.76382827758789,
+      "logps/chosen": -0.306942343711853,
+      "logps/rejected": -0.8188571929931641,
+      "loss": 0.3212,
+      "nll_loss": 0.2955878674983978,
+      "rewards/accuracies": 0.824999988079071,
+      "rewards/chosen": -0.030694235116243362,
+      "rewards/margins": 0.05119148641824722,
+      "rewards/rejected": -0.08188571780920029,
+      "step": 250
+    },
+    {
+      "epoch": 0.416,
+      "grad_norm": 1.1371525474896846,
+      "learning_rate": 1.9297764858882516e-05,
+      "log_odds_chosen": 1.5188441276550293,
+      "log_odds_ratio": -0.3032824695110321,
+      "logits/chosen": 33.87938690185547,
+      "logits/rejected": 34.950599670410156,
+      "logps/chosen": -0.2652571499347687,
+      "logps/rejected": -0.904030442237854,
+      "loss": 0.282,
+      "nll_loss": 0.2362031489610672,
+      "rewards/accuracies": 0.9125000238418579,
+      "rewards/chosen": -0.026525715366005898,
+      "rewards/margins": 0.06387733668088913,
+      "rewards/rejected": -0.09040305018424988,
+      "step": 260
+    },
+    {
+      "epoch": 0.432,
+      "grad_norm": 0.9722134512964291,
+      "learning_rate": 1.9191353392552346e-05,
+      "log_odds_chosen": 1.6056066751480103,
+      "log_odds_ratio": -0.273274302482605,
+      "logits/chosen": 31.2882080078125,
+      "logits/rejected": 32.222434997558594,
+      "logps/chosen": -0.2462386190891266,
+      "logps/rejected": -0.843704342842102,
+      "loss": 0.2621,
+      "nll_loss": 0.23743471503257751,
+      "rewards/accuracies": 0.949999988079071,
+      "rewards/chosen": -0.02462385967373848,
+      "rewards/margins": 0.059746574610471725,
+      "rewards/rejected": -0.0843704342842102,
+      "step": 270
+    },
+    {
+      "epoch": 0.448,
+      "grad_norm": 0.9536358910314567,
+      "learning_rate": 1.907777478532909e-05,
+      "log_odds_chosen": 1.3453700542449951,
+      "log_odds_ratio": -0.3348150849342346,
+      "logits/chosen": 34.44597625732422,
+      "logits/rejected": 35.61604690551758,
+      "logps/chosen": -0.25559094548225403,
+      "logps/rejected": -0.7815641164779663,
+      "loss": 0.2815,
+      "nll_loss": 0.25891661643981934,
+      "rewards/accuracies": 0.9125000238418579,
+      "rewards/chosen": -0.025559093803167343,
+      "rewards/margins": 0.052597321569919586,
+      "rewards/rejected": -0.07815642654895782,
+      "step": 280
+    },
+    {
+      "epoch": 0.464,
+      "grad_norm": 1.4793964733191685,
+      "learning_rate": 1.895711760239413e-05,
+      "log_odds_chosen": 1.8677561283111572,
+      "log_odds_ratio": -0.2598610520362854,
+      "logits/chosen": 30.939416885375977,
+      "logits/rejected": 32.69816589355469,
+      "logps/chosen": -0.25819864869117737,
+      "logps/rejected": -1.080700159072876,
+      "loss": 0.2526,
+      "nll_loss": 0.24270136654376984,
+      "rewards/accuracies": 0.9375,
+      "rewards/chosen": -0.025819867849349976,
+      "rewards/margins": 0.08225013315677643,
+      "rewards/rejected": -0.1080700010061264,
+      "step": 290
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 1.2223140757678745,
+      "learning_rate": 1.8829475928589272e-05,
+      "log_odds_chosen": 1.873132348060608,
+      "log_odds_ratio": -0.21418969333171844,
+      "logits/chosen": 30.050670623779297,
+      "logits/rejected": 31.51558494567871,
+      "logps/chosen": -0.24227066338062286,
+      "logps/rejected": -1.0071208477020264,
+      "loss": 0.2446,
+      "nll_loss": 0.20659121870994568,
+      "rewards/accuracies": 0.9750000238418579,
+      "rewards/chosen": -0.024227067828178406,
+      "rewards/margins": 0.07648502290248871,
+      "rewards/rejected": -0.10071208328008652,
+      "step": 300
+    },
+    {
+      "epoch": 0.496,
+      "grad_norm": 0.8813575124105774,
+      "learning_rate": 1.869494929505219e-05,
+      "log_odds_chosen": 1.9893770217895508,
+      "log_odds_ratio": -0.22330129146575928,
+      "logits/chosen": 33.173133850097656,
+      "logits/rejected": 34.073055267333984,
+      "logps/chosen": -0.2551695704460144,
+      "logps/rejected": -1.1849584579467773,
+      "loss": 0.232,
+      "nll_loss": 0.2051832675933838,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -0.025516962632536888,
+      "rewards/margins": 0.09297889471054077,
+      "rewards/rejected": -0.11849584430456161,
+      "step": 310
+    },
+    {
+      "epoch": 0.512,
+      "grad_norm": 0.9094503804430618,
+      "learning_rate": 1.855364260160507e-05,
+      "log_odds_chosen": 1.7108672857284546,
+      "log_odds_ratio": -0.2914069592952728,
+      "logits/chosen": 31.070571899414062,
+      "logits/rejected": 31.337520599365234,
+      "logps/chosen": -0.26724526286125183,
+      "logps/rejected": -0.9716242551803589,
+      "loss": 0.2355,
+      "nll_loss": 0.2270585000514984,
+      "rewards/accuracies": 0.8999999761581421,
+      "rewards/chosen": -0.026724526658654213,
+      "rewards/margins": 0.07043790072202682,
+      "rewards/rejected": -0.09716243296861649,
+      "step": 320
+    },
+    {
+      "epoch": 0.528,
+      "grad_norm": 0.9058310290840669,
+      "learning_rate": 1.8405666034956842e-05,
+      "log_odds_chosen": 2.1886684894561768,
+      "log_odds_ratio": -0.19184692203998566,
+      "logits/chosen": 30.936025619506836,
+      "logits/rejected": 31.9810848236084,
+      "logps/chosen": -0.2256220132112503,
+      "logps/rejected": -1.2340004444122314,
+      "loss": 0.2216,
+      "nll_loss": 0.18992918729782104,
+      "rewards/accuracies": 0.9750000238418579,
+      "rewards/chosen": -0.02256220206618309,
+      "rewards/margins": 0.100837841629982,
+      "rewards/rejected": -0.12340004742145538,
+      "step": 330
+    },
+    {
+      "epoch": 0.544,
+      "grad_norm": 0.7923767872883756,
+      "learning_rate": 1.8251134982782952e-05,
+      "log_odds_chosen": 1.8938252925872803,
+      "log_odds_ratio": -0.23439836502075195,
+      "logits/chosen": 30.1374454498291,
+      "logits/rejected": 31.97110939025879,
+      "logps/chosen": -0.25573670864105225,
+      "logps/rejected": -1.058814525604248,
+      "loss": 0.2256,
+      "nll_loss": 0.2036077082157135,
+      "rewards/accuracies": 0.9624999761581421,
+      "rewards/chosen": -0.025573670864105225,
+      "rewards/margins": 0.08030776679515839,
+      "rewards/rejected": -0.1058814525604248,
+      "step": 340
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.941711390597777,
+      "learning_rate": 1.8090169943749477e-05,
+      "log_odds_chosen": 1.890398621559143,
+      "log_odds_ratio": -0.23317229747772217,
+      "logits/chosen": 29.39468002319336,
+      "logits/rejected": 31.831829071044922,
+      "logps/chosen": -0.26384299993515015,
+      "logps/rejected": -1.0387157201766968,
+      "loss": 0.2102,
+      "nll_loss": 0.19437000155448914,
+      "rewards/accuracies": 0.949999988079071,
+      "rewards/chosen": -0.026384299620985985,
+      "rewards/margins": 0.0774872750043869,
+      "rewards/rejected": -0.10387158393859863,
+      "step": 350
+    },
+    {
+      "epoch": 0.576,
+      "grad_norm": 0.9511427594232491,
+      "learning_rate": 1.792289643355191e-05,
+      "log_odds_chosen": 2.1849417686462402,
+      "log_odds_ratio": -0.20272669196128845,
+      "logits/chosen": 28.536510467529297,
+      "logits/rejected": 31.184783935546875,
+      "logps/chosen": -0.21713824570178986,
+      "logps/rejected": -1.1126606464385986,
+      "loss": 0.196,
+      "nll_loss": 0.1743771880865097,
+      "rewards/accuracies": 0.9750000238418579,
+      "rewards/chosen": -0.021713824942708015,
+      "rewards/margins": 0.08955223858356476,
+      "rewards/rejected": -0.11126606166362762,
+      "step": 360
+    },
+    {
+      "epoch": 0.592,
+      "grad_norm": 1.3234030569644144,
+      "learning_rate": 1.7749444887041797e-05,
+      "log_odds_chosen": 2.2127633094787598,
+      "log_odds_ratio": -0.1819957047700882,
+      "logits/chosen": 28.459209442138672,
+      "logits/rejected": 30.699054718017578,
+      "logps/chosen": -0.19446463882923126,
+      "logps/rejected": -1.064205527305603,
+      "loss": 0.2095,
+      "nll_loss": 0.157680481672287,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.019446464255452156,
+      "rewards/margins": 0.08697408437728882,
+      "rewards/rejected": -0.10642055422067642,
+      "step": 370
+    },
+    {
+      "epoch": 0.608,
+      "grad_norm": 0.7887424541285082,
+      "learning_rate": 1.7569950556517566e-05,
+      "log_odds_chosen": 2.0950214862823486,
+      "log_odds_ratio": -0.2056867778301239,
+      "logits/chosen": 26.343616485595703,
+      "logits/rejected": 30.175817489624023,
+      "logps/chosen": -0.22634263336658478,
+      "logps/rejected": -1.0738613605499268,
+      "loss": 0.1995,
+      "nll_loss": 0.1639939546585083,
+      "rewards/accuracies": 0.9750000238418579,
+      "rewards/chosen": -0.022634264081716537,
+      "rewards/margins": 0.08475185930728912,
+      "rewards/rejected": -0.10738612711429596,
+      "step": 380
+    },
+    {
+      "epoch": 0.624,
+      "grad_norm": 0.8925541324230298,
+      "learning_rate": 1.7384553406258842e-05,
+      "log_odds_chosen": 2.156597852706909,
+      "log_odds_ratio": -0.1930965632200241,
+      "logits/chosen": 26.64847755432129,
+      "logits/rejected": 28.833852767944336,
+      "logps/chosen": -0.2270650863647461,
+      "logps/rejected": -1.0798944234848022,
+      "loss": 0.2017,
+      "nll_loss": 0.1784597635269165,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -0.022706512361764908,
+      "rewards/margins": 0.08528293669223785,
+      "rewards/rejected": -0.10798945277929306,
+      "step": 390
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 1.0272557886917288,
+      "learning_rate": 1.7193398003386514e-05,
+      "log_odds_chosen": 2.46358323097229,
+      "log_odds_ratio": -0.17916469275951385,
+      "logits/chosen": 28.102930068969727,
+      "logits/rejected": 29.97959327697754,
+      "logps/chosen": -0.21100203692913055,
+      "logps/rejected": -1.2793248891830444,
+      "loss": 0.1938,
+      "nll_loss": 0.1722893863916397,
+      "rewards/accuracies": 0.9750000238418579,
+      "rewards/chosen": -0.021100204437971115,
+      "rewards/margins": 0.10683228820562363,
+      "rewards/rejected": -0.12793250381946564,
+      "step": 400
+    },
+    {
+      "epoch": 0.656,
+      "grad_norm": 0.7328015065358303,
+      "learning_rate": 1.6996633405133656e-05,
+      "log_odds_chosen": 2.6361804008483887,
+      "log_odds_ratio": -0.14345327019691467,
+      "logits/chosen": 27.147369384765625,
+      "logits/rejected": 29.60892105102539,
+      "logps/chosen": -0.18612393736839294,
+      "logps/rejected": -1.346756935119629,
+      "loss": 0.1836,
+      "nll_loss": 0.14793823659420013,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.018612395972013474,
+      "rewards/margins": 0.11606329679489136,
+      "rewards/rejected": -0.13467569649219513,
+      "step": 410
+    },
+    {
+      "epoch": 0.672,
+      "grad_norm": 0.7570321808098591,
+      "learning_rate": 1.6794413042615168e-05,
+      "log_odds_chosen": 2.324608325958252,
+      "log_odds_ratio": -0.16639328002929688,
+      "logits/chosen": 26.477365493774414,
+      "logits/rejected": 29.83447265625,
+      "logps/chosen": -0.19487398862838745,
+      "logps/rejected": -1.086843729019165,
+      "loss": 0.1708,
+      "nll_loss": 0.15096671879291534,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.019487399607896805,
+      "rewards/margins": 0.08919696509838104,
+      "rewards/rejected": -0.10868436098098755,
+      "step": 420
+    },
+    {
+      "epoch": 0.688,
+      "grad_norm": 0.8589237619818589,
+      "learning_rate": 1.6586894601186804e-05,
+      "log_odds_chosen": 2.4892306327819824,
+      "log_odds_ratio": -0.1399163007736206,
+      "logits/chosen": 26.335521697998047,
+      "logits/rejected": 28.351062774658203,
+      "logps/chosen": -0.185403510928154,
+      "logps/rejected": -1.1550278663635254,
+      "loss": 0.1719,
+      "nll_loss": 0.15061026811599731,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -0.01854034885764122,
+      "rewards/margins": 0.09696242958307266,
+      "rewards/rejected": -0.11550277471542358,
+      "step": 430
+    },
+    {
+      "epoch": 0.704,
+      "grad_norm": 0.9410946432421042,
+      "learning_rate": 1.63742398974869e-05,
+      "log_odds_chosen": 2.6492185592651367,
+      "log_odds_ratio": -0.11772862821817398,
+      "logits/chosen": 25.387847900390625,
+      "logits/rejected": 27.318252563476562,
+      "logps/chosen": -0.18541622161865234,
+      "logps/rejected": -1.3411946296691895,
+      "loss": 0.168,
+      "nll_loss": 0.1478230208158493,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.018541622906923294,
+      "rewards/margins": 0.11557783931493759,
+      "rewards/rejected": -0.13411946594715118,
+      "step": 440
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.913121815013067,
+      "learning_rate": 1.6156614753256583e-05,
+      "log_odds_chosen": 2.5952377319335938,
+      "log_odds_ratio": -0.13427497446537018,
+      "logits/chosen": 25.367713928222656,
+      "logits/rejected": 26.90452003479004,
+      "logps/chosen": -0.1814824938774109,
+      "logps/rejected": -1.2440539598464966,
+      "loss": 0.1621,
+      "nll_loss": 0.13832086324691772,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -0.018148250877857208,
+      "rewards/margins": 0.10625714063644409,
+      "rewards/rejected": -0.1244053840637207,
+      "step": 450
+    },
+    {
+      "epoch": 0.736,
+      "grad_norm": 0.8697507877833889,
+      "learning_rate": 1.5934188866037017e-05,
+      "log_odds_chosen": 2.944627046585083,
+      "log_odds_ratio": -0.11624413728713989,
+      "logits/chosen": 25.267337799072266,
+      "logits/rejected": 29.403518676757812,
+      "logps/chosen": -0.17217525839805603,
+      "logps/rejected": -1.3934593200683594,
+      "loss": 0.1569,
+      "nll_loss": 0.14144101738929749,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -0.017217526212334633,
+      "rewards/margins": 0.12212841212749481,
+      "rewards/rejected": -0.1393459290266037,
+      "step": 460
+    },
+    {
+      "epoch": 0.752,
+      "grad_norm": 0.6621793550764674,
+      "learning_rate": 1.570713567684432e-05,
+      "log_odds_chosen": 2.606139659881592,
+      "log_odds_ratio": -0.1426299810409546,
+      "logits/chosen": 29.288564682006836,
+      "logits/rejected": 29.074045181274414,
+      "logps/chosen": -0.19370333850383759,
+      "logps/rejected": -1.2478258609771729,
+      "loss": 0.1792,
+      "nll_loss": 0.16615137457847595,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -0.019370336085557938,
+      "rewards/margins": 0.10541225969791412,
+      "rewards/rejected": -0.12478259950876236,
+      "step": 470
+    },
+    {
+      "epoch": 0.768,
+      "grad_norm": 0.6543673919224716,
+      "learning_rate": 1.5475632234925505e-05,
+      "log_odds_chosen": 2.9211063385009766,
+      "log_odds_ratio": -0.105687215924263,
+      "logits/chosen": 24.23305892944336,
+      "logits/rejected": 27.091516494750977,
+      "logps/chosen": -0.1637393683195114,
+      "logps/rejected": -1.3047716617584229,
+      "loss": 0.1509,
+      "nll_loss": 0.12331098318099976,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.01637393794953823,
+      "rewards/margins": 0.11410322040319443,
+      "rewards/rejected": -0.1304771602153778,
+      "step": 480
+    },
+    {
+      "epoch": 0.784,
+      "grad_norm": 0.741926124626365,
+      "learning_rate": 1.5239859059700794e-05,
+      "log_odds_chosen": 2.723001003265381,
+      "log_odds_ratio": -0.13461025059223175,
+      "logits/chosen": 25.58415412902832,
+      "logits/rejected": 27.281457901000977,
+      "logps/chosen": -0.17167863249778748,
+      "logps/rejected": -1.1783361434936523,
+      "loss": 0.149,
+      "nll_loss": 0.1352442055940628,
+      "rewards/accuracies": 0.9750000238418579,
+      "rewards/chosen": -0.017167862504720688,
+      "rewards/margins": 0.10066574811935425,
+      "rewards/rejected": -0.11783361434936523,
+      "step": 490
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 1.4262079801129468,
+      "learning_rate": 1.5000000000000002e-05,
+      "log_odds_chosen": 2.723966121673584,
+      "log_odds_ratio": -0.116343654692173,
+      "logits/chosen": 22.94664764404297,
+      "logits/rejected": 26.360061645507812,
+      "logps/chosen": -0.17480744421482086,
+      "logps/rejected": -1.2685492038726807,
+      "loss": 0.145,
+      "nll_loss": 0.13372007012367249,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.017480744048953056,
+      "rewards/margins": 0.10937418043613434,
+      "rewards/rejected": -0.12685491144657135,
+      "step": 500
+    },
+    {
+      "epoch": 0.816,
+      "grad_norm": 1.961973010059407,
+      "learning_rate": 1.4756242090702756e-05,
+      "log_odds_chosen": 2.9430527687072754,
+      "log_odds_ratio": -0.1177729144692421,
+      "logits/chosen": 24.766284942626953,
+      "logits/rejected": 28.823827743530273,
+      "logps/chosen": -0.16511258482933044,
+      "logps/rejected": -1.238128423690796,
+      "loss": 0.1441,
+      "nll_loss": 0.12999919056892395,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -0.016511257737874985,
+      "rewards/margins": 0.10730159282684326,
+      "rewards/rejected": -0.12381285429000854,
+      "step": 510
+    },
+    {
+      "epoch": 0.832,
+      "grad_norm": 0.8171513438848954,
+      "learning_rate": 1.4508775406894308e-05,
+      "log_odds_chosen": 2.880876064300537,
+      "log_odds_ratio": -0.10505475103855133,
+      "logits/chosen": 22.384557723999023,
+      "logits/rejected": 26.74051856994629,
+      "logps/chosen": -0.1716020405292511,
+      "logps/rejected": -1.2871472835540771,
+      "loss": 0.1375,
+      "nll_loss": 0.12949416041374207,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.01716020330786705,
+      "rewards/margins": 0.11155451834201813,
+      "rewards/rejected": -0.12871472537517548,
+      "step": 520
+    },
+    {
+      "epoch": 0.848,
+      "grad_norm": 0.6656893252356583,
+      "learning_rate": 1.4257792915650728e-05,
+      "log_odds_chosen": 3.0356669425964355,
+      "log_odds_ratio": -0.10586954653263092,
+      "logits/chosen": 19.979793548583984,
+      "logits/rejected": 23.590118408203125,
+      "logps/chosen": -0.1698727309703827,
+      "logps/rejected": -1.3184887170791626,
+      "loss": 0.127,
+      "nll_loss": 0.12099827826023102,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.0169872734695673,
+      "rewards/margins": 0.11486159265041351,
+      "rewards/rejected": -0.13184888660907745,
+      "step": 530
+    },
+    {
+      "epoch": 0.864,
+      "grad_norm": 0.5859755094542713,
+      "learning_rate": 1.4003490325568953e-05,
+      "log_odds_chosen": 3.2319984436035156,
+      "log_odds_ratio": -0.09660033136606216,
+      "logits/chosen": 21.624605178833008,
+      "logits/rejected": 24.835023880004883,
+      "logps/chosen": -0.1984686404466629,
+      "logps/rejected": -1.5682591199874878,
+      "loss": 0.1372,
+      "nll_loss": 0.12878955900669098,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -0.01984686590731144,
+      "rewards/margins": 0.13697902858257294,
+      "rewards/rejected": -0.1568259298801422,
+      "step": 540
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.779222909675585,
+      "learning_rate": 1.3746065934159123e-05,
+      "log_odds_chosen": 3.039996862411499,
+      "log_odds_ratio": -0.10127159208059311,
+      "logits/chosen": 25.058979034423828,
+      "logits/rejected": 27.096914291381836,
+      "logps/chosen": -0.18606023490428925,
+      "logps/rejected": -1.4767307043075562,
+      "loss": 0.137,
+      "nll_loss": 0.14917424321174622,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.018606023862957954,
+      "rewards/margins": 0.12906703352928162,
+      "rewards/rejected": -0.1476730853319168,
+      "step": 550
+    },
+    {
+      "epoch": 0.896,
+      "grad_norm": 0.5131532913955095,
+      "learning_rate": 1.3485720473218153e-05,
+      "log_odds_chosen": 3.5183768272399902,
+      "log_odds_ratio": -0.08511906862258911,
+      "logits/chosen": 21.076614379882812,
+      "logits/rejected": 24.55303955078125,
+      "logps/chosen": -0.15598443150520325,
+      "logps/rejected": -1.6431341171264648,
+      "loss": 0.1361,
+      "nll_loss": 0.12268916517496109,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.015598444268107414,
+      "rewards/margins": 0.14871497452259064,
+      "rewards/rejected": -0.164313405752182,
+      "step": 560
+    },
+    {
+      "epoch": 0.912,
+      "grad_norm": 0.7652230384669911,
+      "learning_rate": 1.3222656952305113e-05,
+      "log_odds_chosen": 3.6853690147399902,
+      "log_odds_ratio": -0.08712838590145111,
+      "logits/chosen": 20.190637588500977,
+      "logits/rejected": 25.091556549072266,
+      "logps/chosen": -0.13156656920909882,
+      "logps/rejected": -1.6401069164276123,
+      "loss": 0.121,
+      "nll_loss": 0.10304377228021622,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.01315665990114212,
+      "rewards/margins": 0.15085402131080627,
+      "rewards/rejected": -0.1640106737613678,
+      "step": 570
+    },
+    {
+      "epoch": 0.928,
+      "grad_norm": 0.7300290151409546,
+      "learning_rate": 1.2957080500440469e-05,
+      "log_odds_chosen": 3.642732620239258,
+      "log_odds_ratio": -0.07864442467689514,
+      "logits/chosen": 19.066701889038086,
+      "logits/rejected": 23.404861450195312,
+      "logps/chosen": -0.14534644782543182,
+      "logps/rejected": -1.6337239742279053,
+      "loss": 0.1262,
+      "nll_loss": 0.11143051087856293,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.014534644782543182,
+      "rewards/margins": 0.14883777499198914,
+      "rewards/rejected": -0.16337242722511292,
+      "step": 580
+    },
+    {
+      "epoch": 0.944,
+      "grad_norm": 0.581668451170283,
+      "learning_rate": 1.2689198206152657e-05,
+      "log_odds_chosen": 3.713766098022461,
+      "log_odds_ratio": -0.0588177926838398,
+      "logits/chosen": 22.885316848754883,
+      "logits/rejected": 25.525924682617188,
+      "logps/chosen": -0.14779958128929138,
+      "logps/rejected": -1.8394057750701904,
+      "loss": 0.1234,
+      "nll_loss": 0.11549869924783707,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.014779957942664623,
+      "rewards/margins": 0.1691606342792511,
+      "rewards/rejected": -0.1839405745267868,
+      "step": 590
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.566646644489029,
+      "learning_rate": 1.2419218955996677e-05,
+      "log_odds_chosen": 3.757345199584961,
+      "log_odds_ratio": -0.06075813248753548,
+      "logits/chosen": 17.20478057861328,
+      "logits/rejected": 22.157577514648438,
+      "logps/chosen": -0.11853750050067902,
+      "logps/rejected": -1.5277284383773804,
+      "loss": 0.1127,
+      "nll_loss": 0.09783687442541122,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.011853748932480812,
+      "rewards/margins": 0.14091908931732178,
+      "rewards/rejected": -0.15277281403541565,
+      "step": 600
+    },
+    {
+      "epoch": 0.976,
+      "grad_norm": 0.6375581513651442,
+      "learning_rate": 1.2147353271670634e-05,
+      "log_odds_chosen": 3.8227837085723877,
+      "log_odds_ratio": -0.09637900441884995,
+      "logits/chosen": 16.64263916015625,
+      "logits/rejected": 21.630008697509766,
+      "logps/chosen": -0.1534954011440277,
+      "logps/rejected": -1.8045285940170288,
+      "loss": 0.1165,
+      "nll_loss": 0.11545213311910629,
+      "rewards/accuracies": 0.9750000238418579,
+      "rewards/chosen": -0.015349539928138256,
+      "rewards/margins": 0.16510331630706787,
+      "rewards/rejected": -0.1804528534412384,
+      "step": 610
+    },
+    {
+      "epoch": 0.992,
+      "grad_norm": 0.5653677883458513,
+      "learning_rate": 1.187381314585725e-05,
+      "log_odds_chosen": 3.73738431930542,
+      "log_odds_ratio": -0.06889805942773819,
+      "logits/chosen": 20.336971282958984,
+      "logits/rejected": 23.924381256103516,
+      "logps/chosen": -0.12372555583715439,
+      "logps/rejected": -1.6133959293365479,
+      "loss": 0.1249,
+      "nll_loss": 0.1038857102394104,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.012372555211186409,
+      "rewards/margins": 0.14896704256534576,
+      "rewards/rejected": -0.16133958101272583,
+      "step": 620
+    },
+    {
+      "epoch": 1.008,
+      "grad_norm": 0.41947247295916856,
+      "learning_rate": 1.159881187691835e-05,
+      "log_odds_chosen": 3.937540054321289,
+      "log_odds_ratio": -0.05745882913470268,
+      "logits/chosen": 18.518375396728516,
+      "logits/rejected": 22.057506561279297,
+      "logps/chosen": -0.12291697412729263,
+      "logps/rejected": -1.742640495300293,
+      "loss": 0.1059,
+      "nll_loss": 0.09989026188850403,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.012291697785258293,
+      "rewards/margins": 0.16197235882282257,
+      "rewards/rejected": -0.17426404356956482,
+      "step": 630
+    },
+    {
+      "epoch": 1.024,
+      "grad_norm": 0.6597379471767634,
+      "learning_rate": 1.1322563902571227e-05,
+      "log_odds_chosen": 4.9311842918396,
+      "log_odds_ratio": -0.028991302475333214,
+      "logits/chosen": 12.143542289733887,
+      "logits/rejected": 17.73841094970703,
+      "logps/chosen": -0.06773394346237183,
+      "logps/rejected": -2.039699077606201,
+      "loss": 0.0792,
+      "nll_loss": 0.07453403621912003,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.0067733945325016975,
+      "rewards/margins": 0.19719651341438293,
+      "rewards/rejected": -0.20396992564201355,
+      "step": 640
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.8720926641059171,
+      "learning_rate": 1.1045284632676535e-05,
+      "log_odds_chosen": 4.683058261871338,
+      "log_odds_ratio": -0.025352224707603455,
+      "logits/chosen": 13.8761625289917,
+      "logits/rejected": 19.721921920776367,
+      "logps/chosen": -0.08715043216943741,
+      "logps/rejected": -1.9542983770370483,
+      "loss": 0.0855,
+      "nll_loss": 0.0856931135058403,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.008715043775737286,
+      "rewards/margins": 0.18671482801437378,
+      "rewards/rejected": -0.19542984664440155,
+      "step": 650
+    },
+    {
+      "epoch": 1.056,
+      "grad_norm": 0.5401242210481009,
+      "learning_rate": 1.0767190281268187e-05,
+      "log_odds_chosen": 4.751006603240967,
+      "log_odds_ratio": -0.03235021233558655,
+      "logits/chosen": 14.378767013549805,
+      "logits/rejected": 19.829517364501953,
+      "logps/chosen": -0.07587035000324249,
+      "logps/rejected": -2.011232614517212,
+      "loss": 0.0829,
+      "nll_loss": 0.07639533281326294,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.007587035186588764,
+      "rewards/margins": 0.19353622198104858,
+      "rewards/rejected": -0.20112323760986328,
+      "step": 660
+    },
+    {
+      "epoch": 1.072,
+      "grad_norm": 0.4975498251334291,
+      "learning_rate": 1.0488497697956134e-05,
+      "log_odds_chosen": 4.578000068664551,
+      "log_odds_ratio": -0.03245697170495987,
+      "logits/chosen": 13.391039848327637,
+      "logits/rejected": 18.039913177490234,
+      "logps/chosen": -0.08072780072689056,
+      "logps/rejected": -1.7768867015838623,
+      "loss": 0.0835,
+      "nll_loss": 0.07963033765554428,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.008072780445218086,
+      "rewards/margins": 0.16961589455604553,
+      "rewards/rejected": -0.17768868803977966,
+      "step": 670
+    },
+    {
+      "epoch": 1.088,
+      "grad_norm": 0.4579496342491343,
+      "learning_rate": 1.0209424198833571e-05,
+      "log_odds_chosen": 4.71366024017334,
+      "log_odds_ratio": -0.029907289892435074,
+      "logits/chosen": 11.174861907958984,
+      "logits/rejected": 16.83877182006836,
+      "logps/chosen": -0.08100271224975586,
+      "logps/rejected": -1.965436339378357,
+      "loss": 0.0782,
+      "nll_loss": 0.07603878527879715,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.008100271224975586,
+      "rewards/margins": 0.1884433925151825,
+      "rewards/rejected": -0.19654366374015808,
+      "step": 680
+    },
+    {
+      "epoch": 1.104,
+      "grad_norm": 0.41635397248386163,
+      "learning_rate": 9.930187397020385e-06,
+      "log_odds_chosen": 4.987896919250488,
+      "log_odds_ratio": -0.029992487281560898,
+      "logits/chosen": 10.937190055847168,
+      "logits/rejected": 16.221263885498047,
+      "logps/chosen": -0.0843561664223671,
+      "logps/rejected": -1.950840950012207,
+      "loss": 0.0802,
+      "nll_loss": 0.07738173007965088,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.00843561626970768,
+      "rewards/margins": 0.18664847314357758,
+      "rewards/rejected": -0.1950840950012207,
+      "step": 690
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.42432934191231536,
+      "learning_rate": 9.651005032974994e-06,
+      "log_odds_chosen": 5.254538059234619,
+      "log_odds_ratio": -0.019603287801146507,
+      "logits/chosen": 12.815790176391602,
+      "logits/rejected": 16.598247528076172,
+      "logps/chosen": -0.07913420349359512,
+      "logps/rejected": -2.306891918182373,
+      "loss": 0.0795,
+      "nll_loss": 0.07783981412649155,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.007913419976830482,
+      "rewards/margins": 0.22277574241161346,
+      "rewards/rejected": -0.23068919777870178,
+      "step": 700
+    },
+    {
+      "epoch": 1.1360000000000001,
+      "grad_norm": 0.5144220041987062,
+      "learning_rate": 9.372094804706867e-06,
+      "log_odds_chosen": 5.051201820373535,
+      "log_odds_ratio": -0.022446967661380768,
+      "logits/chosen": 8.465599060058594,
+      "logits/rejected": 15.262280464172363,
+      "logps/chosen": -0.04941285401582718,
+      "logps/rejected": -1.7998939752578735,
+      "loss": 0.0699,
+      "nll_loss": 0.06099081039428711,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.0049412851221859455,
+      "rewards/margins": 0.1750481128692627,
+      "rewards/rejected": -0.17998939752578735,
+      "step": 710
+    },
+    {
+      "epoch": 1.152,
+      "grad_norm": 0.4355953213178483,
+      "learning_rate": 9.093674198022201e-06,
+      "log_odds_chosen": 4.678675651550293,
+      "log_odds_ratio": -0.03710417076945305,
+      "logits/chosen": 9.093629837036133,
+      "logits/rejected": 15.341288566589355,
+      "logps/chosen": -0.0850113034248352,
+      "logps/rejected": -1.8611046075820923,
+      "loss": 0.0802,
+      "nll_loss": 0.08102207630872726,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -0.008501130156219006,
+      "rewards/margins": 0.17760933935642242,
+      "rewards/rejected": -0.1861104667186737,
+      "step": 720
+    },
+    {
+      "epoch": 1.168,
+      "grad_norm": 0.4183273277614356,
+      "learning_rate": 8.815960316934991e-06,
+      "log_odds_chosen": 4.754403591156006,
+      "log_odds_ratio": -0.026562413200736046,
+      "logits/chosen": 11.237001419067383,
+      "logits/rejected": 16.33772087097168,
+      "logps/chosen": -0.0855894535779953,
+      "logps/rejected": -2.021867275238037,
+      "loss": 0.074,
+      "nll_loss": 0.07607786357402802,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.00855894573032856,
+      "rewards/margins": 0.1936277598142624,
+      "rewards/rejected": -0.2021867036819458,
+      "step": 730
+    },
+    {
+      "epoch": 1.184,
+      "grad_norm": 0.5708367601297668,
+      "learning_rate": 8.539169714375885e-06,
+      "log_odds_chosen": 5.499794960021973,
+      "log_odds_ratio": -0.015643080696463585,
+      "logits/chosen": 11.671289443969727,
+      "logits/rejected": 15.871813774108887,
+      "logps/chosen": -0.0698837861418724,
+      "logps/rejected": -2.4058656692504883,
+      "loss": 0.0758,
+      "nll_loss": 0.06816570460796356,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.006988379172980785,
+      "rewards/margins": 0.23359815776348114,
+      "rewards/rejected": -0.2405865490436554,
+      "step": 740
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.46974989560288,
+      "learning_rate": 8.263518223330698e-06,
+      "log_odds_chosen": 5.374692916870117,
+      "log_odds_ratio": -0.01755271479487419,
+      "logits/chosen": 10.17614459991455,
+      "logits/rejected": 15.561047554016113,
+      "logps/chosen": -0.06151513010263443,
+      "logps/rejected": -2.1926753520965576,
+      "loss": 0.0765,
+      "nll_loss": 0.06682710349559784,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.006151513196527958,
+      "rewards/margins": 0.21311601996421814,
+      "rewards/rejected": -0.21926753222942352,
+      "step": 750
+    },
+    {
+      "epoch": 1.216,
+      "grad_norm": 0.536798190903821,
+      "learning_rate": 7.989220788540356e-06,
+      "log_odds_chosen": 5.090784549713135,
+      "log_odds_ratio": -0.024079417809844017,
+      "logits/chosen": 10.42772102355957,
+      "logits/rejected": 16.83907127380371,
+      "logps/chosen": -0.0742868185043335,
+      "logps/rejected": -1.9404573440551758,
+      "loss": 0.0767,
+      "nll_loss": 0.07289760559797287,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.007428681943565607,
+      "rewards/margins": 0.18661704659461975,
+      "rewards/rejected": -0.19404573738574982,
+      "step": 760
+    },
+    {
+      "epoch": 1.232,
+      "grad_norm": 0.7535095663117326,
+      "learning_rate": 7.716491298893443e-06,
+      "log_odds_chosen": 5.1247100830078125,
+      "log_odds_ratio": -0.027670254930853844,
+      "logits/chosen": 13.019775390625,
+      "logits/rejected": 18.80875015258789,
+      "logps/chosen": -0.08515826612710953,
+      "logps/rejected": -2.227242946624756,
+      "loss": 0.0799,
+      "nll_loss": 0.08172322809696198,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.008515825495123863,
+      "rewards/margins": 0.2142084836959839,
+      "rewards/rejected": -0.2227243185043335,
+      "step": 770
+    },
+    {
+      "epoch": 1.248,
+      "grad_norm": 0.43802198454777963,
+      "learning_rate": 7.445542420642097e-06,
+      "log_odds_chosen": 4.944367408752441,
+      "log_odds_ratio": -0.019510487094521523,
+      "logits/chosen": 12.673688888549805,
+      "logits/rejected": 17.701494216918945,
+      "logps/chosen": -0.06077063828706741,
+      "logps/rejected": -1.8956024646759033,
+      "loss": 0.0722,
+      "nll_loss": 0.06931096315383911,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.0060770632699131966,
+      "rewards/margins": 0.18348316848278046,
+      "rewards/rejected": -0.18956024944782257,
+      "step": 780
+    },
+    {
+      "epoch": 1.264,
+      "grad_norm": 0.5918184881161849,
+      "learning_rate": 7.176585431571235e-06,
+      "log_odds_chosen": 4.934049606323242,
+      "log_odds_ratio": -0.022184232249855995,
+      "logits/chosen": 10.117755889892578,
+      "logits/rejected": 16.32207489013672,
+      "logps/chosen": -0.06841601431369781,
+      "logps/rejected": -1.8420988321304321,
+      "loss": 0.076,
+      "nll_loss": 0.07008203864097595,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.006841603666543961,
+      "rewards/margins": 0.17736826837062836,
+      "rewards/rejected": -0.1842098832130432,
+      "step": 790
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.4109794535608094,
+      "learning_rate": 6.909830056250527e-06,
+      "log_odds_chosen": 5.385331153869629,
+      "log_odds_ratio": -0.01698954775929451,
+      "logits/chosen": 9.808797836303711,
+      "logits/rejected": 15.439534187316895,
+      "logps/chosen": -0.06334672123193741,
+      "logps/rejected": -2.0740480422973633,
+      "loss": 0.0735,
+      "nll_loss": 0.06666533648967743,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.006334672216325998,
+      "rewards/margins": 0.2010701447725296,
+      "rewards/rejected": -0.20740482211112976,
+      "step": 800
+    },
+    {
+      "epoch": 1.296,
+      "grad_norm": 0.5786524901744241,
+      "learning_rate": 6.645484302497452e-06,
+      "log_odds_chosen": 5.305611610412598,
+      "log_odds_ratio": -0.015638932585716248,
+      "logits/chosen": 10.026269912719727,
+      "logits/rejected": 16.038047790527344,
+      "logps/chosen": -0.058640025556087494,
+      "logps/rejected": -2.095346212387085,
+      "loss": 0.0687,
+      "nll_loss": 0.0625784695148468,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.005864003207534552,
+      "rewards/margins": 0.20367059111595154,
+      "rewards/rejected": -0.20953461527824402,
+      "step": 810
+    },
+    {
+      "epoch": 1.312,
+      "grad_norm": 0.413245918888085,
+      "learning_rate": 6.383754299179079e-06,
+      "log_odds_chosen": 5.338342189788818,
+      "log_odds_ratio": -0.016953352838754654,
+      "logits/chosen": 10.80897331237793,
+      "logits/rejected": 16.324369430541992,
+      "logps/chosen": -0.07062120735645294,
+      "logps/rejected": -2.1152849197387695,
+      "loss": 0.0753,
+      "nll_loss": 0.07253970950841904,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.007062119897454977,
+      "rewards/margins": 0.20446637272834778,
+      "rewards/rejected": -0.2115284949541092,
+      "step": 820
+    },
+    {
+      "epoch": 1.328,
+      "grad_norm": 0.45473358832367533,
+      "learning_rate": 6.124844135478971e-06,
+      "log_odds_chosen": 4.9816741943359375,
+      "log_odds_ratio": -0.016600508242845535,
+      "logits/chosen": 12.60502815246582,
+      "logits/rejected": 17.482601165771484,
+      "logps/chosen": -0.07652293890714645,
+      "logps/rejected": -1.9486392736434937,
+      "loss": 0.0753,
+      "nll_loss": 0.0775555893778801,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.007652295287698507,
+      "rewards/margins": 0.18721160292625427,
+      "rewards/rejected": -0.1948639154434204,
+      "step": 830
+    },
+    {
+      "epoch": 1.3439999999999999,
+      "grad_norm": 0.43221707053355657,
+      "learning_rate": 5.868955701754584e-06,
+      "log_odds_chosen": 5.058526039123535,
+      "log_odds_ratio": -0.02234785258769989,
+      "logits/chosen": 10.739975929260254,
+      "logits/rejected": 14.592979431152344,
+      "logps/chosen": -0.08031310141086578,
+      "logps/rejected": -2.022430896759033,
+      "loss": 0.071,
+      "nll_loss": 0.07450385391712189,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.008031310513615608,
+      "rewards/margins": 0.19421175122261047,
+      "rewards/rejected": -0.20224305987358093,
+      "step": 840
+    },
+    {
+      "epoch": 1.3599999999999999,
+      "grad_norm": 0.36475019984330276,
+      "learning_rate": 5.616288532109225e-06,
+      "log_odds_chosen": 5.216296672821045,
+      "log_odds_ratio": -0.02249590866267681,
+      "logits/chosen": 10.627111434936523,
+      "logits/rejected": 16.160968780517578,
+      "logps/chosen": -0.08645172417163849,
+      "logps/rejected": -2.255333662033081,
+      "loss": 0.0741,
+      "nll_loss": 0.07864928245544434,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.008645173162221909,
+      "rewards/margins": 0.2168881893157959,
+      "rewards/rejected": -0.2255333662033081,
+      "step": 850
+    },
+    {
+      "epoch": 1.376,
+      "grad_norm": 0.5386028507331131,
+      "learning_rate": 5.367039648801386e-06,
+      "log_odds_chosen": 5.349644184112549,
+      "log_odds_ratio": -0.019762977957725525,
+      "logits/chosen": 11.73155689239502,
+      "logits/rejected": 16.905214309692383,
+      "logps/chosen": -0.07144279778003693,
+      "logps/rejected": -2.1353299617767334,
+      "loss": 0.073,
+      "nll_loss": 0.07140424847602844,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.007144278846681118,
+      "rewards/margins": 0.20638874173164368,
+      "rewards/rejected": -0.21353299915790558,
+      "step": 860
+    },
+    {
+      "epoch": 1.392,
+      "grad_norm": 0.4861620483022773,
+      "learning_rate": 5.121403408612672e-06,
+      "log_odds_chosen": 5.4971466064453125,
+      "log_odds_ratio": -0.013068562373518944,
+      "logits/chosen": 10.185466766357422,
+      "logits/rejected": 15.52686595916748,
+      "logps/chosen": -0.05803658813238144,
+      "logps/rejected": -2.2335634231567383,
+      "loss": 0.071,
+      "nll_loss": 0.06299170106649399,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.005803658626973629,
+      "rewards/margins": 0.21755269169807434,
+      "rewards/rejected": -0.22335636615753174,
+      "step": 870
+    },
+    {
+      "epoch": 1.408,
+      "grad_norm": 0.7221352144813555,
+      "learning_rate": 4.879571351294287e-06,
+      "log_odds_chosen": 5.298310279846191,
+      "log_odds_ratio": -0.015498277731239796,
+      "logits/chosen": 8.6812744140625,
+      "logits/rejected": 14.908788681030273,
+      "logps/chosen": -0.04598275199532509,
+      "logps/rejected": -1.8670889139175415,
+      "loss": 0.0695,
+      "nll_loss": 0.05738530308008194,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.004598274827003479,
+      "rewards/margins": 0.18211060762405396,
+      "rewards/rejected": -0.18670889735221863,
+      "step": 880
+    },
+    {
+      "epoch": 1.424,
+      "grad_norm": 0.4395795057245911,
+      "learning_rate": 4.641732050210032e-06,
+      "log_odds_chosen": 5.754244327545166,
+      "log_odds_ratio": -0.013112393207848072,
+      "logits/chosen": 10.254205703735352,
+      "logits/rejected": 16.640913009643555,
+      "logps/chosen": -0.07648586481809616,
+      "logps/rejected": -2.492318630218506,
+      "loss": 0.0712,
+      "nll_loss": 0.07544175535440445,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.007648587226867676,
+      "rewards/margins": 0.24158325791358948,
+      "rewards/rejected": -0.24923184514045715,
+      "step": 890
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.6571181936107011,
+      "learning_rate": 4.408070965292534e-06,
+      "log_odds_chosen": 5.259117126464844,
+      "log_odds_ratio": -0.016550693660974503,
+      "logits/chosen": 10.593389511108398,
+      "logits/rejected": 15.269793510437012,
+      "logps/chosen": -0.08531410992145538,
+      "logps/rejected": -2.2608280181884766,
+      "loss": 0.0758,
+      "nll_loss": 0.07715515047311783,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.008531411178410053,
+      "rewards/margins": 0.21755139529705048,
+      "rewards/rejected": -0.22608280181884766,
+      "step": 900
+    },
+    {
+      "epoch": 1.456,
+      "grad_norm": 0.45529898147163606,
+      "learning_rate": 4.178770298427107e-06,
+      "log_odds_chosen": 5.506943702697754,
+      "log_odds_ratio": -0.017402973026037216,
+      "logits/chosen": 10.588479995727539,
+      "logits/rejected": 16.51026153564453,
+      "logps/chosen": -0.06945344805717468,
+      "logps/rejected": -2.281757354736328,
+      "loss": 0.0701,
+      "nll_loss": 0.07317786663770676,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.006945345550775528,
+      "rewards/margins": 0.2212304174900055,
+      "rewards/rejected": -0.22817575931549072,
+      "step": 910
+    },
+    {
+      "epoch": 1.472,
+      "grad_norm": 0.4799597661016462,
+      "learning_rate": 3.954008851376252e-06,
+      "log_odds_chosen": 5.671328544616699,
+      "log_odds_ratio": -0.01591522805392742,
+      "logits/chosen": 7.850234031677246,
+      "logits/rejected": 15.029336929321289,
+      "logps/chosen": -0.04982997104525566,
+      "logps/rejected": -2.066762685775757,
+      "loss": 0.0637,
+      "nll_loss": 0.059512246400117874,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.0049829971976578236,
+      "rewards/margins": 0.20169326663017273,
+      "rewards/rejected": -0.20667627453804016,
+      "step": 920
+    },
+    {
+      "epoch": 1.488,
+      "grad_norm": 0.42918774258741327,
+      "learning_rate": 3.7339618863553983e-06,
+      "log_odds_chosen": 5.558573246002197,
+      "log_odds_ratio": -0.014333250001072884,
+      "logits/chosen": 8.254855155944824,
+      "logits/rejected": 14.497201919555664,
+      "logps/chosen": -0.06610764563083649,
+      "logps/rejected": -2.231790781021118,
+      "loss": 0.0638,
+      "nll_loss": 0.06769216060638428,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.0066107651218771935,
+      "rewards/margins": 0.21656832098960876,
+      "rewards/rejected": -0.22317907214164734,
+      "step": 930
+    },
+    {
+      "epoch": 1.504,
+      "grad_norm": 0.39881406218102516,
+      "learning_rate": 3.5188009893686916e-06,
+      "log_odds_chosen": 5.319655418395996,
+      "log_odds_ratio": -0.0184173621237278,
+      "logits/chosen": 10.073064804077148,
+      "logits/rejected": 15.57422924041748,
+      "logps/chosen": -0.07947548478841782,
+      "logps/rejected": -2.1314804553985596,
+      "loss": 0.0699,
+      "nll_loss": 0.07717884331941605,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.007947548292577267,
+      "rewards/margins": 0.20520050823688507,
+      "rewards/rejected": -0.21314804255962372,
+      "step": 940
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.3003778882690616,
+      "learning_rate": 3.308693936411421e-06,
+      "log_odds_chosen": 5.652432441711426,
+      "log_odds_ratio": -0.018848979845643044,
+      "logits/chosen": 7.4497809410095215,
+      "logits/rejected": 14.933306694030762,
+      "logps/chosen": -0.0582701750099659,
+      "logps/rejected": -2.193925142288208,
+      "loss": 0.0677,
+      "nll_loss": 0.06635381281375885,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.005827018059790134,
+      "rewards/margins": 0.21356551349163055,
+      "rewards/rejected": -0.21939250826835632,
+      "step": 950
+    },
+    {
+      "epoch": 1.536,
+      "grad_norm": 0.3009824619938579,
+      "learning_rate": 3.103804562643302e-06,
+      "log_odds_chosen": 5.427401065826416,
+      "log_odds_ratio": -0.014070016331970692,
+      "logits/chosen": 8.830900192260742,
+      "logits/rejected": 14.178878784179688,
+      "logps/chosen": -0.06140875071287155,
+      "logps/rejected": -2.2172341346740723,
+      "loss": 0.0645,
+      "nll_loss": 0.06464984267950058,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.0061408756300807,
+      "rewards/margins": 0.21558256447315216,
+      "rewards/rejected": -0.22172346711158752,
+      "step": 960
+    },
+    {
+      "epoch": 1.552,
+      "grad_norm": 0.4056333088268534,
+      "learning_rate": 2.9042926346347932e-06,
+      "log_odds_chosen": 5.441522598266602,
+      "log_odds_ratio": -0.014954231679439545,
+      "logits/chosen": 9.046590805053711,
+      "logits/rejected": 15.852540969848633,
+      "logps/chosen": -0.06886409223079681,
+      "logps/rejected": -2.157127618789673,
+      "loss": 0.0658,
+      "nll_loss": 0.06967596709728241,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.006886409129947424,
+      "rewards/margins": 0.20882634818553925,
+      "rewards/rejected": -0.2157127559185028,
+      "step": 970
+    },
+    {
+      "epoch": 1.568,
+      "grad_norm": 0.3196857185408296,
+      "learning_rate": 2.7103137257858867e-06,
+      "log_odds_chosen": 5.551312446594238,
+      "log_odds_ratio": -0.011502685025334358,
+      "logits/chosen": 7.783066749572754,
+      "logits/rejected": 14.421549797058105,
+      "logps/chosen": -0.05041471868753433,
+      "logps/rejected": -2.1501784324645996,
+      "loss": 0.0606,
+      "nll_loss": 0.05976518243551254,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.005041471216827631,
+      "rewards/margins": 0.20997636020183563,
+      "rewards/rejected": -0.21501782536506653,
+      "step": 980
+    },
+    {
+      "epoch": 1.584,
+      "grad_norm": 0.5263216429559819,
+      "learning_rate": 2.522019095014683e-06,
+      "log_odds_chosen": 5.743101119995117,
+      "log_odds_ratio": -0.010009285993874073,
+      "logits/chosen": 7.407317161560059,
+      "logits/rejected": 14.168241500854492,
+      "logps/chosen": -0.048953376710414886,
+      "logps/rejected": -2.123490810394287,
+      "loss": 0.0686,
+      "nll_loss": 0.058644842356443405,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.004895338322967291,
+      "rewards/margins": 0.20745377242565155,
+      "rewards/rejected": -0.2123490869998932,
+      "step": 990
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.36561256966761685,
+      "learning_rate": 2.339555568810221e-06,
+      "log_odds_chosen": 5.752957344055176,
+      "log_odds_ratio": -0.008349338546395302,
+      "logits/chosen": 7.5777788162231445,
+      "logits/rejected": 14.192586898803711,
+      "logps/chosen": -0.043931931257247925,
+      "logps/rejected": -2.1989071369171143,
+      "loss": 0.0669,
+      "nll_loss": 0.053385306149721146,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.004393193405121565,
+      "rewards/margins": 0.21549753844738007,
+      "rewards/rejected": -0.21989071369171143,
+      "step": 1000
+    },
+    {
+      "epoch": 1.616,
+      "grad_norm": 0.5870370963384035,
+      "learning_rate": 2.163065426741603e-06,
+      "log_odds_chosen": 5.90677547454834,
+      "log_odds_ratio": -0.012085262686014175,
+      "logits/chosen": 6.974325656890869,
+      "logits/rejected": 14.128486633300781,
+      "logps/chosen": -0.0529697947204113,
+      "logps/rejected": -2.190331220626831,
+      "loss": 0.0645,
+      "nll_loss": 0.05988318473100662,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.0052969790995121,
+      "rewards/margins": 0.2137361317873001,
+      "rewards/rejected": -0.2190331220626831,
+      "step": 1010
+    },
+    {
+      "epoch": 1.6320000000000001,
+      "grad_norm": 0.6243099880494768,
+      "learning_rate": 1.9926862905126663e-06,
+      "log_odds_chosen": 5.65887975692749,
+      "log_odds_ratio": -0.014939057640731335,
+      "logits/chosen": 7.151690483093262,
+      "logits/rejected": 13.71330451965332,
+      "logps/chosen": -0.055140770971775055,
+      "logps/rejected": -2.053590774536133,
+      "loss": 0.0673,
+      "nll_loss": 0.06196935102343559,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.005514076910912991,
+      "rewards/margins": 0.19984501600265503,
+      "rewards/rejected": -0.2053591012954712,
+      "step": 1020
+    },
+    {
+      "epoch": 1.6480000000000001,
+      "grad_norm": 0.405678875969962,
+      "learning_rate": 1.8285510166487154e-06,
+      "log_odds_chosen": 5.765079021453857,
+      "log_odds_ratio": -0.014157906174659729,
+      "logits/chosen": 9.832456588745117,
+      "logits/rejected": 15.122549057006836,
+      "logps/chosen": -0.06559108942747116,
+      "logps/rejected": -2.4253296852111816,
+      "loss": 0.0694,
+      "nll_loss": 0.07027839124202728,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.006559108849614859,
+      "rewards/margins": 0.23597386479377747,
+      "rewards/rejected": -0.24253296852111816,
+      "step": 1030
+    },
+    {
+      "epoch": 1.6640000000000001,
+      "grad_norm": 0.2511510020504382,
+      "learning_rate": 1.6707875928990059e-06,
+      "log_odds_chosen": 5.9670586585998535,
+      "log_odds_ratio": -0.007793957833200693,
+      "logits/chosen": 6.463404178619385,
+      "logits/rejected": 13.94349479675293,
+      "logps/chosen": -0.036587394773960114,
+      "logps/rejected": -2.103496789932251,
+      "loss": 0.0584,
+      "nll_loss": 0.05108030512928963,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.0036587391514331102,
+      "rewards/margins": 0.2066909372806549,
+      "rewards/rejected": -0.2103496789932251,
+      "step": 1040
+    },
+    {
+      "epoch": 1.6800000000000002,
+      "grad_norm": 0.3812260869637052,
+      "learning_rate": 1.5195190384357405e-06,
+      "log_odds_chosen": 5.7870683670043945,
+      "log_odds_ratio": -0.01724204048514366,
+      "logits/chosen": 7.535168647766113,
+      "logits/rejected": 14.668806076049805,
+      "logps/chosen": -0.06202416867017746,
+      "logps/rejected": -2.3643112182617188,
+      "loss": 0.0648,
+      "nll_loss": 0.067660391330719,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.006202416960150003,
+      "rewards/margins": 0.2302287369966507,
+      "rewards/rejected": -0.23643112182617188,
+      "step": 1050
+    },
+    {
+      "epoch": 1.696,
+      "grad_norm": 0.3058840630848455,
+      "learning_rate": 1.3748633079274254e-06,
+      "log_odds_chosen": 5.669142723083496,
+      "log_odds_ratio": -0.010866614989936352,
+      "logits/chosen": 8.615381240844727,
+      "logits/rejected": 14.441973686218262,
+      "logps/chosen": -0.05342826992273331,
+      "logps/rejected": -2.128041982650757,
+      "loss": 0.063,
+      "nll_loss": 0.06444182246923447,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.005342827644199133,
+      "rewards/margins": 0.2074614018201828,
+      "rewards/rejected": -0.21280424296855927,
+      "step": 1060
+    },
+    {
+      "epoch": 1.712,
+      "grad_norm": 0.3235904810469797,
+      "learning_rate": 1.2369331995613664e-06,
+      "log_odds_chosen": 5.627545356750488,
+      "log_odds_ratio": -0.008036193437874317,
+      "logits/chosen": 6.999047756195068,
+      "logits/rejected": 13.646265983581543,
+      "logps/chosen": -0.048848383128643036,
+      "logps/rejected": -2.056873321533203,
+      "loss": 0.0634,
+      "nll_loss": 0.05773681402206421,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.004884839057922363,
+      "rewards/margins": 0.2008024901151657,
+      "rewards/rejected": -0.20568732917308807,
+      "step": 1070
+    },
+    {
+      "epoch": 1.728,
+      "grad_norm": 0.410833487441517,
+      "learning_rate": 1.1058362670870248e-06,
+      "log_odds_chosen": 5.63392448425293,
+      "log_odds_ratio": -0.012561176903545856,
+      "logits/chosen": 7.00423526763916,
+      "logits/rejected": 12.398062705993652,
+      "logps/chosen": -0.06395191699266434,
+      "logps/rejected": -2.2964274883270264,
+      "loss": 0.0622,
+      "nll_loss": 0.06452804803848267,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.006395191885530949,
+      "rewards/margins": 0.22324757277965546,
+      "rewards/rejected": -0.22964277863502502,
+      "step": 1080
+    },
+    {
+      "epoch": 1.744,
+      "grad_norm": 0.34256892405513334,
+      "learning_rate": 9.816747359488632e-07,
+      "log_odds_chosen": 6.084750652313232,
+      "log_odds_ratio": -0.01084104459732771,
+      "logits/chosen": 7.226653099060059,
+      "logits/rejected": 13.381048202514648,
+      "logps/chosen": -0.05891553312540054,
+      "logps/rejected": -2.476435661315918,
+      "loss": 0.0609,
+      "nll_loss": 0.06247776746749878,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.005891552660614252,
+      "rewards/margins": 0.241752028465271,
+      "rewards/rejected": -0.24764356017112732,
+      "step": 1090
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.34625995571735285,
+      "learning_rate": 8.645454235739903e-07,
+      "log_odds_chosen": 5.164887428283691,
+      "log_odds_ratio": -0.01927667111158371,
+      "logits/chosen": 7.622632026672363,
+      "logits/rejected": 15.065897941589355,
+      "logps/chosen": -0.05562591552734375,
+      "logps/rejected": -1.8294832706451416,
+      "loss": 0.0671,
+      "nll_loss": 0.06367551535367966,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.00556259136646986,
+      "rewards/margins": 0.17738574743270874,
+      "rewards/rejected": -0.18294833600521088,
+      "step": 1100
+    },
+    {
+      "epoch": 1.776,
+      "grad_norm": 0.3129225650767717,
+      "learning_rate": 7.545396638768698e-07,
+      "log_odds_chosen": 5.7577362060546875,
+      "log_odds_ratio": -0.009081291034817696,
+      "logits/chosen": 7.138357639312744,
+      "logits/rejected": 13.764391899108887,
+      "logps/chosen": -0.048210419714450836,
+      "logps/rejected": -2.1392695903778076,
+      "loss": 0.0617,
+      "nll_loss": 0.0597044937312603,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.004821042064577341,
+      "rewards/margins": 0.20910592377185822,
+      "rewards/rejected": -0.2139269858598709,
+      "step": 1110
+    },
+    {
+      "epoch": 1.792,
+      "grad_norm": 0.3636030975331139,
+      "learning_rate": 6.517432360398556e-07,
+      "log_odds_chosen": 5.995306015014648,
+      "log_odds_ratio": -0.013278511352837086,
+      "logits/chosen": 7.143318176269531,
+      "logits/rejected": 14.408166885375977,
+      "logps/chosen": -0.048548031598329544,
+      "logps/rejected": -2.2793731689453125,
+      "loss": 0.0626,
+      "nll_loss": 0.06035660579800606,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.0048548029735684395,
+      "rewards/margins": 0.2230825424194336,
+      "rewards/rejected": -0.22793734073638916,
+      "step": 1120
+    },
+    {
+      "epoch": 1.808,
+      "grad_norm": 0.3502514097075585,
+      "learning_rate": 5.562362976251901e-07,
+      "log_odds_chosen": 5.784189224243164,
+      "log_odds_ratio": -0.019179679453372955,
+      "logits/chosen": 7.51906681060791,
+      "logits/rejected": 13.482861518859863,
+      "logps/chosen": -0.050465572625398636,
+      "logps/rejected": -2.069855213165283,
+      "loss": 0.0673,
+      "nll_loss": 0.06206601858139038,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.005046557169407606,
+      "rewards/margins": 0.2019389569759369,
+      "rewards/rejected": -0.20698551833629608,
+      "step": 1130
+    },
+    {
+      "epoch": 1.8239999999999998,
+      "grad_norm": 0.2385312527933045,
+      "learning_rate": 4.6809332207053083e-07,
+      "log_odds_chosen": 6.114804267883301,
+      "log_odds_ratio": -0.01152479462325573,
+      "logits/chosen": 8.802262306213379,
+      "logits/rejected": 14.353776931762695,
+      "logps/chosen": -0.05430366471409798,
+      "logps/rejected": -2.6986145973205566,
+      "loss": 0.0617,
+      "nll_loss": 0.062255967408418655,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.00543036637827754,
+      "rewards/margins": 0.26443108916282654,
+      "rewards/rejected": -0.26986145973205566,
+      "step": 1140
+    },
+    {
+      "epoch": 1.8399999999999999,
+      "grad_norm": 0.5023792704599986,
+      "learning_rate": 3.8738304061681107e-07,
+      "log_odds_chosen": 5.99333381652832,
+      "log_odds_ratio": -0.010073414072394371,
+      "logits/chosen": 6.594714164733887,
+      "logits/rejected": 14.024187088012695,
+      "logps/chosen": -0.04959163814783096,
+      "logps/rejected": -2.3722965717315674,
+      "loss": 0.0664,
+      "nll_loss": 0.05878063291311264,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.004959163721650839,
+      "rewards/margins": 0.2322704792022705,
+      "rewards/rejected": -0.23722966015338898,
+      "step": 1150
+    },
+    {
+      "epoch": 1.8559999999999999,
+      "grad_norm": 0.18925156808932705,
+      "learning_rate": 3.1416838871368925e-07,
+      "log_odds_chosen": 5.59714412689209,
+      "log_odds_ratio": -0.016117483377456665,
+      "logits/chosen": 6.784099578857422,
+      "logits/rejected": 14.297111511230469,
+      "logps/chosen": -0.04300536960363388,
+      "logps/rejected": -1.9598991870880127,
+      "loss": 0.0594,
+      "nll_loss": 0.06088510900735855,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.0043005370534956455,
+      "rewards/margins": 0.1916893571615219,
+      "rewards/rejected": -0.19598989188671112,
+      "step": 1160
+    },
+    {
+      "epoch": 1.8719999999999999,
+      "grad_norm": 0.4752392208294527,
+      "learning_rate": 2.4850645694436736e-07,
+      "log_odds_chosen": 5.687636375427246,
+      "log_odds_ratio": -0.009939742274582386,
+      "logits/chosen": 6.543469429016113,
+      "logits/rejected": 13.751274108886719,
+      "logps/chosen": -0.0570371076464653,
+      "logps/rejected": -2.2096800804138184,
+      "loss": 0.0612,
+      "nll_loss": 0.06289631128311157,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.005703710950911045,
+      "rewards/margins": 0.21526429057121277,
+      "rewards/rejected": -0.22096800804138184,
+      "step": 1170
+    },
+    {
+      "epoch": 1.888,
+      "grad_norm": 0.40463907978157104,
+      "learning_rate": 1.9044844650808468e-07,
+      "log_odds_chosen": 5.943331718444824,
+      "log_odds_ratio": -0.008882846683263779,
+      "logits/chosen": 7.400280952453613,
+      "logits/rejected": 14.994569778442383,
+      "logps/chosen": -0.045132219791412354,
+      "logps/rejected": -2.3415608406066895,
+      "loss": 0.0612,
+      "nll_loss": 0.05737587809562683,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.00451322179287672,
+      "rewards/margins": 0.22964289784431458,
+      "rewards/rejected": -0.23415610194206238,
+      "step": 1180
+    },
+    {
+      "epoch": 1.904,
+      "grad_norm": 0.35720023172691506,
+      "learning_rate": 1.400396292949513e-07,
+      "log_odds_chosen": 5.809309959411621,
+      "log_odds_ratio": -0.008964666165411472,
+      "logits/chosen": 8.318174362182617,
+      "logits/rejected": 13.486407279968262,
+      "logps/chosen": -0.053054939955472946,
+      "logps/rejected": -2.4225754737854004,
+      "loss": 0.0626,
+      "nll_loss": 0.06207244470715523,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.005305493716150522,
+      "rewards/margins": 0.2369520664215088,
+      "rewards/rejected": -0.24225755035877228,
+      "step": 1190
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.45375001244411867,
+      "learning_rate": 9.731931258429638e-08,
+      "log_odds_chosen": 6.011991024017334,
+      "log_odds_ratio": -0.011115940287709236,
+      "logits/chosen": 7.217115879058838,
+      "logits/rejected": 13.6605806350708,
+      "logps/chosen": -0.05640987679362297,
+      "logps/rejected": -2.306795835494995,
+      "loss": 0.0667,
+      "nll_loss": 0.064894899725914,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.005640988238155842,
+      "rewards/margins": 0.22503860294818878,
+      "rewards/rejected": -0.23067960143089294,
+      "step": 1200
+    },
+    {
+      "epoch": 1.936,
+      "grad_norm": 0.3851516384032617,
+      "learning_rate": 6.232080839403631e-08,
+      "log_odds_chosen": 6.033963203430176,
+      "log_odds_ratio": -0.008175736293196678,
+      "logits/chosen": 5.674324035644531,
+      "logits/rejected": 12.042490005493164,
+      "logps/chosen": -0.041214533150196075,
+      "logps/rejected": -2.353848934173584,
+      "loss": 0.0594,
+      "nll_loss": 0.05154932290315628,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.0041214535012841225,
+      "rewards/margins": 0.2312634438276291,
+      "rewards/rejected": -0.23538486659526825,
+      "step": 1210
+    },
+    {
+      "epoch": 1.952,
+      "grad_norm": 0.2301620888800148,
+      "learning_rate": 3.50714075049563e-08,
+      "log_odds_chosen": 5.746167182922363,
+      "log_odds_ratio": -0.01174002606421709,
+      "logits/chosen": 7.780473232269287,
+      "logits/rejected": 14.472818374633789,
+      "logps/chosen": -0.052154283970594406,
+      "logps/rejected": -2.3260200023651123,
+      "loss": 0.0631,
+      "nll_loss": 0.06034677103161812,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.0052154287695884705,
+      "rewards/margins": 0.22738659381866455,
+      "rewards/rejected": -0.23260203003883362,
+      "step": 1220
+    },
+    {
+      "epoch": 1.968,
+      "grad_norm": 0.29743582988851025,
+      "learning_rate": 1.5592358180189782e-08,
+      "log_odds_chosen": 5.381113529205322,
+      "log_odds_ratio": -0.010704567655920982,
+      "logits/chosen": 6.542645454406738,
+      "logits/rejected": 14.09373664855957,
+      "logps/chosen": -0.045278120785951614,
+      "logps/rejected": -1.7775919437408447,
+      "loss": 0.0605,
+      "nll_loss": 0.057972025126218796,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.004527812357991934,
+      "rewards/margins": 0.1732313632965088,
+      "rewards/rejected": -0.17775918543338776,
+      "step": 1230
+    },
+    {
+      "epoch": 1.984,
+      "grad_norm": 0.3648997358256397,
+      "learning_rate": 3.898849596456477e-09,
+      "log_odds_chosen": 5.863680839538574,
+      "log_odds_ratio": -0.009549752809107304,
+      "logits/chosen": 6.157291889190674,
+      "logits/rejected": 12.384984016418457,
+      "logps/chosen": -0.05373894423246384,
+      "logps/rejected": -2.254366397857666,
+      "loss": 0.0613,
+      "nll_loss": 0.06132974475622177,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.005373895168304443,
+      "rewards/margins": 0.2200627624988556,
+      "rewards/rejected": -0.22543664276599884,
+      "step": 1240
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 0.6054356785823999,
+      "learning_rate": 0.0,
+      "log_odds_chosen": 5.457306861877441,
+      "log_odds_ratio": -0.019121190533041954,
+      "logits/chosen": 7.081140041351318,
+      "logits/rejected": 13.820508003234863,
+      "logps/chosen": -0.0638023242354393,
+      "logps/rejected": -1.9536654949188232,
+      "loss": 0.0679,
+      "nll_loss": 0.067476287484169,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -0.006380232982337475,
+      "rewards/margins": 0.18898631632328033,
+      "rewards/rejected": -0.19536656141281128,
+      "step": 1250
+    },
+    {
+      "epoch": 2.0,
+      "step": 1250,
+      "total_flos": 0.0,
+      "train_loss": 0.2130171504020691,
+      "train_runtime": 11716.0982,
+      "train_samples_per_second": 3.414,
+      "train_steps_per_second": 0.107
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 1250,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 2,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 0.0,
+  "train_batch_size": 4,
+  "trial_name": null,
+  "trial_params": null
+}