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.