INNER CODE UNIT · Python

EvoLFQ

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

class EvoLFQ(Module):
    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__()

View source record →