sparseml.pytorch package¶
Subpackages¶
- sparseml.pytorch.datasets package
- Subpackages
- sparseml.pytorch.datasets.classification package
- Submodules
- sparseml.pytorch.datasets.classification.cifar module
- sparseml.pytorch.datasets.classification.imagefolder module
- sparseml.pytorch.datasets.classification.imagenet module
- sparseml.pytorch.datasets.classification.imagenette module
- sparseml.pytorch.datasets.classification.mnist module
- Module contents
- sparseml.pytorch.datasets.detection package
- sparseml.pytorch.datasets.recommendation package
- sparseml.pytorch.datasets.video package
- sparseml.pytorch.datasets.classification package
- Submodules
- sparseml.pytorch.datasets.generic module
- sparseml.pytorch.datasets.registry module
- Module contents
- Subpackages
- sparseml.pytorch.models package
- Subpackages
- sparseml.pytorch.models.classification package
- Submodules
- sparseml.pytorch.models.classification.darknet module
- sparseml.pytorch.models.classification.efficientnet module
- sparseml.pytorch.models.classification.inception_v3 module
- sparseml.pytorch.models.classification.mnist module
- sparseml.pytorch.models.classification.mobilenet module
- sparseml.pytorch.models.classification.mobilenet_v2 module
- sparseml.pytorch.models.classification.resnet module
- sparseml.pytorch.models.classification.vgg module
- Module contents
- sparseml.pytorch.models.detection package
- sparseml.pytorch.models.external package
- sparseml.pytorch.models.recommendation package
- sparseml.pytorch.models.classification package
- Submodules
- sparseml.pytorch.models.registry module
- Module contents
- Subpackages
- sparseml.pytorch.nn package
- sparseml.pytorch.optim package
- Submodules
- sparseml.pytorch.optim.analyzer_as module
- sparseml.pytorch.optim.analyzer_module module
- sparseml.pytorch.optim.analyzer_pruning module
- sparseml.pytorch.optim.manager module
- sparseml.pytorch.optim.mask_creator_pruning module
- sparseml.pytorch.optim.mask_pruning module
- sparseml.pytorch.optim.modifier module
- sparseml.pytorch.optim.modifier_as module
- sparseml.pytorch.optim.modifier_epoch module
- sparseml.pytorch.optim.modifier_lr module
- sparseml.pytorch.optim.modifier_params module
- sparseml.pytorch.optim.modifier_pruning module
- sparseml.pytorch.optim.modifier_quantization module
- sparseml.pytorch.optim.modifier_regularizer module
- sparseml.pytorch.optim.optimizer module
- sparseml.pytorch.optim.sensitivity_as module
- sparseml.pytorch.optim.sensitivity_lr module
- sparseml.pytorch.optim.sensitivity_pruning module
- Module contents
- sparseml.pytorch.utils package
- Subpackages
- Submodules
- sparseml.pytorch.utils.benchmarker module
- sparseml.pytorch.utils.callbacks module
- sparseml.pytorch.utils.exporter module
- sparseml.pytorch.utils.helpers module
- sparseml.pytorch.utils.logger module
- sparseml.pytorch.utils.loss module
- sparseml.pytorch.utils.model module
- sparseml.pytorch.utils.module module
- sparseml.pytorch.utils.ssd_helpers module
- sparseml.pytorch.utils.yolo_helpers module
- Module contents
Module contents¶
Functionality for working with and sparsifying Models in the PyTorch framework