Learn more  » Push, build, and install  RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Bower components Debian packages RPM packages NuGet packages

edgify / torch   python

Repository URL to install this package:

Version: 2.0.1+cpu 

/ ao / pruning / __init__.py

# Variables
from ._mappings import get_dynamic_sparse_quantized_mapping
from ._mappings import get_static_sparse_quantized_mapping

# Sparsifier
from .sparsifier.base_sparsifier import BaseSparsifier
from .sparsifier.weight_norm_sparsifier import WeightNormSparsifier
from .sparsifier.nearly_diagonal_sparsifier import NearlyDiagonalSparsifier

# Scheduler
from .scheduler.base_scheduler import BaseScheduler
from .scheduler.lambda_scheduler import LambdaSL
from .scheduler.cubic_scheduler import CubicSL

# Parametrizations
from .sparsifier.utils import FakeSparsity
from .sparsifier.utils import module_to_fqn
from .sparsifier.utils import fqn_to_module
from .sparsifier.utils import get_arg_info_from_tensor_fqn