baichuan2全参数微调流程

  1. 租用 Autodl/恒源云/智星云 服务器
  2. 选择 100 80G/A800 80G 务器租用(2卡或以上),使用pytorch 1.11.0 python3.8版本镜像
  3. 下载02_llm_code.zip
  4. 解压后创建虚拟环境modelscope_env,pip install modelscope
  5. 安装完成后激活modelscope_env(source activate modelscope_env)
  6. 执行模型下载 llm_download/baichuan2_7B_base_download.py
  7. 下载完成,创建虚拟环境 llm_env
  8. 在 llm_env 中安装requirements.txt(pip install -r requirements.txt)
  9. 安装完成,在training_config/baichuan2_config.json 里面去修改训练参数(例如模型路径)
  10. 根据租用的GPU数量修改train_ft.sh里面的GPU配置
  11. 执行 nohup bash train_ft.sh > 1125.log &
  12. 查看显存利用情况 nvidia-smi
  13. 持续查看显存变化 watch -n -1 nvidia-smi
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容