sparseml.pytorch.models.external package

Submodules

sparseml.pytorch.models.external.torchvision module

Module to register torchvision imagenet classification models with the sparseml.pytorch model registry.

Models will have the key torchvision.model_fn_name. ex: torchvision.mobilenet_v2

torchvision must be installed for models to register

Module contents

Integrations with external model repos to register with the sparseml.pytorch model registry.