Skip to content

dimension for HuBERT base #11

@TonyWangX

Description

@TonyWangX

HuBERT base output has 768 dimensions.

  1. AASIST LL needs to use self.input_dim to specify the input feature dimension?

    self.LL = nn.Linear(1024, 128)

  2. yaml for HuBERT models may be updated. This works for me

AASIST

backend:
  type: AASIST
  args:
    input_dim: 768

MLP

input_dim: 768

TCM

args: {input_dim: 768, emb_size: 128, heads: 4, num_encoders: 1}

Nes2Net

 backend:
   type: Nes2Net
   args:
     input_dim: 768


- {type: CrossEntropy, embedding_dim: 768, n_classes: 2, weight: 1.0}

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