当前位置：首页 » AI答疑

使用MiniMind进行领域定制训练(如医疗问答)的具体流程是什么？

2025-08-28

1.2 K

领域定制需完成以下四阶段操作：

数据准备阶段：
- 收集专业数据并整理为lora_medical.jsonl格式
- 建议保持512 token长度以适配模型架构
参数配置阶段：
- 修改./model/LMConfig.py中的n_layers等参数
- 调整batch_size避免显存溢出（3090建议≤8）
模型训练阶段：
- 执行python train_lora.py启动领域适应训练
- 添加--use_wandb参数监控损失曲线
部署应用阶段：
- 使用serve_openai_api.py启动服务
- 通过curl命令测试医疗问答接口

注意：专业领域训练建议先进行基础预训练（2-3轮）再进行LoRA微调。

快速查询站内AI工具