What problem does it solve?
This Skill addresses the challenge of scaling reinforcement learning (RL) for large language models (LLMs), enabling efficient training and optimization with the slime framework.
Core Features & Use Cases
- LLM Post-Training Framework: Slime provides a streamlined workflow for post-training reinforcement learning with LLMs.
- Custom Data Generation: Supports custom data generation workflows and flexible data buffers.
- Integration: Integrates seamlessly with Megatron-LM for training and SGLang for rollout generation.
- Use Case: Ideal for training GLM models, implementing custom data generation, or when tight integration with Megatron-LM for RL scaling is required.
Quick Start
Run the following command to start the SLIME training process for a GRPO model:
python train.py \
--actor-num-nodes 1 \
--actor-num-gpus-per-node 4 \
--rollout-num-gpus 4 \
--advantage-estimator grpo \
--use-kl-loss --kl-loss-coef 0.001 \
--rollout-batch-size 32 \
--n-samples-per-prompt 8 \
--global-batch-size 256 \
--num-rollout 3000 \
--prompt-data /path/to/data.jsonl