File size: 721 Bytes
561907f |
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 |
exp_name: pretrained
checkpoints_dir: ./checkpoints/${exp_name}/${dataset.dataset_name}
preprocess:
padding: true
use_kinematic: true
model:
motion_encoder: vit_base_patch16_224_in21k
text_encoder: distilbert-base-uncased
train:
batch_size: 256
epoch: 50
optimizer:
motion_lr: 1.0e-05
text_lr: 1.0e-05
head_lr: 1.0e-05
train_motion_encoder: true
train_text_encoder: true
motion_encoder_pretrained: true
patch_size: 16
seed: 42
eval:
eval_train: false
use_best_model: true
dataset:
dataset_name: KIT-ML
data_root: ./data/KIT-ML
motion_dir: ${dataset.data_root}/new_joints
text_dir: ${dataset.data_root}/texts
joints_num: 21
max_motion_length: 224
times: 1
fps: 12.5
|