File size: 663 Bytes
ad5f26a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
# Variables
from ._mappings import (
get_dynamic_sparse_quantized_mapping,
get_static_sparse_quantized_mapping,
)
# Scheduler
from .scheduler.base_scheduler import BaseScheduler
from .scheduler.cubic_scheduler import CubicSL
from .scheduler.lambda_scheduler import LambdaSL
# Sparsifier
from .sparsifier.base_sparsifier import BaseSparsifier
from .sparsifier.nearly_diagonal_sparsifier import NearlyDiagonalSparsifier
# Parametrizations
from .sparsifier.utils import (
FakeSparsity,
fqn_to_module,
get_arg_info_from_tensor_fqn,
module_to_fqn,
)
from .sparsifier.weight_norm_sparsifier import WeightNormSparsifier
|