Huatuo Llama Med Chinese 教程

万万哇  金牌会员 | 昨天 02:33 | 来自手机 | 显示全部楼层 | 阅读模式
打印 上一主题 下一主题

主题 505|帖子 505|积分 1515

Huatuo Llama Med Chinese 教程

  Huatuo-Llama-Med-ChineseRepo for BenTsao [original name: HuaTuo (华驼)], Instruction-tuning Large Language Models with Chinese Medical Knowledge. 本草(原名:华驼)模子仓库,基于中文医学知识的大语言模子指令微调项目地址:https://gitcode.com/gh_mirrors/hu/Huatuo-Llama-Med-Chinese
1. 项目目录结构及介绍

以下是对Huatuo-Llama-Med-Chinese项目的根本目录结构及其作用的概述:
  1. Huatuo-Llama-Med-Chinese/
  2. ├── README.md         # 项目简介和指南
  3. ├── config.yml        # 默认配置文件
  4. ├── requirements.txt  # 依赖项列表
  5. ├── src/
  6. │   ├── model.py       # 模型定义和加载
  7. │   ├── tokenizer.py   # 词法分析器
  8. │   ├── trainer.py     # 训练脚本
  9. │   └── inference.py   # 推断脚本
  10. └── scripts/
  11.     ├── train.sh       # 训练脚本bash文件
  12.     └── infer.sh       # 推断脚本bash文件
复制代码


  • README.md: 项目说明,包括安装和使用指导。
  • config.yml: 存储模子微调和推断的参数设置。
  • requirements.txt: 列出所有必要的Python库,供安装依赖。
  • src/: 代码主体,包罗了模子相干的实现。

    • model.py: 定义Huatuo模子并处置处罚模子加载。
    • tokenizer.py: 实现对输入文本的预处置处罚和tokenization。
    • trainer.py: 包罗模子的训练逻辑。
    • inference.py: 提供模子的推断功能。

  • scripts/: 包罗主动化训练和推断的shell脚本。
2. 项目的启动文件介绍

2.1 训练脚本

启动训练Huatuo模子,你可以运行scripts/train.sh。这个脚本通常会实行以下步调:

  • 根据config.yml配置加载参数。
  • 导入并初始化模子。
  • 加载数据集。
  • 开始模子训练循环,并保存最佳模子。
确保修改train.sh以顺应你的计算环境和训练需求,比方GPU分配和训练轮次。
2.2 推断脚本

对于模子的推断,使用scripts/infer.sh。它会加载预先训练的模子并进行预测:

  • 根据config.yml配置加载参数。
  • 初始化模子,加载最佳权重。
  • 从命令行或指定文件中获取输入文本。
  • 使用模子进行文本处置处罚并输出结果。
同样,根据实际需求调解infer.sh中的路径和参数。
3. 项目的配置文件介绍

config.yml是项目的主要配置文件,包罗模子训练和推断的关键参数。示例内容可能如下:
  1. model:
  2.   name: huatuo_llama_med
  3.   checkpoint_path: ./checkpoints/best_model.ckpt
  4.   max_length: 256
  5.   min_length: 50
  6. data:
  7.   train_data_path: ./data/train.csv
  8.   valid_data_path: ./data/validation.csv
  9.   test_data_path: ./data/test.csv
  10. training:
  11.   batch_size: 8
  12.   epochs: 5
  13.   learning_rate: 5e-5
  14.   weight_decay: 0.01
  15.   early_stopping_patience: 3
  16. inference:
  17.   input_text: "请输入待处理的文本"
  18.   output_file: "./results/output.txt"
复制代码


  • model: 有关模子的部分,包括名称、检查点路径、最大和最小序列长度。
  • data: 数据路径,如训练、验证和测试数据集的位置。
  • training: 训练参数,如批大小、迭代次数、学习率和权重衰减。
  • inference: 推断参数,包括输入文本、输出文件路径。
根据具体应用,你可以修改这些参数来顺应差别的任务和环境。记得在更新配置后重新运行相应的脚本。
  Huatuo-Llama-Med-ChineseRepo for BenTsao [original name: HuaTuo (华驼)], Instruction-tuning Large Language Models with Chinese Medical Knowledge. 本草(原名:华驼)模子仓库,基于中文医学知识的大语言模子指令微调项目地址:https://gitcode.com/gh_mirrors/hu/Huatuo-Llama-Med-Chinese

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

万万哇

金牌会员
这个人很懒什么都没写!

标签云

快速回复 返回顶部 返回列表