sparseml package¶
Subpackages¶
- sparseml.keras package
- Subpackages
- sparseml.keras.datasets package
- sparseml.keras.models package
- sparseml.keras.optim package
- Submodules
- sparseml.keras.optim.manager module
- sparseml.keras.optim.mask_pruning module
- sparseml.keras.optim.mask_pruning_creator module
- sparseml.keras.optim.modifier module
- sparseml.keras.optim.modifier_epoch module
- sparseml.keras.optim.modifier_lr module
- sparseml.keras.optim.modifier_params module
- sparseml.keras.optim.modifier_pruning module
- sparseml.keras.optim.utils module
- Module contents
- sparseml.keras.utils package
- Module contents
- Subpackages
- sparseml.onnx package
- sparseml.optim package
- sparseml.pytorch package
- Subpackages
- sparseml.pytorch.datasets package
- sparseml.pytorch.models package
- 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
- Subpackages
- sparseml.tensorflow_v1 package
- Subpackages
- sparseml.tensorflow_v1.datasets package
- sparseml.tensorflow_v1.models package
- sparseml.tensorflow_v1.nn package
- sparseml.tensorflow_v1.optim package
- Submodules
- sparseml.tensorflow_v1.optim.analyzer_module module
- sparseml.tensorflow_v1.optim.manager module
- sparseml.tensorflow_v1.optim.mask_creator_pruning module
- sparseml.tensorflow_v1.optim.mask_pruning module
- sparseml.tensorflow_v1.optim.modifier module
- sparseml.tensorflow_v1.optim.modifier_epoch module
- sparseml.tensorflow_v1.optim.modifier_lr module
- sparseml.tensorflow_v1.optim.modifier_params module
- sparseml.tensorflow_v1.optim.modifier_pruning module
- sparseml.tensorflow_v1.optim.schedule_lr module
- sparseml.tensorflow_v1.optim.sensitivity_pruning module
- Module contents
- sparseml.tensorflow_v1.utils package
- Module contents
- Subpackages
- sparseml.utils package
Submodules¶
sparseml.log module¶
Root logging file to handle standard logging setups for the package
-
sparseml.log.
get_main_logger
() → logging.Logger[source]¶ - Returns
a main logger that can be used in external scripts for logging in a standard format that is consistent with other loggers in sparseml
Module contents¶
Tooling to help train, test, and optimize models for better performance