End of training

README.md

@@ -1,6 +1,6 @@
 ---
 base_model: Qwen/Qwen2.5-0.5B-Instruct
-datasets: AI-MO/NuminaMath-TIR
+datasets: BytedTsinghua-SIA/DAPO-Math-17k
 library_name: transformers
 model_name: Qwen2.5-0.5B-Instruct-GRPO-test
 tags:
@@ -12,7 +12,7 @@ licence: license
 
 # Model Card for Qwen2.5-0.5B-Instruct-GRPO-test
 
-This model is a fine-tuned version of [Qwen/Qwen2.5-0.5B-Instruct](https://huggingface.co/Qwen/Qwen2.5-0.5B-Instruct) on the [AI-MO/NuminaMath-TIR](https://huggingface.co/datasets/AI-MO/NuminaMath-TIR) dataset.
+This model is a fine-tuned version of [Qwen/Qwen2.5-0.5B-Instruct](https://huggingface.co/Qwen/Qwen2.5-0.5B-Instruct) on the [BytedTsinghua-SIA/DAPO-Math-17k](https://huggingface.co/datasets/BytedTsinghua-SIA/DAPO-Math-17k) dataset.
 It has been trained using [TRL](https://github.com/huggingface/trl).
 
 ## Quick start
@@ -28,14 +28,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/fangwu/grpo_train/runs/eeyiyy9t) 
+[<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/fangwu/grpo_train/runs/to1a36jv) 
 
 
 This model was trained with GRPO, a method introduced in [DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models](https://huggingface.co/papers/2402.03300).
 
 ### Framework versions
 
-- TRL: 0.15.2
+- TRL: 0.16.1
 - Transformers: 4.49.0
 - Pytorch: 2.2.2
 - Datasets: 3.3.2