pytorch
70fc8652 - [ao][sparsity] Support for embeddings and embedding bags in BaseDataSparsifier

Commit
2 years ago
[ao][sparsity] Support for embeddings and embedding bags in BaseDataSparsifier Added the embedding and embedding bags in the supported data types. Currently, the base data sparsifier extracts the weight and stores it as a parameter in the internal module container whose requires_grad=False. The embeddings inside the data sparsifier are non-trainable Test Plan: ```python test/test_ao_sparsity.py TestBaseDataSparsifier``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/79254 Approved by: https://github.com/z-a-f
Author
Committer
Parents
Loading