File size: 436 Bytes
eb339cb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
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'
|