[2025.05.22] add MoE support

This commit is contained in:
coffee
2025-05-22 18:13:01 +08:00
parent 9ae88672f5
commit a7edf87dca
+1 -2
View File
@@ -5,8 +5,7 @@ LLM-Dolly is a custom LLM built from scratch
## 🗓️ 更新日志 ## 🗓️ 更新日志
#### 2025-05-22 #### 2025-05-22
- 📝 添加支持MoE模型,训练资源占用不稳定,测试模型Experts=8, experts_per_tok=4,GPU显存从60%-94%跳动,单个GPU利用率0%-100%跳动。 - 📝 添加支持MoE模型,训练资源占用不稳定,测试模型Experts=8, experts_per_tok=4,GPU显存从60%-94%跳动,单个GPU利用率0%-100%跳动。当调小batch_size后,显存占用稳定在90%,单个GPU利用率稳定在90%以上,偶现30%的利用率。
- 当调小batch_size后,显存占用稳定在90%,单个GPU利用率稳定在90%以上,偶现30%的利用率。
#### 2025-05-20 #### 2025-05-20
- 📝 添加支持jsonl文件训练. - 📝 添加支持jsonl文件训练.