sparseml.pytorch.sparsification package¶
Submodules¶
sparseml.pytorch.sparsification.info module¶
Functionality related to describing availability and information of sparsification algorithms to models within in the PyTorch framework.
-
sparseml.pytorch.sparsification.info.sparsification_info() → sparseml.sparsification.info.SparsificationInfo[source]¶ Load the available setup for sparsifying model within pytorch.
- Returns
The sparsification info for the pytorch framework
- Return type
Module contents¶
Functionality related to applying, describing, and supporting sparsification algorithms to models within in the PyTorch framework.