torch_mist.quantization.factories

Module Contents

Functions

kmeans_quantization(...)
vqvae(→ torch_mist.models.vqvae.VQVAE)
instantiate_quantization(...)

torch_mist.quantization.factories.kmeans_quantization(n_bins: int, n_init='auto', **kwargs) → torch_mist.quantization.functions.ClusterQuantization

torch_mist.quantization.factories.vqvae(input_dim: int, quantization_dim: int, n_bins: int, hidden_dims: List[int], beta: float = 0.2, nonlinearity: Callable | None = None, version: str = 'v2', **train_params) → torch_mist.models.vqvae.VQVAE

torch_mist.quantization.factories.instantiate_quantization(name: str, n_bins: int, **kwargs) → torch_mist.quantization.functions.QuantizationFunction