File size: 473 Bytes
85967a5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | """
DKM: Differentiable K-Means Clustering Layer for Neural Network Compression
Implementation of the ICLR 2022 paper by Cho et al.
(arXiv: 2108.12659)
"""
from dkm.dkm_layer import DKMLayer
from dkm.compressor import DKMCompressor, compress_model
from dkm.utils import compute_model_size, compute_compression_ratio
__version__ = "1.0.0"
__all__ = [
"DKMLayer",
"DKMCompressor",
"compress_model",
"compute_model_size",
"compute_compression_ratio",
]
|