Example of using quantized.Embedding?

I’ve read the quantization doc here: (beta) Static Quantization with Eager Mode in PyTorch — PyTorch Tutorials 2.2.1+cu121 documentation

but I couldn’t find an example for an embedding model to work with quantized.Embedding. Is it documented somewhere?

Simply replace the embedding module, lead to error of empty parameter list error.


see the table above: Quantization — PyTorch 2.2 documentation

if you apply either the fx or eager quantization flows to a model with embeddings, they should get quantized.