Model save

README.md

@@ -2,15 +2,11 @@
 license: mit
 base_model: gpt2
 tags:
-- alignment-handbook
-- trl
-- sft
-- generated_from_trainer
 - trl
 - sft
 - generated_from_trainer
 datasets:
-- shahadalkhalifa/Crypto_Whitepaper_Labeled
+- generator
 model-index:
 - name: gpt2-cpt-white
   results: []
@@ -21,7 +17,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # gpt2-cpt-white
 
-This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on the shahadalkhalifa/Crypto_Whitepaper_Labeled dataset.
+This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on the generator dataset.
 
 ## Model description
 
@@ -45,13 +41,13 @@ The following hyperparameters were used during training:
 - eval_batch_size: 8
 - seed: 42
 - distributed_type: multi-GPU
-- num_devices: 16
-- total_train_batch_size: 256
-- total_eval_batch_size: 128
+- num_devices: 8
+- total_train_batch_size: 128
+- total_eval_batch_size: 64
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: cosine
 - lr_scheduler_warmup_ratio: 0.1
-- num_epochs: 32.0
+- num_epochs: 8.0
 
 ### Training results
 

all_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 32.0,
-    "train_loss": 5.874281942844391,
-    "train_runtime": 26.6871,
+    "epoch": 8.0,
+    "train_loss": 6.084760069847107,
+    "train_runtime": 16.1029,
     "train_samples": 91,
-    "train_samples_per_second": 300.97,
-    "train_steps_per_second": 1.199
+    "train_samples_per_second": 124.698,
+    "train_steps_per_second": 0.994
 }

config.json

@@ -34,6 +34,6 @@
   },
   "torch_dtype": "bfloat16",
   "transformers_version": "4.39.0.dev0",
-  "use_cache": true,
+  "use_cache": false,
   "vocab_size": 50257
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9681ae5acb0c43b99ec0260ed732d7f2fd3962130cdccecaf3e4a746e281b353
+oid sha256:67a4d2b9058ea5b30cd834d46c08a882b87f6b79a5c15f86be3bcea3b929b1af
 size 248894656

train_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 32.0,
-    "train_loss": 5.874281942844391,
-    "train_runtime": 26.6871,
+    "epoch": 8.0,
+    "train_loss": 6.084760069847107,
+    "train_runtime": 16.1029,
     "train_samples": 91,
-    "train_samples_per_second": 300.97,
-    "train_steps_per_second": 1.199
+    "train_samples_per_second": 124.698,
+    "train_steps_per_second": 0.994
 }

trainer_state.json

@@ -1,78 +1,57 @@
 {
   "best_metric": null,
   "best_model_checkpoint": null,
-  "epoch": 32.0,
+  "epoch": 8.0,
   "eval_steps": 500,
-  "global_step": 32,
+  "global_step": 16,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
   "log_history": [
     {
-      "epoch": 1.0,
-      "grad_norm": 6.375,
-      "learning_rate": 5e-05,
-      "loss": 6.4009,
+      "epoch": 0.5,
+      "grad_norm": 6.46875,
+      "learning_rate": 0.0001,
+      "loss": 6.3907,
       "step": 1
     },
     {
-      "epoch": 5.0,
-      "grad_norm": 4.1875,
-      "learning_rate": 0.00019937122098932428,
-      "loss": 6.3103,
-      "step": 5
-    },
-    {
-      "epoch": 10.0,
-      "grad_norm": 2.203125,
+      "epoch": 2.5,
+      "grad_norm": 3.546875,
       "learning_rate": 0.000178183148246803,
-      "loss": 6.0405,
-      "step": 10
-    },
-    {
-      "epoch": 15.0,
-      "grad_norm": 1.703125,
-      "learning_rate": 0.00013302790619551674,
-      "loss": 5.8472,
-      "step": 15
+      "loss": 6.2975,
+      "step": 5
     },
     {
-      "epoch": 20.0,
-      "grad_norm": 1.4296875,
+      "epoch": 5.0,
+      "grad_norm": 2.375,
       "learning_rate": 7.774790660436858e-05,
-      "loss": 5.737,
-      "step": 20
-    },
-    {
-      "epoch": 25.0,
-      "grad_norm": 1.3671875,
-      "learning_rate": 2.9289321881345254e-05,
-      "loss": 5.6888,
-      "step": 25
+      "loss": 6.0287,
+      "step": 10
     },
     {
-      "epoch": 30.0,
-      "grad_norm": 1.359375,
+      "epoch": 7.5,
+      "grad_norm": 2.328125,
       "learning_rate": 2.5072087818176382e-06,
-      "loss": 5.6765,
-      "step": 30
+      "loss": 5.9532,
+      "step": 15
     },
     {
-      "epoch": 32.0,
-      "step": 32,
-      "total_flos": 4281008652288000.0,
-      "train_loss": 5.874281942844391,
-      "train_runtime": 26.6871,
-      "train_samples_per_second": 300.97,
-      "train_steps_per_second": 1.199
+      "epoch": 8.0,
+      "step": 16,
+      "total_flos": 1070252163072000.0,
+      "train_loss": 6.084760069847107,
+      "train_runtime": 16.1029,
+      "train_samples_per_second": 124.698,
+      "train_steps_per_second": 0.994
     }
   ],
   "logging_steps": 5,
-  "max_steps": 32,
+  "max_steps": 16,
   "num_input_tokens_seen": 0,
-  "num_train_epochs": 32,
+  "num_train_epochs": 8,
   "save_steps": 100,
-  "total_flos": 4281008652288000.0,
+  "total_flos": 1070252163072000.0,
   "train_batch_size": 16,
   "trial_name": null,
   "trial_params": null

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e7338dffee09f24d9467fa76f6a06cefe6d0c4191c65797a8d4f4753d6434a12
+oid sha256:a727cd331493721c6b9666c0dc0a0fbc075a5871711e9e5bfcaa1b021a06da2d
 size 4984