Save benchmark: stable-run-sft-step-20

checkpoints/stable-run-sft-step-20/config.json

@@ -0,0 +1,45 @@
+{
+  "_num_labels": 1,
+  "activation_function": "gelu_new",
+  "architectures": [
+    "GPT2LMHeadModel"
+  ],
+  "attn_pdrop": 0.1,
+  "bos_token_id": 50256,
+  "dtype": "float32",
+  "embd_pdrop": 0.1,
+  "eos_token_id": 50256,
+  "id2label": {
+    "0": "LABEL_0"
+  },
+  "initializer_range": 0.02,
+  "label2id": {
+    "LABEL_0": 0
+  },
+  "layer_norm_epsilon": 1e-05,
+  "model_type": "gpt2",
+  "n_ctx": 1024,
+  "n_embd": 768,
+  "n_head": 12,
+  "n_inner": null,
+  "n_layer": 6,
+  "n_positions": 1024,
+  "reorder_and_upcast_attn": false,
+  "resid_pdrop": 0.1,
+  "scale_attn_by_inverse_layer_idx": false,
+  "scale_attn_weights": true,
+  "summary_activation": null,
+  "summary_first_dropout": 0.1,
+  "summary_proj_to_labels": true,
+  "summary_type": "cls_index",
+  "summary_use_proj": true,
+  "task_specific_params": {
+    "text-generation": {
+      "do_sample": true,
+      "max_length": 50
+    }
+  },
+  "transformers_version": "4.56.1",
+  "use_cache": true,
+  "vocab_size": 50257
+}

checkpoints/stable-run-sft-step-20/generation_config.json

@@ -0,0 +1,6 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 50256,
+  "eos_token_id": 50256,
+  "transformers_version": "4.56.1"
+}

checkpoints/stable-run-sft-step-20/hyperparameters.json

@@ -0,0 +1,17 @@
+{
+    "repo_id": "broadfield-dev/savant_2_gsm8k_final",
+    "resume_from_benchmark": "",
+    "benchmark_name_prefix": "stable-run",
+    "benchmark_interval": 20,
+    "sft_epochs": 4,
+    "sft_lr": 2e-05,
+    "sft_warmup_proportion": 0.1,
+    "rl_lr": 1.5e-06,
+    "sft_slice": "0:2000",
+    "rl_slice": "4000:7473",
+    "context_length": 1024,
+    "base_model_name": "distilgpt2",
+    "use_scheduler": true,
+    "sft_batch_size": 1,
+    "grad_accum_steps": 16
+}

checkpoints/stable-run-sft-step-20/model.safetensors

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

checkpoints/stable-run-sft-step-20/optimizer.pt

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

checkpoints/stable-run-sft-step-20/scheduler.pt

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

checkpoints/stable-run-sft-step-20/special_tokens_map.json

@@ -0,0 +1,6 @@
+{
+  "bos_token": "<|endoftext|>",
+  "eos_token": "<|endoftext|>",
+  "pad_token": "<|endoftext|>",
+  "unk_token": "<|endoftext|>"
+}

checkpoints/stable-run-sft-step-20/tokenizer_config.json

@@ -0,0 +1,21 @@
+{
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "50256": {
+      "content": "<|endoftext|>",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "bos_token": "<|endoftext|>",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "<|endoftext|>",
+  "extra_special_tokens": {},
+  "model_max_length": 1024,
+  "pad_token": "<|endoftext|>",
+  "tokenizer_class": "GPT2Tokenizer",
+  "unk_token": "<|endoftext|>"
+}

checkpoints/stable-run-sft-step-20/training_state.json

@@ -0,0 +1,207 @@
+{
+    "phase": "SFT",
+    "total_optimizer_steps": 20,
+    "metrics": [
+        {
+            "Step": 1,
+            "Metric": "SFT Loss",
+            "Value": 10.806696891784668
+        },
+        {
+            "Step": 1,
+            "Metric": "Learning Rate",
+            "Value": 4.0000000000000003e-07
+        },
+        {
+            "Step": 2,
+            "Metric": "SFT Loss",
+            "Value": 20.197086334228516
+        },
+        {
+            "Step": 2,
+            "Metric": "Learning Rate",
+            "Value": 8.000000000000001e-07
+        },
+        {
+            "Step": 3,
+            "Metric": "SFT Loss",
+            "Value": 13.669203758239746
+        },
+        {
+            "Step": 3,
+            "Metric": "Learning Rate",
+            "Value": 1.2000000000000002e-06
+        },
+        {
+            "Step": 4,
+            "Metric": "SFT Loss",
+            "Value": 19.52617835998535
+        },
+        {
+            "Step": 4,
+            "Metric": "Learning Rate",
+            "Value": 1.6000000000000001e-06
+        },
+        {
+            "Step": 5,
+            "Metric": "SFT Loss",
+            "Value": 14.46614933013916
+        },
+        {
+            "Step": 5,
+            "Metric": "Learning Rate",
+            "Value": 2.0000000000000003e-06
+        },
+        {
+            "Step": 6,
+            "Metric": "SFT Loss",
+            "Value": 11.709370613098145
+        },
+        {
+            "Step": 6,
+            "Metric": "Learning Rate",
+            "Value": 2.4000000000000003e-06
+        },
+        {
+            "Step": 7,
+            "Metric": "SFT Loss",
+            "Value": 14.79263687133789
+        },
+        {
+            "Step": 7,
+            "Metric": "Learning Rate",
+            "Value": 2.8000000000000003e-06
+        },
+        {
+            "Step": 8,
+            "Metric": "SFT Loss",
+            "Value": 16.564573287963867
+        },
+        {
+            "Step": 8,
+            "Metric": "Learning Rate",
+            "Value": 3.2000000000000003e-06
+        },
+        {
+            "Step": 9,
+            "Metric": "SFT Loss",
+            "Value": 16.437776565551758
+        },
+        {
+            "Step": 9,
+            "Metric": "Learning Rate",
+            "Value": 3.6000000000000003e-06
+        },
+        {
+            "Step": 10,
+            "Metric": "SFT Loss",
+            "Value": 14.215919494628906
+        },
+        {
+            "Step": 10,
+            "Metric": "Learning Rate",
+            "Value": 4.000000000000001e-06
+        },
+        {
+            "Step": 11,
+            "Metric": "SFT Loss",
+            "Value": 13.728559494018555
+        },
+        {
+            "Step": 11,
+            "Metric": "Learning Rate",
+            "Value": 4.4e-06
+        },
+        {
+            "Step": 12,
+            "Metric": "SFT Loss",
+            "Value": 17.120494842529297
+        },
+        {
+            "Step": 12,
+            "Metric": "Learning Rate",
+            "Value": 4.800000000000001e-06
+        },
+        {
+            "Step": 13,
+            "Metric": "SFT Loss",
+            "Value": 12.443873405456543
+        },
+        {
+            "Step": 13,
+            "Metric": "Learning Rate",
+            "Value": 5.2e-06
+        },
+        {
+            "Step": 14,
+            "Metric": "SFT Loss",
+            "Value": 10.322935104370117
+        },
+        {
+            "Step": 14,
+            "Metric": "Learning Rate",
+            "Value": 5.600000000000001e-06
+        },
+        {
+            "Step": 15,
+            "Metric": "SFT Loss",
+            "Value": 10.010440826416016
+        },
+        {
+            "Step": 15,
+            "Metric": "Learning Rate",
+            "Value": 6e-06
+        },
+        {
+            "Step": 16,
+            "Metric": "SFT Loss",
+            "Value": 9.959354400634766
+        },
+        {
+            "Step": 16,
+            "Metric": "Learning Rate",
+            "Value": 6.4000000000000006e-06
+        },
+        {
+            "Step": 17,
+            "Metric": "SFT Loss",
+            "Value": 9.42132568359375
+        },
+        {
+            "Step": 17,
+            "Metric": "Learning Rate",
+            "Value": 6.800000000000001e-06
+        },
+        {
+            "Step": 18,
+            "Metric": "SFT Loss",
+            "Value": 10.809995651245117
+        },
+        {
+            "Step": 18,
+            "Metric": "Learning Rate",
+            "Value": 7.2000000000000005e-06
+        },
+        {
+            "Step": 19,
+            "Metric": "SFT Loss",
+            "Value": 11.641499519348145
+        },
+        {
+            "Step": 19,
+            "Metric": "Learning Rate",
+            "Value": 7.600000000000001e-06
+        },
+        {
+            "Step": 20,
+            "Metric": "SFT Loss",
+            "Value": 8.137388229370117
+        },
+        {
+            "Step": 20,
+            "Metric": "Learning Rate",
+            "Value": 8.000000000000001e-06
+        }
+    ],
+    "sft_complete": false
+}