File size: 763 Bytes
bbde80b |
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 |
# General
finetune: False
partial_train: null
train_pw3d: True
warmup_h36m: 20
warmup_coco: 100
# Traning
epochs: 60
checkpoint_frequency: 20
batch_size: 128
batch_size_img: 512
dropout: 0.1
lr_backbone: 0.0001
lr_head: 0.0001
weight_decay: 0.01
lr_decay: 0.98
# Model
maxlen: 243
dim_feat: 512
mlp_ratio: 2
depth: 5
dim_rep: 512
num_heads: 8
att_fuse: True
hidden_dim: 1024
# Data
data_root: data/mesh
dt_file_h36m: mesh_det_h36m.pkl
dt_file_coco: mesh_det_coco.pkl
dt_file_pw3d: mesh_det_pw3d.pkl
clip_len: 16
data_stride: 8
sample_stride: 1
num_joints: 17
# Loss
lambda_3d: 0.5
lambda_scale: 0
lambda_3dv: 10
lambda_lv: 0
lambda_lg: 0
lambda_a: 0
lambda_av: 0
lambda_pose: 1000
lambda_shape: 1
lambda_norm: 20
loss_type: 'L1'
# Augmentation
flip: True
|