motius-intermask-interhuman / transformer_opt.txt
ZeyuLing's picture
Add files using upload-large-folder tool
78fa5ce verified
Raw
History Blame Contribute Delete
535 Bytes
------------ Options -------------
batch_size: 52
cache: True
checkpoints_dir: ./checkpoints
cond_drop_prob: 0.1
dataset_name: interhuman
do_eval: True
dropout: 0.2
eval_every_e: 10
ff_size: 1024
force_mask: False
gamma: 0.3333333333333333
gpu_id: 0
gumbel_sample: False
interaction_mask_prob: 0.2
is_continue: False
is_train: True
latent_dim: 384
lr: 0.0002
max_epoch: 500
motion_rep: global
n_heads: 6
n_layers: 6
name: trans_default
seed: 0
step_unroll: 1
test_batch_size: 96
vq_name: vq_default
-------------- End ----------------