motion_vae:
  target: mld.models.architectures.mld_vae.MldVae
  params:
    nfeats: ${DATASET.NFEATS}
    latent_dim: ${model.latent_dim}
    hidden_dim: 256
    force_pre_post_proj: true
    ff_size: 1024
    num_layers: 9
    num_heads: 4
    dropout: 0.1
    arch: 'encoder_decoder'
    normalize_before: false
    norm_eps: 1e-5
    activation: 'gelu'
    norm_post: true
    activation_post: null
    position_embedding: 'learned'