CachedEmbedding icon indicating copy to clipboard operation
CachedEmbedding copied to clipboard

A memory efficient DLRM training solution using ColossalAI

Results 2 CachedEmbedding issues
Sort by recently updated
recently updated
newest added

Upon setting up customized [torchrec](https://github.com/hpcaitech/torchrec) as dependency of `hpcaitech/CachedEmbedding` following the official README.md, `torchrec` does not appear to be properly installed. # Steps to replicate 1. Install `PyTorch` 1.12.1 build...

I need to build custom image with CachedEmbedding. Yet there's no such commit ID for `hpcaitech/torch `as indicated in the `README.md` ``` Install our customized [torchrec](https://github.com/hpcaitech/torchrec) (commit id e8d8eda5e7a0619bd779e35065397679e1536dcd) https://github.com/hpcaitech/torchrec...