Hugging Face's logo

tmeynier
/
hepha-diffusion-100-simple-drawer-5-transformer

Model card Files
xet
 Community
hepha-diffusion-100-simple-drawer-5-transformer
File size: 4,502 Bytes
fe31532
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
{
  "args": {
    "dataset_dir": "datasets/hepha_act_100_simple_drawer_5",
    "output_dir": "runs/hepha_diffusion_100_simple_drawer_5_transformer",
    "epochs": 50,
    "batch_size": 16,
    "num_workers": 4,
    "lr": 0.0001,
    "weight_decay": 1e-06,
    "chunk_size": 50,
    "image_height": 240,
    "image_width": 320,
    "obs_feature_dim": 512,
    "n_obs_steps": 4,
    "n_action_steps": 16,
    "diffusion_step_embed_dim": 256,
    "denoiser": "transformer",
    "down_dims": [
      256,
      512,
      1024
    ],
    "kernel_size": 5,
    "n_groups": 8,
    "cond_predict_scale": true,
    "transformer_layers": 8,
    "transformer_heads": 8,
    "transformer_dim_feedforward": 2048,
    "transformer_cond_layers": 0,
    "transformer_causal_attn": true,
    "diffusion_steps": 100,
    "num_inference_steps": 100,
    "beta_schedule": "squaredcos_cap_v2",
    "prediction_type": "epsilon",
    "backbone": "resnet18",
    "lr_backbone": 1e-05,
    "no_pretrained_backbone": false,
    "crop_height": null,
    "crop_width": null,
    "dropout": 0.1,
    "lr_warmup_steps": 500,
    "val_ratio": 0.1,
    "val_every_steps": 2000,
    "seed": 42,
    "log_every": 50,
    "save_every": 0,
    "save_every_steps": 2000,
    "resume": null,
    "amp": true,
    "device": "cuda",
    "wandb_project": "hepha-diffusion",
    "wandb_entity": null,
    "wandb_name": "hepha-diffusion-100-simple-drawer-5-transformer",
    "wandb_tags": null,
    "wandb_mode": null,
    "wandb_log_checkpoints": false,
    "hf_model_repo_id": "tmeynier/hepha-diffusion-100-simple-drawer-5-transformer",
    "hf_revision": null,
    "hf_private": true,
    "hf_upload_every": null,
    "hf_upload_every_steps": 2000,
    "hf_upload_fail_training": false
  },
  "model_config": {
    "state_dim": 16,
    "action_dim": 15,
    "chunk_size": 50,
    "horizon": 50,
    "n_obs_steps": 4,
    "n_action_steps": 16,
    "obs_feature_dim": 512,
    "diffusion_step_embed_dim": 256,
    "denoiser": "transformer",
    "down_dims": [
      256,
      512,
      1024
    ],
    "kernel_size": 5,
    "n_groups": 8,
    "cond_predict_scale": true,
    "transformer_layers": 8,
    "transformer_heads": 8,
    "transformer_dim_feedforward": 2048,
    "transformer_cond_layers": 0,
    "transformer_causal_attn": true,
    "diffusion_steps": 100,
    "num_inference_steps": 100,
    "beta_start": 0.0001,
    "beta_end": 0.02,
    "beta_schedule": "squaredcos_cap_v2",
    "prediction_type": "epsilon",
    "clip_sample": true,
    "dropout": 0.1,
    "backbone": "resnet18",
    "pretrained_backbone": true,
    "crop_height": 216,
    "crop_width": 288
  },
  "dataset_info": {
    "dataset_dir": "datasets/hepha_act_100_simple_drawer_5",
    "total_frames": 200000,
    "episodes": 100,
    "state_dim": 16,
    "action_dim": 15,
    "normalizer": {
      "mode": "limits",
      "state_offset": [
        0.5153066543862224,
        0.5881318300962448,
        0.7083333283662796,
        0.3882358446717262,
        0.30765194282867014,
        0.5178910288959742,
        0.4271558867767453,
        0.5,
        0.6233630478382111,
        0.20346606895327568,
        0.55288165807724,
        0.4580320715904236,
        0.5001527502172394,
        0.5,
        0.5,
        0.5
      ],
      "state_scale": [
        0.48468315321952105,
        0.3150710016489029,
        0.2916666716337204,
        0.31237538903951645,
        0.3067184032406658,
        0.4759374465793371,
        0.4015634311363101,
        0.5,
        0.2835608422756195,
        0.11999200657010078,
        0.44711834192276,
        0.2851250171661377,
        0.4998472497827606,
        0.5,
        0.5,
        1.0
      ],
      "action_offset": [
        0.5152931213378906,
        0.5875480771064758,
        0.7083333283662796,
        0.39343979954719543,
        0.3081638216972351,
        0.5205638874322176,
        0.455325398594141,
        0.5,
        0.6158838868141174,
        0.20901485532522202,
        0.5,
        0.46368787437677383,
        0.5001716946717352,
        0.5,
        0.5
      ],
      "action_scale": [
        0.4847068786621094,
        0.31383126974105835,
        0.2916666716337204,
        0.3037284314632416,
        0.3081638216972351,
        0.4794361125677824,
        0.36030807718634605,
        0.5,
        0.27900099754333496,
        0.1144387498497963,
        0.5,
        0.27640219777822495,
        0.49982830532826483,
        0.5,
        0.5
      ]
    }
  }
}