多模态示例架构
最后更新:2025/04/28。
介绍
现在,verl 已支持多模态训练。你可以使用 fsdp 和 vllm/sglang 来启动一个多模态 RL 任务。Megatron 的支持也在进行中。
请按照以下步骤快速启动一个多模态 RL 任务。
第一步:准备数据集
# 它将被保存在 $HOME/data/geo3k 文件夹中
python examples/data_preprocess/geo3k.py
第二步:下载模型
# 从 huggingface 下载模型
python3 -c "import transformers; transformers.pipeline(model='Qwen/Qwen2.5-VL-7B-Instruct')"
第三步:使用多模态模型在 Geo3K 数据集上执行 GRPO 训练
# 运行任务
bash examples/grpo_trainer/run_qwen2_5_vl-7b.sh