[论文笔记] chatgpt系列 2.2 DeepSpeed-chat 训练流程脚本

这篇博客详细介绍了ChatGPT系列的训练过程,特别是使用DeepSpeed-Chat进行监督微调、奖励模型构建及强化学习人类反馈(RLHF)的步骤。DeepSpeed的加速技术有效减少了训练成本,而RLHF则通过PPO训练增强了模型性能。
摘要由CSDN通过智能技术生成

 Step 1 - Supervised Fine-Tuning

# Move into the first step of the pipeline cd training/step1_supervised_finetuning/

# Run the training script

bash training_scripts/single_gpu/ run_1.3b.sh

# Evaluate the model

bash evaluation_scripts/ run_prompt.sh

 Step 2 - Reward Model

# Move into the second step of the pipeline cd training/step2_reward_model_finetuning

# Run the training script

bash training_scripts/single_gpu/run_350m.sh

# Evaluate the model

bash evaluation_scripts/ run_eval.sh

 Step 3 - RLHF

作为整个3步InstructGPT管道中最复杂的一步,DeepSpeed Chat的混合动力引擎提供了足够的加速,以避免大量的训练时间(成本)。

强化学习人类反馈(RLHF)获取更多信息。如果您已经有了经过微调的Actor和奖励模型检查点,那么您可以简单地运行以下脚本来启用PPO训练。

# Move into the final step of the pipeline cd training/step3_rthffinetuning/

# Run the training script

bash training_scripts/single_gpu/ run_1.3b.sh

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

心心喵

喵喵(*^▽^*)

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值