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

SparsificationInfo

Module contents

Functionality related to applying, describing, and supporting sparsification algorithms to models within in the PyTorch framework.