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
      • Submodules
      • sparseml.pytorch.datasets.detection.coco module
      • sparseml.pytorch.datasets.detection.helpers module
      • sparseml.pytorch.datasets.detection.voc module
      • Module contents
    • sparseml.pytorch.datasets.recommendation package
      • Module contents
    • sparseml.pytorch.datasets.video package
      • Module contents
  • Submodules
  • sparseml.pytorch.datasets.generic module
  • sparseml.pytorch.datasets.registry module
  • Module contents
• 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
      • Submodules
      • sparseml.pytorch.models.detection.ssd module
      • sparseml.pytorch.models.detection.ssd_lite module
      • sparseml.pytorch.models.detection.ssd_mobilenet module
      • sparseml.pytorch.models.detection.ssd_resnet module
      • sparseml.pytorch.models.detection.yolo_v3 module
      • Module contents
    • sparseml.pytorch.models.external package
      • Submodules
      • sparseml.pytorch.models.external.torchvision module
      • Module contents
    • sparseml.pytorch.models.recommendation package
      • Module contents
  • Submodules
  • sparseml.pytorch.models.registry module
  • Module contents
• sparseml.pytorch.nn package
  • Submodules
  • sparseml.pytorch.nn.activations module
  • sparseml.pytorch.nn.fatrelu module
  • sparseml.pytorch.nn.se module
  • Module contents
• 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
    • sparseml.pytorch.utils.quantization package
      • Submodules
      • sparseml.pytorch.utils.quantization.helpers module
      • sparseml.pytorch.utils.quantization.quantize_qat_export module
      • Module contents
  • 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