Model save

README.md

@@ -0,0 +1,67 @@
+---
+license: apache-2.0
+base_model: jan-hq/LlamaCorn-1.1B-Chat
+tags:
+- trl
+- sft
+- generated_from_trainer
+datasets:
+- generator
+model-index:
+- name: TinyJensen-1.1B
+  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. -->
+
+# TinyJensen-1.1B
+
+This model is a fine-tuned version of [jan-hq/LlamaCorn-1.1B-Chat](https://huggingface.co/jan-hq/LlamaCorn-1.1B-Chat) on the generator dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.8390
+
+## 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: 7e-05
+- train_batch_size: 4
+- eval_batch_size: 4
+- seed: 42
+- distributed_type: multi-GPU
+- gradient_accumulation_steps: 4
+- total_train_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: 3
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss |
+|:-------------:|:-----:|:----:|:---------------:|
+| 0.7614        | 1.0   | 178  | 0.7861          |
+| 0.5033        | 2.0   | 356  | 0.7810          |
+| 0.3561        | 3.0   | 534  | 0.8390          |
+
+
+### Framework versions
+
+- Transformers 4.37.2
+- Pytorch 2.1.2+cu121
+- Datasets 2.14.6
+- Tokenizers 0.15.0

all_results.json

@@ -0,0 +1,13 @@
+{
+    "epoch": 3.0,
+    "eval_loss": 0.8389958739280701,
+    "eval_runtime": 8.8319,
+    "eval_samples": 537,
+    "eval_samples_per_second": 23.551,
+    "eval_steps_per_second": 5.888,
+    "train_loss": 0.5910449749075072,
+    "train_runtime": 1507.9641,
+    "train_samples": 7214,
+    "train_samples_per_second": 5.672,
+    "train_steps_per_second": 0.354
+}

eval_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 3.0,
+    "eval_loss": 0.8389958739280701,
+    "eval_runtime": 8.8319,
+    "eval_samples": 537,
+    "eval_samples_per_second": 23.551,
+    "eval_steps_per_second": 5.888
+}

generation_config.json

@@ -0,0 +1,7 @@
+{
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "max_length": 2048,
+  "pad_token_id": 0,
+  "transformers_version": "4.37.2"
+}

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b200c3ef50530c4cee53471364956457d95524eabb58654ef04f3b1a3fd41178
+oid sha256:990e3a0db5c4b91671e92d86254ef048c41338af6f4768ce1741eae8fe23e3a3
 size 2200119864

runs/Mar13_10-29-23_333df911e7ea/events.out.tfevents.1710326345.333df911e7ea.2445.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:542208fb7aa0a568dab77f0dd7bfb36164a8746539bb5ae3281f9222e43486fc
-size 20789
+oid sha256:1e9867a8bdc52e938322297a21c2c704bee530c421fe860bfdaa6c47c871117c
+size 22356

runs/Mar13_10-29-23_333df911e7ea/events.out.tfevents.1710327864.333df911e7ea.2445.1

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

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 3.0,
+    "train_loss": 0.5910449749075072,
+    "train_runtime": 1507.9641,
+    "train_samples": 7214,
+    "train_samples_per_second": 5.672,
+    "train_steps_per_second": 0.354
+}

trainer_state.json

@@ -0,0 +1,696 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 2.9957924263674616,
+  "eval_steps": 500,
+  "global_step": 534,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.296296296296296e-06,
+      "loss": 1.1525,
+      "step": 1
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6.48148148148148e-06,
+      "loss": 1.1069,
+      "step": 5
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.296296296296296e-05,
+      "loss": 1.0522,
+      "step": 10
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9444444444444442e-05,
+      "loss": 0.977,
+      "step": 15
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.592592592592592e-05,
+      "loss": 0.9307,
+      "step": 20
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 3.240740740740741e-05,
+      "loss": 0.9144,
+      "step": 25
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 3.8888888888888884e-05,
+      "loss": 0.9218,
+      "step": 30
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 4.537037037037037e-05,
+      "loss": 0.8808,
+      "step": 35
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 5.185185185185184e-05,
+      "loss": 0.8981,
+      "step": 40
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 5.833333333333333e-05,
+      "loss": 0.8721,
+      "step": 45
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 6.481481481481482e-05,
+      "loss": 0.8714,
+      "step": 50
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 6.999925035824452e-05,
+      "loss": 0.8616,
+      "step": 55
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 6.99730162684253e-05,
+      "loss": 0.8686,
+      "step": 60
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 6.99093321968073e-05,
+      "loss": 0.823,
+      "step": 65
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 6.980826633788956e-05,
+      "loss": 0.8186,
+      "step": 70
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 6.96699269155252e-05,
+      "loss": 0.8349,
+      "step": 75
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 6.949446206703272e-05,
+      "loss": 0.8436,
+      "step": 80
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 6.928205968456678e-05,
+      "loss": 0.8353,
+      "step": 85
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 6.903294721391867e-05,
+      "loss": 0.8553,
+      "step": 90
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 6.874739141096151e-05,
+      "loss": 0.8244,
+      "step": 95
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 6.842569805600153e-05,
+      "loss": 0.8424,
+      "step": 100
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 6.806821162634073e-05,
+      "loss": 0.8328,
+      "step": 105
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 6.767531492740205e-05,
+      "loss": 0.821,
+      "step": 110
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 6.724742868281173e-05,
+      "loss": 0.7997,
+      "step": 115
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 6.678501108387784e-05,
+      "loss": 0.8393,
+      "step": 120
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 6.628855729894762e-05,
+      "loss": 0.8226,
+      "step": 125
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 6.575859894316877e-05,
+      "loss": 0.7994,
+      "step": 130
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 6.519570350922271e-05,
+      "loss": 0.7988,
+      "step": 135
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 6.460047375963906e-05,
+      "loss": 0.7932,
+      "step": 140
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 6.397354708134257e-05,
+      "loss": 0.786,
+      "step": 145
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 6.331559480312315e-05,
+      "loss": 0.7943,
+      "step": 150
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 6.262732147676025e-05,
+      "loss": 0.8187,
+      "step": 155
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 6.190946412257108e-05,
+      "loss": 0.7774,
+      "step": 160
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 6.116279144019089e-05,
+      "loss": 0.7925,
+      "step": 165
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 6.038810298543006e-05,
+      "loss": 0.7877,
+      "step": 170
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 5.9586228314089804e-05,
+      "loss": 0.7614,
+      "step": 175
+    },
+    {
+      "epoch": 1.0,
+      "eval_loss": 0.7861344218254089,
+      "eval_runtime": 8.8077,
+      "eval_samples_per_second": 23.616,
+      "eval_steps_per_second": 5.904,
+      "step": 178
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 5.8758026093652964e-05,
+      "loss": 0.7171,
+      "step": 180
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 5.790438318380136e-05,
+      "loss": 0.5443,
+      "step": 185
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 5.702621368674431e-05,
+      "loss": 0.5642,
+      "step": 190
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 5.6124457968374994e-05,
+      "loss": 0.5541,
+      "step": 195
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 5.520008165130318e-05,
+      "loss": 0.5469,
+      "step": 200
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 5.425407458084227e-05,
+      "loss": 0.5585,
+      "step": 205
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 5.3287449765058205e-05,
+      "loss": 0.5547,
+      "step": 210
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 5.230124229001506e-05,
+      "loss": 0.5497,
+      "step": 215
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 5.1296508211378896e-05,
+      "loss": 0.5525,
+      "step": 220
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 5.027432342356697e-05,
+      "loss": 0.5793,
+      "step": 225
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 4.9235782507653006e-05,
+      "loss": 0.5504,
+      "step": 230
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 4.818199755926237e-05,
+      "loss": 0.5546,
+      "step": 235
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 4.711409699771225e-05,
+      "loss": 0.5674,
+      "step": 240
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 4.603322435767194e-05,
+      "loss": 0.5422,
+      "step": 245
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 4.49405370646373e-05,
+      "loss": 0.5694,
+      "step": 250
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 4.3837205195530523e-05,
+      "loss": 0.5359,
+      "step": 255
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 4.2724410225752544e-05,
+      "loss": 0.5271,
+      "step": 260
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 4.160334376402943e-05,
+      "loss": 0.5414,
+      "step": 265
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 4.047520627640808e-05,
+      "loss": 0.5243,
+      "step": 270
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 3.934120580076699e-05,
+      "loss": 0.5551,
+      "step": 275
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 3.820255665321909e-05,
+      "loss": 0.5291,
+      "step": 280
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 3.706047812779161e-05,
+      "loss": 0.5604,
+      "step": 285
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 3.5916193190775565e-05,
+      "loss": 0.531,
+      "step": 290
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 3.4770927171142685e-05,
+      "loss": 0.5531,
+      "step": 295
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 3.362590644843261e-05,
+      "loss": 0.5574,
+      "step": 300
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 3.248235713951481e-05,
+      "loss": 0.5286,
+      "step": 305
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 3.134150378563213e-05,
+      "loss": 0.5286,
+      "step": 310
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 3.020456804113112e-05,
+      "loss": 0.5335,
+      "step": 315
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.9072767365284138e-05,
+      "loss": 0.5593,
+      "step": 320
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.7947313718603398e-05,
+      "loss": 0.537,
+      "step": 325
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.6829412265043318e-05,
+      "loss": 0.535,
+      "step": 330
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.5720260081480713e-05,
+      "loss": 0.5187,
+      "step": 335
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4621044875855016e-05,
+      "loss": 0.5157,
+      "step": 340
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.353294371534073e-05,
+      "loss": 0.5321,
+      "step": 345
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.245712176591449e-05,
+      "loss": 0.5577,
+      "step": 350
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.139473104466615e-05,
+      "loss": 0.5033,
+      "step": 355
+    },
+    {
+      "epoch": 2.0,
+      "eval_loss": 0.7809656858444214,
+      "eval_runtime": 8.805,
+      "eval_samples_per_second": 23.623,
+      "eval_steps_per_second": 5.906,
+      "step": 356
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.0346909186190027e-05,
+      "loss": 0.433,
+      "step": 360
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 1.931477822437721e-05,
+      "loss": 0.3788,
+      "step": 365
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 1.8299443390913702e-05,
+      "loss": 0.3838,
+      "step": 370
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 1.7301991931770543e-05,
+      "loss": 0.3723,
+      "step": 375
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 1.6323491942953597e-05,
+      "loss": 0.3813,
+      "step": 380
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 1.5364991226759663e-05,
+      "loss": 0.3772,
+      "step": 385
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 1.4427516169763444e-05,
+      "loss": 0.3917,
+      "step": 390
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 1.351207064373717e-05,
+      "loss": 0.3745,
+      "step": 395
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 1.2619634930679534e-05,
+      "loss": 0.3923,
+      "step": 400
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 1.175116467310515e-05,
+      "loss": 0.3817,
+      "step": 405
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 1.0907589850718605e-05,
+      "loss": 0.4024,
+      "step": 410
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 1.0089813784569004e-05,
+      "loss": 0.3807,
+      "step": 415
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 9.298712169751012e-06,
+      "loss": 0.3739,
+      "step": 420
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 8.535132137688704e-06,
+      "loss": 0.3415,
+      "step": 425
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 7.799891349006018e-06,
+      "loss": 0.3652,
+      "step": 430
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 7.093777117955255e-06,
+      "loss": 0.3785,
+      "step": 435
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 6.417545569341238e-06,
+      "loss": 0.369,
+      "step": 440
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 5.771920828843995e-06,
+      "loss": 0.3693,
+      "step": 445
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 5.157594247606773e-06,
+      "loss": 0.3763,
+      "step": 450
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 4.575223661919969e-06,
+      "loss": 0.3589,
+      "step": 455
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 4.025432688793538e-06,
+      "loss": 0.372,
+      "step": 460
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 3.5088100581722544e-06,
+      "loss": 0.3694,
+      "step": 465
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 3.0259089825089655e-06,
+      "loss": 0.3791,
+      "step": 470
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.577246564370866e-06,
+      "loss": 0.3838,
+      "step": 475
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.163303242713058e-06,
+      "loss": 0.3786,
+      "step": 480
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 1.7845222784125152e-06,
+      "loss": 0.4005,
+      "step": 485
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 1.4413092796132431e-06,
+      "loss": 0.358,
+      "step": 490
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 1.1340317673909082e-06,
+      "loss": 0.3938,
+      "step": 495
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 8.630187822020567e-07,
+      "loss": 0.3751,
+      "step": 500
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 6.28560531539366e-07,
+      "loss": 0.3627,
+      "step": 505
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 4.309080791701819e-07,
+      "loss": 0.3779,
+      "step": 510
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.702730762911531e-07,
+      "loss": 0.3841,
+      "step": 515
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 1.4682753488684196e-07,
+      "loss": 0.3566,
+      "step": 520
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 6.070364353494617e-08,
+      "loss": 0.3684,
+      "step": 525
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 1.1993625855495903e-08,
+      "loss": 0.3561,
+      "step": 530
+    },
+    {
+      "epoch": 3.0,
+      "eval_loss": 0.8389958739280701,
+      "eval_runtime": 8.8169,
+      "eval_samples_per_second": 23.591,
+      "eval_steps_per_second": 5.898,
+      "step": 534
+    },
+    {
+      "epoch": 3.0,
+      "step": 534,
+      "total_flos": 1.0858665717058765e+17,
+      "train_loss": 0.5910449749075072,
+      "train_runtime": 1507.9641,
+      "train_samples_per_second": 5.672,
+      "train_steps_per_second": 0.354
+    }
+  ],
+  "logging_steps": 5,
+  "max_steps": 534,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 3,
+  "save_steps": 100,
+  "total_flos": 1.0858665717058765e+17,
+  "train_batch_size": 4,
+  "trial_name": null,
+  "trial_params": null
+}