File size: 1,088 Bytes
3df66ec |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
exp_name: bert_neg
model_save_dir: ./output/${exp_name}/
save_dir: ./output/${exp_name}/contrastive_results/
mean_path: ./stats/humanml3d/guoh3dfeats/mean.pt
std_path: ./stats/humanml3d/guoh3dfeats/std.pt
model:
text_model_name: distilbert-base-uncased
temperature: 0.1
threshold_selfsim: 0.8
threshold_selfsim_metrics: 0.95
token_num: 128
noneg: false
lmd:
recons: 1.0
latent: 1.0e-05
kl: 1.0e-05
contrast: 0.1
recon: true
motion_encoder:
nfeats: 263
vae: true
latent_dim: 256
ff_size: 1024
num_layers: 6
num_heads: 4
dropout: 0.1
activation: gelu
text_encoder:
nfeats: 768
vae: true
latent_dim: 256
ff_size: 1024
num_layers: 6
num_heads: 4
dropout: 0.1
activation: gelu
motion_decoder:
nfeats: 263
latent_dim: 256
ff_size: 1024
num_layers: 6
num_heads: 4
dropout: 0.1
activation: gelu
dataset:
data_dir: ./data/humanml3d/
max_motion_length: 224
padding: true
ev2ev: true
dataloader:
batch_size: 32
num_workers: 8
train:
seed: 1234
epochs: 200
lr: 0.0001
langlr: 1.0e-05
train_text_encoder: false
|