What problem does it solve?
This Skill addresses the challenges of post-training reinforcement learning (RL) for large language models (LLMs) by providing a robust framework for scaling and integration with existing infrastructure.
Core Features & Use Cases
- LLM Post-Training with RL: Facilitates the integration of reinforcement learning techniques for optimizing LLM performance.
- Custom Data Generation Workflows: Supports flexible data buffers and custom data generation for tailored training sessions.
- Megatron-LM Integration: Ensures seamless integration with the Megatron-LM framework for scalable training.
- Use Case: For a research team developing a new LLM, this Skill can be used to fine-tune a model with RL techniques, improving its performance in specific tasks.
Quick Start
Run the following command to start a GRPO training session with the slime framework:
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 \
${MODEL_ARGS[@]} ${CKPT_ARGS[@]}