github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[embeddings] add doc in readme (#1711)

pull/1712/head
Jiarui Fang 2022-10-16 21:57:50 +08:00 committed by GitHub
parent 22a115406b
commit b0a23dc4fc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
README-zh-Hans.md
View File

@ -148,6 +148,11 @@ Colossal-AI 为您提供了一系列并行组件。我们的目标是让您的
请访问我们的 [文档](https://www.colossalai.org/) 和 [例程](https://github.com/hpcaitech/ColossalAI-Examples) 以了解详情。 请访问我们的 [文档](https://www.colossalai.org/) 和 [例程](https://github.com/hpcaitech/ColossalAI-Examples) 以了解详情。
### 推荐系统模型
- [Cached Embedding](https://github.com/hpcaitech/FreqCacheEmbedding), 使用软件Cache实现Embeddings，用更少GPU显存训练更大的模型。
<p align="right">(<a href="#top">返回顶端</a>)</p> <p align="right">(<a href="#top">返回顶端</a>)</p>
## 单GPU训练样例展示 ## 单GPU训练样例展示

3
README.md
View File

@ -153,6 +153,9 @@ distributed training and inference in a few lines.
Please visit our [documentation](https://www.colossalai.org/) and [examples](https://github.com/hpcaitech/ColossalAI-Examples) for more details. Please visit our [documentation](https://www.colossalai.org/) and [examples](https://github.com/hpcaitech/ColossalAI-Examples) for more details.
### Recommendation System Models
- [Cached Embedding](https://github.com/hpcaitech/FreqCacheEmbedding), utilize software cache to train larger embedding tables with a smaller GPU memory budget.
<p align="right">(<a href="#top">back to top</a>)</p> <p align="right">(<a href="#top">back to top</a>)</p>
## Single GPU Training Demo ## Single GPU Training Demo