Skip to content

[BUG] ResidualVQ always initializes MLPs #160

@hummat

Description

@hummat

MLPs in ResidualVQ are always initialized, even if implicit_neural_codebook=False.

self.mlps = ModuleList([MLP(dim = codebook_dim, l2norm_output = first(self.layers).use_cosine_sim, **mlp_kwargs) for _ in range(num_quantizers - 1)])

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions