[embeddings] add doc in readme (hpcaitech#1711)

erhaa233 · Oct 16, 2022 · b0a23dc · b0a23dc
1 parent 22a1154
commit b0a23dc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/README-zh-Hans.md b/README-zh-Hans.md
@@ -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训练样例展示

diff --git a/README.md b/README.md
@@ -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