Update README.md
Browse files
README.md
CHANGED
|
@@ -18,6 +18,10 @@ tags:
|
|
| 18 |
|
| 19 |
# 春梦蝶最小示例模型集合
|
| 20 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
## 项目简介
|
| 22 |
|
| 23 |
这是一个从零开始、逐级递进的**最小示例模型集合**，展示了如何用最少的代码和数据，在手机上训练出能“说话”的猫娘模型。
|
|
@@ -45,7 +49,7 @@ tags:
|
|
| 45 |
|
| 46 |
- v1、v2、v3：提供训练完成的最终权重（`final.pth`）
|
| 47 |
- v4：因手机 CPU 训练效果差，只提供代码，不提供权重
|
| 48 |
-
- v5：因参数量过大（2100万），超出手机性能，只提供代码参考
|
| 49 |
|
| 50 |
> 中间 checkpoint 因文件大小和存储限制未包含在仓库中，如需完整训练日志或中间版本，可自行运行训练脚本生成。
|
| 51 |
|
|
|
|
| 18 |
|
| 19 |
# 春梦蝶最小示例模型集合
|
| 20 |
|
| 21 |
+
## 关于作者
|
| 22 |
+
|
| 23 |
+
一个热爱 AI 的中学生。欢迎交流学习。
|
| 24 |
+
|
| 25 |
## 项目简介
|
| 26 |
|
| 27 |
这是一个从零开始、逐级递进的**最小示例模型集合**，展示了如何用最少的代码和数据，在手机上训练出能“说话”的猫娘模型。
|
|
|
|
| 49 |
|
| 50 |
- v1、v2、v3：提供训练完成的最终权重（`final.pth`）
|
| 51 |
- v4：因手机 CPU 训练效果差，只提供代码，不提供权重
|
| 52 |
+
- v5：因参数量过大（2100万），超出手机性能，只提供代码参考（如果想尝试，建议手动加上CUDA或者找更好的手机）
|
| 53 |
|
| 54 |
> 中间 checkpoint 因文件大小和存储限制未包含在仓库中，如需完整训练日志或中间版本，可自行运行训练脚本生成。
|
| 55 |
|