INNER CODE UNIT · Python

__init__

lucidrains/vector-quantize-pytorch · vector_quantize_pytorch/evo_lookup_free_quantization.py:30

    def __init__(
        self,
        encoder: Module,
        decoder: Module,
        lfq: LFQ | None = None,
        *,
        dim: int | None = None,
        codebook_size: int | None = None,
        num_codebooks: int = 1,
        pop_size: int = 64,
        mutation_rate: float = 0.02,
        tournament_size: int = 2,
        elitism_count: int = 1,
        generations: int = 50,
        **lfq_kwargs
    ):
        super().__init__()
        assert pop_size > elitism_count, 'pop_size must be greater than elitism_count'

View source record →