Hugging Face's logo

Spaces:
HorizonRobotics
/
DIPO
Running on Zero

App Files Community
DIPO
File size: 2,117 Bytes 
c28dddb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
 
name: sys_origin
exp_dir: ./exps/dipo/denoiser
data_root: /horizon-bucket/robot_lab/users/ruiqi.wu/robot/dataset/blender
n_time_samples: 16
loss_fg_weight: 0.01
img_drop_prob: 0.1
guidance_scaler: 0.5
graph_drop_prob: 0.5
model:
  name: denoiser
  in_ch: 6
  attn_dim: 128
  n_head: 4
  n_layers: 6
  dropout: 0.1
  K: 32
  mode_num: 5
  img_emb_dims:
  - 768
  - 128
  cat_drop_prob: 0.5
scheduler:
  name: ddpm
  config:
    num_train_timesteps: 1000
    beta_schedule: linear
    prediction_type: epsilon
lr_scheduler_adapter:
  name: LinearWarmupCosineAnnealingLR
  warmup_epochs: 3
  max_epochs: 200
  warmup_start_lr: 1.0e-06
  eta_min: 1.0e-05
optimizer_adapter:
  name: AdamW
  args:
    lr: 0.0005
    betas:
    - 0.9
    - 0.99
    eps: 1.0e-15
lr_scheduler_cage:
  name: LinearWarmupCosineAnnealingLR
  warmup_epochs: 3
  max_epochs: 200
  warmup_start_lr: 1.0e-06
  eta_min: 1.0e-05
optimizer_cage:
  name: AdamW
  args:
    lr: 5.0e-05
    betas:
    - 0.9
    - 0.99
    eps: 1.0e-15
hparams:
  name: sys_origin
  exp_dir: ./exps/dipo/denoiser
  data_root: /horizon-bucket/robot_lab/users/ruiqi.wu/robot/dataset/blender
  n_time_samples: 16
  loss_fg_weight: 0.01
  img_drop_prob: 0.1
  guidance_scaler: 0.5
  graph_drop_prob: 0.5
  model:
    name: denoiser
    in_ch: 6
    attn_dim: 128
    n_head: 4
    n_layers: 6
    dropout: 0.1
    K: 32
    mode_num: 5
    img_emb_dims:
    - 768
    - 128
    cat_drop_prob: 0.5
  scheduler:
    name: ddpm
    config:
      num_train_timesteps: 1000
      beta_schedule: linear
      prediction_type: epsilon
  lr_scheduler_adapter:
    name: LinearWarmupCosineAnnealingLR
    warmup_epochs: 3
    max_epochs: 200
    warmup_start_lr: 1.0e-06
    eta_min: 1.0e-05
  optimizer_adapter:
    name: AdamW
    args:
      lr: 0.0005
      betas:
      - 0.9
      - 0.99
      eps: 1.0e-15
  lr_scheduler_cage:
    name: LinearWarmupCosineAnnealingLR
    warmup_epochs: 3
    max_epochs: 200
    warmup_start_lr: 1.0e-06
    eta_min: 1.0e-05
  optimizer_cage:
    name: AdamW
    args:
      lr: 5.0e-05
      betas:
      - 0.9
      - 0.99
      eps: 1.0e-15