Model save

README.md

@@ -1,11 +1,9 @@
 ---
 base_model: Qwen/Qwen2.5-1.5B-Instruct
-datasets: HuggingFaceH4/Bespoke-Stratos-17k
 library_name: transformers
 model_name: Qwen2.5-1.5B-Open-R1-Distill
 tags:
 - generated_from_trainer
-- open-r1
 - trl
 - sft
 licence: license
@@ -13,7 +11,7 @@ licence: license
 
 # Model Card for Qwen2.5-1.5B-Open-R1-Distill
 
-This model is a fine-tuned version of [Qwen/Qwen2.5-1.5B-Instruct](https://huggingface.co/Qwen/Qwen2.5-1.5B-Instruct) on the [HuggingFaceH4/Bespoke-Stratos-17k](https://huggingface.co/datasets/HuggingFaceH4/Bespoke-Stratos-17k) dataset.
+This model is a fine-tuned version of [Qwen/Qwen2.5-1.5B-Instruct](https://huggingface.co/Qwen/Qwen2.5-1.5B-Instruct).
 It has been trained using [TRL](https://github.com/huggingface/trl).
 
 ## Quick start
@@ -29,14 +27,14 @@ print(output["generated_text"])
 
 ## Training procedure
 
-[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/yh-yao/huggingface/runs/y2h2hjrz) 
+[<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/yh-yao/huggingface/runs/rweu490i) 
 
 
 This model was trained with SFT.
 
 ### Framework versions
 
-- TRL: 0.14.0
+- TRL: 0.15.0.dev0
 - Transformers: 4.49.0.dev0
 - Pytorch: 2.5.1
 - Datasets: 3.2.0

all_results.json

@@ -5,10 +5,10 @@
     "eval_samples": 100,
     "eval_samples_per_second": 162.832,
     "eval_steps_per_second": 5.089,
-    "total_flos": 5.2247958791389184e+17,
-    "train_loss": 0.8026480646528436,
-    "train_runtime": 497.857,
+    "total_flos": 76916824473600.0,
+    "train_loss": 0.8026409276138396,
+    "train_runtime": 500.1417,
     "train_samples": 16610,
-    "train_samples_per_second": 43.406,
-    "train_steps_per_second": 0.339
+    "train_samples_per_second": 43.208,
+    "train_steps_per_second": 0.338
 }

config.json

@@ -23,7 +23,7 @@
   "tie_word_embeddings": true,
   "torch_dtype": "bfloat16",
   "transformers_version": "4.49.0.dev0",
-  "use_cache": true,
+  "use_cache": false,
   "use_sliding_window": false,
   "vocab_size": 151936
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3f52fbee5cd23caea4cf13a6534a707c400d798cfb4549d037349616a7263899
+oid sha256:0909704212b2cefa2c538346dae8eef898976666d2321dc8341347dc0c3f3b8b
 size 3087467144

train_results.json

@@ -1,9 +1,9 @@
 {
     "epoch": 1.0,
-    "total_flos": 5.2247958791389184e+17,
-    "train_loss": 0.8026480646528436,
-    "train_runtime": 497.857,
+    "total_flos": 76916824473600.0,
+    "train_loss": 0.8026409276138396,
+    "train_runtime": 500.1417,
     "train_samples": 16610,
-    "train_samples_per_second": 43.406,
-    "train_steps_per_second": 0.339
+    "train_samples_per_second": 43.208,
+    "train_steps_per_second": 0.338
 }

trainer_state.json

@@ -10,239 +10,239 @@
   "log_history": [
     {
       "epoch": 0.029585798816568046,
-      "grad_norm": 2.3450317910704577,
+      "grad_norm": 2.344439472222262,
       "learning_rate": 5.882352941176471e-06,
       "loss": 1.0993,
       "step": 5
     },
     {
       "epoch": 0.05917159763313609,
-      "grad_norm": 1.633061904367079,
+      "grad_norm": 1.63601885296728,
       "learning_rate": 1.1764705882352942e-05,
       "loss": 1.0402,
       "step": 10
     },
     {
       "epoch": 0.08875739644970414,
-      "grad_norm": 0.8877456959606497,
+      "grad_norm": 0.8879200743942525,
       "learning_rate": 1.7647058823529414e-05,
       "loss": 0.9533,
       "step": 15
     },
     {
       "epoch": 0.11834319526627218,
-      "grad_norm": 0.606798846508287,
+      "grad_norm": 0.6062571598207454,
       "learning_rate": 1.9980782984658682e-05,
       "loss": 0.8923,
       "step": 20
     },
     {
       "epoch": 0.14792899408284024,
-      "grad_norm": 0.5388811885900797,
+      "grad_norm": 0.5398319056179057,
       "learning_rate": 1.9863613034027224e-05,
       "loss": 0.8552,
       "step": 25
     },
     {
       "epoch": 0.17751479289940827,
-      "grad_norm": 0.4250915517708058,
+      "grad_norm": 0.4257339924474833,
       "learning_rate": 1.9641197940012136e-05,
       "loss": 0.8283,
       "step": 30
     },
     {
       "epoch": 0.20710059171597633,
-      "grad_norm": 0.37726357010572764,
+      "grad_norm": 0.3758029788005641,
       "learning_rate": 1.9315910880512792e-05,
-      "loss": 0.8229,
+      "loss": 0.823,
       "step": 35
     },
     {
       "epoch": 0.23668639053254437,
-      "grad_norm": 0.4017368360482797,
+      "grad_norm": 0.3990910798541474,
       "learning_rate": 1.8891222681391853e-05,
       "loss": 0.8225,
       "step": 40
     },
     {
       "epoch": 0.26627218934911245,
-      "grad_norm": 0.3431204044336116,
+      "grad_norm": 0.34475189059262795,
       "learning_rate": 1.8371664782625287e-05,
-      "loss": 0.8073,
+      "loss": 0.8074,
       "step": 45
     },
     {
       "epoch": 0.2958579881656805,
-      "grad_norm": 0.33341329906969774,
+      "grad_norm": 0.333866593937156,
       "learning_rate": 1.7762780887657576e-05,
       "loss": 0.7978,
       "step": 50
     },
     {
       "epoch": 0.3254437869822485,
-      "grad_norm": 0.36841342218843565,
+      "grad_norm": 0.36564553792757043,
       "learning_rate": 1.7071067811865477e-05,
       "loss": 0.7877,
       "step": 55
     },
     {
       "epoch": 0.35502958579881655,
-      "grad_norm": 0.376161654126924,
+      "grad_norm": 0.3991199908593827,
       "learning_rate": 1.6303906161279554e-05,
       "loss": 0.7981,
       "step": 60
     },
     {
       "epoch": 0.38461538461538464,
-      "grad_norm": 0.3429901732874326,
+      "grad_norm": 0.34298175038858175,
       "learning_rate": 1.5469481581224274e-05,
       "loss": 0.7721,
       "step": 65
     },
     {
       "epoch": 0.41420118343195267,
-      "grad_norm": 0.3449881548396526,
+      "grad_norm": 0.3456184403820059,
       "learning_rate": 1.4576697415156818e-05,
       "loss": 0.7741,
       "step": 70
     },
     {
       "epoch": 0.4437869822485207,
-      "grad_norm": 0.3373615559482272,
+      "grad_norm": 0.3381633650630839,
       "learning_rate": 1.3635079705638298e-05,
       "loss": 0.7852,
       "step": 75
     },
     {
       "epoch": 0.47337278106508873,
-      "grad_norm": 0.35169218399134256,
+      "grad_norm": 0.35047408988206236,
       "learning_rate": 1.2654675551080724e-05,
       "loss": 0.7666,
       "step": 80
     },
     {
       "epoch": 0.5029585798816568,
-      "grad_norm": 0.3611334142860622,
+      "grad_norm": 0.36085121845408574,
       "learning_rate": 1.164594590280734e-05,
       "loss": 0.7706,
       "step": 85
     },
     {
       "epoch": 0.5325443786982249,
-      "grad_norm": 0.36061127549275923,
+      "grad_norm": 0.35902845961657487,
       "learning_rate": 1.0619653946285948e-05,
       "loss": 0.77,
       "step": 90
     },
     {
       "epoch": 0.5621301775147929,
-      "grad_norm": 0.3750103303716175,
+      "grad_norm": 0.36863631725357704,
       "learning_rate": 9.586750257511868e-06,
       "loss": 0.7771,
       "step": 95
     },
     {
       "epoch": 0.591715976331361,
-      "grad_norm": 0.33357288560635495,
+      "grad_norm": 0.3368221799612284,
       "learning_rate": 8.558255959926533e-06,
       "loss": 0.753,
       "step": 100
     },
     {
       "epoch": 0.591715976331361,
-      "eval_loss": 0.7880157232284546,
-      "eval_runtime": 0.7824,
-      "eval_samples_per_second": 163.595,
-      "eval_steps_per_second": 5.112,
+      "eval_loss": 0.7879893779754639,
+      "eval_runtime": 0.786,
+      "eval_samples_per_second": 162.857,
+      "eval_steps_per_second": 5.089,
       "step": 100
     },
     {
       "epoch": 0.621301775147929,
-      "grad_norm": 0.334580352799432,
+      "grad_norm": 0.3337128215279273,
       "learning_rate": 7.545145128592009e-06,
       "loss": 0.7569,
       "step": 105
     },
     {
       "epoch": 0.650887573964497,
-      "grad_norm": 0.3457940719409492,
+      "grad_norm": 0.34455276378156463,
       "learning_rate": 6.558227696373617e-06,
-      "loss": 0.7681,
+      "loss": 0.768,
       "step": 110
     },
     {
       "epoch": 0.6804733727810651,
-      "grad_norm": 0.3080395598137145,
+      "grad_norm": 0.30772420523279587,
       "learning_rate": 5.608034111526298e-06,
       "loss": 0.7623,
       "step": 115
     },
     {
       "epoch": 0.7100591715976331,
-      "grad_norm": 0.2972896798042334,
+      "grad_norm": 0.29781261348903965,
       "learning_rate": 4.704702977392914e-06,
-      "loss": 0.7514,
+      "loss": 0.7513,
       "step": 120
     },
     {
       "epoch": 0.7396449704142012,
-      "grad_norm": 0.34366581285699394,
+      "grad_norm": 0.34294568370060885,
       "learning_rate": 3.857872873103322e-06,
       "loss": 0.7538,
       "step": 125
     },
     {
       "epoch": 0.7692307692307693,
-      "grad_norm": 0.3114970039276949,
+      "grad_norm": 0.3123527643773362,
       "learning_rate": 3.0765795095517026e-06,
       "loss": 0.7555,
       "step": 130
     },
     {
       "epoch": 0.7988165680473372,
-      "grad_norm": 0.32338821484371644,
+      "grad_norm": 0.3221614581767522,
       "learning_rate": 2.369159318001937e-06,
       "loss": 0.7584,
       "step": 135
     },
     {
       "epoch": 0.8284023668639053,
-      "grad_norm": 0.2993687193760852,
+      "grad_norm": 0.29845427650643824,
       "learning_rate": 1.743160500034443e-06,
       "loss": 0.7498,
       "step": 140
     },
     {
       "epoch": 0.8579881656804734,
-      "grad_norm": 0.31118464680620006,
+      "grad_norm": 0.3105750284295203,
       "learning_rate": 1.2052624879351105e-06,
       "loss": 0.7566,
       "step": 145
     },
     {
       "epoch": 0.8875739644970414,
-      "grad_norm": 0.29520967964613526,
+      "grad_norm": 0.2941366532979867,
       "learning_rate": 7.612046748871327e-07,
       "loss": 0.7665,
       "step": 150
     },
     {
       "epoch": 0.9171597633136095,
-      "grad_norm": 0.2965665861654801,
+      "grad_norm": 0.2970058759706501,
       "learning_rate": 4.1572517541747294e-07,
       "loss": 0.7613,
       "step": 155
     },
     {
       "epoch": 0.9467455621301775,
-      "grad_norm": 0.285789430294503,
+      "grad_norm": 0.28596540886894695,
       "learning_rate": 1.7251026952640583e-07,
-      "loss": 0.7607,
+      "loss": 0.7606,
       "step": 160
     },
     {
       "epoch": 0.9763313609467456,
-      "grad_norm": 0.27681573266768217,
+      "grad_norm": 0.2764995113045065,
       "learning_rate": 3.4155069933301535e-08,
       "loss": 0.7445,
       "step": 165
@@ -250,11 +250,11 @@
     {
       "epoch": 1.0,
       "step": 169,
-      "total_flos": 5.2247958791389184e+17,
-      "train_loss": 0.8026480646528436,
-      "train_runtime": 497.857,
-      "train_samples_per_second": 43.406,
-      "train_steps_per_second": 0.339
+      "total_flos": 76916824473600.0,
+      "train_loss": 0.8026409276138396,
+      "train_runtime": 500.1417,
+      "train_samples_per_second": 43.208,
+      "train_steps_per_second": 0.338
     }
   ],
   "logging_steps": 5,
@@ -274,7 +274,7 @@
       "attributes": {}
     }
   },
-  "total_flos": 5.2247958791389184e+17,
+  "total_flos": 76916824473600.0,
   "train_batch_size": 4,
   "trial_name": null,
   "trial_params": null

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bb626b66b49ea77e8a67fa44a8a0a656bac2735ba7f30e25e289e491b4481d1d
+oid sha256:6c75eb157f86ab6474612e5f9b9d34ac42ec3740174caf5a923e5a7ae4f8be53
 size 7352