search

OpenMOSS / CoLLiE

Collaborative Training of Large Language Models in an Efficient Way
https://openlmlab-collie.readthedocs.io
Apache License 2.0
410 stars 58 forks source link

重构ChatGLM2并增加对应测试用例 #129

Closed MorningForest closed 1 year ago

MorningForest commented 1 year ago
  1. 修改model中的chatglm2模型架构
  2. 测试pp,dp,pp+tp的文本生成
  3. 测试pp,dp,dp+tp的文本分类任务
  4. 修改了accuracy的返回的dict中存在类型为numpy的bug