policy

train_grpo.py

kossisoroyce · PyTorch

or hover any field below to flag it

Overview

Name
train_grpo.py
Author
kossisoroyce
Framework
PyTorch
License
GPL-3.0
Skill type
other
Evidence level
untested
Task description
GRPO Training Script for Qwen Model on GSM8K Dataset. This script trains a Qwen model using the GRPO (Generalized Reinforcement Policy Optimization) method on the GSM8K (Generalized Math 8K) dataset. The script leverages transformers, PEFT (Parameter-Efficient Fine-Tuning), and TRL (Transformers Rei

Spaces

Action space
other · 0-dim · 0Hz
Observation space
  • type: other

Links

HuggingFace repo
null
Paper (arXiv)
null

Compatible robots

3+17 mentioned but not in catalog yet

Compatible environments

0

No environments list train_grpo.py yet.

Datasets that reference this policy

0

No datasets reference train_grpo.py yet.