基于DeepSeek-R1 15b微调训练自主的领域大模型,附带工程代码

 (1)基础模型DeepSeek-R1-Distill-Qwen-1.5B与微调训练后的deepseek-1.5b-lora模型使用对比。

  (2)微调训练模型的过程及Python代码。

 

image

2.   测试DeepSeek-R1 15B原生模型

2.1    提示词1:iNeuOS是什么?

结论:回答内容与iNeuOS不相关。

提问:iNeuOS是什么?

回答:如下图:

 

1

2.2    提示词2:iNeuOS工业互联网操作系统的特点?

结论:回答内容说了一大堆废话,与提问内容不相关,浪费资源。

提问:iNeuOS工业互联网操作系统的特点?

回答:如下图:

 

1

2.3    提示词3:iNeuOS工业互联网操作系统的功能介绍?

结论:回答内容说了一大堆英文,与提问内容不相关,浪费资源。

提问:iNeuOS工业互联网操作系统的功能介绍?

回答:如下图:

 

1

3.   测试微调训练后的大模型

3.1    提示词1:iNeuOS是什么?

结论:回答的内容与iNeuOS有少许的相关性,但是回答的内容不成体系,可能与由于提问太过笼统、训练模型的数据集都有关系。

提问:iNeuOS是什么?

回答:如下图:

 

1

3.2    提示词2:iNeuOS工业互联网操作系统的特点?

结论:相比提示词1,回答的内容与iNeuOS相关的内容更多了,回答的准确度提高了,内容更体系化。

提问:iNeuOS工业互联网操作系统的特点?

回答:如下图:

 

1

3.3    提示词3:iNeuOS工业互联网操作系统的功能介绍?

结论:回答的内容大部分与训练模型的数据集相关,内容更有逻辑感。

提问:iNeuOS工业互联网操作系统的功能介绍?

回答:如下图:

 

1

3.4    提示词4:写一份物联网的方案,包含数据采集、数据可视化、数据表报等

结论:增加回答难度,回答的内容基本与训练模型的数据集相关,如果提示词更完善,那么回答的内容更准确。

提问:依据iNeuOS工业互联网操作系统相关内容,写一份物联网的方案

回答:如下图:

 

1

4.   微调模型训练过程及代码

4.1    代码工程目录

  下载代码:https://github.com/wxzz/iNeuOS_LLM

LLM/

├───📂 deepseek-1.5b-lora-final/    //微调训练后的模型

├───📂 deepseek-1.5b-lora-merged/  //微调后模型与基础模型合并的模型

├───📂 deepseek-ai/

│   ├───📂 DeepSeek-R1-Distill-Qwen-1.5B/   //基础模型

├───📂 docx_input/  //生成数据集的基础Word文件

├───datasets.json/  //微调训练模型的输入数据集文件

├───merged_model.py/  //用于合并基础模型与微调后的模型

├───test_base_model.py/  //测试基础模型

├───test_merged_model.py/  //测试合并后的模型

├───test_train_model.py/  //测试微调后的模型

├───train_model.py/  //微调训练模型

├───training_dataset.json/  //转换Word生成的数据集文件

├───word_to_dataset.py/  //用于转换Word生成的数据集

4.2    下载基础模型DeepSeek-R1-Distill-Qwen-1.5B

  使用huggingface-cli下载基础模型,工具下载地址:https://huggingface.co/deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B

4.3    准备数据集

(1)       把准备好的Word文件,放入docx_input文件夹。一共157个文件。如下图:

(2)       运行word_to_dataset.py代码,会生成training_dataset.json数据集文件,把training_dataset.json文件名称改为datasets.json。数据集有很大的调优空间。

4.4    训练模型

(1)       运行train_model.py,训练模型,以我的笔记本配置,训练完成datasets.json数据集需要将近4个小时。

(2)       训练完成的模型保存在deepseek-1.5b-lora-final目录下。

(3)       使用test_train_model.py代码测试基础模型与训练后的模型。

4.5    合并模型

(1)       运行merged_model.py代码,把基础模型与训练后的模型合并成一个整体的模型,保存在deepseek-1.5b-lora-merged目录下。

(2)       运行test_merged_model.py代码,测试合并后的模型。应用效果参见本文章节:测试微调训练后的大模型。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值