[embeddings] add doc in readme (#1711)

pull/1712/head
Jiarui Fang 2 years ago committed by GitHub
parent 22a115406b
commit b0a23dc4fc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -148,6 +148,11 @@ Colossal-AI 为您提供了一系列并行组件。我们的目标是让您的
请访问我们的 [文档](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>
## 单GPU训练样例展示

@ -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.
### 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>
## Single GPU Training Demo

Loading…
Cancel
Save