| traj_encoder: | |
| target: mld.models.architectures.mld_traj_encoder.MldTrajEncoder | |
| params: | |
| nfeats: ${DATASET.NJOINTS} | |
| latent_dim: ${model.latent_dim} | |
| hidden_dim: 256 | |
| force_post_proj: true | |
| ff_size: 1024 | |
| num_layers: 9 | |
| num_heads: 4 | |
| dropout: 0.1 | |
| normalize_before: false | |
| norm_eps: 1e-5 | |
| activation: 'gelu' | |
| norm_post: true | |
| activation_post: null | |
| position_embedding: 'learned' | |