Hugging Face's logo

Spaces:
jnnan
/
trumans
Paused

App Files Community
2
trumans
File size: 1,674 Bytes 
aeba71c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
 

exp_name: Test
train: false
batch_size: 1
device: cuda
interp_s: 3

len_pre: 4
len_act: 2
action_type: 'none'
scene_name: 'background'
action_id: 1
stay_and_act: false

method_name: Test
continue_last: false

#exp_dir: ${oc.env:ROOT_DIR}/Experiments/${exp_name}
ckpt_dir: ./Checkpoints
smpl_dir: ./smpl_models
#test_dir: ${oc.env:ROOT_DIR}/Test_settings

num_gpus: 1
num_workers: 0

dataset:
  folder: ./Data_blocks_motion_all
  device: cuda
  batch_size: 1
  seq_len: 16
  step: 3
  nb_voxels: 32
  mesh_grid: [ -0.6, 0.6, 0, 1.2, -0.6, 0.6 ]
  train: false
  load_scene: true
  load_action: true

  joints_ind: [ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 25, 40 ]
  nb_joints: 24
  nb_actions: 10

guidance:
  pelvis:
    seq_len: 16
    step: 3
    mask_ind: 0
    fixed_frame: 2
    mask_y: false
    emb_f: -1
    no_scene: false
    no_action: false
    fix_mode: true

model:
  model_smplx:
    input_dim: 72
    output_dim: 132
    hidden_dim: 64
    ckpt: ./Checkpoints/model_joints_to_smpl_wrist.pth
  synhsi_body:
    dim_model: 512
    num_heads: 16
    num_layers: 8
    dropout_p: 0.1
    nb_voxels: 32
    free_p: 0
    ac_type: last_add_first_token
    dim_input: 72
    dim_output: 72
    nb_actions: 10
    no_scene: false
    no_action: false
    ckpt: ./Checkpoints/TRUMANS_mask_ind[0]_timesteps[100]_fixed_frame[2]_ac_type[last_add_first_token]_no_scene[False]_no_action[False]_batch_size[256]_epoch360.pth

sampler:
  pelvis:
    _target_: models.synhsi.Sampler
    device: cuda
    mask_ind: 0
    emb_f: -1
    batch_size: 1
    seq_len: 16
    channel: 72
    fix_mode: true
    timesteps: 100
    fixed_frame: 2