Model save

README.md

@@ -0,0 +1,68 @@
+---
+license: gemma
+base_model: scottsuk0306/easylm-helpsteer-sft-gemma-2-2b
+tags:
+- trl
+- reward-trainer
+- generated_from_trainer
+metrics:
+- accuracy
+model-index:
+- name: easylm-helpsteer-rm-gemma-2-2b
+  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. -->
+
+# easylm-helpsteer-rm-gemma-2-2b
+
+This model is a fine-tuned version of [scottsuk0306/easylm-helpsteer-sft-gemma-2-2b](https://huggingface.co/scottsuk0306/easylm-helpsteer-sft-gemma-2-2b) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 2.9740
+- Accuracy: 0.6324
+
+## 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: 1e-06
+- train_batch_size: 1
+- eval_batch_size: 1
+- seed: 42
+- distributed_type: multi-GPU
+- num_devices: 4
+- total_train_batch_size: 4
+- total_eval_batch_size: 4
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: cosine
+- num_epochs: 1
+
+### Training results
+
+| Training Loss | Epoch  | Step | Validation Loss | Accuracy |
+|:-------------:|:------:|:----:|:---------------:|:--------:|
+| 0.5003        | 0.3115 | 500  | 2.7140          | 0.6294   |
+| 0.3939        | 0.6231 | 1000 | 3.3430          | 0.65     |
+| 0.065         | 0.9346 | 1500 | 2.9740          | 0.6324   |
+
+
+### Framework versions
+
+- Transformers 4.43.3
+- Pytorch 2.3.0+cu121
+- Datasets 2.20.0
+- Tokenizers 0.19.1

all_results.json

@@ -0,0 +1,9 @@
+{
+    "epoch": 1.0,
+    "total_flos": 0.0,
+    "train_loss": 0.3749335529650852,
+    "train_runtime": 2207.4532,
+    "train_samples": 6981,
+    "train_samples_per_second": 2.908,
+    "train_steps_per_second": 0.727
+}

model-00001-of-00002.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cfbe8a4bd99713a83e40129e92ad67cc32deb7855547ff87d73d1608542c1ee5
+oid sha256:a7081c20bd986eee89a2567d9aa2af615991dcaff7ddc91aef7fb8cdbea04c38
 size 4988025760

model-00002-of-00002.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5e8bf548392ea8026ab6a237533819b3a6af92320f656b2fe98b62341b2054f6
+oid sha256:73be8580be7c2160242d37a96f2142f1d637eb7fa70801b1815e1ded2c7a4996
 size 240701032

train_results.json

@@ -0,0 +1,9 @@
+{
+    "epoch": 1.0,
+    "total_flos": 0.0,
+    "train_loss": 0.3749335529650852,
+    "train_runtime": 2207.4532,
+    "train_samples": 6981,
+    "train_samples_per_second": 2.908,
+    "train_steps_per_second": 0.727
+}

trainer_state.json

@@ -0,0 +1,293 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 1.0,
+  "eval_steps": 500,
+  "global_step": 1605,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.03115264797507788,
+      "grad_norm": 57.86160659790039,
+      "learning_rate": 9.976073288648913e-07,
+      "loss": 0.7289,
+      "step": 50
+    },
+    {
+      "epoch": 0.06230529595015576,
+      "grad_norm": 60.39614486694336,
+      "learning_rate": 9.904522149602086e-07,
+      "loss": 0.6972,
+      "step": 100
+    },
+    {
+      "epoch": 0.09345794392523364,
+      "grad_norm": 151.33900451660156,
+      "learning_rate": 9.78603137623984e-07,
+      "loss": 0.5746,
+      "step": 150
+    },
+    {
+      "epoch": 0.12461059190031153,
+      "grad_norm": 52.87775802612305,
+      "learning_rate": 9.621735006374983e-07,
+      "loss": 0.4166,
+      "step": 200
+    },
+    {
+      "epoch": 0.1557632398753894,
+      "grad_norm": 50.08223342895508,
+      "learning_rate": 9.413205468734626e-07,
+      "loss": 0.4558,
+      "step": 250
+    },
+    {
+      "epoch": 0.18691588785046728,
+      "grad_norm": 517.3528442382812,
+      "learning_rate": 9.162438533740891e-07,
+      "loss": 0.3246,
+      "step": 300
+    },
+    {
+      "epoch": 0.21806853582554517,
+      "grad_norm": 1556.693359375,
+      "learning_rate": 8.871834212621772e-07,
+      "loss": 0.4801,
+      "step": 350
+    },
+    {
+      "epoch": 0.24922118380062305,
+      "grad_norm": 16.822763442993164,
+      "learning_rate": 8.544173787660787e-07,
+      "loss": 0.2867,
+      "step": 400
+    },
+    {
+      "epoch": 0.2803738317757009,
+      "grad_norm": 2.0897529125213623,
+      "learning_rate": 8.182593193421624e-07,
+      "loss": 0.4674,
+      "step": 450
+    },
+    {
+      "epoch": 0.3115264797507788,
+      "grad_norm": 19.26500129699707,
+      "learning_rate": 7.790553003707689e-07,
+      "loss": 0.5003,
+      "step": 500
+    },
+    {
+      "epoch": 0.3115264797507788,
+      "eval_accuracy": 0.6294117647058823,
+      "eval_loss": 2.713998317718506,
+      "eval_runtime": 9.3954,
+      "eval_samples_per_second": 36.188,
+      "eval_steps_per_second": 9.047,
+      "step": 500
+    },
+    {
+      "epoch": 0.3426791277258567,
+      "grad_norm": 0.8728576898574829,
+      "learning_rate": 7.371805311501904e-07,
+      "loss": 0.5136,
+      "step": 550
+    },
+    {
+      "epoch": 0.37383177570093457,
+      "grad_norm": 167.8850860595703,
+      "learning_rate": 6.930357818868407e-07,
+      "loss": 0.2441,
+      "step": 600
+    },
+    {
+      "epoch": 0.40498442367601245,
+      "grad_norm": 177.43258666992188,
+      "learning_rate": 6.470435480500362e-07,
+      "loss": 0.4716,
+      "step": 650
+    },
+    {
+      "epoch": 0.43613707165109034,
+      "grad_norm": 289.399658203125,
+      "learning_rate": 5.996440068011383e-07,
+      "loss": 0.3897,
+      "step": 700
+    },
+    {
+      "epoch": 0.4672897196261682,
+      "grad_norm": 787.5913696289062,
+      "learning_rate": 5.512908041968018e-07,
+      "loss": 0.4029,
+      "step": 750
+    },
+    {
+      "epoch": 0.4984423676012461,
+      "grad_norm": 1.4492032960333745e-06,
+      "learning_rate": 5.024467134856725e-07,
+      "loss": 0.3285,
+      "step": 800
+    },
+    {
+      "epoch": 0.5295950155763239,
+      "grad_norm": 3.293706356544135e-07,
+      "learning_rate": 4.5357920605161114e-07,
+      "loss": 0.435,
+      "step": 850
+    },
+    {
+      "epoch": 0.5607476635514018,
+      "grad_norm": 1.7141993045806885,
+      "learning_rate": 4.0515597739254617e-07,
+      "loss": 0.3973,
+      "step": 900
+    },
+    {
+      "epoch": 0.5919003115264797,
+      "grad_norm": 2.922891617629375e-08,
+      "learning_rate": 3.576404709544031e-07,
+      "loss": 0.2163,
+      "step": 950
+    },
+    {
+      "epoch": 0.6230529595015576,
+      "grad_norm": 560.4365844726562,
+      "learning_rate": 3.114874426600802e-07,
+      "loss": 0.3939,
+      "step": 1000
+    },
+    {
+      "epoch": 0.6230529595015576,
+      "eval_accuracy": 0.65,
+      "eval_loss": 3.343003988265991,
+      "eval_runtime": 8.9452,
+      "eval_samples_per_second": 38.009,
+      "eval_steps_per_second": 9.502,
+      "step": 1000
+    },
+    {
+      "epoch": 0.6542056074766355,
+      "grad_norm": 0.0010744519531726837,
+      "learning_rate": 2.671386085839682e-07,
+      "loss": 0.2876,
+      "step": 1050
+    },
+    {
+      "epoch": 0.6853582554517134,
+      "grad_norm": 2111.46875,
+      "learning_rate": 2.2501841742674588e-07,
+      "loss": 0.3456,
+      "step": 1100
+    },
+    {
+      "epoch": 0.7165109034267912,
+      "grad_norm": 0.022361410781741142,
+      "learning_rate": 1.855299882507616e-07,
+      "loss": 0.3517,
+      "step": 1150
+    },
+    {
+      "epoch": 0.7476635514018691,
+      "grad_norm": 2.9614204422045987e-08,
+      "learning_rate": 1.490512523546559e-07,
+      "loss": 0.4218,
+      "step": 1200
+    },
+    {
+      "epoch": 0.778816199376947,
+      "grad_norm": 1.0811158546175648e-07,
+      "learning_rate": 1.1593133621212453e-07,
+      "loss": 0.2904,
+      "step": 1250
+    },
+    {
+      "epoch": 0.8099688473520249,
+      "grad_norm": 1.012383222579956,
+      "learning_rate": 8.648722009257315e-08,
+      "loss": 0.4572,
+      "step": 1300
+    },
+    {
+      "epoch": 0.8411214953271028,
+      "grad_norm": 3.705447113588889e-07,
+      "learning_rate": 6.10007043429538e-08,
+      "loss": 0.187,
+      "step": 1350
+    },
+    {
+      "epoch": 0.8722741433021807,
+      "grad_norm": 9.330016837338917e-06,
+      "learning_rate": 3.971571236554117e-08,
+      "loss": 0.1274,
+      "step": 1400
+    },
+    {
+      "epoch": 0.9034267912772586,
+      "grad_norm": 0.08515036106109619,
+      "learning_rate": 2.2835956103996524e-08,
+      "loss": 0.2892,
+      "step": 1450
+    },
+    {
+      "epoch": 0.9345794392523364,
+      "grad_norm": 0.008006788790225983,
+      "learning_rate": 1.0522986380618604e-08,
+      "loss": 0.065,
+      "step": 1500
+    },
+    {
+      "epoch": 0.9345794392523364,
+      "eval_accuracy": 0.6323529411764706,
+      "eval_loss": 2.9739811420440674,
+      "eval_runtime": 8.9019,
+      "eval_samples_per_second": 38.194,
+      "eval_steps_per_second": 9.549,
+      "step": 1500
+    },
+    {
+      "epoch": 0.9657320872274143,
+      "grad_norm": 1.7190651533383061e-06,
+      "learning_rate": 2.894646744385887e-09,
+      "loss": 0.288,
+      "step": 1550
+    },
+    {
+      "epoch": 0.9968847352024922,
+      "grad_norm": 909.2741088867188,
+      "learning_rate": 2.3945627523891398e-11,
+      "loss": 0.1994,
+      "step": 1600
+    },
+    {
+      "epoch": 1.0,
+      "step": 1605,
+      "total_flos": 0.0,
+      "train_loss": 0.3749335529650852,
+      "train_runtime": 2207.4532,
+      "train_samples_per_second": 2.908,
+      "train_steps_per_second": 0.727
+    }
+  ],
+  "logging_steps": 50,
+  "max_steps": 1605,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 1,
+  "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": 1,
+  "trial_name": null,
+  "trial_params": null
+}