---
library_name: transformers
tags:
- trl
- grpo
---

# SmolGRPO-135M

This is a fine-tune of [HuggingFaceTB/SmolLM2-135M-Instruct](https://huggingface.co/HuggingFaceTB/SmolLM2-135M-Instruct) using GRPO on [mlabonne/smoltldr](https://huggingface.co/datasets/mlabonne/smoltldr) (2k samples).
It is designed to summarize Reddit posts using ~50 characters.

You can reproduce this training using this [colab notebook](https://colab.research.google.com/drive/13mRqgRIvMGGgkQfJL4CS0lzcL4Vl9xUN?usp=sharing). It takes about 40 minutes to train the model.

**Takeaways from these experiments**:
* Adding a system prompt like "Summarize the following text concisely" doesn't help
* You can get faster convergence with a higher learning rate and fewer samples, but it's prone to overshooting your target.
* I tried many reward functions to play with reward shaping but it didn't seem to help

## Example

Input:
> SUBREDDIT: r/Advice
>
> TITLE: I have big dreams and goals, but they are kind of cloudy.[m20]
>
> POST: I live at home with my family right now and I don't go to school. I went to college for a year and decided to stop. My best friend convinced me that I don't need college to do what I want to do. Besides, I hated taking classes I wasn't interested in. The things I want to do in life (I know it seems like too much) included producing music, making a cartoon, making comics, designing clothes and shoes, and other smaller things related to that. I grew up with a good family with a father that had similar dreams. He niw has a job he's been working for 20+ years that he doesn't like. I too am afraid of falling into that path. I've been pretty down and frustrated and feeling things are quite impossible although I know there's always hope. I don't really want to do anything else, but I'm stuck. I'll be pretty down for a week, then by the next week I'm in high spirits with a game plan that always fails. I've been doing this for quite a while now. My parents are starting to get on my case now and when they ask what I'm gonna do in life I don't know how to respond. Maybe I should try looking for lessons in nyc to get me out of the house? I practice drawing and making music a lot, but I can never feel satisfied and feel like I'm moving in the right direction.  Everything seems like a scary cycle of ups and downs. I have faith I can turn it around, but I just don't know how.
>
> TL;DR:

Output:
> I have big dreams and goals, but they are kind of cloudy. I'm going to try to figure out a plan to get out of that house, but I'm not sure what that plan will be.