yamle.pruning package

yamle.pruning.pruner_factory(pruner_type=None)

This function is used to create a pruner instance based on the pruner type.

Parameters:

pruner_type (str) – The type of pruner to create.

Return type:

Type[Callable]

Subpackages

• yamle.pruning.unstructured package
  • Submodules
    • yamle.pruning.unstructured.magnitude module
      • UnstructuredMagnitudePruner

Submodules

• yamle.pruning.pruner module
  • BasePruner
    • BasePruner.prune_parameter()
    • BasePruner.recover_parameter()
    • BasePruner.recover()
    • BasePruner.add_specific_args()
    • BasePruner.summary()
  • DummyPruner
    • DummyPruner.summary()