mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
03979a599e
Part of #123062 - #123062 Pull Request resolved: https://github.com/pytorch/pytorch/pull/128862 Approved by: https://github.com/ezyang ghstack dependencies: #128861
24 lines
640 B
Python
24 lines
640 B
Python
# 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
|