aleleanza commited on
Commit
a80edea
·
verified ·
1 Parent(s): 75f6eb7

Add files using upload-large-folder tool

Browse files
vac_post_vis/action_stats.json ADDED
@@ -0,0 +1,297 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "method": "min_max",
3
+ "output_min": -0.999999,
4
+ "output_max": 0.999999,
5
+ "action_mode": "post_velcmd_hand_binary",
6
+ "action_components": [
7
+ "vel_cmd_x",
8
+ "vel_cmd_y",
9
+ "vel_cmd_z",
10
+ "vel_cmd_rx",
11
+ "vel_cmd_ry",
12
+ "vel_cmd_rz",
13
+ "hand_open_binary"
14
+ ],
15
+ "num_frames": 124722,
16
+ "min": [
17
+ -44.80923843383789,
18
+ -78.85905456542969,
19
+ -272.0136413574219,
20
+ -41.36355972290039,
21
+ -4.46674108505249,
22
+ -50.74555969238281,
23
+ 0.0
24
+ ],
25
+ "max": [
26
+ 60.49591827392578,
27
+ 81.3885269165039,
28
+ 499.7237243652344,
29
+ 40.38335418701172,
30
+ 6.112980842590332,
31
+ 49.7142219543457,
32
+ 1.0
33
+ ],
34
+ "mean": [
35
+ 0.08373736590147018,
36
+ 6.001084804534912,
37
+ -2.738046646118164,
38
+ -0.005926239304244518,
39
+ -0.001013437518849969,
40
+ 0.04403281956911087,
41
+ 0.428376704454422
42
+ ],
43
+ "std": [
44
+ 7.180497646331787,
45
+ 26.540620803833008,
46
+ 46.47587966918945,
47
+ 0.791343629360199,
48
+ 0.21718677878379822,
49
+ 3.934755325317383,
50
+ 0.4948435127735138
51
+ ],
52
+ "scale": [
53
+ 52.65263366699219,
54
+ 80.1238784790039,
55
+ 385.86907958984375,
56
+ 40.87350082397461,
57
+ 5.289865970611572,
58
+ 50.22993850708008,
59
+ 0.500000536441803
60
+ ],
61
+ "offset": [
62
+ 7.843341827392578,
63
+ 1.264739990234375,
64
+ 113.85504150390625,
65
+ -0.4901008605957031,
66
+ 0.8231196403503418,
67
+ -0.5156707763671875,
68
+ 0.5
69
+ ],
70
+ "hand_source": "action.absolute",
71
+ "hand_active_dims": [
72
+ 8,
73
+ 9,
74
+ 10,
75
+ 11
76
+ ],
77
+ "hand_reduce": "mean",
78
+ "hand_threshold": 0.85,
79
+ "hand_high_value": 1.0,
80
+ "hand_low_value": 0.0,
81
+ "dataset": "/workspace/datasets/dataset_pipe_02_06",
82
+ "observation_mode_examples": [
83
+ "vision",
84
+ "vision_wrench",
85
+ "vision_wrench_state"
86
+ ],
87
+ "train_source_episodes": [
88
+ 0,
89
+ 1,
90
+ 2,
91
+ 3,
92
+ 4,
93
+ 5,
94
+ 6,
95
+ 7,
96
+ 8,
97
+ 9,
98
+ 10,
99
+ 11,
100
+ 12,
101
+ 13,
102
+ 14,
103
+ 15,
104
+ 16,
105
+ 17,
106
+ 18,
107
+ 19,
108
+ 20,
109
+ 21,
110
+ 22,
111
+ 23,
112
+ 24,
113
+ 25,
114
+ 26,
115
+ 27,
116
+ 28,
117
+ 29,
118
+ 30,
119
+ 31,
120
+ 32,
121
+ 33,
122
+ 34,
123
+ 35,
124
+ 36,
125
+ 37,
126
+ 38,
127
+ 39,
128
+ 40,
129
+ 41,
130
+ 42,
131
+ 43,
132
+ 44,
133
+ 45,
134
+ 46,
135
+ 47,
136
+ 48,
137
+ 49,
138
+ 50,
139
+ 51,
140
+ 52,
141
+ 53,
142
+ 54,
143
+ 55,
144
+ 56,
145
+ 57,
146
+ 58,
147
+ 59,
148
+ 60,
149
+ 61,
150
+ 62,
151
+ 63,
152
+ 64,
153
+ 65,
154
+ 66,
155
+ 67,
156
+ 68,
157
+ 69,
158
+ 70,
159
+ 71,
160
+ 72,
161
+ 73,
162
+ 74,
163
+ 75,
164
+ 76,
165
+ 77,
166
+ 78,
167
+ 79,
168
+ 80,
169
+ 81,
170
+ 82,
171
+ 83,
172
+ 84,
173
+ 85,
174
+ 86,
175
+ 87,
176
+ 88,
177
+ 89,
178
+ 90,
179
+ 91,
180
+ 92,
181
+ 93,
182
+ 94,
183
+ 95,
184
+ 96,
185
+ 97,
186
+ 98,
187
+ 99,
188
+ 100,
189
+ 101,
190
+ 102,
191
+ 103,
192
+ 104,
193
+ 105,
194
+ 106,
195
+ 107,
196
+ 108,
197
+ 109,
198
+ 110,
199
+ 111,
200
+ 112,
201
+ 113,
202
+ 114,
203
+ 115,
204
+ 116,
205
+ 117,
206
+ 118,
207
+ 119,
208
+ 120,
209
+ 121,
210
+ 122,
211
+ 123,
212
+ 124,
213
+ 125,
214
+ 126,
215
+ 127,
216
+ 128,
217
+ 129,
218
+ 130,
219
+ 131,
220
+ 132,
221
+ 133,
222
+ 134,
223
+ 135,
224
+ 136,
225
+ 137,
226
+ 138,
227
+ 139,
228
+ 140,
229
+ 141,
230
+ 142,
231
+ 143,
232
+ 144,
233
+ 145,
234
+ 146,
235
+ 147,
236
+ 148,
237
+ 149,
238
+ 150,
239
+ 151,
240
+ 152,
241
+ 153,
242
+ 154,
243
+ 155,
244
+ 156,
245
+ 157,
246
+ 158,
247
+ 159,
248
+ 160,
249
+ 161,
250
+ 162,
251
+ 163,
252
+ 164,
253
+ 165,
254
+ 166,
255
+ 167,
256
+ 168,
257
+ 169,
258
+ 170,
259
+ 171
260
+ ],
261
+ "valid_source_episodes": [
262
+ 172,
263
+ 173,
264
+ 174,
265
+ 175,
266
+ 176,
267
+ 177,
268
+ 178,
269
+ 179,
270
+ 180,
271
+ 181,
272
+ 182,
273
+ 183,
274
+ 184,
275
+ 185,
276
+ 186,
277
+ 187,
278
+ 188,
279
+ 189,
280
+ 190,
281
+ 191,
282
+ 192,
283
+ 193,
284
+ 194,
285
+ 195,
286
+ 196,
287
+ 197,
288
+ 198,
289
+ 199,
290
+ 200,
291
+ 201
292
+ ],
293
+ "image_size": [
294
+ 84,
295
+ 84
296
+ ]
297
+ }
vac_post_vis/best.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9cb2858908bff9d02175253f1a3e8e928ce40fdc006021ff4032fe81823d98dc
3
+ size 1252482813
vac_post_vis/config.json ADDED
@@ -0,0 +1,195 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "algo_name": "diffusion_policy",
3
+ "experiment": {
4
+ "name": "vac_post_vis",
5
+ "validate": true,
6
+ "logging": {
7
+ "terminal_output_to_txt": false,
8
+ "log_tb": true,
9
+ "log_wandb": false,
10
+ "wandb_proj_name": "debug"
11
+ },
12
+ "save": {
13
+ "enabled": true,
14
+ "every_n_seconds": null,
15
+ "every_n_epochs": 50,
16
+ "epochs": [],
17
+ "on_best_validation": true,
18
+ "on_best_rollout_return": false,
19
+ "on_best_rollout_success_rate": false
20
+ },
21
+ "epoch_every_n_steps": 100,
22
+ "validation_epoch_every_n_steps": 10,
23
+ "env": null,
24
+ "additional_envs": null,
25
+ "render": false,
26
+ "render_video": false,
27
+ "keep_all_videos": false,
28
+ "video_skip": 5,
29
+ "rollout": {
30
+ "enabled": false,
31
+ "n": 0,
32
+ "horizon": 50,
33
+ "rate": 50,
34
+ "warmstart": 0,
35
+ "terminate_on_success": true
36
+ },
37
+ "env_meta_update_dict": {},
38
+ "ckpt_path": null
39
+ },
40
+ "train": {
41
+ "data": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_post_vis/vac_post_vis.hdf5",
42
+ "output_dir": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_post_vis/robomimic_runs",
43
+ "normalize_weights_by_ds_size": false,
44
+ "num_data_workers": 0,
45
+ "hdf5_cache_mode": "all",
46
+ "hdf5_use_swmr": true,
47
+ "hdf5_load_next_obs": false,
48
+ "hdf5_normalize_obs": false,
49
+ "hdf5_filter_key": "train",
50
+ "hdf5_validation_filter_key": "valid",
51
+ "seq_length": 8,
52
+ "pad_seq_length": true,
53
+ "frame_stack": 2,
54
+ "pad_frame_stack": true,
55
+ "dataset_keys": [
56
+ "actions",
57
+ "rewards",
58
+ "dones"
59
+ ],
60
+ "action_keys": [
61
+ "actions"
62
+ ],
63
+ "action_config": {
64
+ "actions": {
65
+ "normalization": "min_max"
66
+ }
67
+ },
68
+ "goal_mode": null,
69
+ "cuda": true,
70
+ "batch_size": 16,
71
+ "num_epochs": 300,
72
+ "seed": 42,
73
+ "max_grad_norm": null
74
+ },
75
+ "algo": {
76
+ "optim_params": {
77
+ "policy": {
78
+ "optimizer_type": "adamw",
79
+ "learning_rate": {
80
+ "initial": 0.0001,
81
+ "decay_factor": 0.1,
82
+ "step_every_batch": true,
83
+ "scheduler_type": "cosine",
84
+ "num_cycles": 0.5,
85
+ "warmup_steps": 500,
86
+ "epoch_schedule": []
87
+ },
88
+ "regularization": {
89
+ "L2": 1e-06
90
+ }
91
+ }
92
+ },
93
+ "horizon": {
94
+ "observation_horizon": 2,
95
+ "action_horizon": 4,
96
+ "prediction_horizon": 8
97
+ },
98
+ "unet": {
99
+ "enabled": true,
100
+ "diffusion_step_embed_dim": 256,
101
+ "down_dims": [
102
+ 128,
103
+ 256,
104
+ 512
105
+ ],
106
+ "kernel_size": 5,
107
+ "n_groups": 8
108
+ },
109
+ "ema": {
110
+ "enabled": true,
111
+ "power": 0.75
112
+ },
113
+ "ddpm": {
114
+ "enabled": true,
115
+ "num_train_timesteps": 50,
116
+ "num_inference_timesteps": 50,
117
+ "beta_schedule": "squaredcos_cap_v2",
118
+ "clip_sample": true,
119
+ "prediction_type": "epsilon"
120
+ },
121
+ "ddim": {
122
+ "enabled": false,
123
+ "num_train_timesteps": 50,
124
+ "num_inference_timesteps": 10,
125
+ "beta_schedule": "squaredcos_cap_v2",
126
+ "clip_sample": true,
127
+ "set_alpha_to_one": true,
128
+ "steps_offset": 0,
129
+ "prediction_type": "epsilon"
130
+ }
131
+ },
132
+ "observation": {
133
+ "modalities": {
134
+ "obs": {
135
+ "low_dim": [],
136
+ "rgb": [
137
+ "front_rgb"
138
+ ],
139
+ "depth": [],
140
+ "scan": []
141
+ },
142
+ "goal": {
143
+ "low_dim": [],
144
+ "rgb": [],
145
+ "depth": [],
146
+ "scan": []
147
+ }
148
+ },
149
+ "encoder": {
150
+ "low_dim": {
151
+ "core_class": null,
152
+ "core_kwargs": {},
153
+ "obs_randomizer_class": null,
154
+ "obs_randomizer_kwargs": {}
155
+ },
156
+ "rgb": {
157
+ "core_class": "VisualCore",
158
+ "core_kwargs": {
159
+ "backbone_class": "ResNet18Conv",
160
+ "backbone_kwargs": {
161
+ "pretrained": false,
162
+ "input_coord_conv": false
163
+ },
164
+ "pool_class": "SpatialSoftmax",
165
+ "pool_kwargs": {
166
+ "num_kp": 32,
167
+ "learnable_temperature": false,
168
+ "temperature": 1.0,
169
+ "noise_std": 0.0
170
+ },
171
+ "feature_dimension": 64
172
+ },
173
+ "obs_randomizer_class": null,
174
+ "obs_randomizer_kwargs": {}
175
+ },
176
+ "depth": {
177
+ "core_class": "VisualCore",
178
+ "core_kwargs": {},
179
+ "obs_randomizer_class": null,
180
+ "obs_randomizer_kwargs": {}
181
+ },
182
+ "scan": {
183
+ "core_class": "ScanCore",
184
+ "core_kwargs": {},
185
+ "obs_randomizer_class": null,
186
+ "obs_randomizer_kwargs": {}
187
+ }
188
+ }
189
+ },
190
+ "meta": {
191
+ "hp_base_config_file": null,
192
+ "hp_keys": [],
193
+ "hp_values": []
194
+ }
195
+ }
vac_post_vis/dataset_summary.json ADDED
@@ -0,0 +1,2032 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "path": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_post_vis/vac_post_vis.hdf5",
3
+ "total_frames": 145712,
4
+ "train_episodes": 172,
5
+ "valid_episodes": 30,
6
+ "train_source_episodes": [
7
+ 0,
8
+ 1,
9
+ 2,
10
+ 3,
11
+ 4,
12
+ 5,
13
+ 6,
14
+ 7,
15
+ 8,
16
+ 9,
17
+ 10,
18
+ 11,
19
+ 12,
20
+ 13,
21
+ 14,
22
+ 15,
23
+ 16,
24
+ 17,
25
+ 18,
26
+ 19,
27
+ 20,
28
+ 21,
29
+ 22,
30
+ 23,
31
+ 24,
32
+ 25,
33
+ 26,
34
+ 27,
35
+ 28,
36
+ 29,
37
+ 30,
38
+ 31,
39
+ 32,
40
+ 33,
41
+ 34,
42
+ 35,
43
+ 36,
44
+ 37,
45
+ 38,
46
+ 39,
47
+ 40,
48
+ 41,
49
+ 42,
50
+ 43,
51
+ 44,
52
+ 45,
53
+ 46,
54
+ 47,
55
+ 48,
56
+ 49,
57
+ 50,
58
+ 51,
59
+ 52,
60
+ 53,
61
+ 54,
62
+ 55,
63
+ 56,
64
+ 57,
65
+ 58,
66
+ 59,
67
+ 60,
68
+ 61,
69
+ 62,
70
+ 63,
71
+ 64,
72
+ 65,
73
+ 66,
74
+ 67,
75
+ 68,
76
+ 69,
77
+ 70,
78
+ 71,
79
+ 72,
80
+ 73,
81
+ 74,
82
+ 75,
83
+ 76,
84
+ 77,
85
+ 78,
86
+ 79,
87
+ 80,
88
+ 81,
89
+ 82,
90
+ 83,
91
+ 84,
92
+ 85,
93
+ 86,
94
+ 87,
95
+ 88,
96
+ 89,
97
+ 90,
98
+ 91,
99
+ 92,
100
+ 93,
101
+ 94,
102
+ 95,
103
+ 96,
104
+ 97,
105
+ 98,
106
+ 99,
107
+ 100,
108
+ 101,
109
+ 102,
110
+ 103,
111
+ 104,
112
+ 105,
113
+ 106,
114
+ 107,
115
+ 108,
116
+ 109,
117
+ 110,
118
+ 111,
119
+ 112,
120
+ 113,
121
+ 114,
122
+ 115,
123
+ 116,
124
+ 117,
125
+ 118,
126
+ 119,
127
+ 120,
128
+ 121,
129
+ 122,
130
+ 123,
131
+ 124,
132
+ 125,
133
+ 126,
134
+ 127,
135
+ 128,
136
+ 129,
137
+ 130,
138
+ 131,
139
+ 132,
140
+ 133,
141
+ 134,
142
+ 135,
143
+ 136,
144
+ 137,
145
+ 138,
146
+ 139,
147
+ 140,
148
+ 141,
149
+ 142,
150
+ 143,
151
+ 144,
152
+ 145,
153
+ 146,
154
+ 147,
155
+ 148,
156
+ 149,
157
+ 150,
158
+ 151,
159
+ 152,
160
+ 153,
161
+ 154,
162
+ 155,
163
+ 156,
164
+ 157,
165
+ 158,
166
+ 159,
167
+ 160,
168
+ 161,
169
+ 162,
170
+ 163,
171
+ 164,
172
+ 165,
173
+ 166,
174
+ 167,
175
+ 168,
176
+ 169,
177
+ 170,
178
+ 171
179
+ ],
180
+ "valid_source_episodes": [
181
+ 172,
182
+ 173,
183
+ 174,
184
+ 175,
185
+ 176,
186
+ 177,
187
+ 178,
188
+ 179,
189
+ 180,
190
+ 181,
191
+ 182,
192
+ 183,
193
+ 184,
194
+ 185,
195
+ 186,
196
+ 187,
197
+ 188,
198
+ 189,
199
+ 190,
200
+ 191,
201
+ 192,
202
+ 193,
203
+ 194,
204
+ 195,
205
+ 196,
206
+ 197,
207
+ 198,
208
+ 199,
209
+ 200,
210
+ 201
211
+ ],
212
+ "episodes": [
213
+ {
214
+ "hdf5_key": "ep_00000",
215
+ "source_episode_index": 0,
216
+ "split": "train",
217
+ "num_samples": 663,
218
+ "action_dim": 7,
219
+ "obs_mode": "vision",
220
+ "action_mode": "post_velcmd_hand_binary"
221
+ },
222
+ {
223
+ "hdf5_key": "ep_00001",
224
+ "source_episode_index": 1,
225
+ "split": "train",
226
+ "num_samples": 706,
227
+ "action_dim": 7,
228
+ "obs_mode": "vision",
229
+ "action_mode": "post_velcmd_hand_binary"
230
+ },
231
+ {
232
+ "hdf5_key": "ep_00002",
233
+ "source_episode_index": 2,
234
+ "split": "train",
235
+ "num_samples": 1006,
236
+ "action_dim": 7,
237
+ "obs_mode": "vision",
238
+ "action_mode": "post_velcmd_hand_binary"
239
+ },
240
+ {
241
+ "hdf5_key": "ep_00003",
242
+ "source_episode_index": 3,
243
+ "split": "train",
244
+ "num_samples": 709,
245
+ "action_dim": 7,
246
+ "obs_mode": "vision",
247
+ "action_mode": "post_velcmd_hand_binary"
248
+ },
249
+ {
250
+ "hdf5_key": "ep_00004",
251
+ "source_episode_index": 4,
252
+ "split": "train",
253
+ "num_samples": 816,
254
+ "action_dim": 7,
255
+ "obs_mode": "vision",
256
+ "action_mode": "post_velcmd_hand_binary"
257
+ },
258
+ {
259
+ "hdf5_key": "ep_00005",
260
+ "source_episode_index": 5,
261
+ "split": "train",
262
+ "num_samples": 882,
263
+ "action_dim": 7,
264
+ "obs_mode": "vision",
265
+ "action_mode": "post_velcmd_hand_binary"
266
+ },
267
+ {
268
+ "hdf5_key": "ep_00006",
269
+ "source_episode_index": 6,
270
+ "split": "train",
271
+ "num_samples": 699,
272
+ "action_dim": 7,
273
+ "obs_mode": "vision",
274
+ "action_mode": "post_velcmd_hand_binary"
275
+ },
276
+ {
277
+ "hdf5_key": "ep_00007",
278
+ "source_episode_index": 7,
279
+ "split": "train",
280
+ "num_samples": 851,
281
+ "action_dim": 7,
282
+ "obs_mode": "vision",
283
+ "action_mode": "post_velcmd_hand_binary"
284
+ },
285
+ {
286
+ "hdf5_key": "ep_00008",
287
+ "source_episode_index": 8,
288
+ "split": "train",
289
+ "num_samples": 728,
290
+ "action_dim": 7,
291
+ "obs_mode": "vision",
292
+ "action_mode": "post_velcmd_hand_binary"
293
+ },
294
+ {
295
+ "hdf5_key": "ep_00009",
296
+ "source_episode_index": 9,
297
+ "split": "train",
298
+ "num_samples": 770,
299
+ "action_dim": 7,
300
+ "obs_mode": "vision",
301
+ "action_mode": "post_velcmd_hand_binary"
302
+ },
303
+ {
304
+ "hdf5_key": "ep_00010",
305
+ "source_episode_index": 10,
306
+ "split": "train",
307
+ "num_samples": 821,
308
+ "action_dim": 7,
309
+ "obs_mode": "vision",
310
+ "action_mode": "post_velcmd_hand_binary"
311
+ },
312
+ {
313
+ "hdf5_key": "ep_00011",
314
+ "source_episode_index": 11,
315
+ "split": "train",
316
+ "num_samples": 561,
317
+ "action_dim": 7,
318
+ "obs_mode": "vision",
319
+ "action_mode": "post_velcmd_hand_binary"
320
+ },
321
+ {
322
+ "hdf5_key": "ep_00012",
323
+ "source_episode_index": 12,
324
+ "split": "train",
325
+ "num_samples": 517,
326
+ "action_dim": 7,
327
+ "obs_mode": "vision",
328
+ "action_mode": "post_velcmd_hand_binary"
329
+ },
330
+ {
331
+ "hdf5_key": "ep_00013",
332
+ "source_episode_index": 13,
333
+ "split": "train",
334
+ "num_samples": 690,
335
+ "action_dim": 7,
336
+ "obs_mode": "vision",
337
+ "action_mode": "post_velcmd_hand_binary"
338
+ },
339
+ {
340
+ "hdf5_key": "ep_00014",
341
+ "source_episode_index": 14,
342
+ "split": "train",
343
+ "num_samples": 559,
344
+ "action_dim": 7,
345
+ "obs_mode": "vision",
346
+ "action_mode": "post_velcmd_hand_binary"
347
+ },
348
+ {
349
+ "hdf5_key": "ep_00015",
350
+ "source_episode_index": 15,
351
+ "split": "train",
352
+ "num_samples": 889,
353
+ "action_dim": 7,
354
+ "obs_mode": "vision",
355
+ "action_mode": "post_velcmd_hand_binary"
356
+ },
357
+ {
358
+ "hdf5_key": "ep_00016",
359
+ "source_episode_index": 16,
360
+ "split": "train",
361
+ "num_samples": 619,
362
+ "action_dim": 7,
363
+ "obs_mode": "vision",
364
+ "action_mode": "post_velcmd_hand_binary"
365
+ },
366
+ {
367
+ "hdf5_key": "ep_00017",
368
+ "source_episode_index": 17,
369
+ "split": "train",
370
+ "num_samples": 640,
371
+ "action_dim": 7,
372
+ "obs_mode": "vision",
373
+ "action_mode": "post_velcmd_hand_binary"
374
+ },
375
+ {
376
+ "hdf5_key": "ep_00018",
377
+ "source_episode_index": 18,
378
+ "split": "train",
379
+ "num_samples": 612,
380
+ "action_dim": 7,
381
+ "obs_mode": "vision",
382
+ "action_mode": "post_velcmd_hand_binary"
383
+ },
384
+ {
385
+ "hdf5_key": "ep_00019",
386
+ "source_episode_index": 19,
387
+ "split": "train",
388
+ "num_samples": 702,
389
+ "action_dim": 7,
390
+ "obs_mode": "vision",
391
+ "action_mode": "post_velcmd_hand_binary"
392
+ },
393
+ {
394
+ "hdf5_key": "ep_00020",
395
+ "source_episode_index": 20,
396
+ "split": "train",
397
+ "num_samples": 645,
398
+ "action_dim": 7,
399
+ "obs_mode": "vision",
400
+ "action_mode": "post_velcmd_hand_binary"
401
+ },
402
+ {
403
+ "hdf5_key": "ep_00021",
404
+ "source_episode_index": 21,
405
+ "split": "train",
406
+ "num_samples": 751,
407
+ "action_dim": 7,
408
+ "obs_mode": "vision",
409
+ "action_mode": "post_velcmd_hand_binary"
410
+ },
411
+ {
412
+ "hdf5_key": "ep_00022",
413
+ "source_episode_index": 22,
414
+ "split": "train",
415
+ "num_samples": 810,
416
+ "action_dim": 7,
417
+ "obs_mode": "vision",
418
+ "action_mode": "post_velcmd_hand_binary"
419
+ },
420
+ {
421
+ "hdf5_key": "ep_00023",
422
+ "source_episode_index": 23,
423
+ "split": "train",
424
+ "num_samples": 734,
425
+ "action_dim": 7,
426
+ "obs_mode": "vision",
427
+ "action_mode": "post_velcmd_hand_binary"
428
+ },
429
+ {
430
+ "hdf5_key": "ep_00024",
431
+ "source_episode_index": 24,
432
+ "split": "train",
433
+ "num_samples": 637,
434
+ "action_dim": 7,
435
+ "obs_mode": "vision",
436
+ "action_mode": "post_velcmd_hand_binary"
437
+ },
438
+ {
439
+ "hdf5_key": "ep_00025",
440
+ "source_episode_index": 25,
441
+ "split": "train",
442
+ "num_samples": 640,
443
+ "action_dim": 7,
444
+ "obs_mode": "vision",
445
+ "action_mode": "post_velcmd_hand_binary"
446
+ },
447
+ {
448
+ "hdf5_key": "ep_00026",
449
+ "source_episode_index": 26,
450
+ "split": "train",
451
+ "num_samples": 957,
452
+ "action_dim": 7,
453
+ "obs_mode": "vision",
454
+ "action_mode": "post_velcmd_hand_binary"
455
+ },
456
+ {
457
+ "hdf5_key": "ep_00027",
458
+ "source_episode_index": 27,
459
+ "split": "train",
460
+ "num_samples": 617,
461
+ "action_dim": 7,
462
+ "obs_mode": "vision",
463
+ "action_mode": "post_velcmd_hand_binary"
464
+ },
465
+ {
466
+ "hdf5_key": "ep_00028",
467
+ "source_episode_index": 28,
468
+ "split": "train",
469
+ "num_samples": 772,
470
+ "action_dim": 7,
471
+ "obs_mode": "vision",
472
+ "action_mode": "post_velcmd_hand_binary"
473
+ },
474
+ {
475
+ "hdf5_key": "ep_00029",
476
+ "source_episode_index": 29,
477
+ "split": "train",
478
+ "num_samples": 616,
479
+ "action_dim": 7,
480
+ "obs_mode": "vision",
481
+ "action_mode": "post_velcmd_hand_binary"
482
+ },
483
+ {
484
+ "hdf5_key": "ep_00030",
485
+ "source_episode_index": 30,
486
+ "split": "train",
487
+ "num_samples": 732,
488
+ "action_dim": 7,
489
+ "obs_mode": "vision",
490
+ "action_mode": "post_velcmd_hand_binary"
491
+ },
492
+ {
493
+ "hdf5_key": "ep_00031",
494
+ "source_episode_index": 31,
495
+ "split": "train",
496
+ "num_samples": 761,
497
+ "action_dim": 7,
498
+ "obs_mode": "vision",
499
+ "action_mode": "post_velcmd_hand_binary"
500
+ },
501
+ {
502
+ "hdf5_key": "ep_00032",
503
+ "source_episode_index": 32,
504
+ "split": "train",
505
+ "num_samples": 804,
506
+ "action_dim": 7,
507
+ "obs_mode": "vision",
508
+ "action_mode": "post_velcmd_hand_binary"
509
+ },
510
+ {
511
+ "hdf5_key": "ep_00033",
512
+ "source_episode_index": 33,
513
+ "split": "train",
514
+ "num_samples": 735,
515
+ "action_dim": 7,
516
+ "obs_mode": "vision",
517
+ "action_mode": "post_velcmd_hand_binary"
518
+ },
519
+ {
520
+ "hdf5_key": "ep_00034",
521
+ "source_episode_index": 34,
522
+ "split": "train",
523
+ "num_samples": 721,
524
+ "action_dim": 7,
525
+ "obs_mode": "vision",
526
+ "action_mode": "post_velcmd_hand_binary"
527
+ },
528
+ {
529
+ "hdf5_key": "ep_00035",
530
+ "source_episode_index": 35,
531
+ "split": "train",
532
+ "num_samples": 612,
533
+ "action_dim": 7,
534
+ "obs_mode": "vision",
535
+ "action_mode": "post_velcmd_hand_binary"
536
+ },
537
+ {
538
+ "hdf5_key": "ep_00036",
539
+ "source_episode_index": 36,
540
+ "split": "train",
541
+ "num_samples": 685,
542
+ "action_dim": 7,
543
+ "obs_mode": "vision",
544
+ "action_mode": "post_velcmd_hand_binary"
545
+ },
546
+ {
547
+ "hdf5_key": "ep_00037",
548
+ "source_episode_index": 37,
549
+ "split": "train",
550
+ "num_samples": 685,
551
+ "action_dim": 7,
552
+ "obs_mode": "vision",
553
+ "action_mode": "post_velcmd_hand_binary"
554
+ },
555
+ {
556
+ "hdf5_key": "ep_00038",
557
+ "source_episode_index": 38,
558
+ "split": "train",
559
+ "num_samples": 663,
560
+ "action_dim": 7,
561
+ "obs_mode": "vision",
562
+ "action_mode": "post_velcmd_hand_binary"
563
+ },
564
+ {
565
+ "hdf5_key": "ep_00039",
566
+ "source_episode_index": 39,
567
+ "split": "train",
568
+ "num_samples": 855,
569
+ "action_dim": 7,
570
+ "obs_mode": "vision",
571
+ "action_mode": "post_velcmd_hand_binary"
572
+ },
573
+ {
574
+ "hdf5_key": "ep_00040",
575
+ "source_episode_index": 40,
576
+ "split": "train",
577
+ "num_samples": 810,
578
+ "action_dim": 7,
579
+ "obs_mode": "vision",
580
+ "action_mode": "post_velcmd_hand_binary"
581
+ },
582
+ {
583
+ "hdf5_key": "ep_00041",
584
+ "source_episode_index": 41,
585
+ "split": "train",
586
+ "num_samples": 655,
587
+ "action_dim": 7,
588
+ "obs_mode": "vision",
589
+ "action_mode": "post_velcmd_hand_binary"
590
+ },
591
+ {
592
+ "hdf5_key": "ep_00042",
593
+ "source_episode_index": 42,
594
+ "split": "train",
595
+ "num_samples": 946,
596
+ "action_dim": 7,
597
+ "obs_mode": "vision",
598
+ "action_mode": "post_velcmd_hand_binary"
599
+ },
600
+ {
601
+ "hdf5_key": "ep_00043",
602
+ "source_episode_index": 43,
603
+ "split": "train",
604
+ "num_samples": 703,
605
+ "action_dim": 7,
606
+ "obs_mode": "vision",
607
+ "action_mode": "post_velcmd_hand_binary"
608
+ },
609
+ {
610
+ "hdf5_key": "ep_00044",
611
+ "source_episode_index": 44,
612
+ "split": "train",
613
+ "num_samples": 606,
614
+ "action_dim": 7,
615
+ "obs_mode": "vision",
616
+ "action_mode": "post_velcmd_hand_binary"
617
+ },
618
+ {
619
+ "hdf5_key": "ep_00045",
620
+ "source_episode_index": 45,
621
+ "split": "train",
622
+ "num_samples": 914,
623
+ "action_dim": 7,
624
+ "obs_mode": "vision",
625
+ "action_mode": "post_velcmd_hand_binary"
626
+ },
627
+ {
628
+ "hdf5_key": "ep_00046",
629
+ "source_episode_index": 46,
630
+ "split": "train",
631
+ "num_samples": 949,
632
+ "action_dim": 7,
633
+ "obs_mode": "vision",
634
+ "action_mode": "post_velcmd_hand_binary"
635
+ },
636
+ {
637
+ "hdf5_key": "ep_00047",
638
+ "source_episode_index": 47,
639
+ "split": "train",
640
+ "num_samples": 818,
641
+ "action_dim": 7,
642
+ "obs_mode": "vision",
643
+ "action_mode": "post_velcmd_hand_binary"
644
+ },
645
+ {
646
+ "hdf5_key": "ep_00048",
647
+ "source_episode_index": 48,
648
+ "split": "train",
649
+ "num_samples": 913,
650
+ "action_dim": 7,
651
+ "obs_mode": "vision",
652
+ "action_mode": "post_velcmd_hand_binary"
653
+ },
654
+ {
655
+ "hdf5_key": "ep_00049",
656
+ "source_episode_index": 49,
657
+ "split": "train",
658
+ "num_samples": 904,
659
+ "action_dim": 7,
660
+ "obs_mode": "vision",
661
+ "action_mode": "post_velcmd_hand_binary"
662
+ },
663
+ {
664
+ "hdf5_key": "ep_00050",
665
+ "source_episode_index": 50,
666
+ "split": "train",
667
+ "num_samples": 923,
668
+ "action_dim": 7,
669
+ "obs_mode": "vision",
670
+ "action_mode": "post_velcmd_hand_binary"
671
+ },
672
+ {
673
+ "hdf5_key": "ep_00051",
674
+ "source_episode_index": 51,
675
+ "split": "train",
676
+ "num_samples": 824,
677
+ "action_dim": 7,
678
+ "obs_mode": "vision",
679
+ "action_mode": "post_velcmd_hand_binary"
680
+ },
681
+ {
682
+ "hdf5_key": "ep_00052",
683
+ "source_episode_index": 52,
684
+ "split": "train",
685
+ "num_samples": 777,
686
+ "action_dim": 7,
687
+ "obs_mode": "vision",
688
+ "action_mode": "post_velcmd_hand_binary"
689
+ },
690
+ {
691
+ "hdf5_key": "ep_00053",
692
+ "source_episode_index": 53,
693
+ "split": "train",
694
+ "num_samples": 714,
695
+ "action_dim": 7,
696
+ "obs_mode": "vision",
697
+ "action_mode": "post_velcmd_hand_binary"
698
+ },
699
+ {
700
+ "hdf5_key": "ep_00054",
701
+ "source_episode_index": 54,
702
+ "split": "train",
703
+ "num_samples": 771,
704
+ "action_dim": 7,
705
+ "obs_mode": "vision",
706
+ "action_mode": "post_velcmd_hand_binary"
707
+ },
708
+ {
709
+ "hdf5_key": "ep_00055",
710
+ "source_episode_index": 55,
711
+ "split": "train",
712
+ "num_samples": 684,
713
+ "action_dim": 7,
714
+ "obs_mode": "vision",
715
+ "action_mode": "post_velcmd_hand_binary"
716
+ },
717
+ {
718
+ "hdf5_key": "ep_00056",
719
+ "source_episode_index": 56,
720
+ "split": "train",
721
+ "num_samples": 766,
722
+ "action_dim": 7,
723
+ "obs_mode": "vision",
724
+ "action_mode": "post_velcmd_hand_binary"
725
+ },
726
+ {
727
+ "hdf5_key": "ep_00057",
728
+ "source_episode_index": 57,
729
+ "split": "train",
730
+ "num_samples": 890,
731
+ "action_dim": 7,
732
+ "obs_mode": "vision",
733
+ "action_mode": "post_velcmd_hand_binary"
734
+ },
735
+ {
736
+ "hdf5_key": "ep_00058",
737
+ "source_episode_index": 58,
738
+ "split": "train",
739
+ "num_samples": 774,
740
+ "action_dim": 7,
741
+ "obs_mode": "vision",
742
+ "action_mode": "post_velcmd_hand_binary"
743
+ },
744
+ {
745
+ "hdf5_key": "ep_00059",
746
+ "source_episode_index": 59,
747
+ "split": "train",
748
+ "num_samples": 781,
749
+ "action_dim": 7,
750
+ "obs_mode": "vision",
751
+ "action_mode": "post_velcmd_hand_binary"
752
+ },
753
+ {
754
+ "hdf5_key": "ep_00060",
755
+ "source_episode_index": 60,
756
+ "split": "train",
757
+ "num_samples": 1093,
758
+ "action_dim": 7,
759
+ "obs_mode": "vision",
760
+ "action_mode": "post_velcmd_hand_binary"
761
+ },
762
+ {
763
+ "hdf5_key": "ep_00061",
764
+ "source_episode_index": 61,
765
+ "split": "train",
766
+ "num_samples": 900,
767
+ "action_dim": 7,
768
+ "obs_mode": "vision",
769
+ "action_mode": "post_velcmd_hand_binary"
770
+ },
771
+ {
772
+ "hdf5_key": "ep_00062",
773
+ "source_episode_index": 62,
774
+ "split": "train",
775
+ "num_samples": 814,
776
+ "action_dim": 7,
777
+ "obs_mode": "vision",
778
+ "action_mode": "post_velcmd_hand_binary"
779
+ },
780
+ {
781
+ "hdf5_key": "ep_00063",
782
+ "source_episode_index": 63,
783
+ "split": "train",
784
+ "num_samples": 1108,
785
+ "action_dim": 7,
786
+ "obs_mode": "vision",
787
+ "action_mode": "post_velcmd_hand_binary"
788
+ },
789
+ {
790
+ "hdf5_key": "ep_00064",
791
+ "source_episode_index": 64,
792
+ "split": "train",
793
+ "num_samples": 816,
794
+ "action_dim": 7,
795
+ "obs_mode": "vision",
796
+ "action_mode": "post_velcmd_hand_binary"
797
+ },
798
+ {
799
+ "hdf5_key": "ep_00065",
800
+ "source_episode_index": 65,
801
+ "split": "train",
802
+ "num_samples": 963,
803
+ "action_dim": 7,
804
+ "obs_mode": "vision",
805
+ "action_mode": "post_velcmd_hand_binary"
806
+ },
807
+ {
808
+ "hdf5_key": "ep_00066",
809
+ "source_episode_index": 66,
810
+ "split": "train",
811
+ "num_samples": 895,
812
+ "action_dim": 7,
813
+ "obs_mode": "vision",
814
+ "action_mode": "post_velcmd_hand_binary"
815
+ },
816
+ {
817
+ "hdf5_key": "ep_00067",
818
+ "source_episode_index": 67,
819
+ "split": "train",
820
+ "num_samples": 572,
821
+ "action_dim": 7,
822
+ "obs_mode": "vision",
823
+ "action_mode": "post_velcmd_hand_binary"
824
+ },
825
+ {
826
+ "hdf5_key": "ep_00068",
827
+ "source_episode_index": 68,
828
+ "split": "train",
829
+ "num_samples": 648,
830
+ "action_dim": 7,
831
+ "obs_mode": "vision",
832
+ "action_mode": "post_velcmd_hand_binary"
833
+ },
834
+ {
835
+ "hdf5_key": "ep_00069",
836
+ "source_episode_index": 69,
837
+ "split": "train",
838
+ "num_samples": 772,
839
+ "action_dim": 7,
840
+ "obs_mode": "vision",
841
+ "action_mode": "post_velcmd_hand_binary"
842
+ },
843
+ {
844
+ "hdf5_key": "ep_00070",
845
+ "source_episode_index": 70,
846
+ "split": "train",
847
+ "num_samples": 702,
848
+ "action_dim": 7,
849
+ "obs_mode": "vision",
850
+ "action_mode": "post_velcmd_hand_binary"
851
+ },
852
+ {
853
+ "hdf5_key": "ep_00071",
854
+ "source_episode_index": 71,
855
+ "split": "train",
856
+ "num_samples": 786,
857
+ "action_dim": 7,
858
+ "obs_mode": "vision",
859
+ "action_mode": "post_velcmd_hand_binary"
860
+ },
861
+ {
862
+ "hdf5_key": "ep_00072",
863
+ "source_episode_index": 72,
864
+ "split": "train",
865
+ "num_samples": 843,
866
+ "action_dim": 7,
867
+ "obs_mode": "vision",
868
+ "action_mode": "post_velcmd_hand_binary"
869
+ },
870
+ {
871
+ "hdf5_key": "ep_00073",
872
+ "source_episode_index": 73,
873
+ "split": "train",
874
+ "num_samples": 767,
875
+ "action_dim": 7,
876
+ "obs_mode": "vision",
877
+ "action_mode": "post_velcmd_hand_binary"
878
+ },
879
+ {
880
+ "hdf5_key": "ep_00074",
881
+ "source_episode_index": 74,
882
+ "split": "train",
883
+ "num_samples": 764,
884
+ "action_dim": 7,
885
+ "obs_mode": "vision",
886
+ "action_mode": "post_velcmd_hand_binary"
887
+ },
888
+ {
889
+ "hdf5_key": "ep_00075",
890
+ "source_episode_index": 75,
891
+ "split": "train",
892
+ "num_samples": 750,
893
+ "action_dim": 7,
894
+ "obs_mode": "vision",
895
+ "action_mode": "post_velcmd_hand_binary"
896
+ },
897
+ {
898
+ "hdf5_key": "ep_00076",
899
+ "source_episode_index": 76,
900
+ "split": "train",
901
+ "num_samples": 652,
902
+ "action_dim": 7,
903
+ "obs_mode": "vision",
904
+ "action_mode": "post_velcmd_hand_binary"
905
+ },
906
+ {
907
+ "hdf5_key": "ep_00077",
908
+ "source_episode_index": 77,
909
+ "split": "train",
910
+ "num_samples": 812,
911
+ "action_dim": 7,
912
+ "obs_mode": "vision",
913
+ "action_mode": "post_velcmd_hand_binary"
914
+ },
915
+ {
916
+ "hdf5_key": "ep_00078",
917
+ "source_episode_index": 78,
918
+ "split": "train",
919
+ "num_samples": 722,
920
+ "action_dim": 7,
921
+ "obs_mode": "vision",
922
+ "action_mode": "post_velcmd_hand_binary"
923
+ },
924
+ {
925
+ "hdf5_key": "ep_00079",
926
+ "source_episode_index": 79,
927
+ "split": "train",
928
+ "num_samples": 723,
929
+ "action_dim": 7,
930
+ "obs_mode": "vision",
931
+ "action_mode": "post_velcmd_hand_binary"
932
+ },
933
+ {
934
+ "hdf5_key": "ep_00080",
935
+ "source_episode_index": 80,
936
+ "split": "train",
937
+ "num_samples": 765,
938
+ "action_dim": 7,
939
+ "obs_mode": "vision",
940
+ "action_mode": "post_velcmd_hand_binary"
941
+ },
942
+ {
943
+ "hdf5_key": "ep_00081",
944
+ "source_episode_index": 81,
945
+ "split": "train",
946
+ "num_samples": 981,
947
+ "action_dim": 7,
948
+ "obs_mode": "vision",
949
+ "action_mode": "post_velcmd_hand_binary"
950
+ },
951
+ {
952
+ "hdf5_key": "ep_00082",
953
+ "source_episode_index": 82,
954
+ "split": "train",
955
+ "num_samples": 959,
956
+ "action_dim": 7,
957
+ "obs_mode": "vision",
958
+ "action_mode": "post_velcmd_hand_binary"
959
+ },
960
+ {
961
+ "hdf5_key": "ep_00083",
962
+ "source_episode_index": 83,
963
+ "split": "train",
964
+ "num_samples": 655,
965
+ "action_dim": 7,
966
+ "obs_mode": "vision",
967
+ "action_mode": "post_velcmd_hand_binary"
968
+ },
969
+ {
970
+ "hdf5_key": "ep_00084",
971
+ "source_episode_index": 84,
972
+ "split": "train",
973
+ "num_samples": 746,
974
+ "action_dim": 7,
975
+ "obs_mode": "vision",
976
+ "action_mode": "post_velcmd_hand_binary"
977
+ },
978
+ {
979
+ "hdf5_key": "ep_00085",
980
+ "source_episode_index": 85,
981
+ "split": "train",
982
+ "num_samples": 660,
983
+ "action_dim": 7,
984
+ "obs_mode": "vision",
985
+ "action_mode": "post_velcmd_hand_binary"
986
+ },
987
+ {
988
+ "hdf5_key": "ep_00086",
989
+ "source_episode_index": 86,
990
+ "split": "train",
991
+ "num_samples": 617,
992
+ "action_dim": 7,
993
+ "obs_mode": "vision",
994
+ "action_mode": "post_velcmd_hand_binary"
995
+ },
996
+ {
997
+ "hdf5_key": "ep_00087",
998
+ "source_episode_index": 87,
999
+ "split": "train",
1000
+ "num_samples": 819,
1001
+ "action_dim": 7,
1002
+ "obs_mode": "vision",
1003
+ "action_mode": "post_velcmd_hand_binary"
1004
+ },
1005
+ {
1006
+ "hdf5_key": "ep_00088",
1007
+ "source_episode_index": 88,
1008
+ "split": "train",
1009
+ "num_samples": 696,
1010
+ "action_dim": 7,
1011
+ "obs_mode": "vision",
1012
+ "action_mode": "post_velcmd_hand_binary"
1013
+ },
1014
+ {
1015
+ "hdf5_key": "ep_00089",
1016
+ "source_episode_index": 89,
1017
+ "split": "train",
1018
+ "num_samples": 855,
1019
+ "action_dim": 7,
1020
+ "obs_mode": "vision",
1021
+ "action_mode": "post_velcmd_hand_binary"
1022
+ },
1023
+ {
1024
+ "hdf5_key": "ep_00090",
1025
+ "source_episode_index": 90,
1026
+ "split": "train",
1027
+ "num_samples": 678,
1028
+ "action_dim": 7,
1029
+ "obs_mode": "vision",
1030
+ "action_mode": "post_velcmd_hand_binary"
1031
+ },
1032
+ {
1033
+ "hdf5_key": "ep_00091",
1034
+ "source_episode_index": 91,
1035
+ "split": "train",
1036
+ "num_samples": 711,
1037
+ "action_dim": 7,
1038
+ "obs_mode": "vision",
1039
+ "action_mode": "post_velcmd_hand_binary"
1040
+ },
1041
+ {
1042
+ "hdf5_key": "ep_00092",
1043
+ "source_episode_index": 92,
1044
+ "split": "train",
1045
+ "num_samples": 763,
1046
+ "action_dim": 7,
1047
+ "obs_mode": "vision",
1048
+ "action_mode": "post_velcmd_hand_binary"
1049
+ },
1050
+ {
1051
+ "hdf5_key": "ep_00093",
1052
+ "source_episode_index": 93,
1053
+ "split": "train",
1054
+ "num_samples": 723,
1055
+ "action_dim": 7,
1056
+ "obs_mode": "vision",
1057
+ "action_mode": "post_velcmd_hand_binary"
1058
+ },
1059
+ {
1060
+ "hdf5_key": "ep_00094",
1061
+ "source_episode_index": 94,
1062
+ "split": "train",
1063
+ "num_samples": 723,
1064
+ "action_dim": 7,
1065
+ "obs_mode": "vision",
1066
+ "action_mode": "post_velcmd_hand_binary"
1067
+ },
1068
+ {
1069
+ "hdf5_key": "ep_00095",
1070
+ "source_episode_index": 95,
1071
+ "split": "train",
1072
+ "num_samples": 865,
1073
+ "action_dim": 7,
1074
+ "obs_mode": "vision",
1075
+ "action_mode": "post_velcmd_hand_binary"
1076
+ },
1077
+ {
1078
+ "hdf5_key": "ep_00096",
1079
+ "source_episode_index": 96,
1080
+ "split": "train",
1081
+ "num_samples": 495,
1082
+ "action_dim": 7,
1083
+ "obs_mode": "vision",
1084
+ "action_mode": "post_velcmd_hand_binary"
1085
+ },
1086
+ {
1087
+ "hdf5_key": "ep_00097",
1088
+ "source_episode_index": 97,
1089
+ "split": "train",
1090
+ "num_samples": 451,
1091
+ "action_dim": 7,
1092
+ "obs_mode": "vision",
1093
+ "action_mode": "post_velcmd_hand_binary"
1094
+ },
1095
+ {
1096
+ "hdf5_key": "ep_00098",
1097
+ "source_episode_index": 98,
1098
+ "split": "train",
1099
+ "num_samples": 690,
1100
+ "action_dim": 7,
1101
+ "obs_mode": "vision",
1102
+ "action_mode": "post_velcmd_hand_binary"
1103
+ },
1104
+ {
1105
+ "hdf5_key": "ep_00099",
1106
+ "source_episode_index": 99,
1107
+ "split": "train",
1108
+ "num_samples": 690,
1109
+ "action_dim": 7,
1110
+ "obs_mode": "vision",
1111
+ "action_mode": "post_velcmd_hand_binary"
1112
+ },
1113
+ {
1114
+ "hdf5_key": "ep_00100",
1115
+ "source_episode_index": 100,
1116
+ "split": "train",
1117
+ "num_samples": 615,
1118
+ "action_dim": 7,
1119
+ "obs_mode": "vision",
1120
+ "action_mode": "post_velcmd_hand_binary"
1121
+ },
1122
+ {
1123
+ "hdf5_key": "ep_00101",
1124
+ "source_episode_index": 101,
1125
+ "split": "train",
1126
+ "num_samples": 662,
1127
+ "action_dim": 7,
1128
+ "obs_mode": "vision",
1129
+ "action_mode": "post_velcmd_hand_binary"
1130
+ },
1131
+ {
1132
+ "hdf5_key": "ep_00102",
1133
+ "source_episode_index": 102,
1134
+ "split": "train",
1135
+ "num_samples": 614,
1136
+ "action_dim": 7,
1137
+ "obs_mode": "vision",
1138
+ "action_mode": "post_velcmd_hand_binary"
1139
+ },
1140
+ {
1141
+ "hdf5_key": "ep_00103",
1142
+ "source_episode_index": 103,
1143
+ "split": "train",
1144
+ "num_samples": 730,
1145
+ "action_dim": 7,
1146
+ "obs_mode": "vision",
1147
+ "action_mode": "post_velcmd_hand_binary"
1148
+ },
1149
+ {
1150
+ "hdf5_key": "ep_00104",
1151
+ "source_episode_index": 104,
1152
+ "split": "train",
1153
+ "num_samples": 683,
1154
+ "action_dim": 7,
1155
+ "obs_mode": "vision",
1156
+ "action_mode": "post_velcmd_hand_binary"
1157
+ },
1158
+ {
1159
+ "hdf5_key": "ep_00105",
1160
+ "source_episode_index": 105,
1161
+ "split": "train",
1162
+ "num_samples": 602,
1163
+ "action_dim": 7,
1164
+ "obs_mode": "vision",
1165
+ "action_mode": "post_velcmd_hand_binary"
1166
+ },
1167
+ {
1168
+ "hdf5_key": "ep_00106",
1169
+ "source_episode_index": 106,
1170
+ "split": "train",
1171
+ "num_samples": 652,
1172
+ "action_dim": 7,
1173
+ "obs_mode": "vision",
1174
+ "action_mode": "post_velcmd_hand_binary"
1175
+ },
1176
+ {
1177
+ "hdf5_key": "ep_00107",
1178
+ "source_episode_index": 107,
1179
+ "split": "train",
1180
+ "num_samples": 722,
1181
+ "action_dim": 7,
1182
+ "obs_mode": "vision",
1183
+ "action_mode": "post_velcmd_hand_binary"
1184
+ },
1185
+ {
1186
+ "hdf5_key": "ep_00108",
1187
+ "source_episode_index": 108,
1188
+ "split": "train",
1189
+ "num_samples": 605,
1190
+ "action_dim": 7,
1191
+ "obs_mode": "vision",
1192
+ "action_mode": "post_velcmd_hand_binary"
1193
+ },
1194
+ {
1195
+ "hdf5_key": "ep_00109",
1196
+ "source_episode_index": 109,
1197
+ "split": "train",
1198
+ "num_samples": 944,
1199
+ "action_dim": 7,
1200
+ "obs_mode": "vision",
1201
+ "action_mode": "post_velcmd_hand_binary"
1202
+ },
1203
+ {
1204
+ "hdf5_key": "ep_00110",
1205
+ "source_episode_index": 110,
1206
+ "split": "train",
1207
+ "num_samples": 661,
1208
+ "action_dim": 7,
1209
+ "obs_mode": "vision",
1210
+ "action_mode": "post_velcmd_hand_binary"
1211
+ },
1212
+ {
1213
+ "hdf5_key": "ep_00111",
1214
+ "source_episode_index": 111,
1215
+ "split": "train",
1216
+ "num_samples": 644,
1217
+ "action_dim": 7,
1218
+ "obs_mode": "vision",
1219
+ "action_mode": "post_velcmd_hand_binary"
1220
+ },
1221
+ {
1222
+ "hdf5_key": "ep_00112",
1223
+ "source_episode_index": 112,
1224
+ "split": "train",
1225
+ "num_samples": 614,
1226
+ "action_dim": 7,
1227
+ "obs_mode": "vision",
1228
+ "action_mode": "post_velcmd_hand_binary"
1229
+ },
1230
+ {
1231
+ "hdf5_key": "ep_00113",
1232
+ "source_episode_index": 113,
1233
+ "split": "train",
1234
+ "num_samples": 548,
1235
+ "action_dim": 7,
1236
+ "obs_mode": "vision",
1237
+ "action_mode": "post_velcmd_hand_binary"
1238
+ },
1239
+ {
1240
+ "hdf5_key": "ep_00114",
1241
+ "source_episode_index": 114,
1242
+ "split": "train",
1243
+ "num_samples": 568,
1244
+ "action_dim": 7,
1245
+ "obs_mode": "vision",
1246
+ "action_mode": "post_velcmd_hand_binary"
1247
+ },
1248
+ {
1249
+ "hdf5_key": "ep_00115",
1250
+ "source_episode_index": 115,
1251
+ "split": "train",
1252
+ "num_samples": 730,
1253
+ "action_dim": 7,
1254
+ "obs_mode": "vision",
1255
+ "action_mode": "post_velcmd_hand_binary"
1256
+ },
1257
+ {
1258
+ "hdf5_key": "ep_00116",
1259
+ "source_episode_index": 116,
1260
+ "split": "train",
1261
+ "num_samples": 481,
1262
+ "action_dim": 7,
1263
+ "obs_mode": "vision",
1264
+ "action_mode": "post_velcmd_hand_binary"
1265
+ },
1266
+ {
1267
+ "hdf5_key": "ep_00117",
1268
+ "source_episode_index": 117,
1269
+ "split": "train",
1270
+ "num_samples": 687,
1271
+ "action_dim": 7,
1272
+ "obs_mode": "vision",
1273
+ "action_mode": "post_velcmd_hand_binary"
1274
+ },
1275
+ {
1276
+ "hdf5_key": "ep_00118",
1277
+ "source_episode_index": 118,
1278
+ "split": "train",
1279
+ "num_samples": 618,
1280
+ "action_dim": 7,
1281
+ "obs_mode": "vision",
1282
+ "action_mode": "post_velcmd_hand_binary"
1283
+ },
1284
+ {
1285
+ "hdf5_key": "ep_00119",
1286
+ "source_episode_index": 119,
1287
+ "split": "train",
1288
+ "num_samples": 768,
1289
+ "action_dim": 7,
1290
+ "obs_mode": "vision",
1291
+ "action_mode": "post_velcmd_hand_binary"
1292
+ },
1293
+ {
1294
+ "hdf5_key": "ep_00120",
1295
+ "source_episode_index": 120,
1296
+ "split": "train",
1297
+ "num_samples": 734,
1298
+ "action_dim": 7,
1299
+ "obs_mode": "vision",
1300
+ "action_mode": "post_velcmd_hand_binary"
1301
+ },
1302
+ {
1303
+ "hdf5_key": "ep_00121",
1304
+ "source_episode_index": 121,
1305
+ "split": "train",
1306
+ "num_samples": 604,
1307
+ "action_dim": 7,
1308
+ "obs_mode": "vision",
1309
+ "action_mode": "post_velcmd_hand_binary"
1310
+ },
1311
+ {
1312
+ "hdf5_key": "ep_00122",
1313
+ "source_episode_index": 122,
1314
+ "split": "train",
1315
+ "num_samples": 610,
1316
+ "action_dim": 7,
1317
+ "obs_mode": "vision",
1318
+ "action_mode": "post_velcmd_hand_binary"
1319
+ },
1320
+ {
1321
+ "hdf5_key": "ep_00123",
1322
+ "source_episode_index": 123,
1323
+ "split": "train",
1324
+ "num_samples": 602,
1325
+ "action_dim": 7,
1326
+ "obs_mode": "vision",
1327
+ "action_mode": "post_velcmd_hand_binary"
1328
+ },
1329
+ {
1330
+ "hdf5_key": "ep_00124",
1331
+ "source_episode_index": 124,
1332
+ "split": "train",
1333
+ "num_samples": 630,
1334
+ "action_dim": 7,
1335
+ "obs_mode": "vision",
1336
+ "action_mode": "post_velcmd_hand_binary"
1337
+ },
1338
+ {
1339
+ "hdf5_key": "ep_00125",
1340
+ "source_episode_index": 125,
1341
+ "split": "train",
1342
+ "num_samples": 621,
1343
+ "action_dim": 7,
1344
+ "obs_mode": "vision",
1345
+ "action_mode": "post_velcmd_hand_binary"
1346
+ },
1347
+ {
1348
+ "hdf5_key": "ep_00126",
1349
+ "source_episode_index": 126,
1350
+ "split": "train",
1351
+ "num_samples": 600,
1352
+ "action_dim": 7,
1353
+ "obs_mode": "vision",
1354
+ "action_mode": "post_velcmd_hand_binary"
1355
+ },
1356
+ {
1357
+ "hdf5_key": "ep_00127",
1358
+ "source_episode_index": 127,
1359
+ "split": "train",
1360
+ "num_samples": 478,
1361
+ "action_dim": 7,
1362
+ "obs_mode": "vision",
1363
+ "action_mode": "post_velcmd_hand_binary"
1364
+ },
1365
+ {
1366
+ "hdf5_key": "ep_00128",
1367
+ "source_episode_index": 128,
1368
+ "split": "train",
1369
+ "num_samples": 606,
1370
+ "action_dim": 7,
1371
+ "obs_mode": "vision",
1372
+ "action_mode": "post_velcmd_hand_binary"
1373
+ },
1374
+ {
1375
+ "hdf5_key": "ep_00129",
1376
+ "source_episode_index": 129,
1377
+ "split": "train",
1378
+ "num_samples": 666,
1379
+ "action_dim": 7,
1380
+ "obs_mode": "vision",
1381
+ "action_mode": "post_velcmd_hand_binary"
1382
+ },
1383
+ {
1384
+ "hdf5_key": "ep_00130",
1385
+ "source_episode_index": 130,
1386
+ "split": "train",
1387
+ "num_samples": 655,
1388
+ "action_dim": 7,
1389
+ "obs_mode": "vision",
1390
+ "action_mode": "post_velcmd_hand_binary"
1391
+ },
1392
+ {
1393
+ "hdf5_key": "ep_00131",
1394
+ "source_episode_index": 131,
1395
+ "split": "train",
1396
+ "num_samples": 588,
1397
+ "action_dim": 7,
1398
+ "obs_mode": "vision",
1399
+ "action_mode": "post_velcmd_hand_binary"
1400
+ },
1401
+ {
1402
+ "hdf5_key": "ep_00132",
1403
+ "source_episode_index": 132,
1404
+ "split": "train",
1405
+ "num_samples": 806,
1406
+ "action_dim": 7,
1407
+ "obs_mode": "vision",
1408
+ "action_mode": "post_velcmd_hand_binary"
1409
+ },
1410
+ {
1411
+ "hdf5_key": "ep_00133",
1412
+ "source_episode_index": 133,
1413
+ "split": "train",
1414
+ "num_samples": 586,
1415
+ "action_dim": 7,
1416
+ "obs_mode": "vision",
1417
+ "action_mode": "post_velcmd_hand_binary"
1418
+ },
1419
+ {
1420
+ "hdf5_key": "ep_00134",
1421
+ "source_episode_index": 134,
1422
+ "split": "train",
1423
+ "num_samples": 714,
1424
+ "action_dim": 7,
1425
+ "obs_mode": "vision",
1426
+ "action_mode": "post_velcmd_hand_binary"
1427
+ },
1428
+ {
1429
+ "hdf5_key": "ep_00135",
1430
+ "source_episode_index": 135,
1431
+ "split": "train",
1432
+ "num_samples": 574,
1433
+ "action_dim": 7,
1434
+ "obs_mode": "vision",
1435
+ "action_mode": "post_velcmd_hand_binary"
1436
+ },
1437
+ {
1438
+ "hdf5_key": "ep_00136",
1439
+ "source_episode_index": 136,
1440
+ "split": "train",
1441
+ "num_samples": 577,
1442
+ "action_dim": 7,
1443
+ "obs_mode": "vision",
1444
+ "action_mode": "post_velcmd_hand_binary"
1445
+ },
1446
+ {
1447
+ "hdf5_key": "ep_00137",
1448
+ "source_episode_index": 137,
1449
+ "split": "train",
1450
+ "num_samples": 663,
1451
+ "action_dim": 7,
1452
+ "obs_mode": "vision",
1453
+ "action_mode": "post_velcmd_hand_binary"
1454
+ },
1455
+ {
1456
+ "hdf5_key": "ep_00138",
1457
+ "source_episode_index": 138,
1458
+ "split": "train",
1459
+ "num_samples": 720,
1460
+ "action_dim": 7,
1461
+ "obs_mode": "vision",
1462
+ "action_mode": "post_velcmd_hand_binary"
1463
+ },
1464
+ {
1465
+ "hdf5_key": "ep_00139",
1466
+ "source_episode_index": 139,
1467
+ "split": "train",
1468
+ "num_samples": 640,
1469
+ "action_dim": 7,
1470
+ "obs_mode": "vision",
1471
+ "action_mode": "post_velcmd_hand_binary"
1472
+ },
1473
+ {
1474
+ "hdf5_key": "ep_00140",
1475
+ "source_episode_index": 140,
1476
+ "split": "train",
1477
+ "num_samples": 544,
1478
+ "action_dim": 7,
1479
+ "obs_mode": "vision",
1480
+ "action_mode": "post_velcmd_hand_binary"
1481
+ },
1482
+ {
1483
+ "hdf5_key": "ep_00141",
1484
+ "source_episode_index": 141,
1485
+ "split": "train",
1486
+ "num_samples": 539,
1487
+ "action_dim": 7,
1488
+ "obs_mode": "vision",
1489
+ "action_mode": "post_velcmd_hand_binary"
1490
+ },
1491
+ {
1492
+ "hdf5_key": "ep_00142",
1493
+ "source_episode_index": 142,
1494
+ "split": "train",
1495
+ "num_samples": 592,
1496
+ "action_dim": 7,
1497
+ "obs_mode": "vision",
1498
+ "action_mode": "post_velcmd_hand_binary"
1499
+ },
1500
+ {
1501
+ "hdf5_key": "ep_00143",
1502
+ "source_episode_index": 143,
1503
+ "split": "train",
1504
+ "num_samples": 625,
1505
+ "action_dim": 7,
1506
+ "obs_mode": "vision",
1507
+ "action_mode": "post_velcmd_hand_binary"
1508
+ },
1509
+ {
1510
+ "hdf5_key": "ep_00144",
1511
+ "source_episode_index": 144,
1512
+ "split": "train",
1513
+ "num_samples": 684,
1514
+ "action_dim": 7,
1515
+ "obs_mode": "vision",
1516
+ "action_mode": "post_velcmd_hand_binary"
1517
+ },
1518
+ {
1519
+ "hdf5_key": "ep_00145",
1520
+ "source_episode_index": 145,
1521
+ "split": "train",
1522
+ "num_samples": 559,
1523
+ "action_dim": 7,
1524
+ "obs_mode": "vision",
1525
+ "action_mode": "post_velcmd_hand_binary"
1526
+ },
1527
+ {
1528
+ "hdf5_key": "ep_00146",
1529
+ "source_episode_index": 146,
1530
+ "split": "train",
1531
+ "num_samples": 677,
1532
+ "action_dim": 7,
1533
+ "obs_mode": "vision",
1534
+ "action_mode": "post_velcmd_hand_binary"
1535
+ },
1536
+ {
1537
+ "hdf5_key": "ep_00147",
1538
+ "source_episode_index": 147,
1539
+ "split": "train",
1540
+ "num_samples": 846,
1541
+ "action_dim": 7,
1542
+ "obs_mode": "vision",
1543
+ "action_mode": "post_velcmd_hand_binary"
1544
+ },
1545
+ {
1546
+ "hdf5_key": "ep_00148",
1547
+ "source_episode_index": 148,
1548
+ "split": "train",
1549
+ "num_samples": 615,
1550
+ "action_dim": 7,
1551
+ "obs_mode": "vision",
1552
+ "action_mode": "post_velcmd_hand_binary"
1553
+ },
1554
+ {
1555
+ "hdf5_key": "ep_00149",
1556
+ "source_episode_index": 149,
1557
+ "split": "train",
1558
+ "num_samples": 517,
1559
+ "action_dim": 7,
1560
+ "obs_mode": "vision",
1561
+ "action_mode": "post_velcmd_hand_binary"
1562
+ },
1563
+ {
1564
+ "hdf5_key": "ep_00150",
1565
+ "source_episode_index": 150,
1566
+ "split": "train",
1567
+ "num_samples": 817,
1568
+ "action_dim": 7,
1569
+ "obs_mode": "vision",
1570
+ "action_mode": "post_velcmd_hand_binary"
1571
+ },
1572
+ {
1573
+ "hdf5_key": "ep_00151",
1574
+ "source_episode_index": 151,
1575
+ "split": "train",
1576
+ "num_samples": 814,
1577
+ "action_dim": 7,
1578
+ "obs_mode": "vision",
1579
+ "action_mode": "post_velcmd_hand_binary"
1580
+ },
1581
+ {
1582
+ "hdf5_key": "ep_00152",
1583
+ "source_episode_index": 152,
1584
+ "split": "train",
1585
+ "num_samples": 828,
1586
+ "action_dim": 7,
1587
+ "obs_mode": "vision",
1588
+ "action_mode": "post_velcmd_hand_binary"
1589
+ },
1590
+ {
1591
+ "hdf5_key": "ep_00153",
1592
+ "source_episode_index": 153,
1593
+ "split": "train",
1594
+ "num_samples": 736,
1595
+ "action_dim": 7,
1596
+ "obs_mode": "vision",
1597
+ "action_mode": "post_velcmd_hand_binary"
1598
+ },
1599
+ {
1600
+ "hdf5_key": "ep_00154",
1601
+ "source_episode_index": 154,
1602
+ "split": "train",
1603
+ "num_samples": 860,
1604
+ "action_dim": 7,
1605
+ "obs_mode": "vision",
1606
+ "action_mode": "post_velcmd_hand_binary"
1607
+ },
1608
+ {
1609
+ "hdf5_key": "ep_00155",
1610
+ "source_episode_index": 155,
1611
+ "split": "train",
1612
+ "num_samples": 786,
1613
+ "action_dim": 7,
1614
+ "obs_mode": "vision",
1615
+ "action_mode": "post_velcmd_hand_binary"
1616
+ },
1617
+ {
1618
+ "hdf5_key": "ep_00156",
1619
+ "source_episode_index": 156,
1620
+ "split": "train",
1621
+ "num_samples": 725,
1622
+ "action_dim": 7,
1623
+ "obs_mode": "vision",
1624
+ "action_mode": "post_velcmd_hand_binary"
1625
+ },
1626
+ {
1627
+ "hdf5_key": "ep_00157",
1628
+ "source_episode_index": 157,
1629
+ "split": "train",
1630
+ "num_samples": 834,
1631
+ "action_dim": 7,
1632
+ "obs_mode": "vision",
1633
+ "action_mode": "post_velcmd_hand_binary"
1634
+ },
1635
+ {
1636
+ "hdf5_key": "ep_00158",
1637
+ "source_episode_index": 158,
1638
+ "split": "train",
1639
+ "num_samples": 652,
1640
+ "action_dim": 7,
1641
+ "obs_mode": "vision",
1642
+ "action_mode": "post_velcmd_hand_binary"
1643
+ },
1644
+ {
1645
+ "hdf5_key": "ep_00159",
1646
+ "source_episode_index": 159,
1647
+ "split": "train",
1648
+ "num_samples": 699,
1649
+ "action_dim": 7,
1650
+ "obs_mode": "vision",
1651
+ "action_mode": "post_velcmd_hand_binary"
1652
+ },
1653
+ {
1654
+ "hdf5_key": "ep_00160",
1655
+ "source_episode_index": 160,
1656
+ "split": "train",
1657
+ "num_samples": 812,
1658
+ "action_dim": 7,
1659
+ "obs_mode": "vision",
1660
+ "action_mode": "post_velcmd_hand_binary"
1661
+ },
1662
+ {
1663
+ "hdf5_key": "ep_00161",
1664
+ "source_episode_index": 161,
1665
+ "split": "train",
1666
+ "num_samples": 782,
1667
+ "action_dim": 7,
1668
+ "obs_mode": "vision",
1669
+ "action_mode": "post_velcmd_hand_binary"
1670
+ },
1671
+ {
1672
+ "hdf5_key": "ep_00162",
1673
+ "source_episode_index": 162,
1674
+ "split": "train",
1675
+ "num_samples": 681,
1676
+ "action_dim": 7,
1677
+ "obs_mode": "vision",
1678
+ "action_mode": "post_velcmd_hand_binary"
1679
+ },
1680
+ {
1681
+ "hdf5_key": "ep_00163",
1682
+ "source_episode_index": 163,
1683
+ "split": "train",
1684
+ "num_samples": 742,
1685
+ "action_dim": 7,
1686
+ "obs_mode": "vision",
1687
+ "action_mode": "post_velcmd_hand_binary"
1688
+ },
1689
+ {
1690
+ "hdf5_key": "ep_00164",
1691
+ "source_episode_index": 164,
1692
+ "split": "train",
1693
+ "num_samples": 766,
1694
+ "action_dim": 7,
1695
+ "obs_mode": "vision",
1696
+ "action_mode": "post_velcmd_hand_binary"
1697
+ },
1698
+ {
1699
+ "hdf5_key": "ep_00165",
1700
+ "source_episode_index": 165,
1701
+ "split": "train",
1702
+ "num_samples": 821,
1703
+ "action_dim": 7,
1704
+ "obs_mode": "vision",
1705
+ "action_mode": "post_velcmd_hand_binary"
1706
+ },
1707
+ {
1708
+ "hdf5_key": "ep_00166",
1709
+ "source_episode_index": 166,
1710
+ "split": "train",
1711
+ "num_samples": 1028,
1712
+ "action_dim": 7,
1713
+ "obs_mode": "vision",
1714
+ "action_mode": "post_velcmd_hand_binary"
1715
+ },
1716
+ {
1717
+ "hdf5_key": "ep_00167",
1718
+ "source_episode_index": 167,
1719
+ "split": "train",
1720
+ "num_samples": 720,
1721
+ "action_dim": 7,
1722
+ "obs_mode": "vision",
1723
+ "action_mode": "post_velcmd_hand_binary"
1724
+ },
1725
+ {
1726
+ "hdf5_key": "ep_00168",
1727
+ "source_episode_index": 168,
1728
+ "split": "train",
1729
+ "num_samples": 773,
1730
+ "action_dim": 7,
1731
+ "obs_mode": "vision",
1732
+ "action_mode": "post_velcmd_hand_binary"
1733
+ },
1734
+ {
1735
+ "hdf5_key": "ep_00169",
1736
+ "source_episode_index": 169,
1737
+ "split": "train",
1738
+ "num_samples": 690,
1739
+ "action_dim": 7,
1740
+ "obs_mode": "vision",
1741
+ "action_mode": "post_velcmd_hand_binary"
1742
+ },
1743
+ {
1744
+ "hdf5_key": "ep_00170",
1745
+ "source_episode_index": 170,
1746
+ "split": "train",
1747
+ "num_samples": 756,
1748
+ "action_dim": 7,
1749
+ "obs_mode": "vision",
1750
+ "action_mode": "post_velcmd_hand_binary"
1751
+ },
1752
+ {
1753
+ "hdf5_key": "ep_00171",
1754
+ "source_episode_index": 171,
1755
+ "split": "train",
1756
+ "num_samples": 742,
1757
+ "action_dim": 7,
1758
+ "obs_mode": "vision",
1759
+ "action_mode": "post_velcmd_hand_binary"
1760
+ },
1761
+ {
1762
+ "hdf5_key": "ep_00172",
1763
+ "source_episode_index": 172,
1764
+ "split": "valid",
1765
+ "num_samples": 689,
1766
+ "action_dim": 7,
1767
+ "obs_mode": "vision",
1768
+ "action_mode": "post_velcmd_hand_binary"
1769
+ },
1770
+ {
1771
+ "hdf5_key": "ep_00173",
1772
+ "source_episode_index": 173,
1773
+ "split": "valid",
1774
+ "num_samples": 639,
1775
+ "action_dim": 7,
1776
+ "obs_mode": "vision",
1777
+ "action_mode": "post_velcmd_hand_binary"
1778
+ },
1779
+ {
1780
+ "hdf5_key": "ep_00174",
1781
+ "source_episode_index": 174,
1782
+ "split": "valid",
1783
+ "num_samples": 694,
1784
+ "action_dim": 7,
1785
+ "obs_mode": "vision",
1786
+ "action_mode": "post_velcmd_hand_binary"
1787
+ },
1788
+ {
1789
+ "hdf5_key": "ep_00175",
1790
+ "source_episode_index": 175,
1791
+ "split": "valid",
1792
+ "num_samples": 532,
1793
+ "action_dim": 7,
1794
+ "obs_mode": "vision",
1795
+ "action_mode": "post_velcmd_hand_binary"
1796
+ },
1797
+ {
1798
+ "hdf5_key": "ep_00176",
1799
+ "source_episode_index": 176,
1800
+ "split": "valid",
1801
+ "num_samples": 665,
1802
+ "action_dim": 7,
1803
+ "obs_mode": "vision",
1804
+ "action_mode": "post_velcmd_hand_binary"
1805
+ },
1806
+ {
1807
+ "hdf5_key": "ep_00177",
1808
+ "source_episode_index": 177,
1809
+ "split": "valid",
1810
+ "num_samples": 771,
1811
+ "action_dim": 7,
1812
+ "obs_mode": "vision",
1813
+ "action_mode": "post_velcmd_hand_binary"
1814
+ },
1815
+ {
1816
+ "hdf5_key": "ep_00178",
1817
+ "source_episode_index": 178,
1818
+ "split": "valid",
1819
+ "num_samples": 602,
1820
+ "action_dim": 7,
1821
+ "obs_mode": "vision",
1822
+ "action_mode": "post_velcmd_hand_binary"
1823
+ },
1824
+ {
1825
+ "hdf5_key": "ep_00179",
1826
+ "source_episode_index": 179,
1827
+ "split": "valid",
1828
+ "num_samples": 700,
1829
+ "action_dim": 7,
1830
+ "obs_mode": "vision",
1831
+ "action_mode": "post_velcmd_hand_binary"
1832
+ },
1833
+ {
1834
+ "hdf5_key": "ep_00180",
1835
+ "source_episode_index": 180,
1836
+ "split": "valid",
1837
+ "num_samples": 880,
1838
+ "action_dim": 7,
1839
+ "obs_mode": "vision",
1840
+ "action_mode": "post_velcmd_hand_binary"
1841
+ },
1842
+ {
1843
+ "hdf5_key": "ep_00181",
1844
+ "source_episode_index": 181,
1845
+ "split": "valid",
1846
+ "num_samples": 522,
1847
+ "action_dim": 7,
1848
+ "obs_mode": "vision",
1849
+ "action_mode": "post_velcmd_hand_binary"
1850
+ },
1851
+ {
1852
+ "hdf5_key": "ep_00182",
1853
+ "source_episode_index": 182,
1854
+ "split": "valid",
1855
+ "num_samples": 697,
1856
+ "action_dim": 7,
1857
+ "obs_mode": "vision",
1858
+ "action_mode": "post_velcmd_hand_binary"
1859
+ },
1860
+ {
1861
+ "hdf5_key": "ep_00183",
1862
+ "source_episode_index": 183,
1863
+ "split": "valid",
1864
+ "num_samples": 621,
1865
+ "action_dim": 7,
1866
+ "obs_mode": "vision",
1867
+ "action_mode": "post_velcmd_hand_binary"
1868
+ },
1869
+ {
1870
+ "hdf5_key": "ep_00184",
1871
+ "source_episode_index": 184,
1872
+ "split": "valid",
1873
+ "num_samples": 710,
1874
+ "action_dim": 7,
1875
+ "obs_mode": "vision",
1876
+ "action_mode": "post_velcmd_hand_binary"
1877
+ },
1878
+ {
1879
+ "hdf5_key": "ep_00185",
1880
+ "source_episode_index": 185,
1881
+ "split": "valid",
1882
+ "num_samples": 774,
1883
+ "action_dim": 7,
1884
+ "obs_mode": "vision",
1885
+ "action_mode": "post_velcmd_hand_binary"
1886
+ },
1887
+ {
1888
+ "hdf5_key": "ep_00186",
1889
+ "source_episode_index": 186,
1890
+ "split": "valid",
1891
+ "num_samples": 682,
1892
+ "action_dim": 7,
1893
+ "obs_mode": "vision",
1894
+ "action_mode": "post_velcmd_hand_binary"
1895
+ },
1896
+ {
1897
+ "hdf5_key": "ep_00187",
1898
+ "source_episode_index": 187,
1899
+ "split": "valid",
1900
+ "num_samples": 715,
1901
+ "action_dim": 7,
1902
+ "obs_mode": "vision",
1903
+ "action_mode": "post_velcmd_hand_binary"
1904
+ },
1905
+ {
1906
+ "hdf5_key": "ep_00188",
1907
+ "source_episode_index": 188,
1908
+ "split": "valid",
1909
+ "num_samples": 798,
1910
+ "action_dim": 7,
1911
+ "obs_mode": "vision",
1912
+ "action_mode": "post_velcmd_hand_binary"
1913
+ },
1914
+ {
1915
+ "hdf5_key": "ep_00189",
1916
+ "source_episode_index": 189,
1917
+ "split": "valid",
1918
+ "num_samples": 786,
1919
+ "action_dim": 7,
1920
+ "obs_mode": "vision",
1921
+ "action_mode": "post_velcmd_hand_binary"
1922
+ },
1923
+ {
1924
+ "hdf5_key": "ep_00190",
1925
+ "source_episode_index": 190,
1926
+ "split": "valid",
1927
+ "num_samples": 759,
1928
+ "action_dim": 7,
1929
+ "obs_mode": "vision",
1930
+ "action_mode": "post_velcmd_hand_binary"
1931
+ },
1932
+ {
1933
+ "hdf5_key": "ep_00191",
1934
+ "source_episode_index": 191,
1935
+ "split": "valid",
1936
+ "num_samples": 812,
1937
+ "action_dim": 7,
1938
+ "obs_mode": "vision",
1939
+ "action_mode": "post_velcmd_hand_binary"
1940
+ },
1941
+ {
1942
+ "hdf5_key": "ep_00192",
1943
+ "source_episode_index": 192,
1944
+ "split": "valid",
1945
+ "num_samples": 803,
1946
+ "action_dim": 7,
1947
+ "obs_mode": "vision",
1948
+ "action_mode": "post_velcmd_hand_binary"
1949
+ },
1950
+ {
1951
+ "hdf5_key": "ep_00193",
1952
+ "source_episode_index": 193,
1953
+ "split": "valid",
1954
+ "num_samples": 780,
1955
+ "action_dim": 7,
1956
+ "obs_mode": "vision",
1957
+ "action_mode": "post_velcmd_hand_binary"
1958
+ },
1959
+ {
1960
+ "hdf5_key": "ep_00194",
1961
+ "source_episode_index": 194,
1962
+ "split": "valid",
1963
+ "num_samples": 853,
1964
+ "action_dim": 7,
1965
+ "obs_mode": "vision",
1966
+ "action_mode": "post_velcmd_hand_binary"
1967
+ },
1968
+ {
1969
+ "hdf5_key": "ep_00195",
1970
+ "source_episode_index": 195,
1971
+ "split": "valid",
1972
+ "num_samples": 710,
1973
+ "action_dim": 7,
1974
+ "obs_mode": "vision",
1975
+ "action_mode": "post_velcmd_hand_binary"
1976
+ },
1977
+ {
1978
+ "hdf5_key": "ep_00196",
1979
+ "source_episode_index": 196,
1980
+ "split": "valid",
1981
+ "num_samples": 855,
1982
+ "action_dim": 7,
1983
+ "obs_mode": "vision",
1984
+ "action_mode": "post_velcmd_hand_binary"
1985
+ },
1986
+ {
1987
+ "hdf5_key": "ep_00197",
1988
+ "source_episode_index": 197,
1989
+ "split": "valid",
1990
+ "num_samples": 707,
1991
+ "action_dim": 7,
1992
+ "obs_mode": "vision",
1993
+ "action_mode": "post_velcmd_hand_binary"
1994
+ },
1995
+ {
1996
+ "hdf5_key": "ep_00198",
1997
+ "source_episode_index": 198,
1998
+ "split": "valid",
1999
+ "num_samples": 557,
2000
+ "action_dim": 7,
2001
+ "obs_mode": "vision",
2002
+ "action_mode": "post_velcmd_hand_binary"
2003
+ },
2004
+ {
2005
+ "hdf5_key": "ep_00199",
2006
+ "source_episode_index": 199,
2007
+ "split": "valid",
2008
+ "num_samples": 522,
2009
+ "action_dim": 7,
2010
+ "obs_mode": "vision",
2011
+ "action_mode": "post_velcmd_hand_binary"
2012
+ },
2013
+ {
2014
+ "hdf5_key": "ep_00200",
2015
+ "source_episode_index": 200,
2016
+ "split": "valid",
2017
+ "num_samples": 581,
2018
+ "action_dim": 7,
2019
+ "obs_mode": "vision",
2020
+ "action_mode": "post_velcmd_hand_binary"
2021
+ },
2022
+ {
2023
+ "hdf5_key": "ep_00201",
2024
+ "source_episode_index": 201,
2025
+ "split": "valid",
2026
+ "num_samples": 574,
2027
+ "action_dim": 7,
2028
+ "obs_mode": "vision",
2029
+ "action_mode": "post_velcmd_hand_binary"
2030
+ }
2031
+ ]
2032
+ }
vac_post_vis/last.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7769b00bc8f0cab6c58c1ce953c8262ec5f199745864625f9d29cc76b5238e33
3
+ size 1252378845
vac_post_vis_wrench/action_stats.json ADDED
@@ -0,0 +1,297 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "method": "min_max",
3
+ "output_min": -0.999999,
4
+ "output_max": 0.999999,
5
+ "action_mode": "post_velcmd_hand_binary",
6
+ "action_components": [
7
+ "vel_cmd_x",
8
+ "vel_cmd_y",
9
+ "vel_cmd_z",
10
+ "vel_cmd_rx",
11
+ "vel_cmd_ry",
12
+ "vel_cmd_rz",
13
+ "hand_open_binary"
14
+ ],
15
+ "num_frames": 124722,
16
+ "min": [
17
+ -44.80923843383789,
18
+ -78.85905456542969,
19
+ -272.0136413574219,
20
+ -41.36355972290039,
21
+ -4.46674108505249,
22
+ -50.74555969238281,
23
+ 0.0
24
+ ],
25
+ "max": [
26
+ 60.49591827392578,
27
+ 81.3885269165039,
28
+ 499.7237243652344,
29
+ 40.38335418701172,
30
+ 6.112980842590332,
31
+ 49.7142219543457,
32
+ 1.0
33
+ ],
34
+ "mean": [
35
+ 0.08373736590147018,
36
+ 6.001084804534912,
37
+ -2.738046646118164,
38
+ -0.005926239304244518,
39
+ -0.001013437518849969,
40
+ 0.04403281956911087,
41
+ 0.428376704454422
42
+ ],
43
+ "std": [
44
+ 7.180497646331787,
45
+ 26.540620803833008,
46
+ 46.47587966918945,
47
+ 0.791343629360199,
48
+ 0.21718677878379822,
49
+ 3.934755325317383,
50
+ 0.4948435127735138
51
+ ],
52
+ "scale": [
53
+ 52.65263366699219,
54
+ 80.1238784790039,
55
+ 385.86907958984375,
56
+ 40.87350082397461,
57
+ 5.289865970611572,
58
+ 50.22993850708008,
59
+ 0.500000536441803
60
+ ],
61
+ "offset": [
62
+ 7.843341827392578,
63
+ 1.264739990234375,
64
+ 113.85504150390625,
65
+ -0.4901008605957031,
66
+ 0.8231196403503418,
67
+ -0.5156707763671875,
68
+ 0.5
69
+ ],
70
+ "hand_source": "action.absolute",
71
+ "hand_active_dims": [
72
+ 8,
73
+ 9,
74
+ 10,
75
+ 11
76
+ ],
77
+ "hand_reduce": "mean",
78
+ "hand_threshold": 0.85,
79
+ "hand_high_value": 1.0,
80
+ "hand_low_value": 0.0,
81
+ "dataset": "/workspace/datasets/dataset_pipe_02_06",
82
+ "observation_mode_examples": [
83
+ "vision",
84
+ "vision_wrench",
85
+ "vision_wrench_state"
86
+ ],
87
+ "train_source_episodes": [
88
+ 0,
89
+ 1,
90
+ 2,
91
+ 3,
92
+ 4,
93
+ 5,
94
+ 6,
95
+ 7,
96
+ 8,
97
+ 9,
98
+ 10,
99
+ 11,
100
+ 12,
101
+ 13,
102
+ 14,
103
+ 15,
104
+ 16,
105
+ 17,
106
+ 18,
107
+ 19,
108
+ 20,
109
+ 21,
110
+ 22,
111
+ 23,
112
+ 24,
113
+ 25,
114
+ 26,
115
+ 27,
116
+ 28,
117
+ 29,
118
+ 30,
119
+ 31,
120
+ 32,
121
+ 33,
122
+ 34,
123
+ 35,
124
+ 36,
125
+ 37,
126
+ 38,
127
+ 39,
128
+ 40,
129
+ 41,
130
+ 42,
131
+ 43,
132
+ 44,
133
+ 45,
134
+ 46,
135
+ 47,
136
+ 48,
137
+ 49,
138
+ 50,
139
+ 51,
140
+ 52,
141
+ 53,
142
+ 54,
143
+ 55,
144
+ 56,
145
+ 57,
146
+ 58,
147
+ 59,
148
+ 60,
149
+ 61,
150
+ 62,
151
+ 63,
152
+ 64,
153
+ 65,
154
+ 66,
155
+ 67,
156
+ 68,
157
+ 69,
158
+ 70,
159
+ 71,
160
+ 72,
161
+ 73,
162
+ 74,
163
+ 75,
164
+ 76,
165
+ 77,
166
+ 78,
167
+ 79,
168
+ 80,
169
+ 81,
170
+ 82,
171
+ 83,
172
+ 84,
173
+ 85,
174
+ 86,
175
+ 87,
176
+ 88,
177
+ 89,
178
+ 90,
179
+ 91,
180
+ 92,
181
+ 93,
182
+ 94,
183
+ 95,
184
+ 96,
185
+ 97,
186
+ 98,
187
+ 99,
188
+ 100,
189
+ 101,
190
+ 102,
191
+ 103,
192
+ 104,
193
+ 105,
194
+ 106,
195
+ 107,
196
+ 108,
197
+ 109,
198
+ 110,
199
+ 111,
200
+ 112,
201
+ 113,
202
+ 114,
203
+ 115,
204
+ 116,
205
+ 117,
206
+ 118,
207
+ 119,
208
+ 120,
209
+ 121,
210
+ 122,
211
+ 123,
212
+ 124,
213
+ 125,
214
+ 126,
215
+ 127,
216
+ 128,
217
+ 129,
218
+ 130,
219
+ 131,
220
+ 132,
221
+ 133,
222
+ 134,
223
+ 135,
224
+ 136,
225
+ 137,
226
+ 138,
227
+ 139,
228
+ 140,
229
+ 141,
230
+ 142,
231
+ 143,
232
+ 144,
233
+ 145,
234
+ 146,
235
+ 147,
236
+ 148,
237
+ 149,
238
+ 150,
239
+ 151,
240
+ 152,
241
+ 153,
242
+ 154,
243
+ 155,
244
+ 156,
245
+ 157,
246
+ 158,
247
+ 159,
248
+ 160,
249
+ 161,
250
+ 162,
251
+ 163,
252
+ 164,
253
+ 165,
254
+ 166,
255
+ 167,
256
+ 168,
257
+ 169,
258
+ 170,
259
+ 171
260
+ ],
261
+ "valid_source_episodes": [
262
+ 172,
263
+ 173,
264
+ 174,
265
+ 175,
266
+ 176,
267
+ 177,
268
+ 178,
269
+ 179,
270
+ 180,
271
+ 181,
272
+ 182,
273
+ 183,
274
+ 184,
275
+ 185,
276
+ 186,
277
+ 187,
278
+ 188,
279
+ 189,
280
+ 190,
281
+ 191,
282
+ 192,
283
+ 193,
284
+ 194,
285
+ 195,
286
+ 196,
287
+ 197,
288
+ 198,
289
+ 199,
290
+ 200,
291
+ 201
292
+ ],
293
+ "image_size": [
294
+ 84,
295
+ 84
296
+ ]
297
+ }
vac_post_vis_wrench/best.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf806cd106fc738dba5fd663ed2edb8108225b18bb295ac823d5b75a52e73e64
3
+ size 1255236653
vac_post_vis_wrench/config.json ADDED
@@ -0,0 +1,197 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "algo_name": "diffusion_policy",
3
+ "experiment": {
4
+ "name": "vac_post_vis_wrench",
5
+ "validate": true,
6
+ "logging": {
7
+ "terminal_output_to_txt": false,
8
+ "log_tb": true,
9
+ "log_wandb": false,
10
+ "wandb_proj_name": "debug"
11
+ },
12
+ "save": {
13
+ "enabled": true,
14
+ "every_n_seconds": null,
15
+ "every_n_epochs": 50,
16
+ "epochs": [],
17
+ "on_best_validation": true,
18
+ "on_best_rollout_return": false,
19
+ "on_best_rollout_success_rate": false
20
+ },
21
+ "epoch_every_n_steps": 100,
22
+ "validation_epoch_every_n_steps": 10,
23
+ "env": null,
24
+ "additional_envs": null,
25
+ "render": false,
26
+ "render_video": false,
27
+ "keep_all_videos": false,
28
+ "video_skip": 5,
29
+ "rollout": {
30
+ "enabled": false,
31
+ "n": 0,
32
+ "horizon": 50,
33
+ "rate": 50,
34
+ "warmstart": 0,
35
+ "terminate_on_success": true
36
+ },
37
+ "env_meta_update_dict": {},
38
+ "ckpt_path": null
39
+ },
40
+ "train": {
41
+ "data": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_post_vis_wrench/vac_post_vis_wrench.hdf5",
42
+ "output_dir": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_post_vis_wrench/robomimic_runs",
43
+ "normalize_weights_by_ds_size": false,
44
+ "num_data_workers": 0,
45
+ "hdf5_cache_mode": "all",
46
+ "hdf5_use_swmr": true,
47
+ "hdf5_load_next_obs": false,
48
+ "hdf5_normalize_obs": false,
49
+ "hdf5_filter_key": "train",
50
+ "hdf5_validation_filter_key": "valid",
51
+ "seq_length": 8,
52
+ "pad_seq_length": true,
53
+ "frame_stack": 2,
54
+ "pad_frame_stack": true,
55
+ "dataset_keys": [
56
+ "actions",
57
+ "rewards",
58
+ "dones"
59
+ ],
60
+ "action_keys": [
61
+ "actions"
62
+ ],
63
+ "action_config": {
64
+ "actions": {
65
+ "normalization": "min_max"
66
+ }
67
+ },
68
+ "goal_mode": null,
69
+ "cuda": true,
70
+ "batch_size": 16,
71
+ "num_epochs": 300,
72
+ "seed": 42,
73
+ "max_grad_norm": null
74
+ },
75
+ "algo": {
76
+ "optim_params": {
77
+ "policy": {
78
+ "optimizer_type": "adamw",
79
+ "learning_rate": {
80
+ "initial": 0.0001,
81
+ "decay_factor": 0.1,
82
+ "step_every_batch": true,
83
+ "scheduler_type": "cosine",
84
+ "num_cycles": 0.5,
85
+ "warmup_steps": 500,
86
+ "epoch_schedule": []
87
+ },
88
+ "regularization": {
89
+ "L2": 1e-06
90
+ }
91
+ }
92
+ },
93
+ "horizon": {
94
+ "observation_horizon": 2,
95
+ "action_horizon": 4,
96
+ "prediction_horizon": 8
97
+ },
98
+ "unet": {
99
+ "enabled": true,
100
+ "diffusion_step_embed_dim": 256,
101
+ "down_dims": [
102
+ 128,
103
+ 256,
104
+ 512
105
+ ],
106
+ "kernel_size": 5,
107
+ "n_groups": 8
108
+ },
109
+ "ema": {
110
+ "enabled": true,
111
+ "power": 0.75
112
+ },
113
+ "ddpm": {
114
+ "enabled": true,
115
+ "num_train_timesteps": 50,
116
+ "num_inference_timesteps": 50,
117
+ "beta_schedule": "squaredcos_cap_v2",
118
+ "clip_sample": true,
119
+ "prediction_type": "epsilon"
120
+ },
121
+ "ddim": {
122
+ "enabled": false,
123
+ "num_train_timesteps": 50,
124
+ "num_inference_timesteps": 10,
125
+ "beta_schedule": "squaredcos_cap_v2",
126
+ "clip_sample": true,
127
+ "set_alpha_to_one": true,
128
+ "steps_offset": 0,
129
+ "prediction_type": "epsilon"
130
+ }
131
+ },
132
+ "observation": {
133
+ "modalities": {
134
+ "obs": {
135
+ "low_dim": [
136
+ "wrench"
137
+ ],
138
+ "rgb": [
139
+ "front_rgb"
140
+ ],
141
+ "depth": [],
142
+ "scan": []
143
+ },
144
+ "goal": {
145
+ "low_dim": [],
146
+ "rgb": [],
147
+ "depth": [],
148
+ "scan": []
149
+ }
150
+ },
151
+ "encoder": {
152
+ "low_dim": {
153
+ "core_class": null,
154
+ "core_kwargs": {},
155
+ "obs_randomizer_class": null,
156
+ "obs_randomizer_kwargs": {}
157
+ },
158
+ "rgb": {
159
+ "core_class": "VisualCore",
160
+ "core_kwargs": {
161
+ "backbone_class": "ResNet18Conv",
162
+ "backbone_kwargs": {
163
+ "pretrained": false,
164
+ "input_coord_conv": false
165
+ },
166
+ "pool_class": "SpatialSoftmax",
167
+ "pool_kwargs": {
168
+ "num_kp": 32,
169
+ "learnable_temperature": false,
170
+ "temperature": 1.0,
171
+ "noise_std": 0.0
172
+ },
173
+ "feature_dimension": 64
174
+ },
175
+ "obs_randomizer_class": null,
176
+ "obs_randomizer_kwargs": {}
177
+ },
178
+ "depth": {
179
+ "core_class": "VisualCore",
180
+ "core_kwargs": {},
181
+ "obs_randomizer_class": null,
182
+ "obs_randomizer_kwargs": {}
183
+ },
184
+ "scan": {
185
+ "core_class": "ScanCore",
186
+ "core_kwargs": {},
187
+ "obs_randomizer_class": null,
188
+ "obs_randomizer_kwargs": {}
189
+ }
190
+ }
191
+ },
192
+ "meta": {
193
+ "hp_base_config_file": null,
194
+ "hp_keys": [],
195
+ "hp_values": []
196
+ }
197
+ }
vac_post_vis_wrench/dataset_summary.json ADDED
@@ -0,0 +1,2032 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "path": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_post_vis_wrench/vac_post_vis_wrench.hdf5",
3
+ "total_frames": 145712,
4
+ "train_episodes": 172,
5
+ "valid_episodes": 30,
6
+ "train_source_episodes": [
7
+ 0,
8
+ 1,
9
+ 2,
10
+ 3,
11
+ 4,
12
+ 5,
13
+ 6,
14
+ 7,
15
+ 8,
16
+ 9,
17
+ 10,
18
+ 11,
19
+ 12,
20
+ 13,
21
+ 14,
22
+ 15,
23
+ 16,
24
+ 17,
25
+ 18,
26
+ 19,
27
+ 20,
28
+ 21,
29
+ 22,
30
+ 23,
31
+ 24,
32
+ 25,
33
+ 26,
34
+ 27,
35
+ 28,
36
+ 29,
37
+ 30,
38
+ 31,
39
+ 32,
40
+ 33,
41
+ 34,
42
+ 35,
43
+ 36,
44
+ 37,
45
+ 38,
46
+ 39,
47
+ 40,
48
+ 41,
49
+ 42,
50
+ 43,
51
+ 44,
52
+ 45,
53
+ 46,
54
+ 47,
55
+ 48,
56
+ 49,
57
+ 50,
58
+ 51,
59
+ 52,
60
+ 53,
61
+ 54,
62
+ 55,
63
+ 56,
64
+ 57,
65
+ 58,
66
+ 59,
67
+ 60,
68
+ 61,
69
+ 62,
70
+ 63,
71
+ 64,
72
+ 65,
73
+ 66,
74
+ 67,
75
+ 68,
76
+ 69,
77
+ 70,
78
+ 71,
79
+ 72,
80
+ 73,
81
+ 74,
82
+ 75,
83
+ 76,
84
+ 77,
85
+ 78,
86
+ 79,
87
+ 80,
88
+ 81,
89
+ 82,
90
+ 83,
91
+ 84,
92
+ 85,
93
+ 86,
94
+ 87,
95
+ 88,
96
+ 89,
97
+ 90,
98
+ 91,
99
+ 92,
100
+ 93,
101
+ 94,
102
+ 95,
103
+ 96,
104
+ 97,
105
+ 98,
106
+ 99,
107
+ 100,
108
+ 101,
109
+ 102,
110
+ 103,
111
+ 104,
112
+ 105,
113
+ 106,
114
+ 107,
115
+ 108,
116
+ 109,
117
+ 110,
118
+ 111,
119
+ 112,
120
+ 113,
121
+ 114,
122
+ 115,
123
+ 116,
124
+ 117,
125
+ 118,
126
+ 119,
127
+ 120,
128
+ 121,
129
+ 122,
130
+ 123,
131
+ 124,
132
+ 125,
133
+ 126,
134
+ 127,
135
+ 128,
136
+ 129,
137
+ 130,
138
+ 131,
139
+ 132,
140
+ 133,
141
+ 134,
142
+ 135,
143
+ 136,
144
+ 137,
145
+ 138,
146
+ 139,
147
+ 140,
148
+ 141,
149
+ 142,
150
+ 143,
151
+ 144,
152
+ 145,
153
+ 146,
154
+ 147,
155
+ 148,
156
+ 149,
157
+ 150,
158
+ 151,
159
+ 152,
160
+ 153,
161
+ 154,
162
+ 155,
163
+ 156,
164
+ 157,
165
+ 158,
166
+ 159,
167
+ 160,
168
+ 161,
169
+ 162,
170
+ 163,
171
+ 164,
172
+ 165,
173
+ 166,
174
+ 167,
175
+ 168,
176
+ 169,
177
+ 170,
178
+ 171
179
+ ],
180
+ "valid_source_episodes": [
181
+ 172,
182
+ 173,
183
+ 174,
184
+ 175,
185
+ 176,
186
+ 177,
187
+ 178,
188
+ 179,
189
+ 180,
190
+ 181,
191
+ 182,
192
+ 183,
193
+ 184,
194
+ 185,
195
+ 186,
196
+ 187,
197
+ 188,
198
+ 189,
199
+ 190,
200
+ 191,
201
+ 192,
202
+ 193,
203
+ 194,
204
+ 195,
205
+ 196,
206
+ 197,
207
+ 198,
208
+ 199,
209
+ 200,
210
+ 201
211
+ ],
212
+ "episodes": [
213
+ {
214
+ "hdf5_key": "ep_00000",
215
+ "source_episode_index": 0,
216
+ "split": "train",
217
+ "num_samples": 663,
218
+ "action_dim": 7,
219
+ "obs_mode": "vision_wrench",
220
+ "action_mode": "post_velcmd_hand_binary"
221
+ },
222
+ {
223
+ "hdf5_key": "ep_00001",
224
+ "source_episode_index": 1,
225
+ "split": "train",
226
+ "num_samples": 706,
227
+ "action_dim": 7,
228
+ "obs_mode": "vision_wrench",
229
+ "action_mode": "post_velcmd_hand_binary"
230
+ },
231
+ {
232
+ "hdf5_key": "ep_00002",
233
+ "source_episode_index": 2,
234
+ "split": "train",
235
+ "num_samples": 1006,
236
+ "action_dim": 7,
237
+ "obs_mode": "vision_wrench",
238
+ "action_mode": "post_velcmd_hand_binary"
239
+ },
240
+ {
241
+ "hdf5_key": "ep_00003",
242
+ "source_episode_index": 3,
243
+ "split": "train",
244
+ "num_samples": 709,
245
+ "action_dim": 7,
246
+ "obs_mode": "vision_wrench",
247
+ "action_mode": "post_velcmd_hand_binary"
248
+ },
249
+ {
250
+ "hdf5_key": "ep_00004",
251
+ "source_episode_index": 4,
252
+ "split": "train",
253
+ "num_samples": 816,
254
+ "action_dim": 7,
255
+ "obs_mode": "vision_wrench",
256
+ "action_mode": "post_velcmd_hand_binary"
257
+ },
258
+ {
259
+ "hdf5_key": "ep_00005",
260
+ "source_episode_index": 5,
261
+ "split": "train",
262
+ "num_samples": 882,
263
+ "action_dim": 7,
264
+ "obs_mode": "vision_wrench",
265
+ "action_mode": "post_velcmd_hand_binary"
266
+ },
267
+ {
268
+ "hdf5_key": "ep_00006",
269
+ "source_episode_index": 6,
270
+ "split": "train",
271
+ "num_samples": 699,
272
+ "action_dim": 7,
273
+ "obs_mode": "vision_wrench",
274
+ "action_mode": "post_velcmd_hand_binary"
275
+ },
276
+ {
277
+ "hdf5_key": "ep_00007",
278
+ "source_episode_index": 7,
279
+ "split": "train",
280
+ "num_samples": 851,
281
+ "action_dim": 7,
282
+ "obs_mode": "vision_wrench",
283
+ "action_mode": "post_velcmd_hand_binary"
284
+ },
285
+ {
286
+ "hdf5_key": "ep_00008",
287
+ "source_episode_index": 8,
288
+ "split": "train",
289
+ "num_samples": 728,
290
+ "action_dim": 7,
291
+ "obs_mode": "vision_wrench",
292
+ "action_mode": "post_velcmd_hand_binary"
293
+ },
294
+ {
295
+ "hdf5_key": "ep_00009",
296
+ "source_episode_index": 9,
297
+ "split": "train",
298
+ "num_samples": 770,
299
+ "action_dim": 7,
300
+ "obs_mode": "vision_wrench",
301
+ "action_mode": "post_velcmd_hand_binary"
302
+ },
303
+ {
304
+ "hdf5_key": "ep_00010",
305
+ "source_episode_index": 10,
306
+ "split": "train",
307
+ "num_samples": 821,
308
+ "action_dim": 7,
309
+ "obs_mode": "vision_wrench",
310
+ "action_mode": "post_velcmd_hand_binary"
311
+ },
312
+ {
313
+ "hdf5_key": "ep_00011",
314
+ "source_episode_index": 11,
315
+ "split": "train",
316
+ "num_samples": 561,
317
+ "action_dim": 7,
318
+ "obs_mode": "vision_wrench",
319
+ "action_mode": "post_velcmd_hand_binary"
320
+ },
321
+ {
322
+ "hdf5_key": "ep_00012",
323
+ "source_episode_index": 12,
324
+ "split": "train",
325
+ "num_samples": 517,
326
+ "action_dim": 7,
327
+ "obs_mode": "vision_wrench",
328
+ "action_mode": "post_velcmd_hand_binary"
329
+ },
330
+ {
331
+ "hdf5_key": "ep_00013",
332
+ "source_episode_index": 13,
333
+ "split": "train",
334
+ "num_samples": 690,
335
+ "action_dim": 7,
336
+ "obs_mode": "vision_wrench",
337
+ "action_mode": "post_velcmd_hand_binary"
338
+ },
339
+ {
340
+ "hdf5_key": "ep_00014",
341
+ "source_episode_index": 14,
342
+ "split": "train",
343
+ "num_samples": 559,
344
+ "action_dim": 7,
345
+ "obs_mode": "vision_wrench",
346
+ "action_mode": "post_velcmd_hand_binary"
347
+ },
348
+ {
349
+ "hdf5_key": "ep_00015",
350
+ "source_episode_index": 15,
351
+ "split": "train",
352
+ "num_samples": 889,
353
+ "action_dim": 7,
354
+ "obs_mode": "vision_wrench",
355
+ "action_mode": "post_velcmd_hand_binary"
356
+ },
357
+ {
358
+ "hdf5_key": "ep_00016",
359
+ "source_episode_index": 16,
360
+ "split": "train",
361
+ "num_samples": 619,
362
+ "action_dim": 7,
363
+ "obs_mode": "vision_wrench",
364
+ "action_mode": "post_velcmd_hand_binary"
365
+ },
366
+ {
367
+ "hdf5_key": "ep_00017",
368
+ "source_episode_index": 17,
369
+ "split": "train",
370
+ "num_samples": 640,
371
+ "action_dim": 7,
372
+ "obs_mode": "vision_wrench",
373
+ "action_mode": "post_velcmd_hand_binary"
374
+ },
375
+ {
376
+ "hdf5_key": "ep_00018",
377
+ "source_episode_index": 18,
378
+ "split": "train",
379
+ "num_samples": 612,
380
+ "action_dim": 7,
381
+ "obs_mode": "vision_wrench",
382
+ "action_mode": "post_velcmd_hand_binary"
383
+ },
384
+ {
385
+ "hdf5_key": "ep_00019",
386
+ "source_episode_index": 19,
387
+ "split": "train",
388
+ "num_samples": 702,
389
+ "action_dim": 7,
390
+ "obs_mode": "vision_wrench",
391
+ "action_mode": "post_velcmd_hand_binary"
392
+ },
393
+ {
394
+ "hdf5_key": "ep_00020",
395
+ "source_episode_index": 20,
396
+ "split": "train",
397
+ "num_samples": 645,
398
+ "action_dim": 7,
399
+ "obs_mode": "vision_wrench",
400
+ "action_mode": "post_velcmd_hand_binary"
401
+ },
402
+ {
403
+ "hdf5_key": "ep_00021",
404
+ "source_episode_index": 21,
405
+ "split": "train",
406
+ "num_samples": 751,
407
+ "action_dim": 7,
408
+ "obs_mode": "vision_wrench",
409
+ "action_mode": "post_velcmd_hand_binary"
410
+ },
411
+ {
412
+ "hdf5_key": "ep_00022",
413
+ "source_episode_index": 22,
414
+ "split": "train",
415
+ "num_samples": 810,
416
+ "action_dim": 7,
417
+ "obs_mode": "vision_wrench",
418
+ "action_mode": "post_velcmd_hand_binary"
419
+ },
420
+ {
421
+ "hdf5_key": "ep_00023",
422
+ "source_episode_index": 23,
423
+ "split": "train",
424
+ "num_samples": 734,
425
+ "action_dim": 7,
426
+ "obs_mode": "vision_wrench",
427
+ "action_mode": "post_velcmd_hand_binary"
428
+ },
429
+ {
430
+ "hdf5_key": "ep_00024",
431
+ "source_episode_index": 24,
432
+ "split": "train",
433
+ "num_samples": 637,
434
+ "action_dim": 7,
435
+ "obs_mode": "vision_wrench",
436
+ "action_mode": "post_velcmd_hand_binary"
437
+ },
438
+ {
439
+ "hdf5_key": "ep_00025",
440
+ "source_episode_index": 25,
441
+ "split": "train",
442
+ "num_samples": 640,
443
+ "action_dim": 7,
444
+ "obs_mode": "vision_wrench",
445
+ "action_mode": "post_velcmd_hand_binary"
446
+ },
447
+ {
448
+ "hdf5_key": "ep_00026",
449
+ "source_episode_index": 26,
450
+ "split": "train",
451
+ "num_samples": 957,
452
+ "action_dim": 7,
453
+ "obs_mode": "vision_wrench",
454
+ "action_mode": "post_velcmd_hand_binary"
455
+ },
456
+ {
457
+ "hdf5_key": "ep_00027",
458
+ "source_episode_index": 27,
459
+ "split": "train",
460
+ "num_samples": 617,
461
+ "action_dim": 7,
462
+ "obs_mode": "vision_wrench",
463
+ "action_mode": "post_velcmd_hand_binary"
464
+ },
465
+ {
466
+ "hdf5_key": "ep_00028",
467
+ "source_episode_index": 28,
468
+ "split": "train",
469
+ "num_samples": 772,
470
+ "action_dim": 7,
471
+ "obs_mode": "vision_wrench",
472
+ "action_mode": "post_velcmd_hand_binary"
473
+ },
474
+ {
475
+ "hdf5_key": "ep_00029",
476
+ "source_episode_index": 29,
477
+ "split": "train",
478
+ "num_samples": 616,
479
+ "action_dim": 7,
480
+ "obs_mode": "vision_wrench",
481
+ "action_mode": "post_velcmd_hand_binary"
482
+ },
483
+ {
484
+ "hdf5_key": "ep_00030",
485
+ "source_episode_index": 30,
486
+ "split": "train",
487
+ "num_samples": 732,
488
+ "action_dim": 7,
489
+ "obs_mode": "vision_wrench",
490
+ "action_mode": "post_velcmd_hand_binary"
491
+ },
492
+ {
493
+ "hdf5_key": "ep_00031",
494
+ "source_episode_index": 31,
495
+ "split": "train",
496
+ "num_samples": 761,
497
+ "action_dim": 7,
498
+ "obs_mode": "vision_wrench",
499
+ "action_mode": "post_velcmd_hand_binary"
500
+ },
501
+ {
502
+ "hdf5_key": "ep_00032",
503
+ "source_episode_index": 32,
504
+ "split": "train",
505
+ "num_samples": 804,
506
+ "action_dim": 7,
507
+ "obs_mode": "vision_wrench",
508
+ "action_mode": "post_velcmd_hand_binary"
509
+ },
510
+ {
511
+ "hdf5_key": "ep_00033",
512
+ "source_episode_index": 33,
513
+ "split": "train",
514
+ "num_samples": 735,
515
+ "action_dim": 7,
516
+ "obs_mode": "vision_wrench",
517
+ "action_mode": "post_velcmd_hand_binary"
518
+ },
519
+ {
520
+ "hdf5_key": "ep_00034",
521
+ "source_episode_index": 34,
522
+ "split": "train",
523
+ "num_samples": 721,
524
+ "action_dim": 7,
525
+ "obs_mode": "vision_wrench",
526
+ "action_mode": "post_velcmd_hand_binary"
527
+ },
528
+ {
529
+ "hdf5_key": "ep_00035",
530
+ "source_episode_index": 35,
531
+ "split": "train",
532
+ "num_samples": 612,
533
+ "action_dim": 7,
534
+ "obs_mode": "vision_wrench",
535
+ "action_mode": "post_velcmd_hand_binary"
536
+ },
537
+ {
538
+ "hdf5_key": "ep_00036",
539
+ "source_episode_index": 36,
540
+ "split": "train",
541
+ "num_samples": 685,
542
+ "action_dim": 7,
543
+ "obs_mode": "vision_wrench",
544
+ "action_mode": "post_velcmd_hand_binary"
545
+ },
546
+ {
547
+ "hdf5_key": "ep_00037",
548
+ "source_episode_index": 37,
549
+ "split": "train",
550
+ "num_samples": 685,
551
+ "action_dim": 7,
552
+ "obs_mode": "vision_wrench",
553
+ "action_mode": "post_velcmd_hand_binary"
554
+ },
555
+ {
556
+ "hdf5_key": "ep_00038",
557
+ "source_episode_index": 38,
558
+ "split": "train",
559
+ "num_samples": 663,
560
+ "action_dim": 7,
561
+ "obs_mode": "vision_wrench",
562
+ "action_mode": "post_velcmd_hand_binary"
563
+ },
564
+ {
565
+ "hdf5_key": "ep_00039",
566
+ "source_episode_index": 39,
567
+ "split": "train",
568
+ "num_samples": 855,
569
+ "action_dim": 7,
570
+ "obs_mode": "vision_wrench",
571
+ "action_mode": "post_velcmd_hand_binary"
572
+ },
573
+ {
574
+ "hdf5_key": "ep_00040",
575
+ "source_episode_index": 40,
576
+ "split": "train",
577
+ "num_samples": 810,
578
+ "action_dim": 7,
579
+ "obs_mode": "vision_wrench",
580
+ "action_mode": "post_velcmd_hand_binary"
581
+ },
582
+ {
583
+ "hdf5_key": "ep_00041",
584
+ "source_episode_index": 41,
585
+ "split": "train",
586
+ "num_samples": 655,
587
+ "action_dim": 7,
588
+ "obs_mode": "vision_wrench",
589
+ "action_mode": "post_velcmd_hand_binary"
590
+ },
591
+ {
592
+ "hdf5_key": "ep_00042",
593
+ "source_episode_index": 42,
594
+ "split": "train",
595
+ "num_samples": 946,
596
+ "action_dim": 7,
597
+ "obs_mode": "vision_wrench",
598
+ "action_mode": "post_velcmd_hand_binary"
599
+ },
600
+ {
601
+ "hdf5_key": "ep_00043",
602
+ "source_episode_index": 43,
603
+ "split": "train",
604
+ "num_samples": 703,
605
+ "action_dim": 7,
606
+ "obs_mode": "vision_wrench",
607
+ "action_mode": "post_velcmd_hand_binary"
608
+ },
609
+ {
610
+ "hdf5_key": "ep_00044",
611
+ "source_episode_index": 44,
612
+ "split": "train",
613
+ "num_samples": 606,
614
+ "action_dim": 7,
615
+ "obs_mode": "vision_wrench",
616
+ "action_mode": "post_velcmd_hand_binary"
617
+ },
618
+ {
619
+ "hdf5_key": "ep_00045",
620
+ "source_episode_index": 45,
621
+ "split": "train",
622
+ "num_samples": 914,
623
+ "action_dim": 7,
624
+ "obs_mode": "vision_wrench",
625
+ "action_mode": "post_velcmd_hand_binary"
626
+ },
627
+ {
628
+ "hdf5_key": "ep_00046",
629
+ "source_episode_index": 46,
630
+ "split": "train",
631
+ "num_samples": 949,
632
+ "action_dim": 7,
633
+ "obs_mode": "vision_wrench",
634
+ "action_mode": "post_velcmd_hand_binary"
635
+ },
636
+ {
637
+ "hdf5_key": "ep_00047",
638
+ "source_episode_index": 47,
639
+ "split": "train",
640
+ "num_samples": 818,
641
+ "action_dim": 7,
642
+ "obs_mode": "vision_wrench",
643
+ "action_mode": "post_velcmd_hand_binary"
644
+ },
645
+ {
646
+ "hdf5_key": "ep_00048",
647
+ "source_episode_index": 48,
648
+ "split": "train",
649
+ "num_samples": 913,
650
+ "action_dim": 7,
651
+ "obs_mode": "vision_wrench",
652
+ "action_mode": "post_velcmd_hand_binary"
653
+ },
654
+ {
655
+ "hdf5_key": "ep_00049",
656
+ "source_episode_index": 49,
657
+ "split": "train",
658
+ "num_samples": 904,
659
+ "action_dim": 7,
660
+ "obs_mode": "vision_wrench",
661
+ "action_mode": "post_velcmd_hand_binary"
662
+ },
663
+ {
664
+ "hdf5_key": "ep_00050",
665
+ "source_episode_index": 50,
666
+ "split": "train",
667
+ "num_samples": 923,
668
+ "action_dim": 7,
669
+ "obs_mode": "vision_wrench",
670
+ "action_mode": "post_velcmd_hand_binary"
671
+ },
672
+ {
673
+ "hdf5_key": "ep_00051",
674
+ "source_episode_index": 51,
675
+ "split": "train",
676
+ "num_samples": 824,
677
+ "action_dim": 7,
678
+ "obs_mode": "vision_wrench",
679
+ "action_mode": "post_velcmd_hand_binary"
680
+ },
681
+ {
682
+ "hdf5_key": "ep_00052",
683
+ "source_episode_index": 52,
684
+ "split": "train",
685
+ "num_samples": 777,
686
+ "action_dim": 7,
687
+ "obs_mode": "vision_wrench",
688
+ "action_mode": "post_velcmd_hand_binary"
689
+ },
690
+ {
691
+ "hdf5_key": "ep_00053",
692
+ "source_episode_index": 53,
693
+ "split": "train",
694
+ "num_samples": 714,
695
+ "action_dim": 7,
696
+ "obs_mode": "vision_wrench",
697
+ "action_mode": "post_velcmd_hand_binary"
698
+ },
699
+ {
700
+ "hdf5_key": "ep_00054",
701
+ "source_episode_index": 54,
702
+ "split": "train",
703
+ "num_samples": 771,
704
+ "action_dim": 7,
705
+ "obs_mode": "vision_wrench",
706
+ "action_mode": "post_velcmd_hand_binary"
707
+ },
708
+ {
709
+ "hdf5_key": "ep_00055",
710
+ "source_episode_index": 55,
711
+ "split": "train",
712
+ "num_samples": 684,
713
+ "action_dim": 7,
714
+ "obs_mode": "vision_wrench",
715
+ "action_mode": "post_velcmd_hand_binary"
716
+ },
717
+ {
718
+ "hdf5_key": "ep_00056",
719
+ "source_episode_index": 56,
720
+ "split": "train",
721
+ "num_samples": 766,
722
+ "action_dim": 7,
723
+ "obs_mode": "vision_wrench",
724
+ "action_mode": "post_velcmd_hand_binary"
725
+ },
726
+ {
727
+ "hdf5_key": "ep_00057",
728
+ "source_episode_index": 57,
729
+ "split": "train",
730
+ "num_samples": 890,
731
+ "action_dim": 7,
732
+ "obs_mode": "vision_wrench",
733
+ "action_mode": "post_velcmd_hand_binary"
734
+ },
735
+ {
736
+ "hdf5_key": "ep_00058",
737
+ "source_episode_index": 58,
738
+ "split": "train",
739
+ "num_samples": 774,
740
+ "action_dim": 7,
741
+ "obs_mode": "vision_wrench",
742
+ "action_mode": "post_velcmd_hand_binary"
743
+ },
744
+ {
745
+ "hdf5_key": "ep_00059",
746
+ "source_episode_index": 59,
747
+ "split": "train",
748
+ "num_samples": 781,
749
+ "action_dim": 7,
750
+ "obs_mode": "vision_wrench",
751
+ "action_mode": "post_velcmd_hand_binary"
752
+ },
753
+ {
754
+ "hdf5_key": "ep_00060",
755
+ "source_episode_index": 60,
756
+ "split": "train",
757
+ "num_samples": 1093,
758
+ "action_dim": 7,
759
+ "obs_mode": "vision_wrench",
760
+ "action_mode": "post_velcmd_hand_binary"
761
+ },
762
+ {
763
+ "hdf5_key": "ep_00061",
764
+ "source_episode_index": 61,
765
+ "split": "train",
766
+ "num_samples": 900,
767
+ "action_dim": 7,
768
+ "obs_mode": "vision_wrench",
769
+ "action_mode": "post_velcmd_hand_binary"
770
+ },
771
+ {
772
+ "hdf5_key": "ep_00062",
773
+ "source_episode_index": 62,
774
+ "split": "train",
775
+ "num_samples": 814,
776
+ "action_dim": 7,
777
+ "obs_mode": "vision_wrench",
778
+ "action_mode": "post_velcmd_hand_binary"
779
+ },
780
+ {
781
+ "hdf5_key": "ep_00063",
782
+ "source_episode_index": 63,
783
+ "split": "train",
784
+ "num_samples": 1108,
785
+ "action_dim": 7,
786
+ "obs_mode": "vision_wrench",
787
+ "action_mode": "post_velcmd_hand_binary"
788
+ },
789
+ {
790
+ "hdf5_key": "ep_00064",
791
+ "source_episode_index": 64,
792
+ "split": "train",
793
+ "num_samples": 816,
794
+ "action_dim": 7,
795
+ "obs_mode": "vision_wrench",
796
+ "action_mode": "post_velcmd_hand_binary"
797
+ },
798
+ {
799
+ "hdf5_key": "ep_00065",
800
+ "source_episode_index": 65,
801
+ "split": "train",
802
+ "num_samples": 963,
803
+ "action_dim": 7,
804
+ "obs_mode": "vision_wrench",
805
+ "action_mode": "post_velcmd_hand_binary"
806
+ },
807
+ {
808
+ "hdf5_key": "ep_00066",
809
+ "source_episode_index": 66,
810
+ "split": "train",
811
+ "num_samples": 895,
812
+ "action_dim": 7,
813
+ "obs_mode": "vision_wrench",
814
+ "action_mode": "post_velcmd_hand_binary"
815
+ },
816
+ {
817
+ "hdf5_key": "ep_00067",
818
+ "source_episode_index": 67,
819
+ "split": "train",
820
+ "num_samples": 572,
821
+ "action_dim": 7,
822
+ "obs_mode": "vision_wrench",
823
+ "action_mode": "post_velcmd_hand_binary"
824
+ },
825
+ {
826
+ "hdf5_key": "ep_00068",
827
+ "source_episode_index": 68,
828
+ "split": "train",
829
+ "num_samples": 648,
830
+ "action_dim": 7,
831
+ "obs_mode": "vision_wrench",
832
+ "action_mode": "post_velcmd_hand_binary"
833
+ },
834
+ {
835
+ "hdf5_key": "ep_00069",
836
+ "source_episode_index": 69,
837
+ "split": "train",
838
+ "num_samples": 772,
839
+ "action_dim": 7,
840
+ "obs_mode": "vision_wrench",
841
+ "action_mode": "post_velcmd_hand_binary"
842
+ },
843
+ {
844
+ "hdf5_key": "ep_00070",
845
+ "source_episode_index": 70,
846
+ "split": "train",
847
+ "num_samples": 702,
848
+ "action_dim": 7,
849
+ "obs_mode": "vision_wrench",
850
+ "action_mode": "post_velcmd_hand_binary"
851
+ },
852
+ {
853
+ "hdf5_key": "ep_00071",
854
+ "source_episode_index": 71,
855
+ "split": "train",
856
+ "num_samples": 786,
857
+ "action_dim": 7,
858
+ "obs_mode": "vision_wrench",
859
+ "action_mode": "post_velcmd_hand_binary"
860
+ },
861
+ {
862
+ "hdf5_key": "ep_00072",
863
+ "source_episode_index": 72,
864
+ "split": "train",
865
+ "num_samples": 843,
866
+ "action_dim": 7,
867
+ "obs_mode": "vision_wrench",
868
+ "action_mode": "post_velcmd_hand_binary"
869
+ },
870
+ {
871
+ "hdf5_key": "ep_00073",
872
+ "source_episode_index": 73,
873
+ "split": "train",
874
+ "num_samples": 767,
875
+ "action_dim": 7,
876
+ "obs_mode": "vision_wrench",
877
+ "action_mode": "post_velcmd_hand_binary"
878
+ },
879
+ {
880
+ "hdf5_key": "ep_00074",
881
+ "source_episode_index": 74,
882
+ "split": "train",
883
+ "num_samples": 764,
884
+ "action_dim": 7,
885
+ "obs_mode": "vision_wrench",
886
+ "action_mode": "post_velcmd_hand_binary"
887
+ },
888
+ {
889
+ "hdf5_key": "ep_00075",
890
+ "source_episode_index": 75,
891
+ "split": "train",
892
+ "num_samples": 750,
893
+ "action_dim": 7,
894
+ "obs_mode": "vision_wrench",
895
+ "action_mode": "post_velcmd_hand_binary"
896
+ },
897
+ {
898
+ "hdf5_key": "ep_00076",
899
+ "source_episode_index": 76,
900
+ "split": "train",
901
+ "num_samples": 652,
902
+ "action_dim": 7,
903
+ "obs_mode": "vision_wrench",
904
+ "action_mode": "post_velcmd_hand_binary"
905
+ },
906
+ {
907
+ "hdf5_key": "ep_00077",
908
+ "source_episode_index": 77,
909
+ "split": "train",
910
+ "num_samples": 812,
911
+ "action_dim": 7,
912
+ "obs_mode": "vision_wrench",
913
+ "action_mode": "post_velcmd_hand_binary"
914
+ },
915
+ {
916
+ "hdf5_key": "ep_00078",
917
+ "source_episode_index": 78,
918
+ "split": "train",
919
+ "num_samples": 722,
920
+ "action_dim": 7,
921
+ "obs_mode": "vision_wrench",
922
+ "action_mode": "post_velcmd_hand_binary"
923
+ },
924
+ {
925
+ "hdf5_key": "ep_00079",
926
+ "source_episode_index": 79,
927
+ "split": "train",
928
+ "num_samples": 723,
929
+ "action_dim": 7,
930
+ "obs_mode": "vision_wrench",
931
+ "action_mode": "post_velcmd_hand_binary"
932
+ },
933
+ {
934
+ "hdf5_key": "ep_00080",
935
+ "source_episode_index": 80,
936
+ "split": "train",
937
+ "num_samples": 765,
938
+ "action_dim": 7,
939
+ "obs_mode": "vision_wrench",
940
+ "action_mode": "post_velcmd_hand_binary"
941
+ },
942
+ {
943
+ "hdf5_key": "ep_00081",
944
+ "source_episode_index": 81,
945
+ "split": "train",
946
+ "num_samples": 981,
947
+ "action_dim": 7,
948
+ "obs_mode": "vision_wrench",
949
+ "action_mode": "post_velcmd_hand_binary"
950
+ },
951
+ {
952
+ "hdf5_key": "ep_00082",
953
+ "source_episode_index": 82,
954
+ "split": "train",
955
+ "num_samples": 959,
956
+ "action_dim": 7,
957
+ "obs_mode": "vision_wrench",
958
+ "action_mode": "post_velcmd_hand_binary"
959
+ },
960
+ {
961
+ "hdf5_key": "ep_00083",
962
+ "source_episode_index": 83,
963
+ "split": "train",
964
+ "num_samples": 655,
965
+ "action_dim": 7,
966
+ "obs_mode": "vision_wrench",
967
+ "action_mode": "post_velcmd_hand_binary"
968
+ },
969
+ {
970
+ "hdf5_key": "ep_00084",
971
+ "source_episode_index": 84,
972
+ "split": "train",
973
+ "num_samples": 746,
974
+ "action_dim": 7,
975
+ "obs_mode": "vision_wrench",
976
+ "action_mode": "post_velcmd_hand_binary"
977
+ },
978
+ {
979
+ "hdf5_key": "ep_00085",
980
+ "source_episode_index": 85,
981
+ "split": "train",
982
+ "num_samples": 660,
983
+ "action_dim": 7,
984
+ "obs_mode": "vision_wrench",
985
+ "action_mode": "post_velcmd_hand_binary"
986
+ },
987
+ {
988
+ "hdf5_key": "ep_00086",
989
+ "source_episode_index": 86,
990
+ "split": "train",
991
+ "num_samples": 617,
992
+ "action_dim": 7,
993
+ "obs_mode": "vision_wrench",
994
+ "action_mode": "post_velcmd_hand_binary"
995
+ },
996
+ {
997
+ "hdf5_key": "ep_00087",
998
+ "source_episode_index": 87,
999
+ "split": "train",
1000
+ "num_samples": 819,
1001
+ "action_dim": 7,
1002
+ "obs_mode": "vision_wrench",
1003
+ "action_mode": "post_velcmd_hand_binary"
1004
+ },
1005
+ {
1006
+ "hdf5_key": "ep_00088",
1007
+ "source_episode_index": 88,
1008
+ "split": "train",
1009
+ "num_samples": 696,
1010
+ "action_dim": 7,
1011
+ "obs_mode": "vision_wrench",
1012
+ "action_mode": "post_velcmd_hand_binary"
1013
+ },
1014
+ {
1015
+ "hdf5_key": "ep_00089",
1016
+ "source_episode_index": 89,
1017
+ "split": "train",
1018
+ "num_samples": 855,
1019
+ "action_dim": 7,
1020
+ "obs_mode": "vision_wrench",
1021
+ "action_mode": "post_velcmd_hand_binary"
1022
+ },
1023
+ {
1024
+ "hdf5_key": "ep_00090",
1025
+ "source_episode_index": 90,
1026
+ "split": "train",
1027
+ "num_samples": 678,
1028
+ "action_dim": 7,
1029
+ "obs_mode": "vision_wrench",
1030
+ "action_mode": "post_velcmd_hand_binary"
1031
+ },
1032
+ {
1033
+ "hdf5_key": "ep_00091",
1034
+ "source_episode_index": 91,
1035
+ "split": "train",
1036
+ "num_samples": 711,
1037
+ "action_dim": 7,
1038
+ "obs_mode": "vision_wrench",
1039
+ "action_mode": "post_velcmd_hand_binary"
1040
+ },
1041
+ {
1042
+ "hdf5_key": "ep_00092",
1043
+ "source_episode_index": 92,
1044
+ "split": "train",
1045
+ "num_samples": 763,
1046
+ "action_dim": 7,
1047
+ "obs_mode": "vision_wrench",
1048
+ "action_mode": "post_velcmd_hand_binary"
1049
+ },
1050
+ {
1051
+ "hdf5_key": "ep_00093",
1052
+ "source_episode_index": 93,
1053
+ "split": "train",
1054
+ "num_samples": 723,
1055
+ "action_dim": 7,
1056
+ "obs_mode": "vision_wrench",
1057
+ "action_mode": "post_velcmd_hand_binary"
1058
+ },
1059
+ {
1060
+ "hdf5_key": "ep_00094",
1061
+ "source_episode_index": 94,
1062
+ "split": "train",
1063
+ "num_samples": 723,
1064
+ "action_dim": 7,
1065
+ "obs_mode": "vision_wrench",
1066
+ "action_mode": "post_velcmd_hand_binary"
1067
+ },
1068
+ {
1069
+ "hdf5_key": "ep_00095",
1070
+ "source_episode_index": 95,
1071
+ "split": "train",
1072
+ "num_samples": 865,
1073
+ "action_dim": 7,
1074
+ "obs_mode": "vision_wrench",
1075
+ "action_mode": "post_velcmd_hand_binary"
1076
+ },
1077
+ {
1078
+ "hdf5_key": "ep_00096",
1079
+ "source_episode_index": 96,
1080
+ "split": "train",
1081
+ "num_samples": 495,
1082
+ "action_dim": 7,
1083
+ "obs_mode": "vision_wrench",
1084
+ "action_mode": "post_velcmd_hand_binary"
1085
+ },
1086
+ {
1087
+ "hdf5_key": "ep_00097",
1088
+ "source_episode_index": 97,
1089
+ "split": "train",
1090
+ "num_samples": 451,
1091
+ "action_dim": 7,
1092
+ "obs_mode": "vision_wrench",
1093
+ "action_mode": "post_velcmd_hand_binary"
1094
+ },
1095
+ {
1096
+ "hdf5_key": "ep_00098",
1097
+ "source_episode_index": 98,
1098
+ "split": "train",
1099
+ "num_samples": 690,
1100
+ "action_dim": 7,
1101
+ "obs_mode": "vision_wrench",
1102
+ "action_mode": "post_velcmd_hand_binary"
1103
+ },
1104
+ {
1105
+ "hdf5_key": "ep_00099",
1106
+ "source_episode_index": 99,
1107
+ "split": "train",
1108
+ "num_samples": 690,
1109
+ "action_dim": 7,
1110
+ "obs_mode": "vision_wrench",
1111
+ "action_mode": "post_velcmd_hand_binary"
1112
+ },
1113
+ {
1114
+ "hdf5_key": "ep_00100",
1115
+ "source_episode_index": 100,
1116
+ "split": "train",
1117
+ "num_samples": 615,
1118
+ "action_dim": 7,
1119
+ "obs_mode": "vision_wrench",
1120
+ "action_mode": "post_velcmd_hand_binary"
1121
+ },
1122
+ {
1123
+ "hdf5_key": "ep_00101",
1124
+ "source_episode_index": 101,
1125
+ "split": "train",
1126
+ "num_samples": 662,
1127
+ "action_dim": 7,
1128
+ "obs_mode": "vision_wrench",
1129
+ "action_mode": "post_velcmd_hand_binary"
1130
+ },
1131
+ {
1132
+ "hdf5_key": "ep_00102",
1133
+ "source_episode_index": 102,
1134
+ "split": "train",
1135
+ "num_samples": 614,
1136
+ "action_dim": 7,
1137
+ "obs_mode": "vision_wrench",
1138
+ "action_mode": "post_velcmd_hand_binary"
1139
+ },
1140
+ {
1141
+ "hdf5_key": "ep_00103",
1142
+ "source_episode_index": 103,
1143
+ "split": "train",
1144
+ "num_samples": 730,
1145
+ "action_dim": 7,
1146
+ "obs_mode": "vision_wrench",
1147
+ "action_mode": "post_velcmd_hand_binary"
1148
+ },
1149
+ {
1150
+ "hdf5_key": "ep_00104",
1151
+ "source_episode_index": 104,
1152
+ "split": "train",
1153
+ "num_samples": 683,
1154
+ "action_dim": 7,
1155
+ "obs_mode": "vision_wrench",
1156
+ "action_mode": "post_velcmd_hand_binary"
1157
+ },
1158
+ {
1159
+ "hdf5_key": "ep_00105",
1160
+ "source_episode_index": 105,
1161
+ "split": "train",
1162
+ "num_samples": 602,
1163
+ "action_dim": 7,
1164
+ "obs_mode": "vision_wrench",
1165
+ "action_mode": "post_velcmd_hand_binary"
1166
+ },
1167
+ {
1168
+ "hdf5_key": "ep_00106",
1169
+ "source_episode_index": 106,
1170
+ "split": "train",
1171
+ "num_samples": 652,
1172
+ "action_dim": 7,
1173
+ "obs_mode": "vision_wrench",
1174
+ "action_mode": "post_velcmd_hand_binary"
1175
+ },
1176
+ {
1177
+ "hdf5_key": "ep_00107",
1178
+ "source_episode_index": 107,
1179
+ "split": "train",
1180
+ "num_samples": 722,
1181
+ "action_dim": 7,
1182
+ "obs_mode": "vision_wrench",
1183
+ "action_mode": "post_velcmd_hand_binary"
1184
+ },
1185
+ {
1186
+ "hdf5_key": "ep_00108",
1187
+ "source_episode_index": 108,
1188
+ "split": "train",
1189
+ "num_samples": 605,
1190
+ "action_dim": 7,
1191
+ "obs_mode": "vision_wrench",
1192
+ "action_mode": "post_velcmd_hand_binary"
1193
+ },
1194
+ {
1195
+ "hdf5_key": "ep_00109",
1196
+ "source_episode_index": 109,
1197
+ "split": "train",
1198
+ "num_samples": 944,
1199
+ "action_dim": 7,
1200
+ "obs_mode": "vision_wrench",
1201
+ "action_mode": "post_velcmd_hand_binary"
1202
+ },
1203
+ {
1204
+ "hdf5_key": "ep_00110",
1205
+ "source_episode_index": 110,
1206
+ "split": "train",
1207
+ "num_samples": 661,
1208
+ "action_dim": 7,
1209
+ "obs_mode": "vision_wrench",
1210
+ "action_mode": "post_velcmd_hand_binary"
1211
+ },
1212
+ {
1213
+ "hdf5_key": "ep_00111",
1214
+ "source_episode_index": 111,
1215
+ "split": "train",
1216
+ "num_samples": 644,
1217
+ "action_dim": 7,
1218
+ "obs_mode": "vision_wrench",
1219
+ "action_mode": "post_velcmd_hand_binary"
1220
+ },
1221
+ {
1222
+ "hdf5_key": "ep_00112",
1223
+ "source_episode_index": 112,
1224
+ "split": "train",
1225
+ "num_samples": 614,
1226
+ "action_dim": 7,
1227
+ "obs_mode": "vision_wrench",
1228
+ "action_mode": "post_velcmd_hand_binary"
1229
+ },
1230
+ {
1231
+ "hdf5_key": "ep_00113",
1232
+ "source_episode_index": 113,
1233
+ "split": "train",
1234
+ "num_samples": 548,
1235
+ "action_dim": 7,
1236
+ "obs_mode": "vision_wrench",
1237
+ "action_mode": "post_velcmd_hand_binary"
1238
+ },
1239
+ {
1240
+ "hdf5_key": "ep_00114",
1241
+ "source_episode_index": 114,
1242
+ "split": "train",
1243
+ "num_samples": 568,
1244
+ "action_dim": 7,
1245
+ "obs_mode": "vision_wrench",
1246
+ "action_mode": "post_velcmd_hand_binary"
1247
+ },
1248
+ {
1249
+ "hdf5_key": "ep_00115",
1250
+ "source_episode_index": 115,
1251
+ "split": "train",
1252
+ "num_samples": 730,
1253
+ "action_dim": 7,
1254
+ "obs_mode": "vision_wrench",
1255
+ "action_mode": "post_velcmd_hand_binary"
1256
+ },
1257
+ {
1258
+ "hdf5_key": "ep_00116",
1259
+ "source_episode_index": 116,
1260
+ "split": "train",
1261
+ "num_samples": 481,
1262
+ "action_dim": 7,
1263
+ "obs_mode": "vision_wrench",
1264
+ "action_mode": "post_velcmd_hand_binary"
1265
+ },
1266
+ {
1267
+ "hdf5_key": "ep_00117",
1268
+ "source_episode_index": 117,
1269
+ "split": "train",
1270
+ "num_samples": 687,
1271
+ "action_dim": 7,
1272
+ "obs_mode": "vision_wrench",
1273
+ "action_mode": "post_velcmd_hand_binary"
1274
+ },
1275
+ {
1276
+ "hdf5_key": "ep_00118",
1277
+ "source_episode_index": 118,
1278
+ "split": "train",
1279
+ "num_samples": 618,
1280
+ "action_dim": 7,
1281
+ "obs_mode": "vision_wrench",
1282
+ "action_mode": "post_velcmd_hand_binary"
1283
+ },
1284
+ {
1285
+ "hdf5_key": "ep_00119",
1286
+ "source_episode_index": 119,
1287
+ "split": "train",
1288
+ "num_samples": 768,
1289
+ "action_dim": 7,
1290
+ "obs_mode": "vision_wrench",
1291
+ "action_mode": "post_velcmd_hand_binary"
1292
+ },
1293
+ {
1294
+ "hdf5_key": "ep_00120",
1295
+ "source_episode_index": 120,
1296
+ "split": "train",
1297
+ "num_samples": 734,
1298
+ "action_dim": 7,
1299
+ "obs_mode": "vision_wrench",
1300
+ "action_mode": "post_velcmd_hand_binary"
1301
+ },
1302
+ {
1303
+ "hdf5_key": "ep_00121",
1304
+ "source_episode_index": 121,
1305
+ "split": "train",
1306
+ "num_samples": 604,
1307
+ "action_dim": 7,
1308
+ "obs_mode": "vision_wrench",
1309
+ "action_mode": "post_velcmd_hand_binary"
1310
+ },
1311
+ {
1312
+ "hdf5_key": "ep_00122",
1313
+ "source_episode_index": 122,
1314
+ "split": "train",
1315
+ "num_samples": 610,
1316
+ "action_dim": 7,
1317
+ "obs_mode": "vision_wrench",
1318
+ "action_mode": "post_velcmd_hand_binary"
1319
+ },
1320
+ {
1321
+ "hdf5_key": "ep_00123",
1322
+ "source_episode_index": 123,
1323
+ "split": "train",
1324
+ "num_samples": 602,
1325
+ "action_dim": 7,
1326
+ "obs_mode": "vision_wrench",
1327
+ "action_mode": "post_velcmd_hand_binary"
1328
+ },
1329
+ {
1330
+ "hdf5_key": "ep_00124",
1331
+ "source_episode_index": 124,
1332
+ "split": "train",
1333
+ "num_samples": 630,
1334
+ "action_dim": 7,
1335
+ "obs_mode": "vision_wrench",
1336
+ "action_mode": "post_velcmd_hand_binary"
1337
+ },
1338
+ {
1339
+ "hdf5_key": "ep_00125",
1340
+ "source_episode_index": 125,
1341
+ "split": "train",
1342
+ "num_samples": 621,
1343
+ "action_dim": 7,
1344
+ "obs_mode": "vision_wrench",
1345
+ "action_mode": "post_velcmd_hand_binary"
1346
+ },
1347
+ {
1348
+ "hdf5_key": "ep_00126",
1349
+ "source_episode_index": 126,
1350
+ "split": "train",
1351
+ "num_samples": 600,
1352
+ "action_dim": 7,
1353
+ "obs_mode": "vision_wrench",
1354
+ "action_mode": "post_velcmd_hand_binary"
1355
+ },
1356
+ {
1357
+ "hdf5_key": "ep_00127",
1358
+ "source_episode_index": 127,
1359
+ "split": "train",
1360
+ "num_samples": 478,
1361
+ "action_dim": 7,
1362
+ "obs_mode": "vision_wrench",
1363
+ "action_mode": "post_velcmd_hand_binary"
1364
+ },
1365
+ {
1366
+ "hdf5_key": "ep_00128",
1367
+ "source_episode_index": 128,
1368
+ "split": "train",
1369
+ "num_samples": 606,
1370
+ "action_dim": 7,
1371
+ "obs_mode": "vision_wrench",
1372
+ "action_mode": "post_velcmd_hand_binary"
1373
+ },
1374
+ {
1375
+ "hdf5_key": "ep_00129",
1376
+ "source_episode_index": 129,
1377
+ "split": "train",
1378
+ "num_samples": 666,
1379
+ "action_dim": 7,
1380
+ "obs_mode": "vision_wrench",
1381
+ "action_mode": "post_velcmd_hand_binary"
1382
+ },
1383
+ {
1384
+ "hdf5_key": "ep_00130",
1385
+ "source_episode_index": 130,
1386
+ "split": "train",
1387
+ "num_samples": 655,
1388
+ "action_dim": 7,
1389
+ "obs_mode": "vision_wrench",
1390
+ "action_mode": "post_velcmd_hand_binary"
1391
+ },
1392
+ {
1393
+ "hdf5_key": "ep_00131",
1394
+ "source_episode_index": 131,
1395
+ "split": "train",
1396
+ "num_samples": 588,
1397
+ "action_dim": 7,
1398
+ "obs_mode": "vision_wrench",
1399
+ "action_mode": "post_velcmd_hand_binary"
1400
+ },
1401
+ {
1402
+ "hdf5_key": "ep_00132",
1403
+ "source_episode_index": 132,
1404
+ "split": "train",
1405
+ "num_samples": 806,
1406
+ "action_dim": 7,
1407
+ "obs_mode": "vision_wrench",
1408
+ "action_mode": "post_velcmd_hand_binary"
1409
+ },
1410
+ {
1411
+ "hdf5_key": "ep_00133",
1412
+ "source_episode_index": 133,
1413
+ "split": "train",
1414
+ "num_samples": 586,
1415
+ "action_dim": 7,
1416
+ "obs_mode": "vision_wrench",
1417
+ "action_mode": "post_velcmd_hand_binary"
1418
+ },
1419
+ {
1420
+ "hdf5_key": "ep_00134",
1421
+ "source_episode_index": 134,
1422
+ "split": "train",
1423
+ "num_samples": 714,
1424
+ "action_dim": 7,
1425
+ "obs_mode": "vision_wrench",
1426
+ "action_mode": "post_velcmd_hand_binary"
1427
+ },
1428
+ {
1429
+ "hdf5_key": "ep_00135",
1430
+ "source_episode_index": 135,
1431
+ "split": "train",
1432
+ "num_samples": 574,
1433
+ "action_dim": 7,
1434
+ "obs_mode": "vision_wrench",
1435
+ "action_mode": "post_velcmd_hand_binary"
1436
+ },
1437
+ {
1438
+ "hdf5_key": "ep_00136",
1439
+ "source_episode_index": 136,
1440
+ "split": "train",
1441
+ "num_samples": 577,
1442
+ "action_dim": 7,
1443
+ "obs_mode": "vision_wrench",
1444
+ "action_mode": "post_velcmd_hand_binary"
1445
+ },
1446
+ {
1447
+ "hdf5_key": "ep_00137",
1448
+ "source_episode_index": 137,
1449
+ "split": "train",
1450
+ "num_samples": 663,
1451
+ "action_dim": 7,
1452
+ "obs_mode": "vision_wrench",
1453
+ "action_mode": "post_velcmd_hand_binary"
1454
+ },
1455
+ {
1456
+ "hdf5_key": "ep_00138",
1457
+ "source_episode_index": 138,
1458
+ "split": "train",
1459
+ "num_samples": 720,
1460
+ "action_dim": 7,
1461
+ "obs_mode": "vision_wrench",
1462
+ "action_mode": "post_velcmd_hand_binary"
1463
+ },
1464
+ {
1465
+ "hdf5_key": "ep_00139",
1466
+ "source_episode_index": 139,
1467
+ "split": "train",
1468
+ "num_samples": 640,
1469
+ "action_dim": 7,
1470
+ "obs_mode": "vision_wrench",
1471
+ "action_mode": "post_velcmd_hand_binary"
1472
+ },
1473
+ {
1474
+ "hdf5_key": "ep_00140",
1475
+ "source_episode_index": 140,
1476
+ "split": "train",
1477
+ "num_samples": 544,
1478
+ "action_dim": 7,
1479
+ "obs_mode": "vision_wrench",
1480
+ "action_mode": "post_velcmd_hand_binary"
1481
+ },
1482
+ {
1483
+ "hdf5_key": "ep_00141",
1484
+ "source_episode_index": 141,
1485
+ "split": "train",
1486
+ "num_samples": 539,
1487
+ "action_dim": 7,
1488
+ "obs_mode": "vision_wrench",
1489
+ "action_mode": "post_velcmd_hand_binary"
1490
+ },
1491
+ {
1492
+ "hdf5_key": "ep_00142",
1493
+ "source_episode_index": 142,
1494
+ "split": "train",
1495
+ "num_samples": 592,
1496
+ "action_dim": 7,
1497
+ "obs_mode": "vision_wrench",
1498
+ "action_mode": "post_velcmd_hand_binary"
1499
+ },
1500
+ {
1501
+ "hdf5_key": "ep_00143",
1502
+ "source_episode_index": 143,
1503
+ "split": "train",
1504
+ "num_samples": 625,
1505
+ "action_dim": 7,
1506
+ "obs_mode": "vision_wrench",
1507
+ "action_mode": "post_velcmd_hand_binary"
1508
+ },
1509
+ {
1510
+ "hdf5_key": "ep_00144",
1511
+ "source_episode_index": 144,
1512
+ "split": "train",
1513
+ "num_samples": 684,
1514
+ "action_dim": 7,
1515
+ "obs_mode": "vision_wrench",
1516
+ "action_mode": "post_velcmd_hand_binary"
1517
+ },
1518
+ {
1519
+ "hdf5_key": "ep_00145",
1520
+ "source_episode_index": 145,
1521
+ "split": "train",
1522
+ "num_samples": 559,
1523
+ "action_dim": 7,
1524
+ "obs_mode": "vision_wrench",
1525
+ "action_mode": "post_velcmd_hand_binary"
1526
+ },
1527
+ {
1528
+ "hdf5_key": "ep_00146",
1529
+ "source_episode_index": 146,
1530
+ "split": "train",
1531
+ "num_samples": 677,
1532
+ "action_dim": 7,
1533
+ "obs_mode": "vision_wrench",
1534
+ "action_mode": "post_velcmd_hand_binary"
1535
+ },
1536
+ {
1537
+ "hdf5_key": "ep_00147",
1538
+ "source_episode_index": 147,
1539
+ "split": "train",
1540
+ "num_samples": 846,
1541
+ "action_dim": 7,
1542
+ "obs_mode": "vision_wrench",
1543
+ "action_mode": "post_velcmd_hand_binary"
1544
+ },
1545
+ {
1546
+ "hdf5_key": "ep_00148",
1547
+ "source_episode_index": 148,
1548
+ "split": "train",
1549
+ "num_samples": 615,
1550
+ "action_dim": 7,
1551
+ "obs_mode": "vision_wrench",
1552
+ "action_mode": "post_velcmd_hand_binary"
1553
+ },
1554
+ {
1555
+ "hdf5_key": "ep_00149",
1556
+ "source_episode_index": 149,
1557
+ "split": "train",
1558
+ "num_samples": 517,
1559
+ "action_dim": 7,
1560
+ "obs_mode": "vision_wrench",
1561
+ "action_mode": "post_velcmd_hand_binary"
1562
+ },
1563
+ {
1564
+ "hdf5_key": "ep_00150",
1565
+ "source_episode_index": 150,
1566
+ "split": "train",
1567
+ "num_samples": 817,
1568
+ "action_dim": 7,
1569
+ "obs_mode": "vision_wrench",
1570
+ "action_mode": "post_velcmd_hand_binary"
1571
+ },
1572
+ {
1573
+ "hdf5_key": "ep_00151",
1574
+ "source_episode_index": 151,
1575
+ "split": "train",
1576
+ "num_samples": 814,
1577
+ "action_dim": 7,
1578
+ "obs_mode": "vision_wrench",
1579
+ "action_mode": "post_velcmd_hand_binary"
1580
+ },
1581
+ {
1582
+ "hdf5_key": "ep_00152",
1583
+ "source_episode_index": 152,
1584
+ "split": "train",
1585
+ "num_samples": 828,
1586
+ "action_dim": 7,
1587
+ "obs_mode": "vision_wrench",
1588
+ "action_mode": "post_velcmd_hand_binary"
1589
+ },
1590
+ {
1591
+ "hdf5_key": "ep_00153",
1592
+ "source_episode_index": 153,
1593
+ "split": "train",
1594
+ "num_samples": 736,
1595
+ "action_dim": 7,
1596
+ "obs_mode": "vision_wrench",
1597
+ "action_mode": "post_velcmd_hand_binary"
1598
+ },
1599
+ {
1600
+ "hdf5_key": "ep_00154",
1601
+ "source_episode_index": 154,
1602
+ "split": "train",
1603
+ "num_samples": 860,
1604
+ "action_dim": 7,
1605
+ "obs_mode": "vision_wrench",
1606
+ "action_mode": "post_velcmd_hand_binary"
1607
+ },
1608
+ {
1609
+ "hdf5_key": "ep_00155",
1610
+ "source_episode_index": 155,
1611
+ "split": "train",
1612
+ "num_samples": 786,
1613
+ "action_dim": 7,
1614
+ "obs_mode": "vision_wrench",
1615
+ "action_mode": "post_velcmd_hand_binary"
1616
+ },
1617
+ {
1618
+ "hdf5_key": "ep_00156",
1619
+ "source_episode_index": 156,
1620
+ "split": "train",
1621
+ "num_samples": 725,
1622
+ "action_dim": 7,
1623
+ "obs_mode": "vision_wrench",
1624
+ "action_mode": "post_velcmd_hand_binary"
1625
+ },
1626
+ {
1627
+ "hdf5_key": "ep_00157",
1628
+ "source_episode_index": 157,
1629
+ "split": "train",
1630
+ "num_samples": 834,
1631
+ "action_dim": 7,
1632
+ "obs_mode": "vision_wrench",
1633
+ "action_mode": "post_velcmd_hand_binary"
1634
+ },
1635
+ {
1636
+ "hdf5_key": "ep_00158",
1637
+ "source_episode_index": 158,
1638
+ "split": "train",
1639
+ "num_samples": 652,
1640
+ "action_dim": 7,
1641
+ "obs_mode": "vision_wrench",
1642
+ "action_mode": "post_velcmd_hand_binary"
1643
+ },
1644
+ {
1645
+ "hdf5_key": "ep_00159",
1646
+ "source_episode_index": 159,
1647
+ "split": "train",
1648
+ "num_samples": 699,
1649
+ "action_dim": 7,
1650
+ "obs_mode": "vision_wrench",
1651
+ "action_mode": "post_velcmd_hand_binary"
1652
+ },
1653
+ {
1654
+ "hdf5_key": "ep_00160",
1655
+ "source_episode_index": 160,
1656
+ "split": "train",
1657
+ "num_samples": 812,
1658
+ "action_dim": 7,
1659
+ "obs_mode": "vision_wrench",
1660
+ "action_mode": "post_velcmd_hand_binary"
1661
+ },
1662
+ {
1663
+ "hdf5_key": "ep_00161",
1664
+ "source_episode_index": 161,
1665
+ "split": "train",
1666
+ "num_samples": 782,
1667
+ "action_dim": 7,
1668
+ "obs_mode": "vision_wrench",
1669
+ "action_mode": "post_velcmd_hand_binary"
1670
+ },
1671
+ {
1672
+ "hdf5_key": "ep_00162",
1673
+ "source_episode_index": 162,
1674
+ "split": "train",
1675
+ "num_samples": 681,
1676
+ "action_dim": 7,
1677
+ "obs_mode": "vision_wrench",
1678
+ "action_mode": "post_velcmd_hand_binary"
1679
+ },
1680
+ {
1681
+ "hdf5_key": "ep_00163",
1682
+ "source_episode_index": 163,
1683
+ "split": "train",
1684
+ "num_samples": 742,
1685
+ "action_dim": 7,
1686
+ "obs_mode": "vision_wrench",
1687
+ "action_mode": "post_velcmd_hand_binary"
1688
+ },
1689
+ {
1690
+ "hdf5_key": "ep_00164",
1691
+ "source_episode_index": 164,
1692
+ "split": "train",
1693
+ "num_samples": 766,
1694
+ "action_dim": 7,
1695
+ "obs_mode": "vision_wrench",
1696
+ "action_mode": "post_velcmd_hand_binary"
1697
+ },
1698
+ {
1699
+ "hdf5_key": "ep_00165",
1700
+ "source_episode_index": 165,
1701
+ "split": "train",
1702
+ "num_samples": 821,
1703
+ "action_dim": 7,
1704
+ "obs_mode": "vision_wrench",
1705
+ "action_mode": "post_velcmd_hand_binary"
1706
+ },
1707
+ {
1708
+ "hdf5_key": "ep_00166",
1709
+ "source_episode_index": 166,
1710
+ "split": "train",
1711
+ "num_samples": 1028,
1712
+ "action_dim": 7,
1713
+ "obs_mode": "vision_wrench",
1714
+ "action_mode": "post_velcmd_hand_binary"
1715
+ },
1716
+ {
1717
+ "hdf5_key": "ep_00167",
1718
+ "source_episode_index": 167,
1719
+ "split": "train",
1720
+ "num_samples": 720,
1721
+ "action_dim": 7,
1722
+ "obs_mode": "vision_wrench",
1723
+ "action_mode": "post_velcmd_hand_binary"
1724
+ },
1725
+ {
1726
+ "hdf5_key": "ep_00168",
1727
+ "source_episode_index": 168,
1728
+ "split": "train",
1729
+ "num_samples": 773,
1730
+ "action_dim": 7,
1731
+ "obs_mode": "vision_wrench",
1732
+ "action_mode": "post_velcmd_hand_binary"
1733
+ },
1734
+ {
1735
+ "hdf5_key": "ep_00169",
1736
+ "source_episode_index": 169,
1737
+ "split": "train",
1738
+ "num_samples": 690,
1739
+ "action_dim": 7,
1740
+ "obs_mode": "vision_wrench",
1741
+ "action_mode": "post_velcmd_hand_binary"
1742
+ },
1743
+ {
1744
+ "hdf5_key": "ep_00170",
1745
+ "source_episode_index": 170,
1746
+ "split": "train",
1747
+ "num_samples": 756,
1748
+ "action_dim": 7,
1749
+ "obs_mode": "vision_wrench",
1750
+ "action_mode": "post_velcmd_hand_binary"
1751
+ },
1752
+ {
1753
+ "hdf5_key": "ep_00171",
1754
+ "source_episode_index": 171,
1755
+ "split": "train",
1756
+ "num_samples": 742,
1757
+ "action_dim": 7,
1758
+ "obs_mode": "vision_wrench",
1759
+ "action_mode": "post_velcmd_hand_binary"
1760
+ },
1761
+ {
1762
+ "hdf5_key": "ep_00172",
1763
+ "source_episode_index": 172,
1764
+ "split": "valid",
1765
+ "num_samples": 689,
1766
+ "action_dim": 7,
1767
+ "obs_mode": "vision_wrench",
1768
+ "action_mode": "post_velcmd_hand_binary"
1769
+ },
1770
+ {
1771
+ "hdf5_key": "ep_00173",
1772
+ "source_episode_index": 173,
1773
+ "split": "valid",
1774
+ "num_samples": 639,
1775
+ "action_dim": 7,
1776
+ "obs_mode": "vision_wrench",
1777
+ "action_mode": "post_velcmd_hand_binary"
1778
+ },
1779
+ {
1780
+ "hdf5_key": "ep_00174",
1781
+ "source_episode_index": 174,
1782
+ "split": "valid",
1783
+ "num_samples": 694,
1784
+ "action_dim": 7,
1785
+ "obs_mode": "vision_wrench",
1786
+ "action_mode": "post_velcmd_hand_binary"
1787
+ },
1788
+ {
1789
+ "hdf5_key": "ep_00175",
1790
+ "source_episode_index": 175,
1791
+ "split": "valid",
1792
+ "num_samples": 532,
1793
+ "action_dim": 7,
1794
+ "obs_mode": "vision_wrench",
1795
+ "action_mode": "post_velcmd_hand_binary"
1796
+ },
1797
+ {
1798
+ "hdf5_key": "ep_00176",
1799
+ "source_episode_index": 176,
1800
+ "split": "valid",
1801
+ "num_samples": 665,
1802
+ "action_dim": 7,
1803
+ "obs_mode": "vision_wrench",
1804
+ "action_mode": "post_velcmd_hand_binary"
1805
+ },
1806
+ {
1807
+ "hdf5_key": "ep_00177",
1808
+ "source_episode_index": 177,
1809
+ "split": "valid",
1810
+ "num_samples": 771,
1811
+ "action_dim": 7,
1812
+ "obs_mode": "vision_wrench",
1813
+ "action_mode": "post_velcmd_hand_binary"
1814
+ },
1815
+ {
1816
+ "hdf5_key": "ep_00178",
1817
+ "source_episode_index": 178,
1818
+ "split": "valid",
1819
+ "num_samples": 602,
1820
+ "action_dim": 7,
1821
+ "obs_mode": "vision_wrench",
1822
+ "action_mode": "post_velcmd_hand_binary"
1823
+ },
1824
+ {
1825
+ "hdf5_key": "ep_00179",
1826
+ "source_episode_index": 179,
1827
+ "split": "valid",
1828
+ "num_samples": 700,
1829
+ "action_dim": 7,
1830
+ "obs_mode": "vision_wrench",
1831
+ "action_mode": "post_velcmd_hand_binary"
1832
+ },
1833
+ {
1834
+ "hdf5_key": "ep_00180",
1835
+ "source_episode_index": 180,
1836
+ "split": "valid",
1837
+ "num_samples": 880,
1838
+ "action_dim": 7,
1839
+ "obs_mode": "vision_wrench",
1840
+ "action_mode": "post_velcmd_hand_binary"
1841
+ },
1842
+ {
1843
+ "hdf5_key": "ep_00181",
1844
+ "source_episode_index": 181,
1845
+ "split": "valid",
1846
+ "num_samples": 522,
1847
+ "action_dim": 7,
1848
+ "obs_mode": "vision_wrench",
1849
+ "action_mode": "post_velcmd_hand_binary"
1850
+ },
1851
+ {
1852
+ "hdf5_key": "ep_00182",
1853
+ "source_episode_index": 182,
1854
+ "split": "valid",
1855
+ "num_samples": 697,
1856
+ "action_dim": 7,
1857
+ "obs_mode": "vision_wrench",
1858
+ "action_mode": "post_velcmd_hand_binary"
1859
+ },
1860
+ {
1861
+ "hdf5_key": "ep_00183",
1862
+ "source_episode_index": 183,
1863
+ "split": "valid",
1864
+ "num_samples": 621,
1865
+ "action_dim": 7,
1866
+ "obs_mode": "vision_wrench",
1867
+ "action_mode": "post_velcmd_hand_binary"
1868
+ },
1869
+ {
1870
+ "hdf5_key": "ep_00184",
1871
+ "source_episode_index": 184,
1872
+ "split": "valid",
1873
+ "num_samples": 710,
1874
+ "action_dim": 7,
1875
+ "obs_mode": "vision_wrench",
1876
+ "action_mode": "post_velcmd_hand_binary"
1877
+ },
1878
+ {
1879
+ "hdf5_key": "ep_00185",
1880
+ "source_episode_index": 185,
1881
+ "split": "valid",
1882
+ "num_samples": 774,
1883
+ "action_dim": 7,
1884
+ "obs_mode": "vision_wrench",
1885
+ "action_mode": "post_velcmd_hand_binary"
1886
+ },
1887
+ {
1888
+ "hdf5_key": "ep_00186",
1889
+ "source_episode_index": 186,
1890
+ "split": "valid",
1891
+ "num_samples": 682,
1892
+ "action_dim": 7,
1893
+ "obs_mode": "vision_wrench",
1894
+ "action_mode": "post_velcmd_hand_binary"
1895
+ },
1896
+ {
1897
+ "hdf5_key": "ep_00187",
1898
+ "source_episode_index": 187,
1899
+ "split": "valid",
1900
+ "num_samples": 715,
1901
+ "action_dim": 7,
1902
+ "obs_mode": "vision_wrench",
1903
+ "action_mode": "post_velcmd_hand_binary"
1904
+ },
1905
+ {
1906
+ "hdf5_key": "ep_00188",
1907
+ "source_episode_index": 188,
1908
+ "split": "valid",
1909
+ "num_samples": 798,
1910
+ "action_dim": 7,
1911
+ "obs_mode": "vision_wrench",
1912
+ "action_mode": "post_velcmd_hand_binary"
1913
+ },
1914
+ {
1915
+ "hdf5_key": "ep_00189",
1916
+ "source_episode_index": 189,
1917
+ "split": "valid",
1918
+ "num_samples": 786,
1919
+ "action_dim": 7,
1920
+ "obs_mode": "vision_wrench",
1921
+ "action_mode": "post_velcmd_hand_binary"
1922
+ },
1923
+ {
1924
+ "hdf5_key": "ep_00190",
1925
+ "source_episode_index": 190,
1926
+ "split": "valid",
1927
+ "num_samples": 759,
1928
+ "action_dim": 7,
1929
+ "obs_mode": "vision_wrench",
1930
+ "action_mode": "post_velcmd_hand_binary"
1931
+ },
1932
+ {
1933
+ "hdf5_key": "ep_00191",
1934
+ "source_episode_index": 191,
1935
+ "split": "valid",
1936
+ "num_samples": 812,
1937
+ "action_dim": 7,
1938
+ "obs_mode": "vision_wrench",
1939
+ "action_mode": "post_velcmd_hand_binary"
1940
+ },
1941
+ {
1942
+ "hdf5_key": "ep_00192",
1943
+ "source_episode_index": 192,
1944
+ "split": "valid",
1945
+ "num_samples": 803,
1946
+ "action_dim": 7,
1947
+ "obs_mode": "vision_wrench",
1948
+ "action_mode": "post_velcmd_hand_binary"
1949
+ },
1950
+ {
1951
+ "hdf5_key": "ep_00193",
1952
+ "source_episode_index": 193,
1953
+ "split": "valid",
1954
+ "num_samples": 780,
1955
+ "action_dim": 7,
1956
+ "obs_mode": "vision_wrench",
1957
+ "action_mode": "post_velcmd_hand_binary"
1958
+ },
1959
+ {
1960
+ "hdf5_key": "ep_00194",
1961
+ "source_episode_index": 194,
1962
+ "split": "valid",
1963
+ "num_samples": 853,
1964
+ "action_dim": 7,
1965
+ "obs_mode": "vision_wrench",
1966
+ "action_mode": "post_velcmd_hand_binary"
1967
+ },
1968
+ {
1969
+ "hdf5_key": "ep_00195",
1970
+ "source_episode_index": 195,
1971
+ "split": "valid",
1972
+ "num_samples": 710,
1973
+ "action_dim": 7,
1974
+ "obs_mode": "vision_wrench",
1975
+ "action_mode": "post_velcmd_hand_binary"
1976
+ },
1977
+ {
1978
+ "hdf5_key": "ep_00196",
1979
+ "source_episode_index": 196,
1980
+ "split": "valid",
1981
+ "num_samples": 855,
1982
+ "action_dim": 7,
1983
+ "obs_mode": "vision_wrench",
1984
+ "action_mode": "post_velcmd_hand_binary"
1985
+ },
1986
+ {
1987
+ "hdf5_key": "ep_00197",
1988
+ "source_episode_index": 197,
1989
+ "split": "valid",
1990
+ "num_samples": 707,
1991
+ "action_dim": 7,
1992
+ "obs_mode": "vision_wrench",
1993
+ "action_mode": "post_velcmd_hand_binary"
1994
+ },
1995
+ {
1996
+ "hdf5_key": "ep_00198",
1997
+ "source_episode_index": 198,
1998
+ "split": "valid",
1999
+ "num_samples": 557,
2000
+ "action_dim": 7,
2001
+ "obs_mode": "vision_wrench",
2002
+ "action_mode": "post_velcmd_hand_binary"
2003
+ },
2004
+ {
2005
+ "hdf5_key": "ep_00199",
2006
+ "source_episode_index": 199,
2007
+ "split": "valid",
2008
+ "num_samples": 522,
2009
+ "action_dim": 7,
2010
+ "obs_mode": "vision_wrench",
2011
+ "action_mode": "post_velcmd_hand_binary"
2012
+ },
2013
+ {
2014
+ "hdf5_key": "ep_00200",
2015
+ "source_episode_index": 200,
2016
+ "split": "valid",
2017
+ "num_samples": 581,
2018
+ "action_dim": 7,
2019
+ "obs_mode": "vision_wrench",
2020
+ "action_mode": "post_velcmd_hand_binary"
2021
+ },
2022
+ {
2023
+ "hdf5_key": "ep_00201",
2024
+ "source_episode_index": 201,
2025
+ "split": "valid",
2026
+ "num_samples": 574,
2027
+ "action_dim": 7,
2028
+ "obs_mode": "vision_wrench",
2029
+ "action_mode": "post_velcmd_hand_binary"
2030
+ }
2031
+ ]
2032
+ }
vac_post_vis_wrench/last.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c567f08fbaeb941442dbd43fb48198511cb1de4b71d663f789453187770b7d1
3
+ size 1255131613
vac_post_vis_wrench_state/action_stats.json ADDED
@@ -0,0 +1,297 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "method": "min_max",
3
+ "output_min": -0.999999,
4
+ "output_max": 0.999999,
5
+ "action_mode": "post_velcmd_hand_binary",
6
+ "action_components": [
7
+ "vel_cmd_x",
8
+ "vel_cmd_y",
9
+ "vel_cmd_z",
10
+ "vel_cmd_rx",
11
+ "vel_cmd_ry",
12
+ "vel_cmd_rz",
13
+ "hand_open_binary"
14
+ ],
15
+ "num_frames": 124722,
16
+ "min": [
17
+ -44.80923843383789,
18
+ -78.85905456542969,
19
+ -272.0136413574219,
20
+ -41.36355972290039,
21
+ -4.46674108505249,
22
+ -50.74555969238281,
23
+ 0.0
24
+ ],
25
+ "max": [
26
+ 60.49591827392578,
27
+ 81.3885269165039,
28
+ 499.7237243652344,
29
+ 40.38335418701172,
30
+ 6.112980842590332,
31
+ 49.7142219543457,
32
+ 1.0
33
+ ],
34
+ "mean": [
35
+ 0.08373736590147018,
36
+ 6.001084804534912,
37
+ -2.738046646118164,
38
+ -0.005926239304244518,
39
+ -0.001013437518849969,
40
+ 0.04403281956911087,
41
+ 0.428376704454422
42
+ ],
43
+ "std": [
44
+ 7.180497646331787,
45
+ 26.540620803833008,
46
+ 46.47587966918945,
47
+ 0.791343629360199,
48
+ 0.21718677878379822,
49
+ 3.934755325317383,
50
+ 0.4948435127735138
51
+ ],
52
+ "scale": [
53
+ 52.65263366699219,
54
+ 80.1238784790039,
55
+ 385.86907958984375,
56
+ 40.87350082397461,
57
+ 5.289865970611572,
58
+ 50.22993850708008,
59
+ 0.500000536441803
60
+ ],
61
+ "offset": [
62
+ 7.843341827392578,
63
+ 1.264739990234375,
64
+ 113.85504150390625,
65
+ -0.4901008605957031,
66
+ 0.8231196403503418,
67
+ -0.5156707763671875,
68
+ 0.5
69
+ ],
70
+ "hand_source": "action.absolute",
71
+ "hand_active_dims": [
72
+ 8,
73
+ 9,
74
+ 10,
75
+ 11
76
+ ],
77
+ "hand_reduce": "mean",
78
+ "hand_threshold": 0.85,
79
+ "hand_high_value": 1.0,
80
+ "hand_low_value": 0.0,
81
+ "dataset": "/workspace/datasets/dataset_pipe_02_06",
82
+ "observation_mode_examples": [
83
+ "vision",
84
+ "vision_wrench",
85
+ "vision_wrench_state"
86
+ ],
87
+ "train_source_episodes": [
88
+ 0,
89
+ 1,
90
+ 2,
91
+ 3,
92
+ 4,
93
+ 5,
94
+ 6,
95
+ 7,
96
+ 8,
97
+ 9,
98
+ 10,
99
+ 11,
100
+ 12,
101
+ 13,
102
+ 14,
103
+ 15,
104
+ 16,
105
+ 17,
106
+ 18,
107
+ 19,
108
+ 20,
109
+ 21,
110
+ 22,
111
+ 23,
112
+ 24,
113
+ 25,
114
+ 26,
115
+ 27,
116
+ 28,
117
+ 29,
118
+ 30,
119
+ 31,
120
+ 32,
121
+ 33,
122
+ 34,
123
+ 35,
124
+ 36,
125
+ 37,
126
+ 38,
127
+ 39,
128
+ 40,
129
+ 41,
130
+ 42,
131
+ 43,
132
+ 44,
133
+ 45,
134
+ 46,
135
+ 47,
136
+ 48,
137
+ 49,
138
+ 50,
139
+ 51,
140
+ 52,
141
+ 53,
142
+ 54,
143
+ 55,
144
+ 56,
145
+ 57,
146
+ 58,
147
+ 59,
148
+ 60,
149
+ 61,
150
+ 62,
151
+ 63,
152
+ 64,
153
+ 65,
154
+ 66,
155
+ 67,
156
+ 68,
157
+ 69,
158
+ 70,
159
+ 71,
160
+ 72,
161
+ 73,
162
+ 74,
163
+ 75,
164
+ 76,
165
+ 77,
166
+ 78,
167
+ 79,
168
+ 80,
169
+ 81,
170
+ 82,
171
+ 83,
172
+ 84,
173
+ 85,
174
+ 86,
175
+ 87,
176
+ 88,
177
+ 89,
178
+ 90,
179
+ 91,
180
+ 92,
181
+ 93,
182
+ 94,
183
+ 95,
184
+ 96,
185
+ 97,
186
+ 98,
187
+ 99,
188
+ 100,
189
+ 101,
190
+ 102,
191
+ 103,
192
+ 104,
193
+ 105,
194
+ 106,
195
+ 107,
196
+ 108,
197
+ 109,
198
+ 110,
199
+ 111,
200
+ 112,
201
+ 113,
202
+ 114,
203
+ 115,
204
+ 116,
205
+ 117,
206
+ 118,
207
+ 119,
208
+ 120,
209
+ 121,
210
+ 122,
211
+ 123,
212
+ 124,
213
+ 125,
214
+ 126,
215
+ 127,
216
+ 128,
217
+ 129,
218
+ 130,
219
+ 131,
220
+ 132,
221
+ 133,
222
+ 134,
223
+ 135,
224
+ 136,
225
+ 137,
226
+ 138,
227
+ 139,
228
+ 140,
229
+ 141,
230
+ 142,
231
+ 143,
232
+ 144,
233
+ 145,
234
+ 146,
235
+ 147,
236
+ 148,
237
+ 149,
238
+ 150,
239
+ 151,
240
+ 152,
241
+ 153,
242
+ 154,
243
+ 155,
244
+ 156,
245
+ 157,
246
+ 158,
247
+ 159,
248
+ 160,
249
+ 161,
250
+ 162,
251
+ 163,
252
+ 164,
253
+ 165,
254
+ 166,
255
+ 167,
256
+ 168,
257
+ 169,
258
+ 170,
259
+ 171
260
+ ],
261
+ "valid_source_episodes": [
262
+ 172,
263
+ 173,
264
+ 174,
265
+ 175,
266
+ 176,
267
+ 177,
268
+ 178,
269
+ 179,
270
+ 180,
271
+ 181,
272
+ 182,
273
+ 183,
274
+ 184,
275
+ 185,
276
+ 186,
277
+ 187,
278
+ 188,
279
+ 189,
280
+ 190,
281
+ 191,
282
+ 192,
283
+ 193,
284
+ 194,
285
+ 195,
286
+ 196,
287
+ 197,
288
+ 198,
289
+ 199,
290
+ 200,
291
+ 201
292
+ ],
293
+ "image_size": [
294
+ 84,
295
+ 84
296
+ ]
297
+ }
vac_post_vis_wrench_state/best.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bccfe065ee1634f1fc94bfc5a54a3d830f2f1702e5bf2aa81892138815549876
3
+ size 1260740861
vac_post_vis_wrench_state/config.json ADDED
@@ -0,0 +1,198 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "algo_name": "diffusion_policy",
3
+ "experiment": {
4
+ "name": "vac_post_vis_wrench_state",
5
+ "validate": true,
6
+ "logging": {
7
+ "terminal_output_to_txt": false,
8
+ "log_tb": true,
9
+ "log_wandb": false,
10
+ "wandb_proj_name": "debug"
11
+ },
12
+ "save": {
13
+ "enabled": true,
14
+ "every_n_seconds": null,
15
+ "every_n_epochs": 50,
16
+ "epochs": [],
17
+ "on_best_validation": true,
18
+ "on_best_rollout_return": false,
19
+ "on_best_rollout_success_rate": false
20
+ },
21
+ "epoch_every_n_steps": 100,
22
+ "validation_epoch_every_n_steps": 10,
23
+ "env": null,
24
+ "additional_envs": null,
25
+ "render": false,
26
+ "render_video": false,
27
+ "keep_all_videos": false,
28
+ "video_skip": 5,
29
+ "rollout": {
30
+ "enabled": false,
31
+ "n": 0,
32
+ "horizon": 50,
33
+ "rate": 50,
34
+ "warmstart": 0,
35
+ "terminate_on_success": true
36
+ },
37
+ "env_meta_update_dict": {},
38
+ "ckpt_path": null
39
+ },
40
+ "train": {
41
+ "data": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_post_vis_wrench_state/vac_post_vis_wrench_state.hdf5",
42
+ "output_dir": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_post_vis_wrench_state/robomimic_runs",
43
+ "normalize_weights_by_ds_size": false,
44
+ "num_data_workers": 0,
45
+ "hdf5_cache_mode": "all",
46
+ "hdf5_use_swmr": true,
47
+ "hdf5_load_next_obs": false,
48
+ "hdf5_normalize_obs": false,
49
+ "hdf5_filter_key": "train",
50
+ "hdf5_validation_filter_key": "valid",
51
+ "seq_length": 8,
52
+ "pad_seq_length": true,
53
+ "frame_stack": 2,
54
+ "pad_frame_stack": true,
55
+ "dataset_keys": [
56
+ "actions",
57
+ "rewards",
58
+ "dones"
59
+ ],
60
+ "action_keys": [
61
+ "actions"
62
+ ],
63
+ "action_config": {
64
+ "actions": {
65
+ "normalization": "min_max"
66
+ }
67
+ },
68
+ "goal_mode": null,
69
+ "cuda": true,
70
+ "batch_size": 16,
71
+ "num_epochs": 300,
72
+ "seed": 42,
73
+ "max_grad_norm": null
74
+ },
75
+ "algo": {
76
+ "optim_params": {
77
+ "policy": {
78
+ "optimizer_type": "adamw",
79
+ "learning_rate": {
80
+ "initial": 0.0001,
81
+ "decay_factor": 0.1,
82
+ "step_every_batch": true,
83
+ "scheduler_type": "cosine",
84
+ "num_cycles": 0.5,
85
+ "warmup_steps": 500,
86
+ "epoch_schedule": []
87
+ },
88
+ "regularization": {
89
+ "L2": 1e-06
90
+ }
91
+ }
92
+ },
93
+ "horizon": {
94
+ "observation_horizon": 2,
95
+ "action_horizon": 4,
96
+ "prediction_horizon": 8
97
+ },
98
+ "unet": {
99
+ "enabled": true,
100
+ "diffusion_step_embed_dim": 256,
101
+ "down_dims": [
102
+ 128,
103
+ 256,
104
+ 512
105
+ ],
106
+ "kernel_size": 5,
107
+ "n_groups": 8
108
+ },
109
+ "ema": {
110
+ "enabled": true,
111
+ "power": 0.75
112
+ },
113
+ "ddpm": {
114
+ "enabled": true,
115
+ "num_train_timesteps": 50,
116
+ "num_inference_timesteps": 50,
117
+ "beta_schedule": "squaredcos_cap_v2",
118
+ "clip_sample": true,
119
+ "prediction_type": "epsilon"
120
+ },
121
+ "ddim": {
122
+ "enabled": false,
123
+ "num_train_timesteps": 50,
124
+ "num_inference_timesteps": 10,
125
+ "beta_schedule": "squaredcos_cap_v2",
126
+ "clip_sample": true,
127
+ "set_alpha_to_one": true,
128
+ "steps_offset": 0,
129
+ "prediction_type": "epsilon"
130
+ }
131
+ },
132
+ "observation": {
133
+ "modalities": {
134
+ "obs": {
135
+ "low_dim": [
136
+ "wrench",
137
+ "state"
138
+ ],
139
+ "rgb": [
140
+ "front_rgb"
141
+ ],
142
+ "depth": [],
143
+ "scan": []
144
+ },
145
+ "goal": {
146
+ "low_dim": [],
147
+ "rgb": [],
148
+ "depth": [],
149
+ "scan": []
150
+ }
151
+ },
152
+ "encoder": {
153
+ "low_dim": {
154
+ "core_class": null,
155
+ "core_kwargs": {},
156
+ "obs_randomizer_class": null,
157
+ "obs_randomizer_kwargs": {}
158
+ },
159
+ "rgb": {
160
+ "core_class": "VisualCore",
161
+ "core_kwargs": {
162
+ "backbone_class": "ResNet18Conv",
163
+ "backbone_kwargs": {
164
+ "pretrained": false,
165
+ "input_coord_conv": false
166
+ },
167
+ "pool_class": "SpatialSoftmax",
168
+ "pool_kwargs": {
169
+ "num_kp": 32,
170
+ "learnable_temperature": false,
171
+ "temperature": 1.0,
172
+ "noise_std": 0.0
173
+ },
174
+ "feature_dimension": 64
175
+ },
176
+ "obs_randomizer_class": null,
177
+ "obs_randomizer_kwargs": {}
178
+ },
179
+ "depth": {
180
+ "core_class": "VisualCore",
181
+ "core_kwargs": {},
182
+ "obs_randomizer_class": null,
183
+ "obs_randomizer_kwargs": {}
184
+ },
185
+ "scan": {
186
+ "core_class": "ScanCore",
187
+ "core_kwargs": {},
188
+ "obs_randomizer_class": null,
189
+ "obs_randomizer_kwargs": {}
190
+ }
191
+ }
192
+ },
193
+ "meta": {
194
+ "hp_base_config_file": null,
195
+ "hp_keys": [],
196
+ "hp_values": []
197
+ }
198
+ }
vac_post_vis_wrench_state/dataset_summary.json ADDED
@@ -0,0 +1,2032 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "path": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_post_vis_wrench_state/vac_post_vis_wrench_state.hdf5",
3
+ "total_frames": 145712,
4
+ "train_episodes": 172,
5
+ "valid_episodes": 30,
6
+ "train_source_episodes": [
7
+ 0,
8
+ 1,
9
+ 2,
10
+ 3,
11
+ 4,
12
+ 5,
13
+ 6,
14
+ 7,
15
+ 8,
16
+ 9,
17
+ 10,
18
+ 11,
19
+ 12,
20
+ 13,
21
+ 14,
22
+ 15,
23
+ 16,
24
+ 17,
25
+ 18,
26
+ 19,
27
+ 20,
28
+ 21,
29
+ 22,
30
+ 23,
31
+ 24,
32
+ 25,
33
+ 26,
34
+ 27,
35
+ 28,
36
+ 29,
37
+ 30,
38
+ 31,
39
+ 32,
40
+ 33,
41
+ 34,
42
+ 35,
43
+ 36,
44
+ 37,
45
+ 38,
46
+ 39,
47
+ 40,
48
+ 41,
49
+ 42,
50
+ 43,
51
+ 44,
52
+ 45,
53
+ 46,
54
+ 47,
55
+ 48,
56
+ 49,
57
+ 50,
58
+ 51,
59
+ 52,
60
+ 53,
61
+ 54,
62
+ 55,
63
+ 56,
64
+ 57,
65
+ 58,
66
+ 59,
67
+ 60,
68
+ 61,
69
+ 62,
70
+ 63,
71
+ 64,
72
+ 65,
73
+ 66,
74
+ 67,
75
+ 68,
76
+ 69,
77
+ 70,
78
+ 71,
79
+ 72,
80
+ 73,
81
+ 74,
82
+ 75,
83
+ 76,
84
+ 77,
85
+ 78,
86
+ 79,
87
+ 80,
88
+ 81,
89
+ 82,
90
+ 83,
91
+ 84,
92
+ 85,
93
+ 86,
94
+ 87,
95
+ 88,
96
+ 89,
97
+ 90,
98
+ 91,
99
+ 92,
100
+ 93,
101
+ 94,
102
+ 95,
103
+ 96,
104
+ 97,
105
+ 98,
106
+ 99,
107
+ 100,
108
+ 101,
109
+ 102,
110
+ 103,
111
+ 104,
112
+ 105,
113
+ 106,
114
+ 107,
115
+ 108,
116
+ 109,
117
+ 110,
118
+ 111,
119
+ 112,
120
+ 113,
121
+ 114,
122
+ 115,
123
+ 116,
124
+ 117,
125
+ 118,
126
+ 119,
127
+ 120,
128
+ 121,
129
+ 122,
130
+ 123,
131
+ 124,
132
+ 125,
133
+ 126,
134
+ 127,
135
+ 128,
136
+ 129,
137
+ 130,
138
+ 131,
139
+ 132,
140
+ 133,
141
+ 134,
142
+ 135,
143
+ 136,
144
+ 137,
145
+ 138,
146
+ 139,
147
+ 140,
148
+ 141,
149
+ 142,
150
+ 143,
151
+ 144,
152
+ 145,
153
+ 146,
154
+ 147,
155
+ 148,
156
+ 149,
157
+ 150,
158
+ 151,
159
+ 152,
160
+ 153,
161
+ 154,
162
+ 155,
163
+ 156,
164
+ 157,
165
+ 158,
166
+ 159,
167
+ 160,
168
+ 161,
169
+ 162,
170
+ 163,
171
+ 164,
172
+ 165,
173
+ 166,
174
+ 167,
175
+ 168,
176
+ 169,
177
+ 170,
178
+ 171
179
+ ],
180
+ "valid_source_episodes": [
181
+ 172,
182
+ 173,
183
+ 174,
184
+ 175,
185
+ 176,
186
+ 177,
187
+ 178,
188
+ 179,
189
+ 180,
190
+ 181,
191
+ 182,
192
+ 183,
193
+ 184,
194
+ 185,
195
+ 186,
196
+ 187,
197
+ 188,
198
+ 189,
199
+ 190,
200
+ 191,
201
+ 192,
202
+ 193,
203
+ 194,
204
+ 195,
205
+ 196,
206
+ 197,
207
+ 198,
208
+ 199,
209
+ 200,
210
+ 201
211
+ ],
212
+ "episodes": [
213
+ {
214
+ "hdf5_key": "ep_00000",
215
+ "source_episode_index": 0,
216
+ "split": "train",
217
+ "num_samples": 663,
218
+ "action_dim": 7,
219
+ "obs_mode": "vision_wrench_state",
220
+ "action_mode": "post_velcmd_hand_binary"
221
+ },
222
+ {
223
+ "hdf5_key": "ep_00001",
224
+ "source_episode_index": 1,
225
+ "split": "train",
226
+ "num_samples": 706,
227
+ "action_dim": 7,
228
+ "obs_mode": "vision_wrench_state",
229
+ "action_mode": "post_velcmd_hand_binary"
230
+ },
231
+ {
232
+ "hdf5_key": "ep_00002",
233
+ "source_episode_index": 2,
234
+ "split": "train",
235
+ "num_samples": 1006,
236
+ "action_dim": 7,
237
+ "obs_mode": "vision_wrench_state",
238
+ "action_mode": "post_velcmd_hand_binary"
239
+ },
240
+ {
241
+ "hdf5_key": "ep_00003",
242
+ "source_episode_index": 3,
243
+ "split": "train",
244
+ "num_samples": 709,
245
+ "action_dim": 7,
246
+ "obs_mode": "vision_wrench_state",
247
+ "action_mode": "post_velcmd_hand_binary"
248
+ },
249
+ {
250
+ "hdf5_key": "ep_00004",
251
+ "source_episode_index": 4,
252
+ "split": "train",
253
+ "num_samples": 816,
254
+ "action_dim": 7,
255
+ "obs_mode": "vision_wrench_state",
256
+ "action_mode": "post_velcmd_hand_binary"
257
+ },
258
+ {
259
+ "hdf5_key": "ep_00005",
260
+ "source_episode_index": 5,
261
+ "split": "train",
262
+ "num_samples": 882,
263
+ "action_dim": 7,
264
+ "obs_mode": "vision_wrench_state",
265
+ "action_mode": "post_velcmd_hand_binary"
266
+ },
267
+ {
268
+ "hdf5_key": "ep_00006",
269
+ "source_episode_index": 6,
270
+ "split": "train",
271
+ "num_samples": 699,
272
+ "action_dim": 7,
273
+ "obs_mode": "vision_wrench_state",
274
+ "action_mode": "post_velcmd_hand_binary"
275
+ },
276
+ {
277
+ "hdf5_key": "ep_00007",
278
+ "source_episode_index": 7,
279
+ "split": "train",
280
+ "num_samples": 851,
281
+ "action_dim": 7,
282
+ "obs_mode": "vision_wrench_state",
283
+ "action_mode": "post_velcmd_hand_binary"
284
+ },
285
+ {
286
+ "hdf5_key": "ep_00008",
287
+ "source_episode_index": 8,
288
+ "split": "train",
289
+ "num_samples": 728,
290
+ "action_dim": 7,
291
+ "obs_mode": "vision_wrench_state",
292
+ "action_mode": "post_velcmd_hand_binary"
293
+ },
294
+ {
295
+ "hdf5_key": "ep_00009",
296
+ "source_episode_index": 9,
297
+ "split": "train",
298
+ "num_samples": 770,
299
+ "action_dim": 7,
300
+ "obs_mode": "vision_wrench_state",
301
+ "action_mode": "post_velcmd_hand_binary"
302
+ },
303
+ {
304
+ "hdf5_key": "ep_00010",
305
+ "source_episode_index": 10,
306
+ "split": "train",
307
+ "num_samples": 821,
308
+ "action_dim": 7,
309
+ "obs_mode": "vision_wrench_state",
310
+ "action_mode": "post_velcmd_hand_binary"
311
+ },
312
+ {
313
+ "hdf5_key": "ep_00011",
314
+ "source_episode_index": 11,
315
+ "split": "train",
316
+ "num_samples": 561,
317
+ "action_dim": 7,
318
+ "obs_mode": "vision_wrench_state",
319
+ "action_mode": "post_velcmd_hand_binary"
320
+ },
321
+ {
322
+ "hdf5_key": "ep_00012",
323
+ "source_episode_index": 12,
324
+ "split": "train",
325
+ "num_samples": 517,
326
+ "action_dim": 7,
327
+ "obs_mode": "vision_wrench_state",
328
+ "action_mode": "post_velcmd_hand_binary"
329
+ },
330
+ {
331
+ "hdf5_key": "ep_00013",
332
+ "source_episode_index": 13,
333
+ "split": "train",
334
+ "num_samples": 690,
335
+ "action_dim": 7,
336
+ "obs_mode": "vision_wrench_state",
337
+ "action_mode": "post_velcmd_hand_binary"
338
+ },
339
+ {
340
+ "hdf5_key": "ep_00014",
341
+ "source_episode_index": 14,
342
+ "split": "train",
343
+ "num_samples": 559,
344
+ "action_dim": 7,
345
+ "obs_mode": "vision_wrench_state",
346
+ "action_mode": "post_velcmd_hand_binary"
347
+ },
348
+ {
349
+ "hdf5_key": "ep_00015",
350
+ "source_episode_index": 15,
351
+ "split": "train",
352
+ "num_samples": 889,
353
+ "action_dim": 7,
354
+ "obs_mode": "vision_wrench_state",
355
+ "action_mode": "post_velcmd_hand_binary"
356
+ },
357
+ {
358
+ "hdf5_key": "ep_00016",
359
+ "source_episode_index": 16,
360
+ "split": "train",
361
+ "num_samples": 619,
362
+ "action_dim": 7,
363
+ "obs_mode": "vision_wrench_state",
364
+ "action_mode": "post_velcmd_hand_binary"
365
+ },
366
+ {
367
+ "hdf5_key": "ep_00017",
368
+ "source_episode_index": 17,
369
+ "split": "train",
370
+ "num_samples": 640,
371
+ "action_dim": 7,
372
+ "obs_mode": "vision_wrench_state",
373
+ "action_mode": "post_velcmd_hand_binary"
374
+ },
375
+ {
376
+ "hdf5_key": "ep_00018",
377
+ "source_episode_index": 18,
378
+ "split": "train",
379
+ "num_samples": 612,
380
+ "action_dim": 7,
381
+ "obs_mode": "vision_wrench_state",
382
+ "action_mode": "post_velcmd_hand_binary"
383
+ },
384
+ {
385
+ "hdf5_key": "ep_00019",
386
+ "source_episode_index": 19,
387
+ "split": "train",
388
+ "num_samples": 702,
389
+ "action_dim": 7,
390
+ "obs_mode": "vision_wrench_state",
391
+ "action_mode": "post_velcmd_hand_binary"
392
+ },
393
+ {
394
+ "hdf5_key": "ep_00020",
395
+ "source_episode_index": 20,
396
+ "split": "train",
397
+ "num_samples": 645,
398
+ "action_dim": 7,
399
+ "obs_mode": "vision_wrench_state",
400
+ "action_mode": "post_velcmd_hand_binary"
401
+ },
402
+ {
403
+ "hdf5_key": "ep_00021",
404
+ "source_episode_index": 21,
405
+ "split": "train",
406
+ "num_samples": 751,
407
+ "action_dim": 7,
408
+ "obs_mode": "vision_wrench_state",
409
+ "action_mode": "post_velcmd_hand_binary"
410
+ },
411
+ {
412
+ "hdf5_key": "ep_00022",
413
+ "source_episode_index": 22,
414
+ "split": "train",
415
+ "num_samples": 810,
416
+ "action_dim": 7,
417
+ "obs_mode": "vision_wrench_state",
418
+ "action_mode": "post_velcmd_hand_binary"
419
+ },
420
+ {
421
+ "hdf5_key": "ep_00023",
422
+ "source_episode_index": 23,
423
+ "split": "train",
424
+ "num_samples": 734,
425
+ "action_dim": 7,
426
+ "obs_mode": "vision_wrench_state",
427
+ "action_mode": "post_velcmd_hand_binary"
428
+ },
429
+ {
430
+ "hdf5_key": "ep_00024",
431
+ "source_episode_index": 24,
432
+ "split": "train",
433
+ "num_samples": 637,
434
+ "action_dim": 7,
435
+ "obs_mode": "vision_wrench_state",
436
+ "action_mode": "post_velcmd_hand_binary"
437
+ },
438
+ {
439
+ "hdf5_key": "ep_00025",
440
+ "source_episode_index": 25,
441
+ "split": "train",
442
+ "num_samples": 640,
443
+ "action_dim": 7,
444
+ "obs_mode": "vision_wrench_state",
445
+ "action_mode": "post_velcmd_hand_binary"
446
+ },
447
+ {
448
+ "hdf5_key": "ep_00026",
449
+ "source_episode_index": 26,
450
+ "split": "train",
451
+ "num_samples": 957,
452
+ "action_dim": 7,
453
+ "obs_mode": "vision_wrench_state",
454
+ "action_mode": "post_velcmd_hand_binary"
455
+ },
456
+ {
457
+ "hdf5_key": "ep_00027",
458
+ "source_episode_index": 27,
459
+ "split": "train",
460
+ "num_samples": 617,
461
+ "action_dim": 7,
462
+ "obs_mode": "vision_wrench_state",
463
+ "action_mode": "post_velcmd_hand_binary"
464
+ },
465
+ {
466
+ "hdf5_key": "ep_00028",
467
+ "source_episode_index": 28,
468
+ "split": "train",
469
+ "num_samples": 772,
470
+ "action_dim": 7,
471
+ "obs_mode": "vision_wrench_state",
472
+ "action_mode": "post_velcmd_hand_binary"
473
+ },
474
+ {
475
+ "hdf5_key": "ep_00029",
476
+ "source_episode_index": 29,
477
+ "split": "train",
478
+ "num_samples": 616,
479
+ "action_dim": 7,
480
+ "obs_mode": "vision_wrench_state",
481
+ "action_mode": "post_velcmd_hand_binary"
482
+ },
483
+ {
484
+ "hdf5_key": "ep_00030",
485
+ "source_episode_index": 30,
486
+ "split": "train",
487
+ "num_samples": 732,
488
+ "action_dim": 7,
489
+ "obs_mode": "vision_wrench_state",
490
+ "action_mode": "post_velcmd_hand_binary"
491
+ },
492
+ {
493
+ "hdf5_key": "ep_00031",
494
+ "source_episode_index": 31,
495
+ "split": "train",
496
+ "num_samples": 761,
497
+ "action_dim": 7,
498
+ "obs_mode": "vision_wrench_state",
499
+ "action_mode": "post_velcmd_hand_binary"
500
+ },
501
+ {
502
+ "hdf5_key": "ep_00032",
503
+ "source_episode_index": 32,
504
+ "split": "train",
505
+ "num_samples": 804,
506
+ "action_dim": 7,
507
+ "obs_mode": "vision_wrench_state",
508
+ "action_mode": "post_velcmd_hand_binary"
509
+ },
510
+ {
511
+ "hdf5_key": "ep_00033",
512
+ "source_episode_index": 33,
513
+ "split": "train",
514
+ "num_samples": 735,
515
+ "action_dim": 7,
516
+ "obs_mode": "vision_wrench_state",
517
+ "action_mode": "post_velcmd_hand_binary"
518
+ },
519
+ {
520
+ "hdf5_key": "ep_00034",
521
+ "source_episode_index": 34,
522
+ "split": "train",
523
+ "num_samples": 721,
524
+ "action_dim": 7,
525
+ "obs_mode": "vision_wrench_state",
526
+ "action_mode": "post_velcmd_hand_binary"
527
+ },
528
+ {
529
+ "hdf5_key": "ep_00035",
530
+ "source_episode_index": 35,
531
+ "split": "train",
532
+ "num_samples": 612,
533
+ "action_dim": 7,
534
+ "obs_mode": "vision_wrench_state",
535
+ "action_mode": "post_velcmd_hand_binary"
536
+ },
537
+ {
538
+ "hdf5_key": "ep_00036",
539
+ "source_episode_index": 36,
540
+ "split": "train",
541
+ "num_samples": 685,
542
+ "action_dim": 7,
543
+ "obs_mode": "vision_wrench_state",
544
+ "action_mode": "post_velcmd_hand_binary"
545
+ },
546
+ {
547
+ "hdf5_key": "ep_00037",
548
+ "source_episode_index": 37,
549
+ "split": "train",
550
+ "num_samples": 685,
551
+ "action_dim": 7,
552
+ "obs_mode": "vision_wrench_state",
553
+ "action_mode": "post_velcmd_hand_binary"
554
+ },
555
+ {
556
+ "hdf5_key": "ep_00038",
557
+ "source_episode_index": 38,
558
+ "split": "train",
559
+ "num_samples": 663,
560
+ "action_dim": 7,
561
+ "obs_mode": "vision_wrench_state",
562
+ "action_mode": "post_velcmd_hand_binary"
563
+ },
564
+ {
565
+ "hdf5_key": "ep_00039",
566
+ "source_episode_index": 39,
567
+ "split": "train",
568
+ "num_samples": 855,
569
+ "action_dim": 7,
570
+ "obs_mode": "vision_wrench_state",
571
+ "action_mode": "post_velcmd_hand_binary"
572
+ },
573
+ {
574
+ "hdf5_key": "ep_00040",
575
+ "source_episode_index": 40,
576
+ "split": "train",
577
+ "num_samples": 810,
578
+ "action_dim": 7,
579
+ "obs_mode": "vision_wrench_state",
580
+ "action_mode": "post_velcmd_hand_binary"
581
+ },
582
+ {
583
+ "hdf5_key": "ep_00041",
584
+ "source_episode_index": 41,
585
+ "split": "train",
586
+ "num_samples": 655,
587
+ "action_dim": 7,
588
+ "obs_mode": "vision_wrench_state",
589
+ "action_mode": "post_velcmd_hand_binary"
590
+ },
591
+ {
592
+ "hdf5_key": "ep_00042",
593
+ "source_episode_index": 42,
594
+ "split": "train",
595
+ "num_samples": 946,
596
+ "action_dim": 7,
597
+ "obs_mode": "vision_wrench_state",
598
+ "action_mode": "post_velcmd_hand_binary"
599
+ },
600
+ {
601
+ "hdf5_key": "ep_00043",
602
+ "source_episode_index": 43,
603
+ "split": "train",
604
+ "num_samples": 703,
605
+ "action_dim": 7,
606
+ "obs_mode": "vision_wrench_state",
607
+ "action_mode": "post_velcmd_hand_binary"
608
+ },
609
+ {
610
+ "hdf5_key": "ep_00044",
611
+ "source_episode_index": 44,
612
+ "split": "train",
613
+ "num_samples": 606,
614
+ "action_dim": 7,
615
+ "obs_mode": "vision_wrench_state",
616
+ "action_mode": "post_velcmd_hand_binary"
617
+ },
618
+ {
619
+ "hdf5_key": "ep_00045",
620
+ "source_episode_index": 45,
621
+ "split": "train",
622
+ "num_samples": 914,
623
+ "action_dim": 7,
624
+ "obs_mode": "vision_wrench_state",
625
+ "action_mode": "post_velcmd_hand_binary"
626
+ },
627
+ {
628
+ "hdf5_key": "ep_00046",
629
+ "source_episode_index": 46,
630
+ "split": "train",
631
+ "num_samples": 949,
632
+ "action_dim": 7,
633
+ "obs_mode": "vision_wrench_state",
634
+ "action_mode": "post_velcmd_hand_binary"
635
+ },
636
+ {
637
+ "hdf5_key": "ep_00047",
638
+ "source_episode_index": 47,
639
+ "split": "train",
640
+ "num_samples": 818,
641
+ "action_dim": 7,
642
+ "obs_mode": "vision_wrench_state",
643
+ "action_mode": "post_velcmd_hand_binary"
644
+ },
645
+ {
646
+ "hdf5_key": "ep_00048",
647
+ "source_episode_index": 48,
648
+ "split": "train",
649
+ "num_samples": 913,
650
+ "action_dim": 7,
651
+ "obs_mode": "vision_wrench_state",
652
+ "action_mode": "post_velcmd_hand_binary"
653
+ },
654
+ {
655
+ "hdf5_key": "ep_00049",
656
+ "source_episode_index": 49,
657
+ "split": "train",
658
+ "num_samples": 904,
659
+ "action_dim": 7,
660
+ "obs_mode": "vision_wrench_state",
661
+ "action_mode": "post_velcmd_hand_binary"
662
+ },
663
+ {
664
+ "hdf5_key": "ep_00050",
665
+ "source_episode_index": 50,
666
+ "split": "train",
667
+ "num_samples": 923,
668
+ "action_dim": 7,
669
+ "obs_mode": "vision_wrench_state",
670
+ "action_mode": "post_velcmd_hand_binary"
671
+ },
672
+ {
673
+ "hdf5_key": "ep_00051",
674
+ "source_episode_index": 51,
675
+ "split": "train",
676
+ "num_samples": 824,
677
+ "action_dim": 7,
678
+ "obs_mode": "vision_wrench_state",
679
+ "action_mode": "post_velcmd_hand_binary"
680
+ },
681
+ {
682
+ "hdf5_key": "ep_00052",
683
+ "source_episode_index": 52,
684
+ "split": "train",
685
+ "num_samples": 777,
686
+ "action_dim": 7,
687
+ "obs_mode": "vision_wrench_state",
688
+ "action_mode": "post_velcmd_hand_binary"
689
+ },
690
+ {
691
+ "hdf5_key": "ep_00053",
692
+ "source_episode_index": 53,
693
+ "split": "train",
694
+ "num_samples": 714,
695
+ "action_dim": 7,
696
+ "obs_mode": "vision_wrench_state",
697
+ "action_mode": "post_velcmd_hand_binary"
698
+ },
699
+ {
700
+ "hdf5_key": "ep_00054",
701
+ "source_episode_index": 54,
702
+ "split": "train",
703
+ "num_samples": 771,
704
+ "action_dim": 7,
705
+ "obs_mode": "vision_wrench_state",
706
+ "action_mode": "post_velcmd_hand_binary"
707
+ },
708
+ {
709
+ "hdf5_key": "ep_00055",
710
+ "source_episode_index": 55,
711
+ "split": "train",
712
+ "num_samples": 684,
713
+ "action_dim": 7,
714
+ "obs_mode": "vision_wrench_state",
715
+ "action_mode": "post_velcmd_hand_binary"
716
+ },
717
+ {
718
+ "hdf5_key": "ep_00056",
719
+ "source_episode_index": 56,
720
+ "split": "train",
721
+ "num_samples": 766,
722
+ "action_dim": 7,
723
+ "obs_mode": "vision_wrench_state",
724
+ "action_mode": "post_velcmd_hand_binary"
725
+ },
726
+ {
727
+ "hdf5_key": "ep_00057",
728
+ "source_episode_index": 57,
729
+ "split": "train",
730
+ "num_samples": 890,
731
+ "action_dim": 7,
732
+ "obs_mode": "vision_wrench_state",
733
+ "action_mode": "post_velcmd_hand_binary"
734
+ },
735
+ {
736
+ "hdf5_key": "ep_00058",
737
+ "source_episode_index": 58,
738
+ "split": "train",
739
+ "num_samples": 774,
740
+ "action_dim": 7,
741
+ "obs_mode": "vision_wrench_state",
742
+ "action_mode": "post_velcmd_hand_binary"
743
+ },
744
+ {
745
+ "hdf5_key": "ep_00059",
746
+ "source_episode_index": 59,
747
+ "split": "train",
748
+ "num_samples": 781,
749
+ "action_dim": 7,
750
+ "obs_mode": "vision_wrench_state",
751
+ "action_mode": "post_velcmd_hand_binary"
752
+ },
753
+ {
754
+ "hdf5_key": "ep_00060",
755
+ "source_episode_index": 60,
756
+ "split": "train",
757
+ "num_samples": 1093,
758
+ "action_dim": 7,
759
+ "obs_mode": "vision_wrench_state",
760
+ "action_mode": "post_velcmd_hand_binary"
761
+ },
762
+ {
763
+ "hdf5_key": "ep_00061",
764
+ "source_episode_index": 61,
765
+ "split": "train",
766
+ "num_samples": 900,
767
+ "action_dim": 7,
768
+ "obs_mode": "vision_wrench_state",
769
+ "action_mode": "post_velcmd_hand_binary"
770
+ },
771
+ {
772
+ "hdf5_key": "ep_00062",
773
+ "source_episode_index": 62,
774
+ "split": "train",
775
+ "num_samples": 814,
776
+ "action_dim": 7,
777
+ "obs_mode": "vision_wrench_state",
778
+ "action_mode": "post_velcmd_hand_binary"
779
+ },
780
+ {
781
+ "hdf5_key": "ep_00063",
782
+ "source_episode_index": 63,
783
+ "split": "train",
784
+ "num_samples": 1108,
785
+ "action_dim": 7,
786
+ "obs_mode": "vision_wrench_state",
787
+ "action_mode": "post_velcmd_hand_binary"
788
+ },
789
+ {
790
+ "hdf5_key": "ep_00064",
791
+ "source_episode_index": 64,
792
+ "split": "train",
793
+ "num_samples": 816,
794
+ "action_dim": 7,
795
+ "obs_mode": "vision_wrench_state",
796
+ "action_mode": "post_velcmd_hand_binary"
797
+ },
798
+ {
799
+ "hdf5_key": "ep_00065",
800
+ "source_episode_index": 65,
801
+ "split": "train",
802
+ "num_samples": 963,
803
+ "action_dim": 7,
804
+ "obs_mode": "vision_wrench_state",
805
+ "action_mode": "post_velcmd_hand_binary"
806
+ },
807
+ {
808
+ "hdf5_key": "ep_00066",
809
+ "source_episode_index": 66,
810
+ "split": "train",
811
+ "num_samples": 895,
812
+ "action_dim": 7,
813
+ "obs_mode": "vision_wrench_state",
814
+ "action_mode": "post_velcmd_hand_binary"
815
+ },
816
+ {
817
+ "hdf5_key": "ep_00067",
818
+ "source_episode_index": 67,
819
+ "split": "train",
820
+ "num_samples": 572,
821
+ "action_dim": 7,
822
+ "obs_mode": "vision_wrench_state",
823
+ "action_mode": "post_velcmd_hand_binary"
824
+ },
825
+ {
826
+ "hdf5_key": "ep_00068",
827
+ "source_episode_index": 68,
828
+ "split": "train",
829
+ "num_samples": 648,
830
+ "action_dim": 7,
831
+ "obs_mode": "vision_wrench_state",
832
+ "action_mode": "post_velcmd_hand_binary"
833
+ },
834
+ {
835
+ "hdf5_key": "ep_00069",
836
+ "source_episode_index": 69,
837
+ "split": "train",
838
+ "num_samples": 772,
839
+ "action_dim": 7,
840
+ "obs_mode": "vision_wrench_state",
841
+ "action_mode": "post_velcmd_hand_binary"
842
+ },
843
+ {
844
+ "hdf5_key": "ep_00070",
845
+ "source_episode_index": 70,
846
+ "split": "train",
847
+ "num_samples": 702,
848
+ "action_dim": 7,
849
+ "obs_mode": "vision_wrench_state",
850
+ "action_mode": "post_velcmd_hand_binary"
851
+ },
852
+ {
853
+ "hdf5_key": "ep_00071",
854
+ "source_episode_index": 71,
855
+ "split": "train",
856
+ "num_samples": 786,
857
+ "action_dim": 7,
858
+ "obs_mode": "vision_wrench_state",
859
+ "action_mode": "post_velcmd_hand_binary"
860
+ },
861
+ {
862
+ "hdf5_key": "ep_00072",
863
+ "source_episode_index": 72,
864
+ "split": "train",
865
+ "num_samples": 843,
866
+ "action_dim": 7,
867
+ "obs_mode": "vision_wrench_state",
868
+ "action_mode": "post_velcmd_hand_binary"
869
+ },
870
+ {
871
+ "hdf5_key": "ep_00073",
872
+ "source_episode_index": 73,
873
+ "split": "train",
874
+ "num_samples": 767,
875
+ "action_dim": 7,
876
+ "obs_mode": "vision_wrench_state",
877
+ "action_mode": "post_velcmd_hand_binary"
878
+ },
879
+ {
880
+ "hdf5_key": "ep_00074",
881
+ "source_episode_index": 74,
882
+ "split": "train",
883
+ "num_samples": 764,
884
+ "action_dim": 7,
885
+ "obs_mode": "vision_wrench_state",
886
+ "action_mode": "post_velcmd_hand_binary"
887
+ },
888
+ {
889
+ "hdf5_key": "ep_00075",
890
+ "source_episode_index": 75,
891
+ "split": "train",
892
+ "num_samples": 750,
893
+ "action_dim": 7,
894
+ "obs_mode": "vision_wrench_state",
895
+ "action_mode": "post_velcmd_hand_binary"
896
+ },
897
+ {
898
+ "hdf5_key": "ep_00076",
899
+ "source_episode_index": 76,
900
+ "split": "train",
901
+ "num_samples": 652,
902
+ "action_dim": 7,
903
+ "obs_mode": "vision_wrench_state",
904
+ "action_mode": "post_velcmd_hand_binary"
905
+ },
906
+ {
907
+ "hdf5_key": "ep_00077",
908
+ "source_episode_index": 77,
909
+ "split": "train",
910
+ "num_samples": 812,
911
+ "action_dim": 7,
912
+ "obs_mode": "vision_wrench_state",
913
+ "action_mode": "post_velcmd_hand_binary"
914
+ },
915
+ {
916
+ "hdf5_key": "ep_00078",
917
+ "source_episode_index": 78,
918
+ "split": "train",
919
+ "num_samples": 722,
920
+ "action_dim": 7,
921
+ "obs_mode": "vision_wrench_state",
922
+ "action_mode": "post_velcmd_hand_binary"
923
+ },
924
+ {
925
+ "hdf5_key": "ep_00079",
926
+ "source_episode_index": 79,
927
+ "split": "train",
928
+ "num_samples": 723,
929
+ "action_dim": 7,
930
+ "obs_mode": "vision_wrench_state",
931
+ "action_mode": "post_velcmd_hand_binary"
932
+ },
933
+ {
934
+ "hdf5_key": "ep_00080",
935
+ "source_episode_index": 80,
936
+ "split": "train",
937
+ "num_samples": 765,
938
+ "action_dim": 7,
939
+ "obs_mode": "vision_wrench_state",
940
+ "action_mode": "post_velcmd_hand_binary"
941
+ },
942
+ {
943
+ "hdf5_key": "ep_00081",
944
+ "source_episode_index": 81,
945
+ "split": "train",
946
+ "num_samples": 981,
947
+ "action_dim": 7,
948
+ "obs_mode": "vision_wrench_state",
949
+ "action_mode": "post_velcmd_hand_binary"
950
+ },
951
+ {
952
+ "hdf5_key": "ep_00082",
953
+ "source_episode_index": 82,
954
+ "split": "train",
955
+ "num_samples": 959,
956
+ "action_dim": 7,
957
+ "obs_mode": "vision_wrench_state",
958
+ "action_mode": "post_velcmd_hand_binary"
959
+ },
960
+ {
961
+ "hdf5_key": "ep_00083",
962
+ "source_episode_index": 83,
963
+ "split": "train",
964
+ "num_samples": 655,
965
+ "action_dim": 7,
966
+ "obs_mode": "vision_wrench_state",
967
+ "action_mode": "post_velcmd_hand_binary"
968
+ },
969
+ {
970
+ "hdf5_key": "ep_00084",
971
+ "source_episode_index": 84,
972
+ "split": "train",
973
+ "num_samples": 746,
974
+ "action_dim": 7,
975
+ "obs_mode": "vision_wrench_state",
976
+ "action_mode": "post_velcmd_hand_binary"
977
+ },
978
+ {
979
+ "hdf5_key": "ep_00085",
980
+ "source_episode_index": 85,
981
+ "split": "train",
982
+ "num_samples": 660,
983
+ "action_dim": 7,
984
+ "obs_mode": "vision_wrench_state",
985
+ "action_mode": "post_velcmd_hand_binary"
986
+ },
987
+ {
988
+ "hdf5_key": "ep_00086",
989
+ "source_episode_index": 86,
990
+ "split": "train",
991
+ "num_samples": 617,
992
+ "action_dim": 7,
993
+ "obs_mode": "vision_wrench_state",
994
+ "action_mode": "post_velcmd_hand_binary"
995
+ },
996
+ {
997
+ "hdf5_key": "ep_00087",
998
+ "source_episode_index": 87,
999
+ "split": "train",
1000
+ "num_samples": 819,
1001
+ "action_dim": 7,
1002
+ "obs_mode": "vision_wrench_state",
1003
+ "action_mode": "post_velcmd_hand_binary"
1004
+ },
1005
+ {
1006
+ "hdf5_key": "ep_00088",
1007
+ "source_episode_index": 88,
1008
+ "split": "train",
1009
+ "num_samples": 696,
1010
+ "action_dim": 7,
1011
+ "obs_mode": "vision_wrench_state",
1012
+ "action_mode": "post_velcmd_hand_binary"
1013
+ },
1014
+ {
1015
+ "hdf5_key": "ep_00089",
1016
+ "source_episode_index": 89,
1017
+ "split": "train",
1018
+ "num_samples": 855,
1019
+ "action_dim": 7,
1020
+ "obs_mode": "vision_wrench_state",
1021
+ "action_mode": "post_velcmd_hand_binary"
1022
+ },
1023
+ {
1024
+ "hdf5_key": "ep_00090",
1025
+ "source_episode_index": 90,
1026
+ "split": "train",
1027
+ "num_samples": 678,
1028
+ "action_dim": 7,
1029
+ "obs_mode": "vision_wrench_state",
1030
+ "action_mode": "post_velcmd_hand_binary"
1031
+ },
1032
+ {
1033
+ "hdf5_key": "ep_00091",
1034
+ "source_episode_index": 91,
1035
+ "split": "train",
1036
+ "num_samples": 711,
1037
+ "action_dim": 7,
1038
+ "obs_mode": "vision_wrench_state",
1039
+ "action_mode": "post_velcmd_hand_binary"
1040
+ },
1041
+ {
1042
+ "hdf5_key": "ep_00092",
1043
+ "source_episode_index": 92,
1044
+ "split": "train",
1045
+ "num_samples": 763,
1046
+ "action_dim": 7,
1047
+ "obs_mode": "vision_wrench_state",
1048
+ "action_mode": "post_velcmd_hand_binary"
1049
+ },
1050
+ {
1051
+ "hdf5_key": "ep_00093",
1052
+ "source_episode_index": 93,
1053
+ "split": "train",
1054
+ "num_samples": 723,
1055
+ "action_dim": 7,
1056
+ "obs_mode": "vision_wrench_state",
1057
+ "action_mode": "post_velcmd_hand_binary"
1058
+ },
1059
+ {
1060
+ "hdf5_key": "ep_00094",
1061
+ "source_episode_index": 94,
1062
+ "split": "train",
1063
+ "num_samples": 723,
1064
+ "action_dim": 7,
1065
+ "obs_mode": "vision_wrench_state",
1066
+ "action_mode": "post_velcmd_hand_binary"
1067
+ },
1068
+ {
1069
+ "hdf5_key": "ep_00095",
1070
+ "source_episode_index": 95,
1071
+ "split": "train",
1072
+ "num_samples": 865,
1073
+ "action_dim": 7,
1074
+ "obs_mode": "vision_wrench_state",
1075
+ "action_mode": "post_velcmd_hand_binary"
1076
+ },
1077
+ {
1078
+ "hdf5_key": "ep_00096",
1079
+ "source_episode_index": 96,
1080
+ "split": "train",
1081
+ "num_samples": 495,
1082
+ "action_dim": 7,
1083
+ "obs_mode": "vision_wrench_state",
1084
+ "action_mode": "post_velcmd_hand_binary"
1085
+ },
1086
+ {
1087
+ "hdf5_key": "ep_00097",
1088
+ "source_episode_index": 97,
1089
+ "split": "train",
1090
+ "num_samples": 451,
1091
+ "action_dim": 7,
1092
+ "obs_mode": "vision_wrench_state",
1093
+ "action_mode": "post_velcmd_hand_binary"
1094
+ },
1095
+ {
1096
+ "hdf5_key": "ep_00098",
1097
+ "source_episode_index": 98,
1098
+ "split": "train",
1099
+ "num_samples": 690,
1100
+ "action_dim": 7,
1101
+ "obs_mode": "vision_wrench_state",
1102
+ "action_mode": "post_velcmd_hand_binary"
1103
+ },
1104
+ {
1105
+ "hdf5_key": "ep_00099",
1106
+ "source_episode_index": 99,
1107
+ "split": "train",
1108
+ "num_samples": 690,
1109
+ "action_dim": 7,
1110
+ "obs_mode": "vision_wrench_state",
1111
+ "action_mode": "post_velcmd_hand_binary"
1112
+ },
1113
+ {
1114
+ "hdf5_key": "ep_00100",
1115
+ "source_episode_index": 100,
1116
+ "split": "train",
1117
+ "num_samples": 615,
1118
+ "action_dim": 7,
1119
+ "obs_mode": "vision_wrench_state",
1120
+ "action_mode": "post_velcmd_hand_binary"
1121
+ },
1122
+ {
1123
+ "hdf5_key": "ep_00101",
1124
+ "source_episode_index": 101,
1125
+ "split": "train",
1126
+ "num_samples": 662,
1127
+ "action_dim": 7,
1128
+ "obs_mode": "vision_wrench_state",
1129
+ "action_mode": "post_velcmd_hand_binary"
1130
+ },
1131
+ {
1132
+ "hdf5_key": "ep_00102",
1133
+ "source_episode_index": 102,
1134
+ "split": "train",
1135
+ "num_samples": 614,
1136
+ "action_dim": 7,
1137
+ "obs_mode": "vision_wrench_state",
1138
+ "action_mode": "post_velcmd_hand_binary"
1139
+ },
1140
+ {
1141
+ "hdf5_key": "ep_00103",
1142
+ "source_episode_index": 103,
1143
+ "split": "train",
1144
+ "num_samples": 730,
1145
+ "action_dim": 7,
1146
+ "obs_mode": "vision_wrench_state",
1147
+ "action_mode": "post_velcmd_hand_binary"
1148
+ },
1149
+ {
1150
+ "hdf5_key": "ep_00104",
1151
+ "source_episode_index": 104,
1152
+ "split": "train",
1153
+ "num_samples": 683,
1154
+ "action_dim": 7,
1155
+ "obs_mode": "vision_wrench_state",
1156
+ "action_mode": "post_velcmd_hand_binary"
1157
+ },
1158
+ {
1159
+ "hdf5_key": "ep_00105",
1160
+ "source_episode_index": 105,
1161
+ "split": "train",
1162
+ "num_samples": 602,
1163
+ "action_dim": 7,
1164
+ "obs_mode": "vision_wrench_state",
1165
+ "action_mode": "post_velcmd_hand_binary"
1166
+ },
1167
+ {
1168
+ "hdf5_key": "ep_00106",
1169
+ "source_episode_index": 106,
1170
+ "split": "train",
1171
+ "num_samples": 652,
1172
+ "action_dim": 7,
1173
+ "obs_mode": "vision_wrench_state",
1174
+ "action_mode": "post_velcmd_hand_binary"
1175
+ },
1176
+ {
1177
+ "hdf5_key": "ep_00107",
1178
+ "source_episode_index": 107,
1179
+ "split": "train",
1180
+ "num_samples": 722,
1181
+ "action_dim": 7,
1182
+ "obs_mode": "vision_wrench_state",
1183
+ "action_mode": "post_velcmd_hand_binary"
1184
+ },
1185
+ {
1186
+ "hdf5_key": "ep_00108",
1187
+ "source_episode_index": 108,
1188
+ "split": "train",
1189
+ "num_samples": 605,
1190
+ "action_dim": 7,
1191
+ "obs_mode": "vision_wrench_state",
1192
+ "action_mode": "post_velcmd_hand_binary"
1193
+ },
1194
+ {
1195
+ "hdf5_key": "ep_00109",
1196
+ "source_episode_index": 109,
1197
+ "split": "train",
1198
+ "num_samples": 944,
1199
+ "action_dim": 7,
1200
+ "obs_mode": "vision_wrench_state",
1201
+ "action_mode": "post_velcmd_hand_binary"
1202
+ },
1203
+ {
1204
+ "hdf5_key": "ep_00110",
1205
+ "source_episode_index": 110,
1206
+ "split": "train",
1207
+ "num_samples": 661,
1208
+ "action_dim": 7,
1209
+ "obs_mode": "vision_wrench_state",
1210
+ "action_mode": "post_velcmd_hand_binary"
1211
+ },
1212
+ {
1213
+ "hdf5_key": "ep_00111",
1214
+ "source_episode_index": 111,
1215
+ "split": "train",
1216
+ "num_samples": 644,
1217
+ "action_dim": 7,
1218
+ "obs_mode": "vision_wrench_state",
1219
+ "action_mode": "post_velcmd_hand_binary"
1220
+ },
1221
+ {
1222
+ "hdf5_key": "ep_00112",
1223
+ "source_episode_index": 112,
1224
+ "split": "train",
1225
+ "num_samples": 614,
1226
+ "action_dim": 7,
1227
+ "obs_mode": "vision_wrench_state",
1228
+ "action_mode": "post_velcmd_hand_binary"
1229
+ },
1230
+ {
1231
+ "hdf5_key": "ep_00113",
1232
+ "source_episode_index": 113,
1233
+ "split": "train",
1234
+ "num_samples": 548,
1235
+ "action_dim": 7,
1236
+ "obs_mode": "vision_wrench_state",
1237
+ "action_mode": "post_velcmd_hand_binary"
1238
+ },
1239
+ {
1240
+ "hdf5_key": "ep_00114",
1241
+ "source_episode_index": 114,
1242
+ "split": "train",
1243
+ "num_samples": 568,
1244
+ "action_dim": 7,
1245
+ "obs_mode": "vision_wrench_state",
1246
+ "action_mode": "post_velcmd_hand_binary"
1247
+ },
1248
+ {
1249
+ "hdf5_key": "ep_00115",
1250
+ "source_episode_index": 115,
1251
+ "split": "train",
1252
+ "num_samples": 730,
1253
+ "action_dim": 7,
1254
+ "obs_mode": "vision_wrench_state",
1255
+ "action_mode": "post_velcmd_hand_binary"
1256
+ },
1257
+ {
1258
+ "hdf5_key": "ep_00116",
1259
+ "source_episode_index": 116,
1260
+ "split": "train",
1261
+ "num_samples": 481,
1262
+ "action_dim": 7,
1263
+ "obs_mode": "vision_wrench_state",
1264
+ "action_mode": "post_velcmd_hand_binary"
1265
+ },
1266
+ {
1267
+ "hdf5_key": "ep_00117",
1268
+ "source_episode_index": 117,
1269
+ "split": "train",
1270
+ "num_samples": 687,
1271
+ "action_dim": 7,
1272
+ "obs_mode": "vision_wrench_state",
1273
+ "action_mode": "post_velcmd_hand_binary"
1274
+ },
1275
+ {
1276
+ "hdf5_key": "ep_00118",
1277
+ "source_episode_index": 118,
1278
+ "split": "train",
1279
+ "num_samples": 618,
1280
+ "action_dim": 7,
1281
+ "obs_mode": "vision_wrench_state",
1282
+ "action_mode": "post_velcmd_hand_binary"
1283
+ },
1284
+ {
1285
+ "hdf5_key": "ep_00119",
1286
+ "source_episode_index": 119,
1287
+ "split": "train",
1288
+ "num_samples": 768,
1289
+ "action_dim": 7,
1290
+ "obs_mode": "vision_wrench_state",
1291
+ "action_mode": "post_velcmd_hand_binary"
1292
+ },
1293
+ {
1294
+ "hdf5_key": "ep_00120",
1295
+ "source_episode_index": 120,
1296
+ "split": "train",
1297
+ "num_samples": 734,
1298
+ "action_dim": 7,
1299
+ "obs_mode": "vision_wrench_state",
1300
+ "action_mode": "post_velcmd_hand_binary"
1301
+ },
1302
+ {
1303
+ "hdf5_key": "ep_00121",
1304
+ "source_episode_index": 121,
1305
+ "split": "train",
1306
+ "num_samples": 604,
1307
+ "action_dim": 7,
1308
+ "obs_mode": "vision_wrench_state",
1309
+ "action_mode": "post_velcmd_hand_binary"
1310
+ },
1311
+ {
1312
+ "hdf5_key": "ep_00122",
1313
+ "source_episode_index": 122,
1314
+ "split": "train",
1315
+ "num_samples": 610,
1316
+ "action_dim": 7,
1317
+ "obs_mode": "vision_wrench_state",
1318
+ "action_mode": "post_velcmd_hand_binary"
1319
+ },
1320
+ {
1321
+ "hdf5_key": "ep_00123",
1322
+ "source_episode_index": 123,
1323
+ "split": "train",
1324
+ "num_samples": 602,
1325
+ "action_dim": 7,
1326
+ "obs_mode": "vision_wrench_state",
1327
+ "action_mode": "post_velcmd_hand_binary"
1328
+ },
1329
+ {
1330
+ "hdf5_key": "ep_00124",
1331
+ "source_episode_index": 124,
1332
+ "split": "train",
1333
+ "num_samples": 630,
1334
+ "action_dim": 7,
1335
+ "obs_mode": "vision_wrench_state",
1336
+ "action_mode": "post_velcmd_hand_binary"
1337
+ },
1338
+ {
1339
+ "hdf5_key": "ep_00125",
1340
+ "source_episode_index": 125,
1341
+ "split": "train",
1342
+ "num_samples": 621,
1343
+ "action_dim": 7,
1344
+ "obs_mode": "vision_wrench_state",
1345
+ "action_mode": "post_velcmd_hand_binary"
1346
+ },
1347
+ {
1348
+ "hdf5_key": "ep_00126",
1349
+ "source_episode_index": 126,
1350
+ "split": "train",
1351
+ "num_samples": 600,
1352
+ "action_dim": 7,
1353
+ "obs_mode": "vision_wrench_state",
1354
+ "action_mode": "post_velcmd_hand_binary"
1355
+ },
1356
+ {
1357
+ "hdf5_key": "ep_00127",
1358
+ "source_episode_index": 127,
1359
+ "split": "train",
1360
+ "num_samples": 478,
1361
+ "action_dim": 7,
1362
+ "obs_mode": "vision_wrench_state",
1363
+ "action_mode": "post_velcmd_hand_binary"
1364
+ },
1365
+ {
1366
+ "hdf5_key": "ep_00128",
1367
+ "source_episode_index": 128,
1368
+ "split": "train",
1369
+ "num_samples": 606,
1370
+ "action_dim": 7,
1371
+ "obs_mode": "vision_wrench_state",
1372
+ "action_mode": "post_velcmd_hand_binary"
1373
+ },
1374
+ {
1375
+ "hdf5_key": "ep_00129",
1376
+ "source_episode_index": 129,
1377
+ "split": "train",
1378
+ "num_samples": 666,
1379
+ "action_dim": 7,
1380
+ "obs_mode": "vision_wrench_state",
1381
+ "action_mode": "post_velcmd_hand_binary"
1382
+ },
1383
+ {
1384
+ "hdf5_key": "ep_00130",
1385
+ "source_episode_index": 130,
1386
+ "split": "train",
1387
+ "num_samples": 655,
1388
+ "action_dim": 7,
1389
+ "obs_mode": "vision_wrench_state",
1390
+ "action_mode": "post_velcmd_hand_binary"
1391
+ },
1392
+ {
1393
+ "hdf5_key": "ep_00131",
1394
+ "source_episode_index": 131,
1395
+ "split": "train",
1396
+ "num_samples": 588,
1397
+ "action_dim": 7,
1398
+ "obs_mode": "vision_wrench_state",
1399
+ "action_mode": "post_velcmd_hand_binary"
1400
+ },
1401
+ {
1402
+ "hdf5_key": "ep_00132",
1403
+ "source_episode_index": 132,
1404
+ "split": "train",
1405
+ "num_samples": 806,
1406
+ "action_dim": 7,
1407
+ "obs_mode": "vision_wrench_state",
1408
+ "action_mode": "post_velcmd_hand_binary"
1409
+ },
1410
+ {
1411
+ "hdf5_key": "ep_00133",
1412
+ "source_episode_index": 133,
1413
+ "split": "train",
1414
+ "num_samples": 586,
1415
+ "action_dim": 7,
1416
+ "obs_mode": "vision_wrench_state",
1417
+ "action_mode": "post_velcmd_hand_binary"
1418
+ },
1419
+ {
1420
+ "hdf5_key": "ep_00134",
1421
+ "source_episode_index": 134,
1422
+ "split": "train",
1423
+ "num_samples": 714,
1424
+ "action_dim": 7,
1425
+ "obs_mode": "vision_wrench_state",
1426
+ "action_mode": "post_velcmd_hand_binary"
1427
+ },
1428
+ {
1429
+ "hdf5_key": "ep_00135",
1430
+ "source_episode_index": 135,
1431
+ "split": "train",
1432
+ "num_samples": 574,
1433
+ "action_dim": 7,
1434
+ "obs_mode": "vision_wrench_state",
1435
+ "action_mode": "post_velcmd_hand_binary"
1436
+ },
1437
+ {
1438
+ "hdf5_key": "ep_00136",
1439
+ "source_episode_index": 136,
1440
+ "split": "train",
1441
+ "num_samples": 577,
1442
+ "action_dim": 7,
1443
+ "obs_mode": "vision_wrench_state",
1444
+ "action_mode": "post_velcmd_hand_binary"
1445
+ },
1446
+ {
1447
+ "hdf5_key": "ep_00137",
1448
+ "source_episode_index": 137,
1449
+ "split": "train",
1450
+ "num_samples": 663,
1451
+ "action_dim": 7,
1452
+ "obs_mode": "vision_wrench_state",
1453
+ "action_mode": "post_velcmd_hand_binary"
1454
+ },
1455
+ {
1456
+ "hdf5_key": "ep_00138",
1457
+ "source_episode_index": 138,
1458
+ "split": "train",
1459
+ "num_samples": 720,
1460
+ "action_dim": 7,
1461
+ "obs_mode": "vision_wrench_state",
1462
+ "action_mode": "post_velcmd_hand_binary"
1463
+ },
1464
+ {
1465
+ "hdf5_key": "ep_00139",
1466
+ "source_episode_index": 139,
1467
+ "split": "train",
1468
+ "num_samples": 640,
1469
+ "action_dim": 7,
1470
+ "obs_mode": "vision_wrench_state",
1471
+ "action_mode": "post_velcmd_hand_binary"
1472
+ },
1473
+ {
1474
+ "hdf5_key": "ep_00140",
1475
+ "source_episode_index": 140,
1476
+ "split": "train",
1477
+ "num_samples": 544,
1478
+ "action_dim": 7,
1479
+ "obs_mode": "vision_wrench_state",
1480
+ "action_mode": "post_velcmd_hand_binary"
1481
+ },
1482
+ {
1483
+ "hdf5_key": "ep_00141",
1484
+ "source_episode_index": 141,
1485
+ "split": "train",
1486
+ "num_samples": 539,
1487
+ "action_dim": 7,
1488
+ "obs_mode": "vision_wrench_state",
1489
+ "action_mode": "post_velcmd_hand_binary"
1490
+ },
1491
+ {
1492
+ "hdf5_key": "ep_00142",
1493
+ "source_episode_index": 142,
1494
+ "split": "train",
1495
+ "num_samples": 592,
1496
+ "action_dim": 7,
1497
+ "obs_mode": "vision_wrench_state",
1498
+ "action_mode": "post_velcmd_hand_binary"
1499
+ },
1500
+ {
1501
+ "hdf5_key": "ep_00143",
1502
+ "source_episode_index": 143,
1503
+ "split": "train",
1504
+ "num_samples": 625,
1505
+ "action_dim": 7,
1506
+ "obs_mode": "vision_wrench_state",
1507
+ "action_mode": "post_velcmd_hand_binary"
1508
+ },
1509
+ {
1510
+ "hdf5_key": "ep_00144",
1511
+ "source_episode_index": 144,
1512
+ "split": "train",
1513
+ "num_samples": 684,
1514
+ "action_dim": 7,
1515
+ "obs_mode": "vision_wrench_state",
1516
+ "action_mode": "post_velcmd_hand_binary"
1517
+ },
1518
+ {
1519
+ "hdf5_key": "ep_00145",
1520
+ "source_episode_index": 145,
1521
+ "split": "train",
1522
+ "num_samples": 559,
1523
+ "action_dim": 7,
1524
+ "obs_mode": "vision_wrench_state",
1525
+ "action_mode": "post_velcmd_hand_binary"
1526
+ },
1527
+ {
1528
+ "hdf5_key": "ep_00146",
1529
+ "source_episode_index": 146,
1530
+ "split": "train",
1531
+ "num_samples": 677,
1532
+ "action_dim": 7,
1533
+ "obs_mode": "vision_wrench_state",
1534
+ "action_mode": "post_velcmd_hand_binary"
1535
+ },
1536
+ {
1537
+ "hdf5_key": "ep_00147",
1538
+ "source_episode_index": 147,
1539
+ "split": "train",
1540
+ "num_samples": 846,
1541
+ "action_dim": 7,
1542
+ "obs_mode": "vision_wrench_state",
1543
+ "action_mode": "post_velcmd_hand_binary"
1544
+ },
1545
+ {
1546
+ "hdf5_key": "ep_00148",
1547
+ "source_episode_index": 148,
1548
+ "split": "train",
1549
+ "num_samples": 615,
1550
+ "action_dim": 7,
1551
+ "obs_mode": "vision_wrench_state",
1552
+ "action_mode": "post_velcmd_hand_binary"
1553
+ },
1554
+ {
1555
+ "hdf5_key": "ep_00149",
1556
+ "source_episode_index": 149,
1557
+ "split": "train",
1558
+ "num_samples": 517,
1559
+ "action_dim": 7,
1560
+ "obs_mode": "vision_wrench_state",
1561
+ "action_mode": "post_velcmd_hand_binary"
1562
+ },
1563
+ {
1564
+ "hdf5_key": "ep_00150",
1565
+ "source_episode_index": 150,
1566
+ "split": "train",
1567
+ "num_samples": 817,
1568
+ "action_dim": 7,
1569
+ "obs_mode": "vision_wrench_state",
1570
+ "action_mode": "post_velcmd_hand_binary"
1571
+ },
1572
+ {
1573
+ "hdf5_key": "ep_00151",
1574
+ "source_episode_index": 151,
1575
+ "split": "train",
1576
+ "num_samples": 814,
1577
+ "action_dim": 7,
1578
+ "obs_mode": "vision_wrench_state",
1579
+ "action_mode": "post_velcmd_hand_binary"
1580
+ },
1581
+ {
1582
+ "hdf5_key": "ep_00152",
1583
+ "source_episode_index": 152,
1584
+ "split": "train",
1585
+ "num_samples": 828,
1586
+ "action_dim": 7,
1587
+ "obs_mode": "vision_wrench_state",
1588
+ "action_mode": "post_velcmd_hand_binary"
1589
+ },
1590
+ {
1591
+ "hdf5_key": "ep_00153",
1592
+ "source_episode_index": 153,
1593
+ "split": "train",
1594
+ "num_samples": 736,
1595
+ "action_dim": 7,
1596
+ "obs_mode": "vision_wrench_state",
1597
+ "action_mode": "post_velcmd_hand_binary"
1598
+ },
1599
+ {
1600
+ "hdf5_key": "ep_00154",
1601
+ "source_episode_index": 154,
1602
+ "split": "train",
1603
+ "num_samples": 860,
1604
+ "action_dim": 7,
1605
+ "obs_mode": "vision_wrench_state",
1606
+ "action_mode": "post_velcmd_hand_binary"
1607
+ },
1608
+ {
1609
+ "hdf5_key": "ep_00155",
1610
+ "source_episode_index": 155,
1611
+ "split": "train",
1612
+ "num_samples": 786,
1613
+ "action_dim": 7,
1614
+ "obs_mode": "vision_wrench_state",
1615
+ "action_mode": "post_velcmd_hand_binary"
1616
+ },
1617
+ {
1618
+ "hdf5_key": "ep_00156",
1619
+ "source_episode_index": 156,
1620
+ "split": "train",
1621
+ "num_samples": 725,
1622
+ "action_dim": 7,
1623
+ "obs_mode": "vision_wrench_state",
1624
+ "action_mode": "post_velcmd_hand_binary"
1625
+ },
1626
+ {
1627
+ "hdf5_key": "ep_00157",
1628
+ "source_episode_index": 157,
1629
+ "split": "train",
1630
+ "num_samples": 834,
1631
+ "action_dim": 7,
1632
+ "obs_mode": "vision_wrench_state",
1633
+ "action_mode": "post_velcmd_hand_binary"
1634
+ },
1635
+ {
1636
+ "hdf5_key": "ep_00158",
1637
+ "source_episode_index": 158,
1638
+ "split": "train",
1639
+ "num_samples": 652,
1640
+ "action_dim": 7,
1641
+ "obs_mode": "vision_wrench_state",
1642
+ "action_mode": "post_velcmd_hand_binary"
1643
+ },
1644
+ {
1645
+ "hdf5_key": "ep_00159",
1646
+ "source_episode_index": 159,
1647
+ "split": "train",
1648
+ "num_samples": 699,
1649
+ "action_dim": 7,
1650
+ "obs_mode": "vision_wrench_state",
1651
+ "action_mode": "post_velcmd_hand_binary"
1652
+ },
1653
+ {
1654
+ "hdf5_key": "ep_00160",
1655
+ "source_episode_index": 160,
1656
+ "split": "train",
1657
+ "num_samples": 812,
1658
+ "action_dim": 7,
1659
+ "obs_mode": "vision_wrench_state",
1660
+ "action_mode": "post_velcmd_hand_binary"
1661
+ },
1662
+ {
1663
+ "hdf5_key": "ep_00161",
1664
+ "source_episode_index": 161,
1665
+ "split": "train",
1666
+ "num_samples": 782,
1667
+ "action_dim": 7,
1668
+ "obs_mode": "vision_wrench_state",
1669
+ "action_mode": "post_velcmd_hand_binary"
1670
+ },
1671
+ {
1672
+ "hdf5_key": "ep_00162",
1673
+ "source_episode_index": 162,
1674
+ "split": "train",
1675
+ "num_samples": 681,
1676
+ "action_dim": 7,
1677
+ "obs_mode": "vision_wrench_state",
1678
+ "action_mode": "post_velcmd_hand_binary"
1679
+ },
1680
+ {
1681
+ "hdf5_key": "ep_00163",
1682
+ "source_episode_index": 163,
1683
+ "split": "train",
1684
+ "num_samples": 742,
1685
+ "action_dim": 7,
1686
+ "obs_mode": "vision_wrench_state",
1687
+ "action_mode": "post_velcmd_hand_binary"
1688
+ },
1689
+ {
1690
+ "hdf5_key": "ep_00164",
1691
+ "source_episode_index": 164,
1692
+ "split": "train",
1693
+ "num_samples": 766,
1694
+ "action_dim": 7,
1695
+ "obs_mode": "vision_wrench_state",
1696
+ "action_mode": "post_velcmd_hand_binary"
1697
+ },
1698
+ {
1699
+ "hdf5_key": "ep_00165",
1700
+ "source_episode_index": 165,
1701
+ "split": "train",
1702
+ "num_samples": 821,
1703
+ "action_dim": 7,
1704
+ "obs_mode": "vision_wrench_state",
1705
+ "action_mode": "post_velcmd_hand_binary"
1706
+ },
1707
+ {
1708
+ "hdf5_key": "ep_00166",
1709
+ "source_episode_index": 166,
1710
+ "split": "train",
1711
+ "num_samples": 1028,
1712
+ "action_dim": 7,
1713
+ "obs_mode": "vision_wrench_state",
1714
+ "action_mode": "post_velcmd_hand_binary"
1715
+ },
1716
+ {
1717
+ "hdf5_key": "ep_00167",
1718
+ "source_episode_index": 167,
1719
+ "split": "train",
1720
+ "num_samples": 720,
1721
+ "action_dim": 7,
1722
+ "obs_mode": "vision_wrench_state",
1723
+ "action_mode": "post_velcmd_hand_binary"
1724
+ },
1725
+ {
1726
+ "hdf5_key": "ep_00168",
1727
+ "source_episode_index": 168,
1728
+ "split": "train",
1729
+ "num_samples": 773,
1730
+ "action_dim": 7,
1731
+ "obs_mode": "vision_wrench_state",
1732
+ "action_mode": "post_velcmd_hand_binary"
1733
+ },
1734
+ {
1735
+ "hdf5_key": "ep_00169",
1736
+ "source_episode_index": 169,
1737
+ "split": "train",
1738
+ "num_samples": 690,
1739
+ "action_dim": 7,
1740
+ "obs_mode": "vision_wrench_state",
1741
+ "action_mode": "post_velcmd_hand_binary"
1742
+ },
1743
+ {
1744
+ "hdf5_key": "ep_00170",
1745
+ "source_episode_index": 170,
1746
+ "split": "train",
1747
+ "num_samples": 756,
1748
+ "action_dim": 7,
1749
+ "obs_mode": "vision_wrench_state",
1750
+ "action_mode": "post_velcmd_hand_binary"
1751
+ },
1752
+ {
1753
+ "hdf5_key": "ep_00171",
1754
+ "source_episode_index": 171,
1755
+ "split": "train",
1756
+ "num_samples": 742,
1757
+ "action_dim": 7,
1758
+ "obs_mode": "vision_wrench_state",
1759
+ "action_mode": "post_velcmd_hand_binary"
1760
+ },
1761
+ {
1762
+ "hdf5_key": "ep_00172",
1763
+ "source_episode_index": 172,
1764
+ "split": "valid",
1765
+ "num_samples": 689,
1766
+ "action_dim": 7,
1767
+ "obs_mode": "vision_wrench_state",
1768
+ "action_mode": "post_velcmd_hand_binary"
1769
+ },
1770
+ {
1771
+ "hdf5_key": "ep_00173",
1772
+ "source_episode_index": 173,
1773
+ "split": "valid",
1774
+ "num_samples": 639,
1775
+ "action_dim": 7,
1776
+ "obs_mode": "vision_wrench_state",
1777
+ "action_mode": "post_velcmd_hand_binary"
1778
+ },
1779
+ {
1780
+ "hdf5_key": "ep_00174",
1781
+ "source_episode_index": 174,
1782
+ "split": "valid",
1783
+ "num_samples": 694,
1784
+ "action_dim": 7,
1785
+ "obs_mode": "vision_wrench_state",
1786
+ "action_mode": "post_velcmd_hand_binary"
1787
+ },
1788
+ {
1789
+ "hdf5_key": "ep_00175",
1790
+ "source_episode_index": 175,
1791
+ "split": "valid",
1792
+ "num_samples": 532,
1793
+ "action_dim": 7,
1794
+ "obs_mode": "vision_wrench_state",
1795
+ "action_mode": "post_velcmd_hand_binary"
1796
+ },
1797
+ {
1798
+ "hdf5_key": "ep_00176",
1799
+ "source_episode_index": 176,
1800
+ "split": "valid",
1801
+ "num_samples": 665,
1802
+ "action_dim": 7,
1803
+ "obs_mode": "vision_wrench_state",
1804
+ "action_mode": "post_velcmd_hand_binary"
1805
+ },
1806
+ {
1807
+ "hdf5_key": "ep_00177",
1808
+ "source_episode_index": 177,
1809
+ "split": "valid",
1810
+ "num_samples": 771,
1811
+ "action_dim": 7,
1812
+ "obs_mode": "vision_wrench_state",
1813
+ "action_mode": "post_velcmd_hand_binary"
1814
+ },
1815
+ {
1816
+ "hdf5_key": "ep_00178",
1817
+ "source_episode_index": 178,
1818
+ "split": "valid",
1819
+ "num_samples": 602,
1820
+ "action_dim": 7,
1821
+ "obs_mode": "vision_wrench_state",
1822
+ "action_mode": "post_velcmd_hand_binary"
1823
+ },
1824
+ {
1825
+ "hdf5_key": "ep_00179",
1826
+ "source_episode_index": 179,
1827
+ "split": "valid",
1828
+ "num_samples": 700,
1829
+ "action_dim": 7,
1830
+ "obs_mode": "vision_wrench_state",
1831
+ "action_mode": "post_velcmd_hand_binary"
1832
+ },
1833
+ {
1834
+ "hdf5_key": "ep_00180",
1835
+ "source_episode_index": 180,
1836
+ "split": "valid",
1837
+ "num_samples": 880,
1838
+ "action_dim": 7,
1839
+ "obs_mode": "vision_wrench_state",
1840
+ "action_mode": "post_velcmd_hand_binary"
1841
+ },
1842
+ {
1843
+ "hdf5_key": "ep_00181",
1844
+ "source_episode_index": 181,
1845
+ "split": "valid",
1846
+ "num_samples": 522,
1847
+ "action_dim": 7,
1848
+ "obs_mode": "vision_wrench_state",
1849
+ "action_mode": "post_velcmd_hand_binary"
1850
+ },
1851
+ {
1852
+ "hdf5_key": "ep_00182",
1853
+ "source_episode_index": 182,
1854
+ "split": "valid",
1855
+ "num_samples": 697,
1856
+ "action_dim": 7,
1857
+ "obs_mode": "vision_wrench_state",
1858
+ "action_mode": "post_velcmd_hand_binary"
1859
+ },
1860
+ {
1861
+ "hdf5_key": "ep_00183",
1862
+ "source_episode_index": 183,
1863
+ "split": "valid",
1864
+ "num_samples": 621,
1865
+ "action_dim": 7,
1866
+ "obs_mode": "vision_wrench_state",
1867
+ "action_mode": "post_velcmd_hand_binary"
1868
+ },
1869
+ {
1870
+ "hdf5_key": "ep_00184",
1871
+ "source_episode_index": 184,
1872
+ "split": "valid",
1873
+ "num_samples": 710,
1874
+ "action_dim": 7,
1875
+ "obs_mode": "vision_wrench_state",
1876
+ "action_mode": "post_velcmd_hand_binary"
1877
+ },
1878
+ {
1879
+ "hdf5_key": "ep_00185",
1880
+ "source_episode_index": 185,
1881
+ "split": "valid",
1882
+ "num_samples": 774,
1883
+ "action_dim": 7,
1884
+ "obs_mode": "vision_wrench_state",
1885
+ "action_mode": "post_velcmd_hand_binary"
1886
+ },
1887
+ {
1888
+ "hdf5_key": "ep_00186",
1889
+ "source_episode_index": 186,
1890
+ "split": "valid",
1891
+ "num_samples": 682,
1892
+ "action_dim": 7,
1893
+ "obs_mode": "vision_wrench_state",
1894
+ "action_mode": "post_velcmd_hand_binary"
1895
+ },
1896
+ {
1897
+ "hdf5_key": "ep_00187",
1898
+ "source_episode_index": 187,
1899
+ "split": "valid",
1900
+ "num_samples": 715,
1901
+ "action_dim": 7,
1902
+ "obs_mode": "vision_wrench_state",
1903
+ "action_mode": "post_velcmd_hand_binary"
1904
+ },
1905
+ {
1906
+ "hdf5_key": "ep_00188",
1907
+ "source_episode_index": 188,
1908
+ "split": "valid",
1909
+ "num_samples": 798,
1910
+ "action_dim": 7,
1911
+ "obs_mode": "vision_wrench_state",
1912
+ "action_mode": "post_velcmd_hand_binary"
1913
+ },
1914
+ {
1915
+ "hdf5_key": "ep_00189",
1916
+ "source_episode_index": 189,
1917
+ "split": "valid",
1918
+ "num_samples": 786,
1919
+ "action_dim": 7,
1920
+ "obs_mode": "vision_wrench_state",
1921
+ "action_mode": "post_velcmd_hand_binary"
1922
+ },
1923
+ {
1924
+ "hdf5_key": "ep_00190",
1925
+ "source_episode_index": 190,
1926
+ "split": "valid",
1927
+ "num_samples": 759,
1928
+ "action_dim": 7,
1929
+ "obs_mode": "vision_wrench_state",
1930
+ "action_mode": "post_velcmd_hand_binary"
1931
+ },
1932
+ {
1933
+ "hdf5_key": "ep_00191",
1934
+ "source_episode_index": 191,
1935
+ "split": "valid",
1936
+ "num_samples": 812,
1937
+ "action_dim": 7,
1938
+ "obs_mode": "vision_wrench_state",
1939
+ "action_mode": "post_velcmd_hand_binary"
1940
+ },
1941
+ {
1942
+ "hdf5_key": "ep_00192",
1943
+ "source_episode_index": 192,
1944
+ "split": "valid",
1945
+ "num_samples": 803,
1946
+ "action_dim": 7,
1947
+ "obs_mode": "vision_wrench_state",
1948
+ "action_mode": "post_velcmd_hand_binary"
1949
+ },
1950
+ {
1951
+ "hdf5_key": "ep_00193",
1952
+ "source_episode_index": 193,
1953
+ "split": "valid",
1954
+ "num_samples": 780,
1955
+ "action_dim": 7,
1956
+ "obs_mode": "vision_wrench_state",
1957
+ "action_mode": "post_velcmd_hand_binary"
1958
+ },
1959
+ {
1960
+ "hdf5_key": "ep_00194",
1961
+ "source_episode_index": 194,
1962
+ "split": "valid",
1963
+ "num_samples": 853,
1964
+ "action_dim": 7,
1965
+ "obs_mode": "vision_wrench_state",
1966
+ "action_mode": "post_velcmd_hand_binary"
1967
+ },
1968
+ {
1969
+ "hdf5_key": "ep_00195",
1970
+ "source_episode_index": 195,
1971
+ "split": "valid",
1972
+ "num_samples": 710,
1973
+ "action_dim": 7,
1974
+ "obs_mode": "vision_wrench_state",
1975
+ "action_mode": "post_velcmd_hand_binary"
1976
+ },
1977
+ {
1978
+ "hdf5_key": "ep_00196",
1979
+ "source_episode_index": 196,
1980
+ "split": "valid",
1981
+ "num_samples": 855,
1982
+ "action_dim": 7,
1983
+ "obs_mode": "vision_wrench_state",
1984
+ "action_mode": "post_velcmd_hand_binary"
1985
+ },
1986
+ {
1987
+ "hdf5_key": "ep_00197",
1988
+ "source_episode_index": 197,
1989
+ "split": "valid",
1990
+ "num_samples": 707,
1991
+ "action_dim": 7,
1992
+ "obs_mode": "vision_wrench_state",
1993
+ "action_mode": "post_velcmd_hand_binary"
1994
+ },
1995
+ {
1996
+ "hdf5_key": "ep_00198",
1997
+ "source_episode_index": 198,
1998
+ "split": "valid",
1999
+ "num_samples": 557,
2000
+ "action_dim": 7,
2001
+ "obs_mode": "vision_wrench_state",
2002
+ "action_mode": "post_velcmd_hand_binary"
2003
+ },
2004
+ {
2005
+ "hdf5_key": "ep_00199",
2006
+ "source_episode_index": 199,
2007
+ "split": "valid",
2008
+ "num_samples": 522,
2009
+ "action_dim": 7,
2010
+ "obs_mode": "vision_wrench_state",
2011
+ "action_mode": "post_velcmd_hand_binary"
2012
+ },
2013
+ {
2014
+ "hdf5_key": "ep_00200",
2015
+ "source_episode_index": 200,
2016
+ "split": "valid",
2017
+ "num_samples": 581,
2018
+ "action_dim": 7,
2019
+ "obs_mode": "vision_wrench_state",
2020
+ "action_mode": "post_velcmd_hand_binary"
2021
+ },
2022
+ {
2023
+ "hdf5_key": "ep_00201",
2024
+ "source_episode_index": 201,
2025
+ "split": "valid",
2026
+ "num_samples": 574,
2027
+ "action_dim": 7,
2028
+ "obs_mode": "vision_wrench_state",
2029
+ "action_mode": "post_velcmd_hand_binary"
2030
+ }
2031
+ ]
2032
+ }
vac_post_vis_wrench_state/last.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2cb19b54dee0d39a9ef94a97e23096aad461c0946b707cb1a01058b156fe2f49
3
+ size 1260636893
vac_pre_vis/action_stats.json ADDED
@@ -0,0 +1,311 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "method": "min_max",
3
+ "output_min": -0.999999,
4
+ "output_max": 0.999999,
5
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary",
6
+ "action_components": [
7
+ "user_cmd_dx",
8
+ "user_cmd_dy",
9
+ "user_cmd_dz",
10
+ "user_cmd_drx",
11
+ "user_cmd_dry",
12
+ "user_cmd_drz",
13
+ "stiffness_cmd",
14
+ "zeta",
15
+ "hand_open_binary"
16
+ ],
17
+ "num_frames": 124722,
18
+ "min": [
19
+ -0.0020000000949949026,
20
+ -0.0020000000949949026,
21
+ -0.0020000000949949026,
22
+ -0.017500000074505806,
23
+ 0.0,
24
+ -0.017500000074505806,
25
+ 3.5638904571533203,
26
+ 0.8021645545959473,
27
+ 0.0
28
+ ],
29
+ "max": [
30
+ 0.0020000000949949026,
31
+ 0.0020000000949949026,
32
+ 0.0020000000949949026,
33
+ 0.017500000074505806,
34
+ 0.0020832535810768604,
35
+ 0.017500000074505806,
36
+ 1997.13330078125,
37
+ 2.5,
38
+ 1.0
39
+ ],
40
+ "mean": [
41
+ -1.295681795454584e-05,
42
+ -4.36740810982883e-05,
43
+ 5.51424891455099e-05,
44
+ -1.8853129404305946e-06,
45
+ 6.276949449102176e-08,
46
+ -1.684121525613591e-05,
47
+ 1533.5157470703125,
48
+ 0.9446797370910645,
49
+ 0.428376704454422
50
+ ],
51
+ "std": [
52
+ 0.00042260822374373674,
53
+ 0.0009382680291309953,
54
+ 0.000991749344393611,
55
+ 0.0003852724621538073,
56
+ 1.1097310562036e-05,
57
+ 0.001960940193384886,
58
+ 461.547119140625,
59
+ 0.24154788255691528,
60
+ 0.4948435127735138
61
+ ],
62
+ "scale": [
63
+ 0.0020000021904706955,
64
+ 0.0020000021904706955,
65
+ 0.0020000021904706955,
66
+ 0.0175000187009573,
67
+ 0.0010416278382763267,
68
+ 0.0175000187009573,
69
+ 996.7857666015625,
70
+ 0.8489185571670532,
71
+ 0.500000536441803
72
+ ],
73
+ "offset": [
74
+ 0.0,
75
+ 0.0,
76
+ 0.0,
77
+ 0.0,
78
+ 0.0010416267905384302,
79
+ 0.0,
80
+ 1000.3486328125,
81
+ 1.6510822772979736,
82
+ 0.5
83
+ ],
84
+ "hand_source": "action.absolute",
85
+ "hand_active_dims": [
86
+ 8,
87
+ 9,
88
+ 10,
89
+ 11
90
+ ],
91
+ "hand_reduce": "mean",
92
+ "hand_threshold": 0.85,
93
+ "hand_high_value": 1.0,
94
+ "hand_low_value": 0.0,
95
+ "dataset": "/workspace/datasets/dataset_pipe_02_06",
96
+ "observation_mode_examples": [
97
+ "vision",
98
+ "vision_wrench",
99
+ "vision_wrench_state"
100
+ ],
101
+ "train_source_episodes": [
102
+ 0,
103
+ 1,
104
+ 2,
105
+ 3,
106
+ 4,
107
+ 5,
108
+ 6,
109
+ 7,
110
+ 8,
111
+ 9,
112
+ 10,
113
+ 11,
114
+ 12,
115
+ 13,
116
+ 14,
117
+ 15,
118
+ 16,
119
+ 17,
120
+ 18,
121
+ 19,
122
+ 20,
123
+ 21,
124
+ 22,
125
+ 23,
126
+ 24,
127
+ 25,
128
+ 26,
129
+ 27,
130
+ 28,
131
+ 29,
132
+ 30,
133
+ 31,
134
+ 32,
135
+ 33,
136
+ 34,
137
+ 35,
138
+ 36,
139
+ 37,
140
+ 38,
141
+ 39,
142
+ 40,
143
+ 41,
144
+ 42,
145
+ 43,
146
+ 44,
147
+ 45,
148
+ 46,
149
+ 47,
150
+ 48,
151
+ 49,
152
+ 50,
153
+ 51,
154
+ 52,
155
+ 53,
156
+ 54,
157
+ 55,
158
+ 56,
159
+ 57,
160
+ 58,
161
+ 59,
162
+ 60,
163
+ 61,
164
+ 62,
165
+ 63,
166
+ 64,
167
+ 65,
168
+ 66,
169
+ 67,
170
+ 68,
171
+ 69,
172
+ 70,
173
+ 71,
174
+ 72,
175
+ 73,
176
+ 74,
177
+ 75,
178
+ 76,
179
+ 77,
180
+ 78,
181
+ 79,
182
+ 80,
183
+ 81,
184
+ 82,
185
+ 83,
186
+ 84,
187
+ 85,
188
+ 86,
189
+ 87,
190
+ 88,
191
+ 89,
192
+ 90,
193
+ 91,
194
+ 92,
195
+ 93,
196
+ 94,
197
+ 95,
198
+ 96,
199
+ 97,
200
+ 98,
201
+ 99,
202
+ 100,
203
+ 101,
204
+ 102,
205
+ 103,
206
+ 104,
207
+ 105,
208
+ 106,
209
+ 107,
210
+ 108,
211
+ 109,
212
+ 110,
213
+ 111,
214
+ 112,
215
+ 113,
216
+ 114,
217
+ 115,
218
+ 116,
219
+ 117,
220
+ 118,
221
+ 119,
222
+ 120,
223
+ 121,
224
+ 122,
225
+ 123,
226
+ 124,
227
+ 125,
228
+ 126,
229
+ 127,
230
+ 128,
231
+ 129,
232
+ 130,
233
+ 131,
234
+ 132,
235
+ 133,
236
+ 134,
237
+ 135,
238
+ 136,
239
+ 137,
240
+ 138,
241
+ 139,
242
+ 140,
243
+ 141,
244
+ 142,
245
+ 143,
246
+ 144,
247
+ 145,
248
+ 146,
249
+ 147,
250
+ 148,
251
+ 149,
252
+ 150,
253
+ 151,
254
+ 152,
255
+ 153,
256
+ 154,
257
+ 155,
258
+ 156,
259
+ 157,
260
+ 158,
261
+ 159,
262
+ 160,
263
+ 161,
264
+ 162,
265
+ 163,
266
+ 164,
267
+ 165,
268
+ 166,
269
+ 167,
270
+ 168,
271
+ 169,
272
+ 170,
273
+ 171
274
+ ],
275
+ "valid_source_episodes": [
276
+ 172,
277
+ 173,
278
+ 174,
279
+ 175,
280
+ 176,
281
+ 177,
282
+ 178,
283
+ 179,
284
+ 180,
285
+ 181,
286
+ 182,
287
+ 183,
288
+ 184,
289
+ 185,
290
+ 186,
291
+ 187,
292
+ 188,
293
+ 189,
294
+ 190,
295
+ 191,
296
+ 192,
297
+ 193,
298
+ 194,
299
+ 195,
300
+ 196,
301
+ 197,
302
+ 198,
303
+ 199,
304
+ 200,
305
+ 201
306
+ ],
307
+ "image_size": [
308
+ 84,
309
+ 84
310
+ ]
311
+ }
vac_pre_vis/best.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a57ddfc5f6228becae363435ffae4c6f159639e99eb935447b56a155d234922
3
+ size 1252540221
vac_pre_vis/config.json ADDED
@@ -0,0 +1,195 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "algo_name": "diffusion_policy",
3
+ "experiment": {
4
+ "name": "vac_pre_vis",
5
+ "validate": true,
6
+ "logging": {
7
+ "terminal_output_to_txt": false,
8
+ "log_tb": true,
9
+ "log_wandb": false,
10
+ "wandb_proj_name": "debug"
11
+ },
12
+ "save": {
13
+ "enabled": true,
14
+ "every_n_seconds": null,
15
+ "every_n_epochs": 50,
16
+ "epochs": [],
17
+ "on_best_validation": true,
18
+ "on_best_rollout_return": false,
19
+ "on_best_rollout_success_rate": false
20
+ },
21
+ "epoch_every_n_steps": 100,
22
+ "validation_epoch_every_n_steps": 10,
23
+ "env": null,
24
+ "additional_envs": null,
25
+ "render": false,
26
+ "render_video": false,
27
+ "keep_all_videos": false,
28
+ "video_skip": 5,
29
+ "rollout": {
30
+ "enabled": false,
31
+ "n": 0,
32
+ "horizon": 50,
33
+ "rate": 50,
34
+ "warmstart": 0,
35
+ "terminate_on_success": true
36
+ },
37
+ "env_meta_update_dict": {},
38
+ "ckpt_path": null
39
+ },
40
+ "train": {
41
+ "data": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_pre_vis/vac_pre_vis.hdf5",
42
+ "output_dir": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_pre_vis/robomimic_runs",
43
+ "normalize_weights_by_ds_size": false,
44
+ "num_data_workers": 0,
45
+ "hdf5_cache_mode": "all",
46
+ "hdf5_use_swmr": true,
47
+ "hdf5_load_next_obs": false,
48
+ "hdf5_normalize_obs": false,
49
+ "hdf5_filter_key": "train",
50
+ "hdf5_validation_filter_key": "valid",
51
+ "seq_length": 8,
52
+ "pad_seq_length": true,
53
+ "frame_stack": 2,
54
+ "pad_frame_stack": true,
55
+ "dataset_keys": [
56
+ "actions",
57
+ "rewards",
58
+ "dones"
59
+ ],
60
+ "action_keys": [
61
+ "actions"
62
+ ],
63
+ "action_config": {
64
+ "actions": {
65
+ "normalization": "min_max"
66
+ }
67
+ },
68
+ "goal_mode": null,
69
+ "cuda": true,
70
+ "batch_size": 16,
71
+ "num_epochs": 300,
72
+ "seed": 42,
73
+ "max_grad_norm": null
74
+ },
75
+ "algo": {
76
+ "optim_params": {
77
+ "policy": {
78
+ "optimizer_type": "adamw",
79
+ "learning_rate": {
80
+ "initial": 0.0001,
81
+ "decay_factor": 0.1,
82
+ "step_every_batch": true,
83
+ "scheduler_type": "cosine",
84
+ "num_cycles": 0.5,
85
+ "warmup_steps": 500,
86
+ "epoch_schedule": []
87
+ },
88
+ "regularization": {
89
+ "L2": 1e-06
90
+ }
91
+ }
92
+ },
93
+ "horizon": {
94
+ "observation_horizon": 2,
95
+ "action_horizon": 4,
96
+ "prediction_horizon": 8
97
+ },
98
+ "unet": {
99
+ "enabled": true,
100
+ "diffusion_step_embed_dim": 256,
101
+ "down_dims": [
102
+ 128,
103
+ 256,
104
+ 512
105
+ ],
106
+ "kernel_size": 5,
107
+ "n_groups": 8
108
+ },
109
+ "ema": {
110
+ "enabled": true,
111
+ "power": 0.75
112
+ },
113
+ "ddpm": {
114
+ "enabled": true,
115
+ "num_train_timesteps": 50,
116
+ "num_inference_timesteps": 50,
117
+ "beta_schedule": "squaredcos_cap_v2",
118
+ "clip_sample": true,
119
+ "prediction_type": "epsilon"
120
+ },
121
+ "ddim": {
122
+ "enabled": false,
123
+ "num_train_timesteps": 50,
124
+ "num_inference_timesteps": 10,
125
+ "beta_schedule": "squaredcos_cap_v2",
126
+ "clip_sample": true,
127
+ "set_alpha_to_one": true,
128
+ "steps_offset": 0,
129
+ "prediction_type": "epsilon"
130
+ }
131
+ },
132
+ "observation": {
133
+ "modalities": {
134
+ "obs": {
135
+ "low_dim": [],
136
+ "rgb": [
137
+ "front_rgb"
138
+ ],
139
+ "depth": [],
140
+ "scan": []
141
+ },
142
+ "goal": {
143
+ "low_dim": [],
144
+ "rgb": [],
145
+ "depth": [],
146
+ "scan": []
147
+ }
148
+ },
149
+ "encoder": {
150
+ "low_dim": {
151
+ "core_class": null,
152
+ "core_kwargs": {},
153
+ "obs_randomizer_class": null,
154
+ "obs_randomizer_kwargs": {}
155
+ },
156
+ "rgb": {
157
+ "core_class": "VisualCore",
158
+ "core_kwargs": {
159
+ "backbone_class": "ResNet18Conv",
160
+ "backbone_kwargs": {
161
+ "pretrained": false,
162
+ "input_coord_conv": false
163
+ },
164
+ "pool_class": "SpatialSoftmax",
165
+ "pool_kwargs": {
166
+ "num_kp": 32,
167
+ "learnable_temperature": false,
168
+ "temperature": 1.0,
169
+ "noise_std": 0.0
170
+ },
171
+ "feature_dimension": 64
172
+ },
173
+ "obs_randomizer_class": null,
174
+ "obs_randomizer_kwargs": {}
175
+ },
176
+ "depth": {
177
+ "core_class": "VisualCore",
178
+ "core_kwargs": {},
179
+ "obs_randomizer_class": null,
180
+ "obs_randomizer_kwargs": {}
181
+ },
182
+ "scan": {
183
+ "core_class": "ScanCore",
184
+ "core_kwargs": {},
185
+ "obs_randomizer_class": null,
186
+ "obs_randomizer_kwargs": {}
187
+ }
188
+ }
189
+ },
190
+ "meta": {
191
+ "hp_base_config_file": null,
192
+ "hp_keys": [],
193
+ "hp_values": []
194
+ }
195
+ }
vac_pre_vis/dataset_summary.json ADDED
@@ -0,0 +1,2032 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "path": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_pre_vis/vac_pre_vis.hdf5",
3
+ "total_frames": 145712,
4
+ "train_episodes": 172,
5
+ "valid_episodes": 30,
6
+ "train_source_episodes": [
7
+ 0,
8
+ 1,
9
+ 2,
10
+ 3,
11
+ 4,
12
+ 5,
13
+ 6,
14
+ 7,
15
+ 8,
16
+ 9,
17
+ 10,
18
+ 11,
19
+ 12,
20
+ 13,
21
+ 14,
22
+ 15,
23
+ 16,
24
+ 17,
25
+ 18,
26
+ 19,
27
+ 20,
28
+ 21,
29
+ 22,
30
+ 23,
31
+ 24,
32
+ 25,
33
+ 26,
34
+ 27,
35
+ 28,
36
+ 29,
37
+ 30,
38
+ 31,
39
+ 32,
40
+ 33,
41
+ 34,
42
+ 35,
43
+ 36,
44
+ 37,
45
+ 38,
46
+ 39,
47
+ 40,
48
+ 41,
49
+ 42,
50
+ 43,
51
+ 44,
52
+ 45,
53
+ 46,
54
+ 47,
55
+ 48,
56
+ 49,
57
+ 50,
58
+ 51,
59
+ 52,
60
+ 53,
61
+ 54,
62
+ 55,
63
+ 56,
64
+ 57,
65
+ 58,
66
+ 59,
67
+ 60,
68
+ 61,
69
+ 62,
70
+ 63,
71
+ 64,
72
+ 65,
73
+ 66,
74
+ 67,
75
+ 68,
76
+ 69,
77
+ 70,
78
+ 71,
79
+ 72,
80
+ 73,
81
+ 74,
82
+ 75,
83
+ 76,
84
+ 77,
85
+ 78,
86
+ 79,
87
+ 80,
88
+ 81,
89
+ 82,
90
+ 83,
91
+ 84,
92
+ 85,
93
+ 86,
94
+ 87,
95
+ 88,
96
+ 89,
97
+ 90,
98
+ 91,
99
+ 92,
100
+ 93,
101
+ 94,
102
+ 95,
103
+ 96,
104
+ 97,
105
+ 98,
106
+ 99,
107
+ 100,
108
+ 101,
109
+ 102,
110
+ 103,
111
+ 104,
112
+ 105,
113
+ 106,
114
+ 107,
115
+ 108,
116
+ 109,
117
+ 110,
118
+ 111,
119
+ 112,
120
+ 113,
121
+ 114,
122
+ 115,
123
+ 116,
124
+ 117,
125
+ 118,
126
+ 119,
127
+ 120,
128
+ 121,
129
+ 122,
130
+ 123,
131
+ 124,
132
+ 125,
133
+ 126,
134
+ 127,
135
+ 128,
136
+ 129,
137
+ 130,
138
+ 131,
139
+ 132,
140
+ 133,
141
+ 134,
142
+ 135,
143
+ 136,
144
+ 137,
145
+ 138,
146
+ 139,
147
+ 140,
148
+ 141,
149
+ 142,
150
+ 143,
151
+ 144,
152
+ 145,
153
+ 146,
154
+ 147,
155
+ 148,
156
+ 149,
157
+ 150,
158
+ 151,
159
+ 152,
160
+ 153,
161
+ 154,
162
+ 155,
163
+ 156,
164
+ 157,
165
+ 158,
166
+ 159,
167
+ 160,
168
+ 161,
169
+ 162,
170
+ 163,
171
+ 164,
172
+ 165,
173
+ 166,
174
+ 167,
175
+ 168,
176
+ 169,
177
+ 170,
178
+ 171
179
+ ],
180
+ "valid_source_episodes": [
181
+ 172,
182
+ 173,
183
+ 174,
184
+ 175,
185
+ 176,
186
+ 177,
187
+ 178,
188
+ 179,
189
+ 180,
190
+ 181,
191
+ 182,
192
+ 183,
193
+ 184,
194
+ 185,
195
+ 186,
196
+ 187,
197
+ 188,
198
+ 189,
199
+ 190,
200
+ 191,
201
+ 192,
202
+ 193,
203
+ 194,
204
+ 195,
205
+ 196,
206
+ 197,
207
+ 198,
208
+ 199,
209
+ 200,
210
+ 201
211
+ ],
212
+ "episodes": [
213
+ {
214
+ "hdf5_key": "ep_00000",
215
+ "source_episode_index": 0,
216
+ "split": "train",
217
+ "num_samples": 663,
218
+ "action_dim": 9,
219
+ "obs_mode": "vision",
220
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
221
+ },
222
+ {
223
+ "hdf5_key": "ep_00001",
224
+ "source_episode_index": 1,
225
+ "split": "train",
226
+ "num_samples": 706,
227
+ "action_dim": 9,
228
+ "obs_mode": "vision",
229
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
230
+ },
231
+ {
232
+ "hdf5_key": "ep_00002",
233
+ "source_episode_index": 2,
234
+ "split": "train",
235
+ "num_samples": 1006,
236
+ "action_dim": 9,
237
+ "obs_mode": "vision",
238
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
239
+ },
240
+ {
241
+ "hdf5_key": "ep_00003",
242
+ "source_episode_index": 3,
243
+ "split": "train",
244
+ "num_samples": 709,
245
+ "action_dim": 9,
246
+ "obs_mode": "vision",
247
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
248
+ },
249
+ {
250
+ "hdf5_key": "ep_00004",
251
+ "source_episode_index": 4,
252
+ "split": "train",
253
+ "num_samples": 816,
254
+ "action_dim": 9,
255
+ "obs_mode": "vision",
256
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
257
+ },
258
+ {
259
+ "hdf5_key": "ep_00005",
260
+ "source_episode_index": 5,
261
+ "split": "train",
262
+ "num_samples": 882,
263
+ "action_dim": 9,
264
+ "obs_mode": "vision",
265
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
266
+ },
267
+ {
268
+ "hdf5_key": "ep_00006",
269
+ "source_episode_index": 6,
270
+ "split": "train",
271
+ "num_samples": 699,
272
+ "action_dim": 9,
273
+ "obs_mode": "vision",
274
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
275
+ },
276
+ {
277
+ "hdf5_key": "ep_00007",
278
+ "source_episode_index": 7,
279
+ "split": "train",
280
+ "num_samples": 851,
281
+ "action_dim": 9,
282
+ "obs_mode": "vision",
283
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
284
+ },
285
+ {
286
+ "hdf5_key": "ep_00008",
287
+ "source_episode_index": 8,
288
+ "split": "train",
289
+ "num_samples": 728,
290
+ "action_dim": 9,
291
+ "obs_mode": "vision",
292
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
293
+ },
294
+ {
295
+ "hdf5_key": "ep_00009",
296
+ "source_episode_index": 9,
297
+ "split": "train",
298
+ "num_samples": 770,
299
+ "action_dim": 9,
300
+ "obs_mode": "vision",
301
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
302
+ },
303
+ {
304
+ "hdf5_key": "ep_00010",
305
+ "source_episode_index": 10,
306
+ "split": "train",
307
+ "num_samples": 821,
308
+ "action_dim": 9,
309
+ "obs_mode": "vision",
310
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
311
+ },
312
+ {
313
+ "hdf5_key": "ep_00011",
314
+ "source_episode_index": 11,
315
+ "split": "train",
316
+ "num_samples": 561,
317
+ "action_dim": 9,
318
+ "obs_mode": "vision",
319
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
320
+ },
321
+ {
322
+ "hdf5_key": "ep_00012",
323
+ "source_episode_index": 12,
324
+ "split": "train",
325
+ "num_samples": 517,
326
+ "action_dim": 9,
327
+ "obs_mode": "vision",
328
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
329
+ },
330
+ {
331
+ "hdf5_key": "ep_00013",
332
+ "source_episode_index": 13,
333
+ "split": "train",
334
+ "num_samples": 690,
335
+ "action_dim": 9,
336
+ "obs_mode": "vision",
337
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
338
+ },
339
+ {
340
+ "hdf5_key": "ep_00014",
341
+ "source_episode_index": 14,
342
+ "split": "train",
343
+ "num_samples": 559,
344
+ "action_dim": 9,
345
+ "obs_mode": "vision",
346
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
347
+ },
348
+ {
349
+ "hdf5_key": "ep_00015",
350
+ "source_episode_index": 15,
351
+ "split": "train",
352
+ "num_samples": 889,
353
+ "action_dim": 9,
354
+ "obs_mode": "vision",
355
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
356
+ },
357
+ {
358
+ "hdf5_key": "ep_00016",
359
+ "source_episode_index": 16,
360
+ "split": "train",
361
+ "num_samples": 619,
362
+ "action_dim": 9,
363
+ "obs_mode": "vision",
364
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
365
+ },
366
+ {
367
+ "hdf5_key": "ep_00017",
368
+ "source_episode_index": 17,
369
+ "split": "train",
370
+ "num_samples": 640,
371
+ "action_dim": 9,
372
+ "obs_mode": "vision",
373
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
374
+ },
375
+ {
376
+ "hdf5_key": "ep_00018",
377
+ "source_episode_index": 18,
378
+ "split": "train",
379
+ "num_samples": 612,
380
+ "action_dim": 9,
381
+ "obs_mode": "vision",
382
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
383
+ },
384
+ {
385
+ "hdf5_key": "ep_00019",
386
+ "source_episode_index": 19,
387
+ "split": "train",
388
+ "num_samples": 702,
389
+ "action_dim": 9,
390
+ "obs_mode": "vision",
391
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
392
+ },
393
+ {
394
+ "hdf5_key": "ep_00020",
395
+ "source_episode_index": 20,
396
+ "split": "train",
397
+ "num_samples": 645,
398
+ "action_dim": 9,
399
+ "obs_mode": "vision",
400
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
401
+ },
402
+ {
403
+ "hdf5_key": "ep_00021",
404
+ "source_episode_index": 21,
405
+ "split": "train",
406
+ "num_samples": 751,
407
+ "action_dim": 9,
408
+ "obs_mode": "vision",
409
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
410
+ },
411
+ {
412
+ "hdf5_key": "ep_00022",
413
+ "source_episode_index": 22,
414
+ "split": "train",
415
+ "num_samples": 810,
416
+ "action_dim": 9,
417
+ "obs_mode": "vision",
418
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
419
+ },
420
+ {
421
+ "hdf5_key": "ep_00023",
422
+ "source_episode_index": 23,
423
+ "split": "train",
424
+ "num_samples": 734,
425
+ "action_dim": 9,
426
+ "obs_mode": "vision",
427
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
428
+ },
429
+ {
430
+ "hdf5_key": "ep_00024",
431
+ "source_episode_index": 24,
432
+ "split": "train",
433
+ "num_samples": 637,
434
+ "action_dim": 9,
435
+ "obs_mode": "vision",
436
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
437
+ },
438
+ {
439
+ "hdf5_key": "ep_00025",
440
+ "source_episode_index": 25,
441
+ "split": "train",
442
+ "num_samples": 640,
443
+ "action_dim": 9,
444
+ "obs_mode": "vision",
445
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
446
+ },
447
+ {
448
+ "hdf5_key": "ep_00026",
449
+ "source_episode_index": 26,
450
+ "split": "train",
451
+ "num_samples": 957,
452
+ "action_dim": 9,
453
+ "obs_mode": "vision",
454
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
455
+ },
456
+ {
457
+ "hdf5_key": "ep_00027",
458
+ "source_episode_index": 27,
459
+ "split": "train",
460
+ "num_samples": 617,
461
+ "action_dim": 9,
462
+ "obs_mode": "vision",
463
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
464
+ },
465
+ {
466
+ "hdf5_key": "ep_00028",
467
+ "source_episode_index": 28,
468
+ "split": "train",
469
+ "num_samples": 772,
470
+ "action_dim": 9,
471
+ "obs_mode": "vision",
472
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
473
+ },
474
+ {
475
+ "hdf5_key": "ep_00029",
476
+ "source_episode_index": 29,
477
+ "split": "train",
478
+ "num_samples": 616,
479
+ "action_dim": 9,
480
+ "obs_mode": "vision",
481
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
482
+ },
483
+ {
484
+ "hdf5_key": "ep_00030",
485
+ "source_episode_index": 30,
486
+ "split": "train",
487
+ "num_samples": 732,
488
+ "action_dim": 9,
489
+ "obs_mode": "vision",
490
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
491
+ },
492
+ {
493
+ "hdf5_key": "ep_00031",
494
+ "source_episode_index": 31,
495
+ "split": "train",
496
+ "num_samples": 761,
497
+ "action_dim": 9,
498
+ "obs_mode": "vision",
499
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
500
+ },
501
+ {
502
+ "hdf5_key": "ep_00032",
503
+ "source_episode_index": 32,
504
+ "split": "train",
505
+ "num_samples": 804,
506
+ "action_dim": 9,
507
+ "obs_mode": "vision",
508
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
509
+ },
510
+ {
511
+ "hdf5_key": "ep_00033",
512
+ "source_episode_index": 33,
513
+ "split": "train",
514
+ "num_samples": 735,
515
+ "action_dim": 9,
516
+ "obs_mode": "vision",
517
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
518
+ },
519
+ {
520
+ "hdf5_key": "ep_00034",
521
+ "source_episode_index": 34,
522
+ "split": "train",
523
+ "num_samples": 721,
524
+ "action_dim": 9,
525
+ "obs_mode": "vision",
526
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
527
+ },
528
+ {
529
+ "hdf5_key": "ep_00035",
530
+ "source_episode_index": 35,
531
+ "split": "train",
532
+ "num_samples": 612,
533
+ "action_dim": 9,
534
+ "obs_mode": "vision",
535
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
536
+ },
537
+ {
538
+ "hdf5_key": "ep_00036",
539
+ "source_episode_index": 36,
540
+ "split": "train",
541
+ "num_samples": 685,
542
+ "action_dim": 9,
543
+ "obs_mode": "vision",
544
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
545
+ },
546
+ {
547
+ "hdf5_key": "ep_00037",
548
+ "source_episode_index": 37,
549
+ "split": "train",
550
+ "num_samples": 685,
551
+ "action_dim": 9,
552
+ "obs_mode": "vision",
553
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
554
+ },
555
+ {
556
+ "hdf5_key": "ep_00038",
557
+ "source_episode_index": 38,
558
+ "split": "train",
559
+ "num_samples": 663,
560
+ "action_dim": 9,
561
+ "obs_mode": "vision",
562
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
563
+ },
564
+ {
565
+ "hdf5_key": "ep_00039",
566
+ "source_episode_index": 39,
567
+ "split": "train",
568
+ "num_samples": 855,
569
+ "action_dim": 9,
570
+ "obs_mode": "vision",
571
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
572
+ },
573
+ {
574
+ "hdf5_key": "ep_00040",
575
+ "source_episode_index": 40,
576
+ "split": "train",
577
+ "num_samples": 810,
578
+ "action_dim": 9,
579
+ "obs_mode": "vision",
580
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
581
+ },
582
+ {
583
+ "hdf5_key": "ep_00041",
584
+ "source_episode_index": 41,
585
+ "split": "train",
586
+ "num_samples": 655,
587
+ "action_dim": 9,
588
+ "obs_mode": "vision",
589
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
590
+ },
591
+ {
592
+ "hdf5_key": "ep_00042",
593
+ "source_episode_index": 42,
594
+ "split": "train",
595
+ "num_samples": 946,
596
+ "action_dim": 9,
597
+ "obs_mode": "vision",
598
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
599
+ },
600
+ {
601
+ "hdf5_key": "ep_00043",
602
+ "source_episode_index": 43,
603
+ "split": "train",
604
+ "num_samples": 703,
605
+ "action_dim": 9,
606
+ "obs_mode": "vision",
607
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
608
+ },
609
+ {
610
+ "hdf5_key": "ep_00044",
611
+ "source_episode_index": 44,
612
+ "split": "train",
613
+ "num_samples": 606,
614
+ "action_dim": 9,
615
+ "obs_mode": "vision",
616
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
617
+ },
618
+ {
619
+ "hdf5_key": "ep_00045",
620
+ "source_episode_index": 45,
621
+ "split": "train",
622
+ "num_samples": 914,
623
+ "action_dim": 9,
624
+ "obs_mode": "vision",
625
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
626
+ },
627
+ {
628
+ "hdf5_key": "ep_00046",
629
+ "source_episode_index": 46,
630
+ "split": "train",
631
+ "num_samples": 949,
632
+ "action_dim": 9,
633
+ "obs_mode": "vision",
634
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
635
+ },
636
+ {
637
+ "hdf5_key": "ep_00047",
638
+ "source_episode_index": 47,
639
+ "split": "train",
640
+ "num_samples": 818,
641
+ "action_dim": 9,
642
+ "obs_mode": "vision",
643
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
644
+ },
645
+ {
646
+ "hdf5_key": "ep_00048",
647
+ "source_episode_index": 48,
648
+ "split": "train",
649
+ "num_samples": 913,
650
+ "action_dim": 9,
651
+ "obs_mode": "vision",
652
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
653
+ },
654
+ {
655
+ "hdf5_key": "ep_00049",
656
+ "source_episode_index": 49,
657
+ "split": "train",
658
+ "num_samples": 904,
659
+ "action_dim": 9,
660
+ "obs_mode": "vision",
661
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
662
+ },
663
+ {
664
+ "hdf5_key": "ep_00050",
665
+ "source_episode_index": 50,
666
+ "split": "train",
667
+ "num_samples": 923,
668
+ "action_dim": 9,
669
+ "obs_mode": "vision",
670
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
671
+ },
672
+ {
673
+ "hdf5_key": "ep_00051",
674
+ "source_episode_index": 51,
675
+ "split": "train",
676
+ "num_samples": 824,
677
+ "action_dim": 9,
678
+ "obs_mode": "vision",
679
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
680
+ },
681
+ {
682
+ "hdf5_key": "ep_00052",
683
+ "source_episode_index": 52,
684
+ "split": "train",
685
+ "num_samples": 777,
686
+ "action_dim": 9,
687
+ "obs_mode": "vision",
688
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
689
+ },
690
+ {
691
+ "hdf5_key": "ep_00053",
692
+ "source_episode_index": 53,
693
+ "split": "train",
694
+ "num_samples": 714,
695
+ "action_dim": 9,
696
+ "obs_mode": "vision",
697
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
698
+ },
699
+ {
700
+ "hdf5_key": "ep_00054",
701
+ "source_episode_index": 54,
702
+ "split": "train",
703
+ "num_samples": 771,
704
+ "action_dim": 9,
705
+ "obs_mode": "vision",
706
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
707
+ },
708
+ {
709
+ "hdf5_key": "ep_00055",
710
+ "source_episode_index": 55,
711
+ "split": "train",
712
+ "num_samples": 684,
713
+ "action_dim": 9,
714
+ "obs_mode": "vision",
715
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
716
+ },
717
+ {
718
+ "hdf5_key": "ep_00056",
719
+ "source_episode_index": 56,
720
+ "split": "train",
721
+ "num_samples": 766,
722
+ "action_dim": 9,
723
+ "obs_mode": "vision",
724
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
725
+ },
726
+ {
727
+ "hdf5_key": "ep_00057",
728
+ "source_episode_index": 57,
729
+ "split": "train",
730
+ "num_samples": 890,
731
+ "action_dim": 9,
732
+ "obs_mode": "vision",
733
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
734
+ },
735
+ {
736
+ "hdf5_key": "ep_00058",
737
+ "source_episode_index": 58,
738
+ "split": "train",
739
+ "num_samples": 774,
740
+ "action_dim": 9,
741
+ "obs_mode": "vision",
742
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
743
+ },
744
+ {
745
+ "hdf5_key": "ep_00059",
746
+ "source_episode_index": 59,
747
+ "split": "train",
748
+ "num_samples": 781,
749
+ "action_dim": 9,
750
+ "obs_mode": "vision",
751
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
752
+ },
753
+ {
754
+ "hdf5_key": "ep_00060",
755
+ "source_episode_index": 60,
756
+ "split": "train",
757
+ "num_samples": 1093,
758
+ "action_dim": 9,
759
+ "obs_mode": "vision",
760
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
761
+ },
762
+ {
763
+ "hdf5_key": "ep_00061",
764
+ "source_episode_index": 61,
765
+ "split": "train",
766
+ "num_samples": 900,
767
+ "action_dim": 9,
768
+ "obs_mode": "vision",
769
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
770
+ },
771
+ {
772
+ "hdf5_key": "ep_00062",
773
+ "source_episode_index": 62,
774
+ "split": "train",
775
+ "num_samples": 814,
776
+ "action_dim": 9,
777
+ "obs_mode": "vision",
778
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
779
+ },
780
+ {
781
+ "hdf5_key": "ep_00063",
782
+ "source_episode_index": 63,
783
+ "split": "train",
784
+ "num_samples": 1108,
785
+ "action_dim": 9,
786
+ "obs_mode": "vision",
787
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
788
+ },
789
+ {
790
+ "hdf5_key": "ep_00064",
791
+ "source_episode_index": 64,
792
+ "split": "train",
793
+ "num_samples": 816,
794
+ "action_dim": 9,
795
+ "obs_mode": "vision",
796
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
797
+ },
798
+ {
799
+ "hdf5_key": "ep_00065",
800
+ "source_episode_index": 65,
801
+ "split": "train",
802
+ "num_samples": 963,
803
+ "action_dim": 9,
804
+ "obs_mode": "vision",
805
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
806
+ },
807
+ {
808
+ "hdf5_key": "ep_00066",
809
+ "source_episode_index": 66,
810
+ "split": "train",
811
+ "num_samples": 895,
812
+ "action_dim": 9,
813
+ "obs_mode": "vision",
814
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
815
+ },
816
+ {
817
+ "hdf5_key": "ep_00067",
818
+ "source_episode_index": 67,
819
+ "split": "train",
820
+ "num_samples": 572,
821
+ "action_dim": 9,
822
+ "obs_mode": "vision",
823
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
824
+ },
825
+ {
826
+ "hdf5_key": "ep_00068",
827
+ "source_episode_index": 68,
828
+ "split": "train",
829
+ "num_samples": 648,
830
+ "action_dim": 9,
831
+ "obs_mode": "vision",
832
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
833
+ },
834
+ {
835
+ "hdf5_key": "ep_00069",
836
+ "source_episode_index": 69,
837
+ "split": "train",
838
+ "num_samples": 772,
839
+ "action_dim": 9,
840
+ "obs_mode": "vision",
841
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
842
+ },
843
+ {
844
+ "hdf5_key": "ep_00070",
845
+ "source_episode_index": 70,
846
+ "split": "train",
847
+ "num_samples": 702,
848
+ "action_dim": 9,
849
+ "obs_mode": "vision",
850
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
851
+ },
852
+ {
853
+ "hdf5_key": "ep_00071",
854
+ "source_episode_index": 71,
855
+ "split": "train",
856
+ "num_samples": 786,
857
+ "action_dim": 9,
858
+ "obs_mode": "vision",
859
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
860
+ },
861
+ {
862
+ "hdf5_key": "ep_00072",
863
+ "source_episode_index": 72,
864
+ "split": "train",
865
+ "num_samples": 843,
866
+ "action_dim": 9,
867
+ "obs_mode": "vision",
868
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
869
+ },
870
+ {
871
+ "hdf5_key": "ep_00073",
872
+ "source_episode_index": 73,
873
+ "split": "train",
874
+ "num_samples": 767,
875
+ "action_dim": 9,
876
+ "obs_mode": "vision",
877
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
878
+ },
879
+ {
880
+ "hdf5_key": "ep_00074",
881
+ "source_episode_index": 74,
882
+ "split": "train",
883
+ "num_samples": 764,
884
+ "action_dim": 9,
885
+ "obs_mode": "vision",
886
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
887
+ },
888
+ {
889
+ "hdf5_key": "ep_00075",
890
+ "source_episode_index": 75,
891
+ "split": "train",
892
+ "num_samples": 750,
893
+ "action_dim": 9,
894
+ "obs_mode": "vision",
895
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
896
+ },
897
+ {
898
+ "hdf5_key": "ep_00076",
899
+ "source_episode_index": 76,
900
+ "split": "train",
901
+ "num_samples": 652,
902
+ "action_dim": 9,
903
+ "obs_mode": "vision",
904
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
905
+ },
906
+ {
907
+ "hdf5_key": "ep_00077",
908
+ "source_episode_index": 77,
909
+ "split": "train",
910
+ "num_samples": 812,
911
+ "action_dim": 9,
912
+ "obs_mode": "vision",
913
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
914
+ },
915
+ {
916
+ "hdf5_key": "ep_00078",
917
+ "source_episode_index": 78,
918
+ "split": "train",
919
+ "num_samples": 722,
920
+ "action_dim": 9,
921
+ "obs_mode": "vision",
922
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
923
+ },
924
+ {
925
+ "hdf5_key": "ep_00079",
926
+ "source_episode_index": 79,
927
+ "split": "train",
928
+ "num_samples": 723,
929
+ "action_dim": 9,
930
+ "obs_mode": "vision",
931
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
932
+ },
933
+ {
934
+ "hdf5_key": "ep_00080",
935
+ "source_episode_index": 80,
936
+ "split": "train",
937
+ "num_samples": 765,
938
+ "action_dim": 9,
939
+ "obs_mode": "vision",
940
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
941
+ },
942
+ {
943
+ "hdf5_key": "ep_00081",
944
+ "source_episode_index": 81,
945
+ "split": "train",
946
+ "num_samples": 981,
947
+ "action_dim": 9,
948
+ "obs_mode": "vision",
949
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
950
+ },
951
+ {
952
+ "hdf5_key": "ep_00082",
953
+ "source_episode_index": 82,
954
+ "split": "train",
955
+ "num_samples": 959,
956
+ "action_dim": 9,
957
+ "obs_mode": "vision",
958
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
959
+ },
960
+ {
961
+ "hdf5_key": "ep_00083",
962
+ "source_episode_index": 83,
963
+ "split": "train",
964
+ "num_samples": 655,
965
+ "action_dim": 9,
966
+ "obs_mode": "vision",
967
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
968
+ },
969
+ {
970
+ "hdf5_key": "ep_00084",
971
+ "source_episode_index": 84,
972
+ "split": "train",
973
+ "num_samples": 746,
974
+ "action_dim": 9,
975
+ "obs_mode": "vision",
976
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
977
+ },
978
+ {
979
+ "hdf5_key": "ep_00085",
980
+ "source_episode_index": 85,
981
+ "split": "train",
982
+ "num_samples": 660,
983
+ "action_dim": 9,
984
+ "obs_mode": "vision",
985
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
986
+ },
987
+ {
988
+ "hdf5_key": "ep_00086",
989
+ "source_episode_index": 86,
990
+ "split": "train",
991
+ "num_samples": 617,
992
+ "action_dim": 9,
993
+ "obs_mode": "vision",
994
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
995
+ },
996
+ {
997
+ "hdf5_key": "ep_00087",
998
+ "source_episode_index": 87,
999
+ "split": "train",
1000
+ "num_samples": 819,
1001
+ "action_dim": 9,
1002
+ "obs_mode": "vision",
1003
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1004
+ },
1005
+ {
1006
+ "hdf5_key": "ep_00088",
1007
+ "source_episode_index": 88,
1008
+ "split": "train",
1009
+ "num_samples": 696,
1010
+ "action_dim": 9,
1011
+ "obs_mode": "vision",
1012
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1013
+ },
1014
+ {
1015
+ "hdf5_key": "ep_00089",
1016
+ "source_episode_index": 89,
1017
+ "split": "train",
1018
+ "num_samples": 855,
1019
+ "action_dim": 9,
1020
+ "obs_mode": "vision",
1021
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1022
+ },
1023
+ {
1024
+ "hdf5_key": "ep_00090",
1025
+ "source_episode_index": 90,
1026
+ "split": "train",
1027
+ "num_samples": 678,
1028
+ "action_dim": 9,
1029
+ "obs_mode": "vision",
1030
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1031
+ },
1032
+ {
1033
+ "hdf5_key": "ep_00091",
1034
+ "source_episode_index": 91,
1035
+ "split": "train",
1036
+ "num_samples": 711,
1037
+ "action_dim": 9,
1038
+ "obs_mode": "vision",
1039
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1040
+ },
1041
+ {
1042
+ "hdf5_key": "ep_00092",
1043
+ "source_episode_index": 92,
1044
+ "split": "train",
1045
+ "num_samples": 763,
1046
+ "action_dim": 9,
1047
+ "obs_mode": "vision",
1048
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1049
+ },
1050
+ {
1051
+ "hdf5_key": "ep_00093",
1052
+ "source_episode_index": 93,
1053
+ "split": "train",
1054
+ "num_samples": 723,
1055
+ "action_dim": 9,
1056
+ "obs_mode": "vision",
1057
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1058
+ },
1059
+ {
1060
+ "hdf5_key": "ep_00094",
1061
+ "source_episode_index": 94,
1062
+ "split": "train",
1063
+ "num_samples": 723,
1064
+ "action_dim": 9,
1065
+ "obs_mode": "vision",
1066
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1067
+ },
1068
+ {
1069
+ "hdf5_key": "ep_00095",
1070
+ "source_episode_index": 95,
1071
+ "split": "train",
1072
+ "num_samples": 865,
1073
+ "action_dim": 9,
1074
+ "obs_mode": "vision",
1075
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1076
+ },
1077
+ {
1078
+ "hdf5_key": "ep_00096",
1079
+ "source_episode_index": 96,
1080
+ "split": "train",
1081
+ "num_samples": 495,
1082
+ "action_dim": 9,
1083
+ "obs_mode": "vision",
1084
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1085
+ },
1086
+ {
1087
+ "hdf5_key": "ep_00097",
1088
+ "source_episode_index": 97,
1089
+ "split": "train",
1090
+ "num_samples": 451,
1091
+ "action_dim": 9,
1092
+ "obs_mode": "vision",
1093
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1094
+ },
1095
+ {
1096
+ "hdf5_key": "ep_00098",
1097
+ "source_episode_index": 98,
1098
+ "split": "train",
1099
+ "num_samples": 690,
1100
+ "action_dim": 9,
1101
+ "obs_mode": "vision",
1102
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1103
+ },
1104
+ {
1105
+ "hdf5_key": "ep_00099",
1106
+ "source_episode_index": 99,
1107
+ "split": "train",
1108
+ "num_samples": 690,
1109
+ "action_dim": 9,
1110
+ "obs_mode": "vision",
1111
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1112
+ },
1113
+ {
1114
+ "hdf5_key": "ep_00100",
1115
+ "source_episode_index": 100,
1116
+ "split": "train",
1117
+ "num_samples": 615,
1118
+ "action_dim": 9,
1119
+ "obs_mode": "vision",
1120
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1121
+ },
1122
+ {
1123
+ "hdf5_key": "ep_00101",
1124
+ "source_episode_index": 101,
1125
+ "split": "train",
1126
+ "num_samples": 662,
1127
+ "action_dim": 9,
1128
+ "obs_mode": "vision",
1129
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1130
+ },
1131
+ {
1132
+ "hdf5_key": "ep_00102",
1133
+ "source_episode_index": 102,
1134
+ "split": "train",
1135
+ "num_samples": 614,
1136
+ "action_dim": 9,
1137
+ "obs_mode": "vision",
1138
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1139
+ },
1140
+ {
1141
+ "hdf5_key": "ep_00103",
1142
+ "source_episode_index": 103,
1143
+ "split": "train",
1144
+ "num_samples": 730,
1145
+ "action_dim": 9,
1146
+ "obs_mode": "vision",
1147
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1148
+ },
1149
+ {
1150
+ "hdf5_key": "ep_00104",
1151
+ "source_episode_index": 104,
1152
+ "split": "train",
1153
+ "num_samples": 683,
1154
+ "action_dim": 9,
1155
+ "obs_mode": "vision",
1156
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1157
+ },
1158
+ {
1159
+ "hdf5_key": "ep_00105",
1160
+ "source_episode_index": 105,
1161
+ "split": "train",
1162
+ "num_samples": 602,
1163
+ "action_dim": 9,
1164
+ "obs_mode": "vision",
1165
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1166
+ },
1167
+ {
1168
+ "hdf5_key": "ep_00106",
1169
+ "source_episode_index": 106,
1170
+ "split": "train",
1171
+ "num_samples": 652,
1172
+ "action_dim": 9,
1173
+ "obs_mode": "vision",
1174
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1175
+ },
1176
+ {
1177
+ "hdf5_key": "ep_00107",
1178
+ "source_episode_index": 107,
1179
+ "split": "train",
1180
+ "num_samples": 722,
1181
+ "action_dim": 9,
1182
+ "obs_mode": "vision",
1183
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1184
+ },
1185
+ {
1186
+ "hdf5_key": "ep_00108",
1187
+ "source_episode_index": 108,
1188
+ "split": "train",
1189
+ "num_samples": 605,
1190
+ "action_dim": 9,
1191
+ "obs_mode": "vision",
1192
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1193
+ },
1194
+ {
1195
+ "hdf5_key": "ep_00109",
1196
+ "source_episode_index": 109,
1197
+ "split": "train",
1198
+ "num_samples": 944,
1199
+ "action_dim": 9,
1200
+ "obs_mode": "vision",
1201
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1202
+ },
1203
+ {
1204
+ "hdf5_key": "ep_00110",
1205
+ "source_episode_index": 110,
1206
+ "split": "train",
1207
+ "num_samples": 661,
1208
+ "action_dim": 9,
1209
+ "obs_mode": "vision",
1210
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1211
+ },
1212
+ {
1213
+ "hdf5_key": "ep_00111",
1214
+ "source_episode_index": 111,
1215
+ "split": "train",
1216
+ "num_samples": 644,
1217
+ "action_dim": 9,
1218
+ "obs_mode": "vision",
1219
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1220
+ },
1221
+ {
1222
+ "hdf5_key": "ep_00112",
1223
+ "source_episode_index": 112,
1224
+ "split": "train",
1225
+ "num_samples": 614,
1226
+ "action_dim": 9,
1227
+ "obs_mode": "vision",
1228
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1229
+ },
1230
+ {
1231
+ "hdf5_key": "ep_00113",
1232
+ "source_episode_index": 113,
1233
+ "split": "train",
1234
+ "num_samples": 548,
1235
+ "action_dim": 9,
1236
+ "obs_mode": "vision",
1237
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1238
+ },
1239
+ {
1240
+ "hdf5_key": "ep_00114",
1241
+ "source_episode_index": 114,
1242
+ "split": "train",
1243
+ "num_samples": 568,
1244
+ "action_dim": 9,
1245
+ "obs_mode": "vision",
1246
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1247
+ },
1248
+ {
1249
+ "hdf5_key": "ep_00115",
1250
+ "source_episode_index": 115,
1251
+ "split": "train",
1252
+ "num_samples": 730,
1253
+ "action_dim": 9,
1254
+ "obs_mode": "vision",
1255
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1256
+ },
1257
+ {
1258
+ "hdf5_key": "ep_00116",
1259
+ "source_episode_index": 116,
1260
+ "split": "train",
1261
+ "num_samples": 481,
1262
+ "action_dim": 9,
1263
+ "obs_mode": "vision",
1264
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1265
+ },
1266
+ {
1267
+ "hdf5_key": "ep_00117",
1268
+ "source_episode_index": 117,
1269
+ "split": "train",
1270
+ "num_samples": 687,
1271
+ "action_dim": 9,
1272
+ "obs_mode": "vision",
1273
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1274
+ },
1275
+ {
1276
+ "hdf5_key": "ep_00118",
1277
+ "source_episode_index": 118,
1278
+ "split": "train",
1279
+ "num_samples": 618,
1280
+ "action_dim": 9,
1281
+ "obs_mode": "vision",
1282
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1283
+ },
1284
+ {
1285
+ "hdf5_key": "ep_00119",
1286
+ "source_episode_index": 119,
1287
+ "split": "train",
1288
+ "num_samples": 768,
1289
+ "action_dim": 9,
1290
+ "obs_mode": "vision",
1291
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1292
+ },
1293
+ {
1294
+ "hdf5_key": "ep_00120",
1295
+ "source_episode_index": 120,
1296
+ "split": "train",
1297
+ "num_samples": 734,
1298
+ "action_dim": 9,
1299
+ "obs_mode": "vision",
1300
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1301
+ },
1302
+ {
1303
+ "hdf5_key": "ep_00121",
1304
+ "source_episode_index": 121,
1305
+ "split": "train",
1306
+ "num_samples": 604,
1307
+ "action_dim": 9,
1308
+ "obs_mode": "vision",
1309
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1310
+ },
1311
+ {
1312
+ "hdf5_key": "ep_00122",
1313
+ "source_episode_index": 122,
1314
+ "split": "train",
1315
+ "num_samples": 610,
1316
+ "action_dim": 9,
1317
+ "obs_mode": "vision",
1318
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1319
+ },
1320
+ {
1321
+ "hdf5_key": "ep_00123",
1322
+ "source_episode_index": 123,
1323
+ "split": "train",
1324
+ "num_samples": 602,
1325
+ "action_dim": 9,
1326
+ "obs_mode": "vision",
1327
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1328
+ },
1329
+ {
1330
+ "hdf5_key": "ep_00124",
1331
+ "source_episode_index": 124,
1332
+ "split": "train",
1333
+ "num_samples": 630,
1334
+ "action_dim": 9,
1335
+ "obs_mode": "vision",
1336
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1337
+ },
1338
+ {
1339
+ "hdf5_key": "ep_00125",
1340
+ "source_episode_index": 125,
1341
+ "split": "train",
1342
+ "num_samples": 621,
1343
+ "action_dim": 9,
1344
+ "obs_mode": "vision",
1345
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1346
+ },
1347
+ {
1348
+ "hdf5_key": "ep_00126",
1349
+ "source_episode_index": 126,
1350
+ "split": "train",
1351
+ "num_samples": 600,
1352
+ "action_dim": 9,
1353
+ "obs_mode": "vision",
1354
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1355
+ },
1356
+ {
1357
+ "hdf5_key": "ep_00127",
1358
+ "source_episode_index": 127,
1359
+ "split": "train",
1360
+ "num_samples": 478,
1361
+ "action_dim": 9,
1362
+ "obs_mode": "vision",
1363
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1364
+ },
1365
+ {
1366
+ "hdf5_key": "ep_00128",
1367
+ "source_episode_index": 128,
1368
+ "split": "train",
1369
+ "num_samples": 606,
1370
+ "action_dim": 9,
1371
+ "obs_mode": "vision",
1372
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1373
+ },
1374
+ {
1375
+ "hdf5_key": "ep_00129",
1376
+ "source_episode_index": 129,
1377
+ "split": "train",
1378
+ "num_samples": 666,
1379
+ "action_dim": 9,
1380
+ "obs_mode": "vision",
1381
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1382
+ },
1383
+ {
1384
+ "hdf5_key": "ep_00130",
1385
+ "source_episode_index": 130,
1386
+ "split": "train",
1387
+ "num_samples": 655,
1388
+ "action_dim": 9,
1389
+ "obs_mode": "vision",
1390
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1391
+ },
1392
+ {
1393
+ "hdf5_key": "ep_00131",
1394
+ "source_episode_index": 131,
1395
+ "split": "train",
1396
+ "num_samples": 588,
1397
+ "action_dim": 9,
1398
+ "obs_mode": "vision",
1399
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1400
+ },
1401
+ {
1402
+ "hdf5_key": "ep_00132",
1403
+ "source_episode_index": 132,
1404
+ "split": "train",
1405
+ "num_samples": 806,
1406
+ "action_dim": 9,
1407
+ "obs_mode": "vision",
1408
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1409
+ },
1410
+ {
1411
+ "hdf5_key": "ep_00133",
1412
+ "source_episode_index": 133,
1413
+ "split": "train",
1414
+ "num_samples": 586,
1415
+ "action_dim": 9,
1416
+ "obs_mode": "vision",
1417
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1418
+ },
1419
+ {
1420
+ "hdf5_key": "ep_00134",
1421
+ "source_episode_index": 134,
1422
+ "split": "train",
1423
+ "num_samples": 714,
1424
+ "action_dim": 9,
1425
+ "obs_mode": "vision",
1426
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1427
+ },
1428
+ {
1429
+ "hdf5_key": "ep_00135",
1430
+ "source_episode_index": 135,
1431
+ "split": "train",
1432
+ "num_samples": 574,
1433
+ "action_dim": 9,
1434
+ "obs_mode": "vision",
1435
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1436
+ },
1437
+ {
1438
+ "hdf5_key": "ep_00136",
1439
+ "source_episode_index": 136,
1440
+ "split": "train",
1441
+ "num_samples": 577,
1442
+ "action_dim": 9,
1443
+ "obs_mode": "vision",
1444
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1445
+ },
1446
+ {
1447
+ "hdf5_key": "ep_00137",
1448
+ "source_episode_index": 137,
1449
+ "split": "train",
1450
+ "num_samples": 663,
1451
+ "action_dim": 9,
1452
+ "obs_mode": "vision",
1453
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1454
+ },
1455
+ {
1456
+ "hdf5_key": "ep_00138",
1457
+ "source_episode_index": 138,
1458
+ "split": "train",
1459
+ "num_samples": 720,
1460
+ "action_dim": 9,
1461
+ "obs_mode": "vision",
1462
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1463
+ },
1464
+ {
1465
+ "hdf5_key": "ep_00139",
1466
+ "source_episode_index": 139,
1467
+ "split": "train",
1468
+ "num_samples": 640,
1469
+ "action_dim": 9,
1470
+ "obs_mode": "vision",
1471
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1472
+ },
1473
+ {
1474
+ "hdf5_key": "ep_00140",
1475
+ "source_episode_index": 140,
1476
+ "split": "train",
1477
+ "num_samples": 544,
1478
+ "action_dim": 9,
1479
+ "obs_mode": "vision",
1480
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1481
+ },
1482
+ {
1483
+ "hdf5_key": "ep_00141",
1484
+ "source_episode_index": 141,
1485
+ "split": "train",
1486
+ "num_samples": 539,
1487
+ "action_dim": 9,
1488
+ "obs_mode": "vision",
1489
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1490
+ },
1491
+ {
1492
+ "hdf5_key": "ep_00142",
1493
+ "source_episode_index": 142,
1494
+ "split": "train",
1495
+ "num_samples": 592,
1496
+ "action_dim": 9,
1497
+ "obs_mode": "vision",
1498
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1499
+ },
1500
+ {
1501
+ "hdf5_key": "ep_00143",
1502
+ "source_episode_index": 143,
1503
+ "split": "train",
1504
+ "num_samples": 625,
1505
+ "action_dim": 9,
1506
+ "obs_mode": "vision",
1507
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1508
+ },
1509
+ {
1510
+ "hdf5_key": "ep_00144",
1511
+ "source_episode_index": 144,
1512
+ "split": "train",
1513
+ "num_samples": 684,
1514
+ "action_dim": 9,
1515
+ "obs_mode": "vision",
1516
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1517
+ },
1518
+ {
1519
+ "hdf5_key": "ep_00145",
1520
+ "source_episode_index": 145,
1521
+ "split": "train",
1522
+ "num_samples": 559,
1523
+ "action_dim": 9,
1524
+ "obs_mode": "vision",
1525
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1526
+ },
1527
+ {
1528
+ "hdf5_key": "ep_00146",
1529
+ "source_episode_index": 146,
1530
+ "split": "train",
1531
+ "num_samples": 677,
1532
+ "action_dim": 9,
1533
+ "obs_mode": "vision",
1534
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1535
+ },
1536
+ {
1537
+ "hdf5_key": "ep_00147",
1538
+ "source_episode_index": 147,
1539
+ "split": "train",
1540
+ "num_samples": 846,
1541
+ "action_dim": 9,
1542
+ "obs_mode": "vision",
1543
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1544
+ },
1545
+ {
1546
+ "hdf5_key": "ep_00148",
1547
+ "source_episode_index": 148,
1548
+ "split": "train",
1549
+ "num_samples": 615,
1550
+ "action_dim": 9,
1551
+ "obs_mode": "vision",
1552
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1553
+ },
1554
+ {
1555
+ "hdf5_key": "ep_00149",
1556
+ "source_episode_index": 149,
1557
+ "split": "train",
1558
+ "num_samples": 517,
1559
+ "action_dim": 9,
1560
+ "obs_mode": "vision",
1561
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1562
+ },
1563
+ {
1564
+ "hdf5_key": "ep_00150",
1565
+ "source_episode_index": 150,
1566
+ "split": "train",
1567
+ "num_samples": 817,
1568
+ "action_dim": 9,
1569
+ "obs_mode": "vision",
1570
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1571
+ },
1572
+ {
1573
+ "hdf5_key": "ep_00151",
1574
+ "source_episode_index": 151,
1575
+ "split": "train",
1576
+ "num_samples": 814,
1577
+ "action_dim": 9,
1578
+ "obs_mode": "vision",
1579
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1580
+ },
1581
+ {
1582
+ "hdf5_key": "ep_00152",
1583
+ "source_episode_index": 152,
1584
+ "split": "train",
1585
+ "num_samples": 828,
1586
+ "action_dim": 9,
1587
+ "obs_mode": "vision",
1588
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1589
+ },
1590
+ {
1591
+ "hdf5_key": "ep_00153",
1592
+ "source_episode_index": 153,
1593
+ "split": "train",
1594
+ "num_samples": 736,
1595
+ "action_dim": 9,
1596
+ "obs_mode": "vision",
1597
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1598
+ },
1599
+ {
1600
+ "hdf5_key": "ep_00154",
1601
+ "source_episode_index": 154,
1602
+ "split": "train",
1603
+ "num_samples": 860,
1604
+ "action_dim": 9,
1605
+ "obs_mode": "vision",
1606
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1607
+ },
1608
+ {
1609
+ "hdf5_key": "ep_00155",
1610
+ "source_episode_index": 155,
1611
+ "split": "train",
1612
+ "num_samples": 786,
1613
+ "action_dim": 9,
1614
+ "obs_mode": "vision",
1615
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1616
+ },
1617
+ {
1618
+ "hdf5_key": "ep_00156",
1619
+ "source_episode_index": 156,
1620
+ "split": "train",
1621
+ "num_samples": 725,
1622
+ "action_dim": 9,
1623
+ "obs_mode": "vision",
1624
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1625
+ },
1626
+ {
1627
+ "hdf5_key": "ep_00157",
1628
+ "source_episode_index": 157,
1629
+ "split": "train",
1630
+ "num_samples": 834,
1631
+ "action_dim": 9,
1632
+ "obs_mode": "vision",
1633
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1634
+ },
1635
+ {
1636
+ "hdf5_key": "ep_00158",
1637
+ "source_episode_index": 158,
1638
+ "split": "train",
1639
+ "num_samples": 652,
1640
+ "action_dim": 9,
1641
+ "obs_mode": "vision",
1642
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1643
+ },
1644
+ {
1645
+ "hdf5_key": "ep_00159",
1646
+ "source_episode_index": 159,
1647
+ "split": "train",
1648
+ "num_samples": 699,
1649
+ "action_dim": 9,
1650
+ "obs_mode": "vision",
1651
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1652
+ },
1653
+ {
1654
+ "hdf5_key": "ep_00160",
1655
+ "source_episode_index": 160,
1656
+ "split": "train",
1657
+ "num_samples": 812,
1658
+ "action_dim": 9,
1659
+ "obs_mode": "vision",
1660
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1661
+ },
1662
+ {
1663
+ "hdf5_key": "ep_00161",
1664
+ "source_episode_index": 161,
1665
+ "split": "train",
1666
+ "num_samples": 782,
1667
+ "action_dim": 9,
1668
+ "obs_mode": "vision",
1669
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1670
+ },
1671
+ {
1672
+ "hdf5_key": "ep_00162",
1673
+ "source_episode_index": 162,
1674
+ "split": "train",
1675
+ "num_samples": 681,
1676
+ "action_dim": 9,
1677
+ "obs_mode": "vision",
1678
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1679
+ },
1680
+ {
1681
+ "hdf5_key": "ep_00163",
1682
+ "source_episode_index": 163,
1683
+ "split": "train",
1684
+ "num_samples": 742,
1685
+ "action_dim": 9,
1686
+ "obs_mode": "vision",
1687
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1688
+ },
1689
+ {
1690
+ "hdf5_key": "ep_00164",
1691
+ "source_episode_index": 164,
1692
+ "split": "train",
1693
+ "num_samples": 766,
1694
+ "action_dim": 9,
1695
+ "obs_mode": "vision",
1696
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1697
+ },
1698
+ {
1699
+ "hdf5_key": "ep_00165",
1700
+ "source_episode_index": 165,
1701
+ "split": "train",
1702
+ "num_samples": 821,
1703
+ "action_dim": 9,
1704
+ "obs_mode": "vision",
1705
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1706
+ },
1707
+ {
1708
+ "hdf5_key": "ep_00166",
1709
+ "source_episode_index": 166,
1710
+ "split": "train",
1711
+ "num_samples": 1028,
1712
+ "action_dim": 9,
1713
+ "obs_mode": "vision",
1714
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1715
+ },
1716
+ {
1717
+ "hdf5_key": "ep_00167",
1718
+ "source_episode_index": 167,
1719
+ "split": "train",
1720
+ "num_samples": 720,
1721
+ "action_dim": 9,
1722
+ "obs_mode": "vision",
1723
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1724
+ },
1725
+ {
1726
+ "hdf5_key": "ep_00168",
1727
+ "source_episode_index": 168,
1728
+ "split": "train",
1729
+ "num_samples": 773,
1730
+ "action_dim": 9,
1731
+ "obs_mode": "vision",
1732
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1733
+ },
1734
+ {
1735
+ "hdf5_key": "ep_00169",
1736
+ "source_episode_index": 169,
1737
+ "split": "train",
1738
+ "num_samples": 690,
1739
+ "action_dim": 9,
1740
+ "obs_mode": "vision",
1741
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1742
+ },
1743
+ {
1744
+ "hdf5_key": "ep_00170",
1745
+ "source_episode_index": 170,
1746
+ "split": "train",
1747
+ "num_samples": 756,
1748
+ "action_dim": 9,
1749
+ "obs_mode": "vision",
1750
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1751
+ },
1752
+ {
1753
+ "hdf5_key": "ep_00171",
1754
+ "source_episode_index": 171,
1755
+ "split": "train",
1756
+ "num_samples": 742,
1757
+ "action_dim": 9,
1758
+ "obs_mode": "vision",
1759
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1760
+ },
1761
+ {
1762
+ "hdf5_key": "ep_00172",
1763
+ "source_episode_index": 172,
1764
+ "split": "valid",
1765
+ "num_samples": 689,
1766
+ "action_dim": 9,
1767
+ "obs_mode": "vision",
1768
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1769
+ },
1770
+ {
1771
+ "hdf5_key": "ep_00173",
1772
+ "source_episode_index": 173,
1773
+ "split": "valid",
1774
+ "num_samples": 639,
1775
+ "action_dim": 9,
1776
+ "obs_mode": "vision",
1777
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1778
+ },
1779
+ {
1780
+ "hdf5_key": "ep_00174",
1781
+ "source_episode_index": 174,
1782
+ "split": "valid",
1783
+ "num_samples": 694,
1784
+ "action_dim": 9,
1785
+ "obs_mode": "vision",
1786
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1787
+ },
1788
+ {
1789
+ "hdf5_key": "ep_00175",
1790
+ "source_episode_index": 175,
1791
+ "split": "valid",
1792
+ "num_samples": 532,
1793
+ "action_dim": 9,
1794
+ "obs_mode": "vision",
1795
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1796
+ },
1797
+ {
1798
+ "hdf5_key": "ep_00176",
1799
+ "source_episode_index": 176,
1800
+ "split": "valid",
1801
+ "num_samples": 665,
1802
+ "action_dim": 9,
1803
+ "obs_mode": "vision",
1804
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1805
+ },
1806
+ {
1807
+ "hdf5_key": "ep_00177",
1808
+ "source_episode_index": 177,
1809
+ "split": "valid",
1810
+ "num_samples": 771,
1811
+ "action_dim": 9,
1812
+ "obs_mode": "vision",
1813
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1814
+ },
1815
+ {
1816
+ "hdf5_key": "ep_00178",
1817
+ "source_episode_index": 178,
1818
+ "split": "valid",
1819
+ "num_samples": 602,
1820
+ "action_dim": 9,
1821
+ "obs_mode": "vision",
1822
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1823
+ },
1824
+ {
1825
+ "hdf5_key": "ep_00179",
1826
+ "source_episode_index": 179,
1827
+ "split": "valid",
1828
+ "num_samples": 700,
1829
+ "action_dim": 9,
1830
+ "obs_mode": "vision",
1831
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1832
+ },
1833
+ {
1834
+ "hdf5_key": "ep_00180",
1835
+ "source_episode_index": 180,
1836
+ "split": "valid",
1837
+ "num_samples": 880,
1838
+ "action_dim": 9,
1839
+ "obs_mode": "vision",
1840
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1841
+ },
1842
+ {
1843
+ "hdf5_key": "ep_00181",
1844
+ "source_episode_index": 181,
1845
+ "split": "valid",
1846
+ "num_samples": 522,
1847
+ "action_dim": 9,
1848
+ "obs_mode": "vision",
1849
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1850
+ },
1851
+ {
1852
+ "hdf5_key": "ep_00182",
1853
+ "source_episode_index": 182,
1854
+ "split": "valid",
1855
+ "num_samples": 697,
1856
+ "action_dim": 9,
1857
+ "obs_mode": "vision",
1858
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1859
+ },
1860
+ {
1861
+ "hdf5_key": "ep_00183",
1862
+ "source_episode_index": 183,
1863
+ "split": "valid",
1864
+ "num_samples": 621,
1865
+ "action_dim": 9,
1866
+ "obs_mode": "vision",
1867
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1868
+ },
1869
+ {
1870
+ "hdf5_key": "ep_00184",
1871
+ "source_episode_index": 184,
1872
+ "split": "valid",
1873
+ "num_samples": 710,
1874
+ "action_dim": 9,
1875
+ "obs_mode": "vision",
1876
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1877
+ },
1878
+ {
1879
+ "hdf5_key": "ep_00185",
1880
+ "source_episode_index": 185,
1881
+ "split": "valid",
1882
+ "num_samples": 774,
1883
+ "action_dim": 9,
1884
+ "obs_mode": "vision",
1885
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1886
+ },
1887
+ {
1888
+ "hdf5_key": "ep_00186",
1889
+ "source_episode_index": 186,
1890
+ "split": "valid",
1891
+ "num_samples": 682,
1892
+ "action_dim": 9,
1893
+ "obs_mode": "vision",
1894
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1895
+ },
1896
+ {
1897
+ "hdf5_key": "ep_00187",
1898
+ "source_episode_index": 187,
1899
+ "split": "valid",
1900
+ "num_samples": 715,
1901
+ "action_dim": 9,
1902
+ "obs_mode": "vision",
1903
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1904
+ },
1905
+ {
1906
+ "hdf5_key": "ep_00188",
1907
+ "source_episode_index": 188,
1908
+ "split": "valid",
1909
+ "num_samples": 798,
1910
+ "action_dim": 9,
1911
+ "obs_mode": "vision",
1912
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1913
+ },
1914
+ {
1915
+ "hdf5_key": "ep_00189",
1916
+ "source_episode_index": 189,
1917
+ "split": "valid",
1918
+ "num_samples": 786,
1919
+ "action_dim": 9,
1920
+ "obs_mode": "vision",
1921
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1922
+ },
1923
+ {
1924
+ "hdf5_key": "ep_00190",
1925
+ "source_episode_index": 190,
1926
+ "split": "valid",
1927
+ "num_samples": 759,
1928
+ "action_dim": 9,
1929
+ "obs_mode": "vision",
1930
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1931
+ },
1932
+ {
1933
+ "hdf5_key": "ep_00191",
1934
+ "source_episode_index": 191,
1935
+ "split": "valid",
1936
+ "num_samples": 812,
1937
+ "action_dim": 9,
1938
+ "obs_mode": "vision",
1939
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1940
+ },
1941
+ {
1942
+ "hdf5_key": "ep_00192",
1943
+ "source_episode_index": 192,
1944
+ "split": "valid",
1945
+ "num_samples": 803,
1946
+ "action_dim": 9,
1947
+ "obs_mode": "vision",
1948
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1949
+ },
1950
+ {
1951
+ "hdf5_key": "ep_00193",
1952
+ "source_episode_index": 193,
1953
+ "split": "valid",
1954
+ "num_samples": 780,
1955
+ "action_dim": 9,
1956
+ "obs_mode": "vision",
1957
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1958
+ },
1959
+ {
1960
+ "hdf5_key": "ep_00194",
1961
+ "source_episode_index": 194,
1962
+ "split": "valid",
1963
+ "num_samples": 853,
1964
+ "action_dim": 9,
1965
+ "obs_mode": "vision",
1966
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1967
+ },
1968
+ {
1969
+ "hdf5_key": "ep_00195",
1970
+ "source_episode_index": 195,
1971
+ "split": "valid",
1972
+ "num_samples": 710,
1973
+ "action_dim": 9,
1974
+ "obs_mode": "vision",
1975
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1976
+ },
1977
+ {
1978
+ "hdf5_key": "ep_00196",
1979
+ "source_episode_index": 196,
1980
+ "split": "valid",
1981
+ "num_samples": 855,
1982
+ "action_dim": 9,
1983
+ "obs_mode": "vision",
1984
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1985
+ },
1986
+ {
1987
+ "hdf5_key": "ep_00197",
1988
+ "source_episode_index": 197,
1989
+ "split": "valid",
1990
+ "num_samples": 707,
1991
+ "action_dim": 9,
1992
+ "obs_mode": "vision",
1993
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1994
+ },
1995
+ {
1996
+ "hdf5_key": "ep_00198",
1997
+ "source_episode_index": 198,
1998
+ "split": "valid",
1999
+ "num_samples": 557,
2000
+ "action_dim": 9,
2001
+ "obs_mode": "vision",
2002
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
2003
+ },
2004
+ {
2005
+ "hdf5_key": "ep_00199",
2006
+ "source_episode_index": 199,
2007
+ "split": "valid",
2008
+ "num_samples": 522,
2009
+ "action_dim": 9,
2010
+ "obs_mode": "vision",
2011
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
2012
+ },
2013
+ {
2014
+ "hdf5_key": "ep_00200",
2015
+ "source_episode_index": 200,
2016
+ "split": "valid",
2017
+ "num_samples": 581,
2018
+ "action_dim": 9,
2019
+ "obs_mode": "vision",
2020
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
2021
+ },
2022
+ {
2023
+ "hdf5_key": "ep_00201",
2024
+ "source_episode_index": 201,
2025
+ "split": "valid",
2026
+ "num_samples": 574,
2027
+ "action_dim": 9,
2028
+ "obs_mode": "vision",
2029
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
2030
+ }
2031
+ ]
2032
+ }
vac_pre_vis/last.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fdfbe4951d55d8cd9a7d09b09fcb69bf77f8e82e4a1c0930b623bf42ee9eb7e4
3
+ size 1252436253
vac_pre_vis_wrench/action_stats.json ADDED
@@ -0,0 +1,311 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "method": "min_max",
3
+ "output_min": -0.999999,
4
+ "output_max": 0.999999,
5
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary",
6
+ "action_components": [
7
+ "user_cmd_dx",
8
+ "user_cmd_dy",
9
+ "user_cmd_dz",
10
+ "user_cmd_drx",
11
+ "user_cmd_dry",
12
+ "user_cmd_drz",
13
+ "stiffness_cmd",
14
+ "zeta",
15
+ "hand_open_binary"
16
+ ],
17
+ "num_frames": 124722,
18
+ "min": [
19
+ -0.0020000000949949026,
20
+ -0.0020000000949949026,
21
+ -0.0020000000949949026,
22
+ -0.017500000074505806,
23
+ 0.0,
24
+ -0.017500000074505806,
25
+ 3.5638904571533203,
26
+ 0.8021645545959473,
27
+ 0.0
28
+ ],
29
+ "max": [
30
+ 0.0020000000949949026,
31
+ 0.0020000000949949026,
32
+ 0.0020000000949949026,
33
+ 0.017500000074505806,
34
+ 0.0020832535810768604,
35
+ 0.017500000074505806,
36
+ 1997.13330078125,
37
+ 2.5,
38
+ 1.0
39
+ ],
40
+ "mean": [
41
+ -1.295681795454584e-05,
42
+ -4.36740810982883e-05,
43
+ 5.51424891455099e-05,
44
+ -1.8853129404305946e-06,
45
+ 6.276949449102176e-08,
46
+ -1.684121525613591e-05,
47
+ 1533.5157470703125,
48
+ 0.9446797370910645,
49
+ 0.428376704454422
50
+ ],
51
+ "std": [
52
+ 0.00042260822374373674,
53
+ 0.0009382680291309953,
54
+ 0.000991749344393611,
55
+ 0.0003852724621538073,
56
+ 1.1097310562036e-05,
57
+ 0.001960940193384886,
58
+ 461.547119140625,
59
+ 0.24154788255691528,
60
+ 0.4948435127735138
61
+ ],
62
+ "scale": [
63
+ 0.0020000021904706955,
64
+ 0.0020000021904706955,
65
+ 0.0020000021904706955,
66
+ 0.0175000187009573,
67
+ 0.0010416278382763267,
68
+ 0.0175000187009573,
69
+ 996.7857666015625,
70
+ 0.8489185571670532,
71
+ 0.500000536441803
72
+ ],
73
+ "offset": [
74
+ 0.0,
75
+ 0.0,
76
+ 0.0,
77
+ 0.0,
78
+ 0.0010416267905384302,
79
+ 0.0,
80
+ 1000.3486328125,
81
+ 1.6510822772979736,
82
+ 0.5
83
+ ],
84
+ "hand_source": "action.absolute",
85
+ "hand_active_dims": [
86
+ 8,
87
+ 9,
88
+ 10,
89
+ 11
90
+ ],
91
+ "hand_reduce": "mean",
92
+ "hand_threshold": 0.85,
93
+ "hand_high_value": 1.0,
94
+ "hand_low_value": 0.0,
95
+ "dataset": "/workspace/datasets/dataset_pipe_02_06",
96
+ "observation_mode_examples": [
97
+ "vision",
98
+ "vision_wrench",
99
+ "vision_wrench_state"
100
+ ],
101
+ "train_source_episodes": [
102
+ 0,
103
+ 1,
104
+ 2,
105
+ 3,
106
+ 4,
107
+ 5,
108
+ 6,
109
+ 7,
110
+ 8,
111
+ 9,
112
+ 10,
113
+ 11,
114
+ 12,
115
+ 13,
116
+ 14,
117
+ 15,
118
+ 16,
119
+ 17,
120
+ 18,
121
+ 19,
122
+ 20,
123
+ 21,
124
+ 22,
125
+ 23,
126
+ 24,
127
+ 25,
128
+ 26,
129
+ 27,
130
+ 28,
131
+ 29,
132
+ 30,
133
+ 31,
134
+ 32,
135
+ 33,
136
+ 34,
137
+ 35,
138
+ 36,
139
+ 37,
140
+ 38,
141
+ 39,
142
+ 40,
143
+ 41,
144
+ 42,
145
+ 43,
146
+ 44,
147
+ 45,
148
+ 46,
149
+ 47,
150
+ 48,
151
+ 49,
152
+ 50,
153
+ 51,
154
+ 52,
155
+ 53,
156
+ 54,
157
+ 55,
158
+ 56,
159
+ 57,
160
+ 58,
161
+ 59,
162
+ 60,
163
+ 61,
164
+ 62,
165
+ 63,
166
+ 64,
167
+ 65,
168
+ 66,
169
+ 67,
170
+ 68,
171
+ 69,
172
+ 70,
173
+ 71,
174
+ 72,
175
+ 73,
176
+ 74,
177
+ 75,
178
+ 76,
179
+ 77,
180
+ 78,
181
+ 79,
182
+ 80,
183
+ 81,
184
+ 82,
185
+ 83,
186
+ 84,
187
+ 85,
188
+ 86,
189
+ 87,
190
+ 88,
191
+ 89,
192
+ 90,
193
+ 91,
194
+ 92,
195
+ 93,
196
+ 94,
197
+ 95,
198
+ 96,
199
+ 97,
200
+ 98,
201
+ 99,
202
+ 100,
203
+ 101,
204
+ 102,
205
+ 103,
206
+ 104,
207
+ 105,
208
+ 106,
209
+ 107,
210
+ 108,
211
+ 109,
212
+ 110,
213
+ 111,
214
+ 112,
215
+ 113,
216
+ 114,
217
+ 115,
218
+ 116,
219
+ 117,
220
+ 118,
221
+ 119,
222
+ 120,
223
+ 121,
224
+ 122,
225
+ 123,
226
+ 124,
227
+ 125,
228
+ 126,
229
+ 127,
230
+ 128,
231
+ 129,
232
+ 130,
233
+ 131,
234
+ 132,
235
+ 133,
236
+ 134,
237
+ 135,
238
+ 136,
239
+ 137,
240
+ 138,
241
+ 139,
242
+ 140,
243
+ 141,
244
+ 142,
245
+ 143,
246
+ 144,
247
+ 145,
248
+ 146,
249
+ 147,
250
+ 148,
251
+ 149,
252
+ 150,
253
+ 151,
254
+ 152,
255
+ 153,
256
+ 154,
257
+ 155,
258
+ 156,
259
+ 157,
260
+ 158,
261
+ 159,
262
+ 160,
263
+ 161,
264
+ 162,
265
+ 163,
266
+ 164,
267
+ 165,
268
+ 166,
269
+ 167,
270
+ 168,
271
+ 169,
272
+ 170,
273
+ 171
274
+ ],
275
+ "valid_source_episodes": [
276
+ 172,
277
+ 173,
278
+ 174,
279
+ 175,
280
+ 176,
281
+ 177,
282
+ 178,
283
+ 179,
284
+ 180,
285
+ 181,
286
+ 182,
287
+ 183,
288
+ 184,
289
+ 185,
290
+ 186,
291
+ 187,
292
+ 188,
293
+ 189,
294
+ 190,
295
+ 191,
296
+ 192,
297
+ 193,
298
+ 194,
299
+ 195,
300
+ 196,
301
+ 197,
302
+ 198,
303
+ 199,
304
+ 200,
305
+ 201
306
+ ],
307
+ "image_size": [
308
+ 84,
309
+ 84
310
+ ]
311
+ }
vac_pre_vis_wrench/best.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:92d6c202e1dc90c77e5148a6035eeb27c90a9249666270ec4ac1abe0240b5cd6
3
+ size 1255291917
vac_pre_vis_wrench/config.json ADDED
@@ -0,0 +1,197 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "algo_name": "diffusion_policy",
3
+ "experiment": {
4
+ "name": "vac_pre_vis_wrench",
5
+ "validate": true,
6
+ "logging": {
7
+ "terminal_output_to_txt": false,
8
+ "log_tb": true,
9
+ "log_wandb": false,
10
+ "wandb_proj_name": "debug"
11
+ },
12
+ "save": {
13
+ "enabled": true,
14
+ "every_n_seconds": null,
15
+ "every_n_epochs": 50,
16
+ "epochs": [],
17
+ "on_best_validation": true,
18
+ "on_best_rollout_return": false,
19
+ "on_best_rollout_success_rate": false
20
+ },
21
+ "epoch_every_n_steps": 100,
22
+ "validation_epoch_every_n_steps": 10,
23
+ "env": null,
24
+ "additional_envs": null,
25
+ "render": false,
26
+ "render_video": false,
27
+ "keep_all_videos": false,
28
+ "video_skip": 5,
29
+ "rollout": {
30
+ "enabled": false,
31
+ "n": 0,
32
+ "horizon": 50,
33
+ "rate": 50,
34
+ "warmstart": 0,
35
+ "terminate_on_success": true
36
+ },
37
+ "env_meta_update_dict": {},
38
+ "ckpt_path": null
39
+ },
40
+ "train": {
41
+ "data": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_pre_vis_wrench/vac_pre_vis_wrench.hdf5",
42
+ "output_dir": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_pre_vis_wrench/robomimic_runs",
43
+ "normalize_weights_by_ds_size": false,
44
+ "num_data_workers": 0,
45
+ "hdf5_cache_mode": "all",
46
+ "hdf5_use_swmr": true,
47
+ "hdf5_load_next_obs": false,
48
+ "hdf5_normalize_obs": false,
49
+ "hdf5_filter_key": "train",
50
+ "hdf5_validation_filter_key": "valid",
51
+ "seq_length": 8,
52
+ "pad_seq_length": true,
53
+ "frame_stack": 2,
54
+ "pad_frame_stack": true,
55
+ "dataset_keys": [
56
+ "actions",
57
+ "rewards",
58
+ "dones"
59
+ ],
60
+ "action_keys": [
61
+ "actions"
62
+ ],
63
+ "action_config": {
64
+ "actions": {
65
+ "normalization": "min_max"
66
+ }
67
+ },
68
+ "goal_mode": null,
69
+ "cuda": true,
70
+ "batch_size": 16,
71
+ "num_epochs": 300,
72
+ "seed": 42,
73
+ "max_grad_norm": null
74
+ },
75
+ "algo": {
76
+ "optim_params": {
77
+ "policy": {
78
+ "optimizer_type": "adamw",
79
+ "learning_rate": {
80
+ "initial": 0.0001,
81
+ "decay_factor": 0.1,
82
+ "step_every_batch": true,
83
+ "scheduler_type": "cosine",
84
+ "num_cycles": 0.5,
85
+ "warmup_steps": 500,
86
+ "epoch_schedule": []
87
+ },
88
+ "regularization": {
89
+ "L2": 1e-06
90
+ }
91
+ }
92
+ },
93
+ "horizon": {
94
+ "observation_horizon": 2,
95
+ "action_horizon": 4,
96
+ "prediction_horizon": 8
97
+ },
98
+ "unet": {
99
+ "enabled": true,
100
+ "diffusion_step_embed_dim": 256,
101
+ "down_dims": [
102
+ 128,
103
+ 256,
104
+ 512
105
+ ],
106
+ "kernel_size": 5,
107
+ "n_groups": 8
108
+ },
109
+ "ema": {
110
+ "enabled": true,
111
+ "power": 0.75
112
+ },
113
+ "ddpm": {
114
+ "enabled": true,
115
+ "num_train_timesteps": 50,
116
+ "num_inference_timesteps": 50,
117
+ "beta_schedule": "squaredcos_cap_v2",
118
+ "clip_sample": true,
119
+ "prediction_type": "epsilon"
120
+ },
121
+ "ddim": {
122
+ "enabled": false,
123
+ "num_train_timesteps": 50,
124
+ "num_inference_timesteps": 10,
125
+ "beta_schedule": "squaredcos_cap_v2",
126
+ "clip_sample": true,
127
+ "set_alpha_to_one": true,
128
+ "steps_offset": 0,
129
+ "prediction_type": "epsilon"
130
+ }
131
+ },
132
+ "observation": {
133
+ "modalities": {
134
+ "obs": {
135
+ "low_dim": [
136
+ "wrench"
137
+ ],
138
+ "rgb": [
139
+ "front_rgb"
140
+ ],
141
+ "depth": [],
142
+ "scan": []
143
+ },
144
+ "goal": {
145
+ "low_dim": [],
146
+ "rgb": [],
147
+ "depth": [],
148
+ "scan": []
149
+ }
150
+ },
151
+ "encoder": {
152
+ "low_dim": {
153
+ "core_class": null,
154
+ "core_kwargs": {},
155
+ "obs_randomizer_class": null,
156
+ "obs_randomizer_kwargs": {}
157
+ },
158
+ "rgb": {
159
+ "core_class": "VisualCore",
160
+ "core_kwargs": {
161
+ "backbone_class": "ResNet18Conv",
162
+ "backbone_kwargs": {
163
+ "pretrained": false,
164
+ "input_coord_conv": false
165
+ },
166
+ "pool_class": "SpatialSoftmax",
167
+ "pool_kwargs": {
168
+ "num_kp": 32,
169
+ "learnable_temperature": false,
170
+ "temperature": 1.0,
171
+ "noise_std": 0.0
172
+ },
173
+ "feature_dimension": 64
174
+ },
175
+ "obs_randomizer_class": null,
176
+ "obs_randomizer_kwargs": {}
177
+ },
178
+ "depth": {
179
+ "core_class": "VisualCore",
180
+ "core_kwargs": {},
181
+ "obs_randomizer_class": null,
182
+ "obs_randomizer_kwargs": {}
183
+ },
184
+ "scan": {
185
+ "core_class": "ScanCore",
186
+ "core_kwargs": {},
187
+ "obs_randomizer_class": null,
188
+ "obs_randomizer_kwargs": {}
189
+ }
190
+ }
191
+ },
192
+ "meta": {
193
+ "hp_base_config_file": null,
194
+ "hp_keys": [],
195
+ "hp_values": []
196
+ }
197
+ }
vac_pre_vis_wrench/dataset_summary.json ADDED
@@ -0,0 +1,2032 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "path": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_pre_vis_wrench/vac_pre_vis_wrench.hdf5",
3
+ "total_frames": 145712,
4
+ "train_episodes": 172,
5
+ "valid_episodes": 30,
6
+ "train_source_episodes": [
7
+ 0,
8
+ 1,
9
+ 2,
10
+ 3,
11
+ 4,
12
+ 5,
13
+ 6,
14
+ 7,
15
+ 8,
16
+ 9,
17
+ 10,
18
+ 11,
19
+ 12,
20
+ 13,
21
+ 14,
22
+ 15,
23
+ 16,
24
+ 17,
25
+ 18,
26
+ 19,
27
+ 20,
28
+ 21,
29
+ 22,
30
+ 23,
31
+ 24,
32
+ 25,
33
+ 26,
34
+ 27,
35
+ 28,
36
+ 29,
37
+ 30,
38
+ 31,
39
+ 32,
40
+ 33,
41
+ 34,
42
+ 35,
43
+ 36,
44
+ 37,
45
+ 38,
46
+ 39,
47
+ 40,
48
+ 41,
49
+ 42,
50
+ 43,
51
+ 44,
52
+ 45,
53
+ 46,
54
+ 47,
55
+ 48,
56
+ 49,
57
+ 50,
58
+ 51,
59
+ 52,
60
+ 53,
61
+ 54,
62
+ 55,
63
+ 56,
64
+ 57,
65
+ 58,
66
+ 59,
67
+ 60,
68
+ 61,
69
+ 62,
70
+ 63,
71
+ 64,
72
+ 65,
73
+ 66,
74
+ 67,
75
+ 68,
76
+ 69,
77
+ 70,
78
+ 71,
79
+ 72,
80
+ 73,
81
+ 74,
82
+ 75,
83
+ 76,
84
+ 77,
85
+ 78,
86
+ 79,
87
+ 80,
88
+ 81,
89
+ 82,
90
+ 83,
91
+ 84,
92
+ 85,
93
+ 86,
94
+ 87,
95
+ 88,
96
+ 89,
97
+ 90,
98
+ 91,
99
+ 92,
100
+ 93,
101
+ 94,
102
+ 95,
103
+ 96,
104
+ 97,
105
+ 98,
106
+ 99,
107
+ 100,
108
+ 101,
109
+ 102,
110
+ 103,
111
+ 104,
112
+ 105,
113
+ 106,
114
+ 107,
115
+ 108,
116
+ 109,
117
+ 110,
118
+ 111,
119
+ 112,
120
+ 113,
121
+ 114,
122
+ 115,
123
+ 116,
124
+ 117,
125
+ 118,
126
+ 119,
127
+ 120,
128
+ 121,
129
+ 122,
130
+ 123,
131
+ 124,
132
+ 125,
133
+ 126,
134
+ 127,
135
+ 128,
136
+ 129,
137
+ 130,
138
+ 131,
139
+ 132,
140
+ 133,
141
+ 134,
142
+ 135,
143
+ 136,
144
+ 137,
145
+ 138,
146
+ 139,
147
+ 140,
148
+ 141,
149
+ 142,
150
+ 143,
151
+ 144,
152
+ 145,
153
+ 146,
154
+ 147,
155
+ 148,
156
+ 149,
157
+ 150,
158
+ 151,
159
+ 152,
160
+ 153,
161
+ 154,
162
+ 155,
163
+ 156,
164
+ 157,
165
+ 158,
166
+ 159,
167
+ 160,
168
+ 161,
169
+ 162,
170
+ 163,
171
+ 164,
172
+ 165,
173
+ 166,
174
+ 167,
175
+ 168,
176
+ 169,
177
+ 170,
178
+ 171
179
+ ],
180
+ "valid_source_episodes": [
181
+ 172,
182
+ 173,
183
+ 174,
184
+ 175,
185
+ 176,
186
+ 177,
187
+ 178,
188
+ 179,
189
+ 180,
190
+ 181,
191
+ 182,
192
+ 183,
193
+ 184,
194
+ 185,
195
+ 186,
196
+ 187,
197
+ 188,
198
+ 189,
199
+ 190,
200
+ 191,
201
+ 192,
202
+ 193,
203
+ 194,
204
+ 195,
205
+ 196,
206
+ 197,
207
+ 198,
208
+ 199,
209
+ 200,
210
+ 201
211
+ ],
212
+ "episodes": [
213
+ {
214
+ "hdf5_key": "ep_00000",
215
+ "source_episode_index": 0,
216
+ "split": "train",
217
+ "num_samples": 663,
218
+ "action_dim": 9,
219
+ "obs_mode": "vision_wrench",
220
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
221
+ },
222
+ {
223
+ "hdf5_key": "ep_00001",
224
+ "source_episode_index": 1,
225
+ "split": "train",
226
+ "num_samples": 706,
227
+ "action_dim": 9,
228
+ "obs_mode": "vision_wrench",
229
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
230
+ },
231
+ {
232
+ "hdf5_key": "ep_00002",
233
+ "source_episode_index": 2,
234
+ "split": "train",
235
+ "num_samples": 1006,
236
+ "action_dim": 9,
237
+ "obs_mode": "vision_wrench",
238
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
239
+ },
240
+ {
241
+ "hdf5_key": "ep_00003",
242
+ "source_episode_index": 3,
243
+ "split": "train",
244
+ "num_samples": 709,
245
+ "action_dim": 9,
246
+ "obs_mode": "vision_wrench",
247
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
248
+ },
249
+ {
250
+ "hdf5_key": "ep_00004",
251
+ "source_episode_index": 4,
252
+ "split": "train",
253
+ "num_samples": 816,
254
+ "action_dim": 9,
255
+ "obs_mode": "vision_wrench",
256
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
257
+ },
258
+ {
259
+ "hdf5_key": "ep_00005",
260
+ "source_episode_index": 5,
261
+ "split": "train",
262
+ "num_samples": 882,
263
+ "action_dim": 9,
264
+ "obs_mode": "vision_wrench",
265
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
266
+ },
267
+ {
268
+ "hdf5_key": "ep_00006",
269
+ "source_episode_index": 6,
270
+ "split": "train",
271
+ "num_samples": 699,
272
+ "action_dim": 9,
273
+ "obs_mode": "vision_wrench",
274
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
275
+ },
276
+ {
277
+ "hdf5_key": "ep_00007",
278
+ "source_episode_index": 7,
279
+ "split": "train",
280
+ "num_samples": 851,
281
+ "action_dim": 9,
282
+ "obs_mode": "vision_wrench",
283
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
284
+ },
285
+ {
286
+ "hdf5_key": "ep_00008",
287
+ "source_episode_index": 8,
288
+ "split": "train",
289
+ "num_samples": 728,
290
+ "action_dim": 9,
291
+ "obs_mode": "vision_wrench",
292
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
293
+ },
294
+ {
295
+ "hdf5_key": "ep_00009",
296
+ "source_episode_index": 9,
297
+ "split": "train",
298
+ "num_samples": 770,
299
+ "action_dim": 9,
300
+ "obs_mode": "vision_wrench",
301
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
302
+ },
303
+ {
304
+ "hdf5_key": "ep_00010",
305
+ "source_episode_index": 10,
306
+ "split": "train",
307
+ "num_samples": 821,
308
+ "action_dim": 9,
309
+ "obs_mode": "vision_wrench",
310
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
311
+ },
312
+ {
313
+ "hdf5_key": "ep_00011",
314
+ "source_episode_index": 11,
315
+ "split": "train",
316
+ "num_samples": 561,
317
+ "action_dim": 9,
318
+ "obs_mode": "vision_wrench",
319
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
320
+ },
321
+ {
322
+ "hdf5_key": "ep_00012",
323
+ "source_episode_index": 12,
324
+ "split": "train",
325
+ "num_samples": 517,
326
+ "action_dim": 9,
327
+ "obs_mode": "vision_wrench",
328
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
329
+ },
330
+ {
331
+ "hdf5_key": "ep_00013",
332
+ "source_episode_index": 13,
333
+ "split": "train",
334
+ "num_samples": 690,
335
+ "action_dim": 9,
336
+ "obs_mode": "vision_wrench",
337
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
338
+ },
339
+ {
340
+ "hdf5_key": "ep_00014",
341
+ "source_episode_index": 14,
342
+ "split": "train",
343
+ "num_samples": 559,
344
+ "action_dim": 9,
345
+ "obs_mode": "vision_wrench",
346
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
347
+ },
348
+ {
349
+ "hdf5_key": "ep_00015",
350
+ "source_episode_index": 15,
351
+ "split": "train",
352
+ "num_samples": 889,
353
+ "action_dim": 9,
354
+ "obs_mode": "vision_wrench",
355
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
356
+ },
357
+ {
358
+ "hdf5_key": "ep_00016",
359
+ "source_episode_index": 16,
360
+ "split": "train",
361
+ "num_samples": 619,
362
+ "action_dim": 9,
363
+ "obs_mode": "vision_wrench",
364
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
365
+ },
366
+ {
367
+ "hdf5_key": "ep_00017",
368
+ "source_episode_index": 17,
369
+ "split": "train",
370
+ "num_samples": 640,
371
+ "action_dim": 9,
372
+ "obs_mode": "vision_wrench",
373
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
374
+ },
375
+ {
376
+ "hdf5_key": "ep_00018",
377
+ "source_episode_index": 18,
378
+ "split": "train",
379
+ "num_samples": 612,
380
+ "action_dim": 9,
381
+ "obs_mode": "vision_wrench",
382
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
383
+ },
384
+ {
385
+ "hdf5_key": "ep_00019",
386
+ "source_episode_index": 19,
387
+ "split": "train",
388
+ "num_samples": 702,
389
+ "action_dim": 9,
390
+ "obs_mode": "vision_wrench",
391
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
392
+ },
393
+ {
394
+ "hdf5_key": "ep_00020",
395
+ "source_episode_index": 20,
396
+ "split": "train",
397
+ "num_samples": 645,
398
+ "action_dim": 9,
399
+ "obs_mode": "vision_wrench",
400
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
401
+ },
402
+ {
403
+ "hdf5_key": "ep_00021",
404
+ "source_episode_index": 21,
405
+ "split": "train",
406
+ "num_samples": 751,
407
+ "action_dim": 9,
408
+ "obs_mode": "vision_wrench",
409
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
410
+ },
411
+ {
412
+ "hdf5_key": "ep_00022",
413
+ "source_episode_index": 22,
414
+ "split": "train",
415
+ "num_samples": 810,
416
+ "action_dim": 9,
417
+ "obs_mode": "vision_wrench",
418
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
419
+ },
420
+ {
421
+ "hdf5_key": "ep_00023",
422
+ "source_episode_index": 23,
423
+ "split": "train",
424
+ "num_samples": 734,
425
+ "action_dim": 9,
426
+ "obs_mode": "vision_wrench",
427
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
428
+ },
429
+ {
430
+ "hdf5_key": "ep_00024",
431
+ "source_episode_index": 24,
432
+ "split": "train",
433
+ "num_samples": 637,
434
+ "action_dim": 9,
435
+ "obs_mode": "vision_wrench",
436
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
437
+ },
438
+ {
439
+ "hdf5_key": "ep_00025",
440
+ "source_episode_index": 25,
441
+ "split": "train",
442
+ "num_samples": 640,
443
+ "action_dim": 9,
444
+ "obs_mode": "vision_wrench",
445
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
446
+ },
447
+ {
448
+ "hdf5_key": "ep_00026",
449
+ "source_episode_index": 26,
450
+ "split": "train",
451
+ "num_samples": 957,
452
+ "action_dim": 9,
453
+ "obs_mode": "vision_wrench",
454
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
455
+ },
456
+ {
457
+ "hdf5_key": "ep_00027",
458
+ "source_episode_index": 27,
459
+ "split": "train",
460
+ "num_samples": 617,
461
+ "action_dim": 9,
462
+ "obs_mode": "vision_wrench",
463
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
464
+ },
465
+ {
466
+ "hdf5_key": "ep_00028",
467
+ "source_episode_index": 28,
468
+ "split": "train",
469
+ "num_samples": 772,
470
+ "action_dim": 9,
471
+ "obs_mode": "vision_wrench",
472
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
473
+ },
474
+ {
475
+ "hdf5_key": "ep_00029",
476
+ "source_episode_index": 29,
477
+ "split": "train",
478
+ "num_samples": 616,
479
+ "action_dim": 9,
480
+ "obs_mode": "vision_wrench",
481
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
482
+ },
483
+ {
484
+ "hdf5_key": "ep_00030",
485
+ "source_episode_index": 30,
486
+ "split": "train",
487
+ "num_samples": 732,
488
+ "action_dim": 9,
489
+ "obs_mode": "vision_wrench",
490
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
491
+ },
492
+ {
493
+ "hdf5_key": "ep_00031",
494
+ "source_episode_index": 31,
495
+ "split": "train",
496
+ "num_samples": 761,
497
+ "action_dim": 9,
498
+ "obs_mode": "vision_wrench",
499
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
500
+ },
501
+ {
502
+ "hdf5_key": "ep_00032",
503
+ "source_episode_index": 32,
504
+ "split": "train",
505
+ "num_samples": 804,
506
+ "action_dim": 9,
507
+ "obs_mode": "vision_wrench",
508
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
509
+ },
510
+ {
511
+ "hdf5_key": "ep_00033",
512
+ "source_episode_index": 33,
513
+ "split": "train",
514
+ "num_samples": 735,
515
+ "action_dim": 9,
516
+ "obs_mode": "vision_wrench",
517
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
518
+ },
519
+ {
520
+ "hdf5_key": "ep_00034",
521
+ "source_episode_index": 34,
522
+ "split": "train",
523
+ "num_samples": 721,
524
+ "action_dim": 9,
525
+ "obs_mode": "vision_wrench",
526
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
527
+ },
528
+ {
529
+ "hdf5_key": "ep_00035",
530
+ "source_episode_index": 35,
531
+ "split": "train",
532
+ "num_samples": 612,
533
+ "action_dim": 9,
534
+ "obs_mode": "vision_wrench",
535
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
536
+ },
537
+ {
538
+ "hdf5_key": "ep_00036",
539
+ "source_episode_index": 36,
540
+ "split": "train",
541
+ "num_samples": 685,
542
+ "action_dim": 9,
543
+ "obs_mode": "vision_wrench",
544
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
545
+ },
546
+ {
547
+ "hdf5_key": "ep_00037",
548
+ "source_episode_index": 37,
549
+ "split": "train",
550
+ "num_samples": 685,
551
+ "action_dim": 9,
552
+ "obs_mode": "vision_wrench",
553
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
554
+ },
555
+ {
556
+ "hdf5_key": "ep_00038",
557
+ "source_episode_index": 38,
558
+ "split": "train",
559
+ "num_samples": 663,
560
+ "action_dim": 9,
561
+ "obs_mode": "vision_wrench",
562
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
563
+ },
564
+ {
565
+ "hdf5_key": "ep_00039",
566
+ "source_episode_index": 39,
567
+ "split": "train",
568
+ "num_samples": 855,
569
+ "action_dim": 9,
570
+ "obs_mode": "vision_wrench",
571
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
572
+ },
573
+ {
574
+ "hdf5_key": "ep_00040",
575
+ "source_episode_index": 40,
576
+ "split": "train",
577
+ "num_samples": 810,
578
+ "action_dim": 9,
579
+ "obs_mode": "vision_wrench",
580
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
581
+ },
582
+ {
583
+ "hdf5_key": "ep_00041",
584
+ "source_episode_index": 41,
585
+ "split": "train",
586
+ "num_samples": 655,
587
+ "action_dim": 9,
588
+ "obs_mode": "vision_wrench",
589
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
590
+ },
591
+ {
592
+ "hdf5_key": "ep_00042",
593
+ "source_episode_index": 42,
594
+ "split": "train",
595
+ "num_samples": 946,
596
+ "action_dim": 9,
597
+ "obs_mode": "vision_wrench",
598
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
599
+ },
600
+ {
601
+ "hdf5_key": "ep_00043",
602
+ "source_episode_index": 43,
603
+ "split": "train",
604
+ "num_samples": 703,
605
+ "action_dim": 9,
606
+ "obs_mode": "vision_wrench",
607
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
608
+ },
609
+ {
610
+ "hdf5_key": "ep_00044",
611
+ "source_episode_index": 44,
612
+ "split": "train",
613
+ "num_samples": 606,
614
+ "action_dim": 9,
615
+ "obs_mode": "vision_wrench",
616
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
617
+ },
618
+ {
619
+ "hdf5_key": "ep_00045",
620
+ "source_episode_index": 45,
621
+ "split": "train",
622
+ "num_samples": 914,
623
+ "action_dim": 9,
624
+ "obs_mode": "vision_wrench",
625
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
626
+ },
627
+ {
628
+ "hdf5_key": "ep_00046",
629
+ "source_episode_index": 46,
630
+ "split": "train",
631
+ "num_samples": 949,
632
+ "action_dim": 9,
633
+ "obs_mode": "vision_wrench",
634
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
635
+ },
636
+ {
637
+ "hdf5_key": "ep_00047",
638
+ "source_episode_index": 47,
639
+ "split": "train",
640
+ "num_samples": 818,
641
+ "action_dim": 9,
642
+ "obs_mode": "vision_wrench",
643
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
644
+ },
645
+ {
646
+ "hdf5_key": "ep_00048",
647
+ "source_episode_index": 48,
648
+ "split": "train",
649
+ "num_samples": 913,
650
+ "action_dim": 9,
651
+ "obs_mode": "vision_wrench",
652
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
653
+ },
654
+ {
655
+ "hdf5_key": "ep_00049",
656
+ "source_episode_index": 49,
657
+ "split": "train",
658
+ "num_samples": 904,
659
+ "action_dim": 9,
660
+ "obs_mode": "vision_wrench",
661
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
662
+ },
663
+ {
664
+ "hdf5_key": "ep_00050",
665
+ "source_episode_index": 50,
666
+ "split": "train",
667
+ "num_samples": 923,
668
+ "action_dim": 9,
669
+ "obs_mode": "vision_wrench",
670
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
671
+ },
672
+ {
673
+ "hdf5_key": "ep_00051",
674
+ "source_episode_index": 51,
675
+ "split": "train",
676
+ "num_samples": 824,
677
+ "action_dim": 9,
678
+ "obs_mode": "vision_wrench",
679
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
680
+ },
681
+ {
682
+ "hdf5_key": "ep_00052",
683
+ "source_episode_index": 52,
684
+ "split": "train",
685
+ "num_samples": 777,
686
+ "action_dim": 9,
687
+ "obs_mode": "vision_wrench",
688
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
689
+ },
690
+ {
691
+ "hdf5_key": "ep_00053",
692
+ "source_episode_index": 53,
693
+ "split": "train",
694
+ "num_samples": 714,
695
+ "action_dim": 9,
696
+ "obs_mode": "vision_wrench",
697
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
698
+ },
699
+ {
700
+ "hdf5_key": "ep_00054",
701
+ "source_episode_index": 54,
702
+ "split": "train",
703
+ "num_samples": 771,
704
+ "action_dim": 9,
705
+ "obs_mode": "vision_wrench",
706
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
707
+ },
708
+ {
709
+ "hdf5_key": "ep_00055",
710
+ "source_episode_index": 55,
711
+ "split": "train",
712
+ "num_samples": 684,
713
+ "action_dim": 9,
714
+ "obs_mode": "vision_wrench",
715
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
716
+ },
717
+ {
718
+ "hdf5_key": "ep_00056",
719
+ "source_episode_index": 56,
720
+ "split": "train",
721
+ "num_samples": 766,
722
+ "action_dim": 9,
723
+ "obs_mode": "vision_wrench",
724
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
725
+ },
726
+ {
727
+ "hdf5_key": "ep_00057",
728
+ "source_episode_index": 57,
729
+ "split": "train",
730
+ "num_samples": 890,
731
+ "action_dim": 9,
732
+ "obs_mode": "vision_wrench",
733
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
734
+ },
735
+ {
736
+ "hdf5_key": "ep_00058",
737
+ "source_episode_index": 58,
738
+ "split": "train",
739
+ "num_samples": 774,
740
+ "action_dim": 9,
741
+ "obs_mode": "vision_wrench",
742
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
743
+ },
744
+ {
745
+ "hdf5_key": "ep_00059",
746
+ "source_episode_index": 59,
747
+ "split": "train",
748
+ "num_samples": 781,
749
+ "action_dim": 9,
750
+ "obs_mode": "vision_wrench",
751
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
752
+ },
753
+ {
754
+ "hdf5_key": "ep_00060",
755
+ "source_episode_index": 60,
756
+ "split": "train",
757
+ "num_samples": 1093,
758
+ "action_dim": 9,
759
+ "obs_mode": "vision_wrench",
760
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
761
+ },
762
+ {
763
+ "hdf5_key": "ep_00061",
764
+ "source_episode_index": 61,
765
+ "split": "train",
766
+ "num_samples": 900,
767
+ "action_dim": 9,
768
+ "obs_mode": "vision_wrench",
769
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
770
+ },
771
+ {
772
+ "hdf5_key": "ep_00062",
773
+ "source_episode_index": 62,
774
+ "split": "train",
775
+ "num_samples": 814,
776
+ "action_dim": 9,
777
+ "obs_mode": "vision_wrench",
778
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
779
+ },
780
+ {
781
+ "hdf5_key": "ep_00063",
782
+ "source_episode_index": 63,
783
+ "split": "train",
784
+ "num_samples": 1108,
785
+ "action_dim": 9,
786
+ "obs_mode": "vision_wrench",
787
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
788
+ },
789
+ {
790
+ "hdf5_key": "ep_00064",
791
+ "source_episode_index": 64,
792
+ "split": "train",
793
+ "num_samples": 816,
794
+ "action_dim": 9,
795
+ "obs_mode": "vision_wrench",
796
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
797
+ },
798
+ {
799
+ "hdf5_key": "ep_00065",
800
+ "source_episode_index": 65,
801
+ "split": "train",
802
+ "num_samples": 963,
803
+ "action_dim": 9,
804
+ "obs_mode": "vision_wrench",
805
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
806
+ },
807
+ {
808
+ "hdf5_key": "ep_00066",
809
+ "source_episode_index": 66,
810
+ "split": "train",
811
+ "num_samples": 895,
812
+ "action_dim": 9,
813
+ "obs_mode": "vision_wrench",
814
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
815
+ },
816
+ {
817
+ "hdf5_key": "ep_00067",
818
+ "source_episode_index": 67,
819
+ "split": "train",
820
+ "num_samples": 572,
821
+ "action_dim": 9,
822
+ "obs_mode": "vision_wrench",
823
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
824
+ },
825
+ {
826
+ "hdf5_key": "ep_00068",
827
+ "source_episode_index": 68,
828
+ "split": "train",
829
+ "num_samples": 648,
830
+ "action_dim": 9,
831
+ "obs_mode": "vision_wrench",
832
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
833
+ },
834
+ {
835
+ "hdf5_key": "ep_00069",
836
+ "source_episode_index": 69,
837
+ "split": "train",
838
+ "num_samples": 772,
839
+ "action_dim": 9,
840
+ "obs_mode": "vision_wrench",
841
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
842
+ },
843
+ {
844
+ "hdf5_key": "ep_00070",
845
+ "source_episode_index": 70,
846
+ "split": "train",
847
+ "num_samples": 702,
848
+ "action_dim": 9,
849
+ "obs_mode": "vision_wrench",
850
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
851
+ },
852
+ {
853
+ "hdf5_key": "ep_00071",
854
+ "source_episode_index": 71,
855
+ "split": "train",
856
+ "num_samples": 786,
857
+ "action_dim": 9,
858
+ "obs_mode": "vision_wrench",
859
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
860
+ },
861
+ {
862
+ "hdf5_key": "ep_00072",
863
+ "source_episode_index": 72,
864
+ "split": "train",
865
+ "num_samples": 843,
866
+ "action_dim": 9,
867
+ "obs_mode": "vision_wrench",
868
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
869
+ },
870
+ {
871
+ "hdf5_key": "ep_00073",
872
+ "source_episode_index": 73,
873
+ "split": "train",
874
+ "num_samples": 767,
875
+ "action_dim": 9,
876
+ "obs_mode": "vision_wrench",
877
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
878
+ },
879
+ {
880
+ "hdf5_key": "ep_00074",
881
+ "source_episode_index": 74,
882
+ "split": "train",
883
+ "num_samples": 764,
884
+ "action_dim": 9,
885
+ "obs_mode": "vision_wrench",
886
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
887
+ },
888
+ {
889
+ "hdf5_key": "ep_00075",
890
+ "source_episode_index": 75,
891
+ "split": "train",
892
+ "num_samples": 750,
893
+ "action_dim": 9,
894
+ "obs_mode": "vision_wrench",
895
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
896
+ },
897
+ {
898
+ "hdf5_key": "ep_00076",
899
+ "source_episode_index": 76,
900
+ "split": "train",
901
+ "num_samples": 652,
902
+ "action_dim": 9,
903
+ "obs_mode": "vision_wrench",
904
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
905
+ },
906
+ {
907
+ "hdf5_key": "ep_00077",
908
+ "source_episode_index": 77,
909
+ "split": "train",
910
+ "num_samples": 812,
911
+ "action_dim": 9,
912
+ "obs_mode": "vision_wrench",
913
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
914
+ },
915
+ {
916
+ "hdf5_key": "ep_00078",
917
+ "source_episode_index": 78,
918
+ "split": "train",
919
+ "num_samples": 722,
920
+ "action_dim": 9,
921
+ "obs_mode": "vision_wrench",
922
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
923
+ },
924
+ {
925
+ "hdf5_key": "ep_00079",
926
+ "source_episode_index": 79,
927
+ "split": "train",
928
+ "num_samples": 723,
929
+ "action_dim": 9,
930
+ "obs_mode": "vision_wrench",
931
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
932
+ },
933
+ {
934
+ "hdf5_key": "ep_00080",
935
+ "source_episode_index": 80,
936
+ "split": "train",
937
+ "num_samples": 765,
938
+ "action_dim": 9,
939
+ "obs_mode": "vision_wrench",
940
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
941
+ },
942
+ {
943
+ "hdf5_key": "ep_00081",
944
+ "source_episode_index": 81,
945
+ "split": "train",
946
+ "num_samples": 981,
947
+ "action_dim": 9,
948
+ "obs_mode": "vision_wrench",
949
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
950
+ },
951
+ {
952
+ "hdf5_key": "ep_00082",
953
+ "source_episode_index": 82,
954
+ "split": "train",
955
+ "num_samples": 959,
956
+ "action_dim": 9,
957
+ "obs_mode": "vision_wrench",
958
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
959
+ },
960
+ {
961
+ "hdf5_key": "ep_00083",
962
+ "source_episode_index": 83,
963
+ "split": "train",
964
+ "num_samples": 655,
965
+ "action_dim": 9,
966
+ "obs_mode": "vision_wrench",
967
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
968
+ },
969
+ {
970
+ "hdf5_key": "ep_00084",
971
+ "source_episode_index": 84,
972
+ "split": "train",
973
+ "num_samples": 746,
974
+ "action_dim": 9,
975
+ "obs_mode": "vision_wrench",
976
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
977
+ },
978
+ {
979
+ "hdf5_key": "ep_00085",
980
+ "source_episode_index": 85,
981
+ "split": "train",
982
+ "num_samples": 660,
983
+ "action_dim": 9,
984
+ "obs_mode": "vision_wrench",
985
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
986
+ },
987
+ {
988
+ "hdf5_key": "ep_00086",
989
+ "source_episode_index": 86,
990
+ "split": "train",
991
+ "num_samples": 617,
992
+ "action_dim": 9,
993
+ "obs_mode": "vision_wrench",
994
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
995
+ },
996
+ {
997
+ "hdf5_key": "ep_00087",
998
+ "source_episode_index": 87,
999
+ "split": "train",
1000
+ "num_samples": 819,
1001
+ "action_dim": 9,
1002
+ "obs_mode": "vision_wrench",
1003
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1004
+ },
1005
+ {
1006
+ "hdf5_key": "ep_00088",
1007
+ "source_episode_index": 88,
1008
+ "split": "train",
1009
+ "num_samples": 696,
1010
+ "action_dim": 9,
1011
+ "obs_mode": "vision_wrench",
1012
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1013
+ },
1014
+ {
1015
+ "hdf5_key": "ep_00089",
1016
+ "source_episode_index": 89,
1017
+ "split": "train",
1018
+ "num_samples": 855,
1019
+ "action_dim": 9,
1020
+ "obs_mode": "vision_wrench",
1021
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1022
+ },
1023
+ {
1024
+ "hdf5_key": "ep_00090",
1025
+ "source_episode_index": 90,
1026
+ "split": "train",
1027
+ "num_samples": 678,
1028
+ "action_dim": 9,
1029
+ "obs_mode": "vision_wrench",
1030
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1031
+ },
1032
+ {
1033
+ "hdf5_key": "ep_00091",
1034
+ "source_episode_index": 91,
1035
+ "split": "train",
1036
+ "num_samples": 711,
1037
+ "action_dim": 9,
1038
+ "obs_mode": "vision_wrench",
1039
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1040
+ },
1041
+ {
1042
+ "hdf5_key": "ep_00092",
1043
+ "source_episode_index": 92,
1044
+ "split": "train",
1045
+ "num_samples": 763,
1046
+ "action_dim": 9,
1047
+ "obs_mode": "vision_wrench",
1048
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1049
+ },
1050
+ {
1051
+ "hdf5_key": "ep_00093",
1052
+ "source_episode_index": 93,
1053
+ "split": "train",
1054
+ "num_samples": 723,
1055
+ "action_dim": 9,
1056
+ "obs_mode": "vision_wrench",
1057
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1058
+ },
1059
+ {
1060
+ "hdf5_key": "ep_00094",
1061
+ "source_episode_index": 94,
1062
+ "split": "train",
1063
+ "num_samples": 723,
1064
+ "action_dim": 9,
1065
+ "obs_mode": "vision_wrench",
1066
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1067
+ },
1068
+ {
1069
+ "hdf5_key": "ep_00095",
1070
+ "source_episode_index": 95,
1071
+ "split": "train",
1072
+ "num_samples": 865,
1073
+ "action_dim": 9,
1074
+ "obs_mode": "vision_wrench",
1075
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1076
+ },
1077
+ {
1078
+ "hdf5_key": "ep_00096",
1079
+ "source_episode_index": 96,
1080
+ "split": "train",
1081
+ "num_samples": 495,
1082
+ "action_dim": 9,
1083
+ "obs_mode": "vision_wrench",
1084
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1085
+ },
1086
+ {
1087
+ "hdf5_key": "ep_00097",
1088
+ "source_episode_index": 97,
1089
+ "split": "train",
1090
+ "num_samples": 451,
1091
+ "action_dim": 9,
1092
+ "obs_mode": "vision_wrench",
1093
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1094
+ },
1095
+ {
1096
+ "hdf5_key": "ep_00098",
1097
+ "source_episode_index": 98,
1098
+ "split": "train",
1099
+ "num_samples": 690,
1100
+ "action_dim": 9,
1101
+ "obs_mode": "vision_wrench",
1102
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1103
+ },
1104
+ {
1105
+ "hdf5_key": "ep_00099",
1106
+ "source_episode_index": 99,
1107
+ "split": "train",
1108
+ "num_samples": 690,
1109
+ "action_dim": 9,
1110
+ "obs_mode": "vision_wrench",
1111
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1112
+ },
1113
+ {
1114
+ "hdf5_key": "ep_00100",
1115
+ "source_episode_index": 100,
1116
+ "split": "train",
1117
+ "num_samples": 615,
1118
+ "action_dim": 9,
1119
+ "obs_mode": "vision_wrench",
1120
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1121
+ },
1122
+ {
1123
+ "hdf5_key": "ep_00101",
1124
+ "source_episode_index": 101,
1125
+ "split": "train",
1126
+ "num_samples": 662,
1127
+ "action_dim": 9,
1128
+ "obs_mode": "vision_wrench",
1129
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1130
+ },
1131
+ {
1132
+ "hdf5_key": "ep_00102",
1133
+ "source_episode_index": 102,
1134
+ "split": "train",
1135
+ "num_samples": 614,
1136
+ "action_dim": 9,
1137
+ "obs_mode": "vision_wrench",
1138
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1139
+ },
1140
+ {
1141
+ "hdf5_key": "ep_00103",
1142
+ "source_episode_index": 103,
1143
+ "split": "train",
1144
+ "num_samples": 730,
1145
+ "action_dim": 9,
1146
+ "obs_mode": "vision_wrench",
1147
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1148
+ },
1149
+ {
1150
+ "hdf5_key": "ep_00104",
1151
+ "source_episode_index": 104,
1152
+ "split": "train",
1153
+ "num_samples": 683,
1154
+ "action_dim": 9,
1155
+ "obs_mode": "vision_wrench",
1156
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1157
+ },
1158
+ {
1159
+ "hdf5_key": "ep_00105",
1160
+ "source_episode_index": 105,
1161
+ "split": "train",
1162
+ "num_samples": 602,
1163
+ "action_dim": 9,
1164
+ "obs_mode": "vision_wrench",
1165
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1166
+ },
1167
+ {
1168
+ "hdf5_key": "ep_00106",
1169
+ "source_episode_index": 106,
1170
+ "split": "train",
1171
+ "num_samples": 652,
1172
+ "action_dim": 9,
1173
+ "obs_mode": "vision_wrench",
1174
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1175
+ },
1176
+ {
1177
+ "hdf5_key": "ep_00107",
1178
+ "source_episode_index": 107,
1179
+ "split": "train",
1180
+ "num_samples": 722,
1181
+ "action_dim": 9,
1182
+ "obs_mode": "vision_wrench",
1183
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1184
+ },
1185
+ {
1186
+ "hdf5_key": "ep_00108",
1187
+ "source_episode_index": 108,
1188
+ "split": "train",
1189
+ "num_samples": 605,
1190
+ "action_dim": 9,
1191
+ "obs_mode": "vision_wrench",
1192
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1193
+ },
1194
+ {
1195
+ "hdf5_key": "ep_00109",
1196
+ "source_episode_index": 109,
1197
+ "split": "train",
1198
+ "num_samples": 944,
1199
+ "action_dim": 9,
1200
+ "obs_mode": "vision_wrench",
1201
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1202
+ },
1203
+ {
1204
+ "hdf5_key": "ep_00110",
1205
+ "source_episode_index": 110,
1206
+ "split": "train",
1207
+ "num_samples": 661,
1208
+ "action_dim": 9,
1209
+ "obs_mode": "vision_wrench",
1210
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1211
+ },
1212
+ {
1213
+ "hdf5_key": "ep_00111",
1214
+ "source_episode_index": 111,
1215
+ "split": "train",
1216
+ "num_samples": 644,
1217
+ "action_dim": 9,
1218
+ "obs_mode": "vision_wrench",
1219
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1220
+ },
1221
+ {
1222
+ "hdf5_key": "ep_00112",
1223
+ "source_episode_index": 112,
1224
+ "split": "train",
1225
+ "num_samples": 614,
1226
+ "action_dim": 9,
1227
+ "obs_mode": "vision_wrench",
1228
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1229
+ },
1230
+ {
1231
+ "hdf5_key": "ep_00113",
1232
+ "source_episode_index": 113,
1233
+ "split": "train",
1234
+ "num_samples": 548,
1235
+ "action_dim": 9,
1236
+ "obs_mode": "vision_wrench",
1237
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1238
+ },
1239
+ {
1240
+ "hdf5_key": "ep_00114",
1241
+ "source_episode_index": 114,
1242
+ "split": "train",
1243
+ "num_samples": 568,
1244
+ "action_dim": 9,
1245
+ "obs_mode": "vision_wrench",
1246
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1247
+ },
1248
+ {
1249
+ "hdf5_key": "ep_00115",
1250
+ "source_episode_index": 115,
1251
+ "split": "train",
1252
+ "num_samples": 730,
1253
+ "action_dim": 9,
1254
+ "obs_mode": "vision_wrench",
1255
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1256
+ },
1257
+ {
1258
+ "hdf5_key": "ep_00116",
1259
+ "source_episode_index": 116,
1260
+ "split": "train",
1261
+ "num_samples": 481,
1262
+ "action_dim": 9,
1263
+ "obs_mode": "vision_wrench",
1264
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1265
+ },
1266
+ {
1267
+ "hdf5_key": "ep_00117",
1268
+ "source_episode_index": 117,
1269
+ "split": "train",
1270
+ "num_samples": 687,
1271
+ "action_dim": 9,
1272
+ "obs_mode": "vision_wrench",
1273
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1274
+ },
1275
+ {
1276
+ "hdf5_key": "ep_00118",
1277
+ "source_episode_index": 118,
1278
+ "split": "train",
1279
+ "num_samples": 618,
1280
+ "action_dim": 9,
1281
+ "obs_mode": "vision_wrench",
1282
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1283
+ },
1284
+ {
1285
+ "hdf5_key": "ep_00119",
1286
+ "source_episode_index": 119,
1287
+ "split": "train",
1288
+ "num_samples": 768,
1289
+ "action_dim": 9,
1290
+ "obs_mode": "vision_wrench",
1291
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1292
+ },
1293
+ {
1294
+ "hdf5_key": "ep_00120",
1295
+ "source_episode_index": 120,
1296
+ "split": "train",
1297
+ "num_samples": 734,
1298
+ "action_dim": 9,
1299
+ "obs_mode": "vision_wrench",
1300
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1301
+ },
1302
+ {
1303
+ "hdf5_key": "ep_00121",
1304
+ "source_episode_index": 121,
1305
+ "split": "train",
1306
+ "num_samples": 604,
1307
+ "action_dim": 9,
1308
+ "obs_mode": "vision_wrench",
1309
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1310
+ },
1311
+ {
1312
+ "hdf5_key": "ep_00122",
1313
+ "source_episode_index": 122,
1314
+ "split": "train",
1315
+ "num_samples": 610,
1316
+ "action_dim": 9,
1317
+ "obs_mode": "vision_wrench",
1318
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1319
+ },
1320
+ {
1321
+ "hdf5_key": "ep_00123",
1322
+ "source_episode_index": 123,
1323
+ "split": "train",
1324
+ "num_samples": 602,
1325
+ "action_dim": 9,
1326
+ "obs_mode": "vision_wrench",
1327
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1328
+ },
1329
+ {
1330
+ "hdf5_key": "ep_00124",
1331
+ "source_episode_index": 124,
1332
+ "split": "train",
1333
+ "num_samples": 630,
1334
+ "action_dim": 9,
1335
+ "obs_mode": "vision_wrench",
1336
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1337
+ },
1338
+ {
1339
+ "hdf5_key": "ep_00125",
1340
+ "source_episode_index": 125,
1341
+ "split": "train",
1342
+ "num_samples": 621,
1343
+ "action_dim": 9,
1344
+ "obs_mode": "vision_wrench",
1345
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1346
+ },
1347
+ {
1348
+ "hdf5_key": "ep_00126",
1349
+ "source_episode_index": 126,
1350
+ "split": "train",
1351
+ "num_samples": 600,
1352
+ "action_dim": 9,
1353
+ "obs_mode": "vision_wrench",
1354
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1355
+ },
1356
+ {
1357
+ "hdf5_key": "ep_00127",
1358
+ "source_episode_index": 127,
1359
+ "split": "train",
1360
+ "num_samples": 478,
1361
+ "action_dim": 9,
1362
+ "obs_mode": "vision_wrench",
1363
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1364
+ },
1365
+ {
1366
+ "hdf5_key": "ep_00128",
1367
+ "source_episode_index": 128,
1368
+ "split": "train",
1369
+ "num_samples": 606,
1370
+ "action_dim": 9,
1371
+ "obs_mode": "vision_wrench",
1372
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1373
+ },
1374
+ {
1375
+ "hdf5_key": "ep_00129",
1376
+ "source_episode_index": 129,
1377
+ "split": "train",
1378
+ "num_samples": 666,
1379
+ "action_dim": 9,
1380
+ "obs_mode": "vision_wrench",
1381
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1382
+ },
1383
+ {
1384
+ "hdf5_key": "ep_00130",
1385
+ "source_episode_index": 130,
1386
+ "split": "train",
1387
+ "num_samples": 655,
1388
+ "action_dim": 9,
1389
+ "obs_mode": "vision_wrench",
1390
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1391
+ },
1392
+ {
1393
+ "hdf5_key": "ep_00131",
1394
+ "source_episode_index": 131,
1395
+ "split": "train",
1396
+ "num_samples": 588,
1397
+ "action_dim": 9,
1398
+ "obs_mode": "vision_wrench",
1399
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1400
+ },
1401
+ {
1402
+ "hdf5_key": "ep_00132",
1403
+ "source_episode_index": 132,
1404
+ "split": "train",
1405
+ "num_samples": 806,
1406
+ "action_dim": 9,
1407
+ "obs_mode": "vision_wrench",
1408
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1409
+ },
1410
+ {
1411
+ "hdf5_key": "ep_00133",
1412
+ "source_episode_index": 133,
1413
+ "split": "train",
1414
+ "num_samples": 586,
1415
+ "action_dim": 9,
1416
+ "obs_mode": "vision_wrench",
1417
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1418
+ },
1419
+ {
1420
+ "hdf5_key": "ep_00134",
1421
+ "source_episode_index": 134,
1422
+ "split": "train",
1423
+ "num_samples": 714,
1424
+ "action_dim": 9,
1425
+ "obs_mode": "vision_wrench",
1426
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1427
+ },
1428
+ {
1429
+ "hdf5_key": "ep_00135",
1430
+ "source_episode_index": 135,
1431
+ "split": "train",
1432
+ "num_samples": 574,
1433
+ "action_dim": 9,
1434
+ "obs_mode": "vision_wrench",
1435
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1436
+ },
1437
+ {
1438
+ "hdf5_key": "ep_00136",
1439
+ "source_episode_index": 136,
1440
+ "split": "train",
1441
+ "num_samples": 577,
1442
+ "action_dim": 9,
1443
+ "obs_mode": "vision_wrench",
1444
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1445
+ },
1446
+ {
1447
+ "hdf5_key": "ep_00137",
1448
+ "source_episode_index": 137,
1449
+ "split": "train",
1450
+ "num_samples": 663,
1451
+ "action_dim": 9,
1452
+ "obs_mode": "vision_wrench",
1453
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1454
+ },
1455
+ {
1456
+ "hdf5_key": "ep_00138",
1457
+ "source_episode_index": 138,
1458
+ "split": "train",
1459
+ "num_samples": 720,
1460
+ "action_dim": 9,
1461
+ "obs_mode": "vision_wrench",
1462
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1463
+ },
1464
+ {
1465
+ "hdf5_key": "ep_00139",
1466
+ "source_episode_index": 139,
1467
+ "split": "train",
1468
+ "num_samples": 640,
1469
+ "action_dim": 9,
1470
+ "obs_mode": "vision_wrench",
1471
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1472
+ },
1473
+ {
1474
+ "hdf5_key": "ep_00140",
1475
+ "source_episode_index": 140,
1476
+ "split": "train",
1477
+ "num_samples": 544,
1478
+ "action_dim": 9,
1479
+ "obs_mode": "vision_wrench",
1480
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1481
+ },
1482
+ {
1483
+ "hdf5_key": "ep_00141",
1484
+ "source_episode_index": 141,
1485
+ "split": "train",
1486
+ "num_samples": 539,
1487
+ "action_dim": 9,
1488
+ "obs_mode": "vision_wrench",
1489
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1490
+ },
1491
+ {
1492
+ "hdf5_key": "ep_00142",
1493
+ "source_episode_index": 142,
1494
+ "split": "train",
1495
+ "num_samples": 592,
1496
+ "action_dim": 9,
1497
+ "obs_mode": "vision_wrench",
1498
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1499
+ },
1500
+ {
1501
+ "hdf5_key": "ep_00143",
1502
+ "source_episode_index": 143,
1503
+ "split": "train",
1504
+ "num_samples": 625,
1505
+ "action_dim": 9,
1506
+ "obs_mode": "vision_wrench",
1507
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1508
+ },
1509
+ {
1510
+ "hdf5_key": "ep_00144",
1511
+ "source_episode_index": 144,
1512
+ "split": "train",
1513
+ "num_samples": 684,
1514
+ "action_dim": 9,
1515
+ "obs_mode": "vision_wrench",
1516
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1517
+ },
1518
+ {
1519
+ "hdf5_key": "ep_00145",
1520
+ "source_episode_index": 145,
1521
+ "split": "train",
1522
+ "num_samples": 559,
1523
+ "action_dim": 9,
1524
+ "obs_mode": "vision_wrench",
1525
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1526
+ },
1527
+ {
1528
+ "hdf5_key": "ep_00146",
1529
+ "source_episode_index": 146,
1530
+ "split": "train",
1531
+ "num_samples": 677,
1532
+ "action_dim": 9,
1533
+ "obs_mode": "vision_wrench",
1534
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1535
+ },
1536
+ {
1537
+ "hdf5_key": "ep_00147",
1538
+ "source_episode_index": 147,
1539
+ "split": "train",
1540
+ "num_samples": 846,
1541
+ "action_dim": 9,
1542
+ "obs_mode": "vision_wrench",
1543
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1544
+ },
1545
+ {
1546
+ "hdf5_key": "ep_00148",
1547
+ "source_episode_index": 148,
1548
+ "split": "train",
1549
+ "num_samples": 615,
1550
+ "action_dim": 9,
1551
+ "obs_mode": "vision_wrench",
1552
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1553
+ },
1554
+ {
1555
+ "hdf5_key": "ep_00149",
1556
+ "source_episode_index": 149,
1557
+ "split": "train",
1558
+ "num_samples": 517,
1559
+ "action_dim": 9,
1560
+ "obs_mode": "vision_wrench",
1561
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1562
+ },
1563
+ {
1564
+ "hdf5_key": "ep_00150",
1565
+ "source_episode_index": 150,
1566
+ "split": "train",
1567
+ "num_samples": 817,
1568
+ "action_dim": 9,
1569
+ "obs_mode": "vision_wrench",
1570
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1571
+ },
1572
+ {
1573
+ "hdf5_key": "ep_00151",
1574
+ "source_episode_index": 151,
1575
+ "split": "train",
1576
+ "num_samples": 814,
1577
+ "action_dim": 9,
1578
+ "obs_mode": "vision_wrench",
1579
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1580
+ },
1581
+ {
1582
+ "hdf5_key": "ep_00152",
1583
+ "source_episode_index": 152,
1584
+ "split": "train",
1585
+ "num_samples": 828,
1586
+ "action_dim": 9,
1587
+ "obs_mode": "vision_wrench",
1588
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1589
+ },
1590
+ {
1591
+ "hdf5_key": "ep_00153",
1592
+ "source_episode_index": 153,
1593
+ "split": "train",
1594
+ "num_samples": 736,
1595
+ "action_dim": 9,
1596
+ "obs_mode": "vision_wrench",
1597
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1598
+ },
1599
+ {
1600
+ "hdf5_key": "ep_00154",
1601
+ "source_episode_index": 154,
1602
+ "split": "train",
1603
+ "num_samples": 860,
1604
+ "action_dim": 9,
1605
+ "obs_mode": "vision_wrench",
1606
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1607
+ },
1608
+ {
1609
+ "hdf5_key": "ep_00155",
1610
+ "source_episode_index": 155,
1611
+ "split": "train",
1612
+ "num_samples": 786,
1613
+ "action_dim": 9,
1614
+ "obs_mode": "vision_wrench",
1615
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1616
+ },
1617
+ {
1618
+ "hdf5_key": "ep_00156",
1619
+ "source_episode_index": 156,
1620
+ "split": "train",
1621
+ "num_samples": 725,
1622
+ "action_dim": 9,
1623
+ "obs_mode": "vision_wrench",
1624
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1625
+ },
1626
+ {
1627
+ "hdf5_key": "ep_00157",
1628
+ "source_episode_index": 157,
1629
+ "split": "train",
1630
+ "num_samples": 834,
1631
+ "action_dim": 9,
1632
+ "obs_mode": "vision_wrench",
1633
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1634
+ },
1635
+ {
1636
+ "hdf5_key": "ep_00158",
1637
+ "source_episode_index": 158,
1638
+ "split": "train",
1639
+ "num_samples": 652,
1640
+ "action_dim": 9,
1641
+ "obs_mode": "vision_wrench",
1642
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1643
+ },
1644
+ {
1645
+ "hdf5_key": "ep_00159",
1646
+ "source_episode_index": 159,
1647
+ "split": "train",
1648
+ "num_samples": 699,
1649
+ "action_dim": 9,
1650
+ "obs_mode": "vision_wrench",
1651
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1652
+ },
1653
+ {
1654
+ "hdf5_key": "ep_00160",
1655
+ "source_episode_index": 160,
1656
+ "split": "train",
1657
+ "num_samples": 812,
1658
+ "action_dim": 9,
1659
+ "obs_mode": "vision_wrench",
1660
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1661
+ },
1662
+ {
1663
+ "hdf5_key": "ep_00161",
1664
+ "source_episode_index": 161,
1665
+ "split": "train",
1666
+ "num_samples": 782,
1667
+ "action_dim": 9,
1668
+ "obs_mode": "vision_wrench",
1669
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1670
+ },
1671
+ {
1672
+ "hdf5_key": "ep_00162",
1673
+ "source_episode_index": 162,
1674
+ "split": "train",
1675
+ "num_samples": 681,
1676
+ "action_dim": 9,
1677
+ "obs_mode": "vision_wrench",
1678
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1679
+ },
1680
+ {
1681
+ "hdf5_key": "ep_00163",
1682
+ "source_episode_index": 163,
1683
+ "split": "train",
1684
+ "num_samples": 742,
1685
+ "action_dim": 9,
1686
+ "obs_mode": "vision_wrench",
1687
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1688
+ },
1689
+ {
1690
+ "hdf5_key": "ep_00164",
1691
+ "source_episode_index": 164,
1692
+ "split": "train",
1693
+ "num_samples": 766,
1694
+ "action_dim": 9,
1695
+ "obs_mode": "vision_wrench",
1696
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1697
+ },
1698
+ {
1699
+ "hdf5_key": "ep_00165",
1700
+ "source_episode_index": 165,
1701
+ "split": "train",
1702
+ "num_samples": 821,
1703
+ "action_dim": 9,
1704
+ "obs_mode": "vision_wrench",
1705
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1706
+ },
1707
+ {
1708
+ "hdf5_key": "ep_00166",
1709
+ "source_episode_index": 166,
1710
+ "split": "train",
1711
+ "num_samples": 1028,
1712
+ "action_dim": 9,
1713
+ "obs_mode": "vision_wrench",
1714
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1715
+ },
1716
+ {
1717
+ "hdf5_key": "ep_00167",
1718
+ "source_episode_index": 167,
1719
+ "split": "train",
1720
+ "num_samples": 720,
1721
+ "action_dim": 9,
1722
+ "obs_mode": "vision_wrench",
1723
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1724
+ },
1725
+ {
1726
+ "hdf5_key": "ep_00168",
1727
+ "source_episode_index": 168,
1728
+ "split": "train",
1729
+ "num_samples": 773,
1730
+ "action_dim": 9,
1731
+ "obs_mode": "vision_wrench",
1732
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1733
+ },
1734
+ {
1735
+ "hdf5_key": "ep_00169",
1736
+ "source_episode_index": 169,
1737
+ "split": "train",
1738
+ "num_samples": 690,
1739
+ "action_dim": 9,
1740
+ "obs_mode": "vision_wrench",
1741
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1742
+ },
1743
+ {
1744
+ "hdf5_key": "ep_00170",
1745
+ "source_episode_index": 170,
1746
+ "split": "train",
1747
+ "num_samples": 756,
1748
+ "action_dim": 9,
1749
+ "obs_mode": "vision_wrench",
1750
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1751
+ },
1752
+ {
1753
+ "hdf5_key": "ep_00171",
1754
+ "source_episode_index": 171,
1755
+ "split": "train",
1756
+ "num_samples": 742,
1757
+ "action_dim": 9,
1758
+ "obs_mode": "vision_wrench",
1759
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1760
+ },
1761
+ {
1762
+ "hdf5_key": "ep_00172",
1763
+ "source_episode_index": 172,
1764
+ "split": "valid",
1765
+ "num_samples": 689,
1766
+ "action_dim": 9,
1767
+ "obs_mode": "vision_wrench",
1768
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1769
+ },
1770
+ {
1771
+ "hdf5_key": "ep_00173",
1772
+ "source_episode_index": 173,
1773
+ "split": "valid",
1774
+ "num_samples": 639,
1775
+ "action_dim": 9,
1776
+ "obs_mode": "vision_wrench",
1777
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1778
+ },
1779
+ {
1780
+ "hdf5_key": "ep_00174",
1781
+ "source_episode_index": 174,
1782
+ "split": "valid",
1783
+ "num_samples": 694,
1784
+ "action_dim": 9,
1785
+ "obs_mode": "vision_wrench",
1786
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1787
+ },
1788
+ {
1789
+ "hdf5_key": "ep_00175",
1790
+ "source_episode_index": 175,
1791
+ "split": "valid",
1792
+ "num_samples": 532,
1793
+ "action_dim": 9,
1794
+ "obs_mode": "vision_wrench",
1795
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1796
+ },
1797
+ {
1798
+ "hdf5_key": "ep_00176",
1799
+ "source_episode_index": 176,
1800
+ "split": "valid",
1801
+ "num_samples": 665,
1802
+ "action_dim": 9,
1803
+ "obs_mode": "vision_wrench",
1804
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1805
+ },
1806
+ {
1807
+ "hdf5_key": "ep_00177",
1808
+ "source_episode_index": 177,
1809
+ "split": "valid",
1810
+ "num_samples": 771,
1811
+ "action_dim": 9,
1812
+ "obs_mode": "vision_wrench",
1813
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1814
+ },
1815
+ {
1816
+ "hdf5_key": "ep_00178",
1817
+ "source_episode_index": 178,
1818
+ "split": "valid",
1819
+ "num_samples": 602,
1820
+ "action_dim": 9,
1821
+ "obs_mode": "vision_wrench",
1822
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1823
+ },
1824
+ {
1825
+ "hdf5_key": "ep_00179",
1826
+ "source_episode_index": 179,
1827
+ "split": "valid",
1828
+ "num_samples": 700,
1829
+ "action_dim": 9,
1830
+ "obs_mode": "vision_wrench",
1831
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1832
+ },
1833
+ {
1834
+ "hdf5_key": "ep_00180",
1835
+ "source_episode_index": 180,
1836
+ "split": "valid",
1837
+ "num_samples": 880,
1838
+ "action_dim": 9,
1839
+ "obs_mode": "vision_wrench",
1840
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1841
+ },
1842
+ {
1843
+ "hdf5_key": "ep_00181",
1844
+ "source_episode_index": 181,
1845
+ "split": "valid",
1846
+ "num_samples": 522,
1847
+ "action_dim": 9,
1848
+ "obs_mode": "vision_wrench",
1849
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1850
+ },
1851
+ {
1852
+ "hdf5_key": "ep_00182",
1853
+ "source_episode_index": 182,
1854
+ "split": "valid",
1855
+ "num_samples": 697,
1856
+ "action_dim": 9,
1857
+ "obs_mode": "vision_wrench",
1858
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1859
+ },
1860
+ {
1861
+ "hdf5_key": "ep_00183",
1862
+ "source_episode_index": 183,
1863
+ "split": "valid",
1864
+ "num_samples": 621,
1865
+ "action_dim": 9,
1866
+ "obs_mode": "vision_wrench",
1867
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1868
+ },
1869
+ {
1870
+ "hdf5_key": "ep_00184",
1871
+ "source_episode_index": 184,
1872
+ "split": "valid",
1873
+ "num_samples": 710,
1874
+ "action_dim": 9,
1875
+ "obs_mode": "vision_wrench",
1876
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1877
+ },
1878
+ {
1879
+ "hdf5_key": "ep_00185",
1880
+ "source_episode_index": 185,
1881
+ "split": "valid",
1882
+ "num_samples": 774,
1883
+ "action_dim": 9,
1884
+ "obs_mode": "vision_wrench",
1885
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1886
+ },
1887
+ {
1888
+ "hdf5_key": "ep_00186",
1889
+ "source_episode_index": 186,
1890
+ "split": "valid",
1891
+ "num_samples": 682,
1892
+ "action_dim": 9,
1893
+ "obs_mode": "vision_wrench",
1894
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1895
+ },
1896
+ {
1897
+ "hdf5_key": "ep_00187",
1898
+ "source_episode_index": 187,
1899
+ "split": "valid",
1900
+ "num_samples": 715,
1901
+ "action_dim": 9,
1902
+ "obs_mode": "vision_wrench",
1903
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1904
+ },
1905
+ {
1906
+ "hdf5_key": "ep_00188",
1907
+ "source_episode_index": 188,
1908
+ "split": "valid",
1909
+ "num_samples": 798,
1910
+ "action_dim": 9,
1911
+ "obs_mode": "vision_wrench",
1912
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1913
+ },
1914
+ {
1915
+ "hdf5_key": "ep_00189",
1916
+ "source_episode_index": 189,
1917
+ "split": "valid",
1918
+ "num_samples": 786,
1919
+ "action_dim": 9,
1920
+ "obs_mode": "vision_wrench",
1921
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1922
+ },
1923
+ {
1924
+ "hdf5_key": "ep_00190",
1925
+ "source_episode_index": 190,
1926
+ "split": "valid",
1927
+ "num_samples": 759,
1928
+ "action_dim": 9,
1929
+ "obs_mode": "vision_wrench",
1930
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1931
+ },
1932
+ {
1933
+ "hdf5_key": "ep_00191",
1934
+ "source_episode_index": 191,
1935
+ "split": "valid",
1936
+ "num_samples": 812,
1937
+ "action_dim": 9,
1938
+ "obs_mode": "vision_wrench",
1939
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1940
+ },
1941
+ {
1942
+ "hdf5_key": "ep_00192",
1943
+ "source_episode_index": 192,
1944
+ "split": "valid",
1945
+ "num_samples": 803,
1946
+ "action_dim": 9,
1947
+ "obs_mode": "vision_wrench",
1948
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1949
+ },
1950
+ {
1951
+ "hdf5_key": "ep_00193",
1952
+ "source_episode_index": 193,
1953
+ "split": "valid",
1954
+ "num_samples": 780,
1955
+ "action_dim": 9,
1956
+ "obs_mode": "vision_wrench",
1957
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1958
+ },
1959
+ {
1960
+ "hdf5_key": "ep_00194",
1961
+ "source_episode_index": 194,
1962
+ "split": "valid",
1963
+ "num_samples": 853,
1964
+ "action_dim": 9,
1965
+ "obs_mode": "vision_wrench",
1966
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1967
+ },
1968
+ {
1969
+ "hdf5_key": "ep_00195",
1970
+ "source_episode_index": 195,
1971
+ "split": "valid",
1972
+ "num_samples": 710,
1973
+ "action_dim": 9,
1974
+ "obs_mode": "vision_wrench",
1975
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1976
+ },
1977
+ {
1978
+ "hdf5_key": "ep_00196",
1979
+ "source_episode_index": 196,
1980
+ "split": "valid",
1981
+ "num_samples": 855,
1982
+ "action_dim": 9,
1983
+ "obs_mode": "vision_wrench",
1984
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1985
+ },
1986
+ {
1987
+ "hdf5_key": "ep_00197",
1988
+ "source_episode_index": 197,
1989
+ "split": "valid",
1990
+ "num_samples": 707,
1991
+ "action_dim": 9,
1992
+ "obs_mode": "vision_wrench",
1993
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1994
+ },
1995
+ {
1996
+ "hdf5_key": "ep_00198",
1997
+ "source_episode_index": 198,
1998
+ "split": "valid",
1999
+ "num_samples": 557,
2000
+ "action_dim": 9,
2001
+ "obs_mode": "vision_wrench",
2002
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
2003
+ },
2004
+ {
2005
+ "hdf5_key": "ep_00199",
2006
+ "source_episode_index": 199,
2007
+ "split": "valid",
2008
+ "num_samples": 522,
2009
+ "action_dim": 9,
2010
+ "obs_mode": "vision_wrench",
2011
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
2012
+ },
2013
+ {
2014
+ "hdf5_key": "ep_00200",
2015
+ "source_episode_index": 200,
2016
+ "split": "valid",
2017
+ "num_samples": 581,
2018
+ "action_dim": 9,
2019
+ "obs_mode": "vision_wrench",
2020
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
2021
+ },
2022
+ {
2023
+ "hdf5_key": "ep_00201",
2024
+ "source_episode_index": 201,
2025
+ "split": "valid",
2026
+ "num_samples": 574,
2027
+ "action_dim": 9,
2028
+ "obs_mode": "vision_wrench",
2029
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
2030
+ }
2031
+ ]
2032
+ }
vac_pre_vis_wrench/last.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:afd7b6c415b7c153856ea67f5d6cc29d33f918222bc3f2ea1c9d8ab5c58a2be2
3
+ size 1255189021
vac_pre_vis_wrench_state/action_stats.json ADDED
@@ -0,0 +1,311 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "method": "min_max",
3
+ "output_min": -0.999999,
4
+ "output_max": 0.999999,
5
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary",
6
+ "action_components": [
7
+ "user_cmd_dx",
8
+ "user_cmd_dy",
9
+ "user_cmd_dz",
10
+ "user_cmd_drx",
11
+ "user_cmd_dry",
12
+ "user_cmd_drz",
13
+ "stiffness_cmd",
14
+ "zeta",
15
+ "hand_open_binary"
16
+ ],
17
+ "num_frames": 124722,
18
+ "min": [
19
+ -0.0020000000949949026,
20
+ -0.0020000000949949026,
21
+ -0.0020000000949949026,
22
+ -0.017500000074505806,
23
+ 0.0,
24
+ -0.017500000074505806,
25
+ 3.5638904571533203,
26
+ 0.8021645545959473,
27
+ 0.0
28
+ ],
29
+ "max": [
30
+ 0.0020000000949949026,
31
+ 0.0020000000949949026,
32
+ 0.0020000000949949026,
33
+ 0.017500000074505806,
34
+ 0.0020832535810768604,
35
+ 0.017500000074505806,
36
+ 1997.13330078125,
37
+ 2.5,
38
+ 1.0
39
+ ],
40
+ "mean": [
41
+ -1.295681795454584e-05,
42
+ -4.36740810982883e-05,
43
+ 5.51424891455099e-05,
44
+ -1.8853129404305946e-06,
45
+ 6.276949449102176e-08,
46
+ -1.684121525613591e-05,
47
+ 1533.5157470703125,
48
+ 0.9446797370910645,
49
+ 0.428376704454422
50
+ ],
51
+ "std": [
52
+ 0.00042260822374373674,
53
+ 0.0009382680291309953,
54
+ 0.000991749344393611,
55
+ 0.0003852724621538073,
56
+ 1.1097310562036e-05,
57
+ 0.001960940193384886,
58
+ 461.547119140625,
59
+ 0.24154788255691528,
60
+ 0.4948435127735138
61
+ ],
62
+ "scale": [
63
+ 0.0020000021904706955,
64
+ 0.0020000021904706955,
65
+ 0.0020000021904706955,
66
+ 0.0175000187009573,
67
+ 0.0010416278382763267,
68
+ 0.0175000187009573,
69
+ 996.7857666015625,
70
+ 0.8489185571670532,
71
+ 0.500000536441803
72
+ ],
73
+ "offset": [
74
+ 0.0,
75
+ 0.0,
76
+ 0.0,
77
+ 0.0,
78
+ 0.0010416267905384302,
79
+ 0.0,
80
+ 1000.3486328125,
81
+ 1.6510822772979736,
82
+ 0.5
83
+ ],
84
+ "hand_source": "action.absolute",
85
+ "hand_active_dims": [
86
+ 8,
87
+ 9,
88
+ 10,
89
+ 11
90
+ ],
91
+ "hand_reduce": "mean",
92
+ "hand_threshold": 0.85,
93
+ "hand_high_value": 1.0,
94
+ "hand_low_value": 0.0,
95
+ "dataset": "/workspace/datasets/dataset_pipe_02_06",
96
+ "observation_mode_examples": [
97
+ "vision",
98
+ "vision_wrench",
99
+ "vision_wrench_state"
100
+ ],
101
+ "train_source_episodes": [
102
+ 0,
103
+ 1,
104
+ 2,
105
+ 3,
106
+ 4,
107
+ 5,
108
+ 6,
109
+ 7,
110
+ 8,
111
+ 9,
112
+ 10,
113
+ 11,
114
+ 12,
115
+ 13,
116
+ 14,
117
+ 15,
118
+ 16,
119
+ 17,
120
+ 18,
121
+ 19,
122
+ 20,
123
+ 21,
124
+ 22,
125
+ 23,
126
+ 24,
127
+ 25,
128
+ 26,
129
+ 27,
130
+ 28,
131
+ 29,
132
+ 30,
133
+ 31,
134
+ 32,
135
+ 33,
136
+ 34,
137
+ 35,
138
+ 36,
139
+ 37,
140
+ 38,
141
+ 39,
142
+ 40,
143
+ 41,
144
+ 42,
145
+ 43,
146
+ 44,
147
+ 45,
148
+ 46,
149
+ 47,
150
+ 48,
151
+ 49,
152
+ 50,
153
+ 51,
154
+ 52,
155
+ 53,
156
+ 54,
157
+ 55,
158
+ 56,
159
+ 57,
160
+ 58,
161
+ 59,
162
+ 60,
163
+ 61,
164
+ 62,
165
+ 63,
166
+ 64,
167
+ 65,
168
+ 66,
169
+ 67,
170
+ 68,
171
+ 69,
172
+ 70,
173
+ 71,
174
+ 72,
175
+ 73,
176
+ 74,
177
+ 75,
178
+ 76,
179
+ 77,
180
+ 78,
181
+ 79,
182
+ 80,
183
+ 81,
184
+ 82,
185
+ 83,
186
+ 84,
187
+ 85,
188
+ 86,
189
+ 87,
190
+ 88,
191
+ 89,
192
+ 90,
193
+ 91,
194
+ 92,
195
+ 93,
196
+ 94,
197
+ 95,
198
+ 96,
199
+ 97,
200
+ 98,
201
+ 99,
202
+ 100,
203
+ 101,
204
+ 102,
205
+ 103,
206
+ 104,
207
+ 105,
208
+ 106,
209
+ 107,
210
+ 108,
211
+ 109,
212
+ 110,
213
+ 111,
214
+ 112,
215
+ 113,
216
+ 114,
217
+ 115,
218
+ 116,
219
+ 117,
220
+ 118,
221
+ 119,
222
+ 120,
223
+ 121,
224
+ 122,
225
+ 123,
226
+ 124,
227
+ 125,
228
+ 126,
229
+ 127,
230
+ 128,
231
+ 129,
232
+ 130,
233
+ 131,
234
+ 132,
235
+ 133,
236
+ 134,
237
+ 135,
238
+ 136,
239
+ 137,
240
+ 138,
241
+ 139,
242
+ 140,
243
+ 141,
244
+ 142,
245
+ 143,
246
+ 144,
247
+ 145,
248
+ 146,
249
+ 147,
250
+ 148,
251
+ 149,
252
+ 150,
253
+ 151,
254
+ 152,
255
+ 153,
256
+ 154,
257
+ 155,
258
+ 156,
259
+ 157,
260
+ 158,
261
+ 159,
262
+ 160,
263
+ 161,
264
+ 162,
265
+ 163,
266
+ 164,
267
+ 165,
268
+ 166,
269
+ 167,
270
+ 168,
271
+ 169,
272
+ 170,
273
+ 171
274
+ ],
275
+ "valid_source_episodes": [
276
+ 172,
277
+ 173,
278
+ 174,
279
+ 175,
280
+ 176,
281
+ 177,
282
+ 178,
283
+ 179,
284
+ 180,
285
+ 181,
286
+ 182,
287
+ 183,
288
+ 184,
289
+ 185,
290
+ 186,
291
+ 187,
292
+ 188,
293
+ 189,
294
+ 190,
295
+ 191,
296
+ 192,
297
+ 193,
298
+ 194,
299
+ 195,
300
+ 196,
301
+ 197,
302
+ 198,
303
+ 199,
304
+ 200,
305
+ 201
306
+ ],
307
+ "image_size": [
308
+ 84,
309
+ 84
310
+ ]
311
+ }
vac_pre_vis_wrench_state/best.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9033420706e947db0022aacf810210910406ae50204646c6350a847a06d8d04
3
+ size 1260797133
vac_pre_vis_wrench_state/config.json ADDED
@@ -0,0 +1,198 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "algo_name": "diffusion_policy",
3
+ "experiment": {
4
+ "name": "vac_pre_vis_wrench_state",
5
+ "validate": true,
6
+ "logging": {
7
+ "terminal_output_to_txt": false,
8
+ "log_tb": true,
9
+ "log_wandb": false,
10
+ "wandb_proj_name": "debug"
11
+ },
12
+ "save": {
13
+ "enabled": true,
14
+ "every_n_seconds": null,
15
+ "every_n_epochs": 50,
16
+ "epochs": [],
17
+ "on_best_validation": true,
18
+ "on_best_rollout_return": false,
19
+ "on_best_rollout_success_rate": false
20
+ },
21
+ "epoch_every_n_steps": 100,
22
+ "validation_epoch_every_n_steps": 10,
23
+ "env": null,
24
+ "additional_envs": null,
25
+ "render": false,
26
+ "render_video": false,
27
+ "keep_all_videos": false,
28
+ "video_skip": 5,
29
+ "rollout": {
30
+ "enabled": false,
31
+ "n": 0,
32
+ "horizon": 50,
33
+ "rate": 50,
34
+ "warmstart": 0,
35
+ "terminate_on_success": true
36
+ },
37
+ "env_meta_update_dict": {},
38
+ "ckpt_path": null
39
+ },
40
+ "train": {
41
+ "data": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_pre_vis_wrench_state/vac_pre_vis_wrench_state.hdf5",
42
+ "output_dir": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_pre_vis_wrench_state/robomimic_runs",
43
+ "normalize_weights_by_ds_size": false,
44
+ "num_data_workers": 0,
45
+ "hdf5_cache_mode": "all",
46
+ "hdf5_use_swmr": true,
47
+ "hdf5_load_next_obs": false,
48
+ "hdf5_normalize_obs": false,
49
+ "hdf5_filter_key": "train",
50
+ "hdf5_validation_filter_key": "valid",
51
+ "seq_length": 8,
52
+ "pad_seq_length": true,
53
+ "frame_stack": 2,
54
+ "pad_frame_stack": true,
55
+ "dataset_keys": [
56
+ "actions",
57
+ "rewards",
58
+ "dones"
59
+ ],
60
+ "action_keys": [
61
+ "actions"
62
+ ],
63
+ "action_config": {
64
+ "actions": {
65
+ "normalization": "min_max"
66
+ }
67
+ },
68
+ "goal_mode": null,
69
+ "cuda": true,
70
+ "batch_size": 16,
71
+ "num_epochs": 300,
72
+ "seed": 42,
73
+ "max_grad_norm": null
74
+ },
75
+ "algo": {
76
+ "optim_params": {
77
+ "policy": {
78
+ "optimizer_type": "adamw",
79
+ "learning_rate": {
80
+ "initial": 0.0001,
81
+ "decay_factor": 0.1,
82
+ "step_every_batch": true,
83
+ "scheduler_type": "cosine",
84
+ "num_cycles": 0.5,
85
+ "warmup_steps": 500,
86
+ "epoch_schedule": []
87
+ },
88
+ "regularization": {
89
+ "L2": 1e-06
90
+ }
91
+ }
92
+ },
93
+ "horizon": {
94
+ "observation_horizon": 2,
95
+ "action_horizon": 4,
96
+ "prediction_horizon": 8
97
+ },
98
+ "unet": {
99
+ "enabled": true,
100
+ "diffusion_step_embed_dim": 256,
101
+ "down_dims": [
102
+ 128,
103
+ 256,
104
+ 512
105
+ ],
106
+ "kernel_size": 5,
107
+ "n_groups": 8
108
+ },
109
+ "ema": {
110
+ "enabled": true,
111
+ "power": 0.75
112
+ },
113
+ "ddpm": {
114
+ "enabled": true,
115
+ "num_train_timesteps": 50,
116
+ "num_inference_timesteps": 50,
117
+ "beta_schedule": "squaredcos_cap_v2",
118
+ "clip_sample": true,
119
+ "prediction_type": "epsilon"
120
+ },
121
+ "ddim": {
122
+ "enabled": false,
123
+ "num_train_timesteps": 50,
124
+ "num_inference_timesteps": 10,
125
+ "beta_schedule": "squaredcos_cap_v2",
126
+ "clip_sample": true,
127
+ "set_alpha_to_one": true,
128
+ "steps_offset": 0,
129
+ "prediction_type": "epsilon"
130
+ }
131
+ },
132
+ "observation": {
133
+ "modalities": {
134
+ "obs": {
135
+ "low_dim": [
136
+ "wrench",
137
+ "state"
138
+ ],
139
+ "rgb": [
140
+ "front_rgb"
141
+ ],
142
+ "depth": [],
143
+ "scan": []
144
+ },
145
+ "goal": {
146
+ "low_dim": [],
147
+ "rgb": [],
148
+ "depth": [],
149
+ "scan": []
150
+ }
151
+ },
152
+ "encoder": {
153
+ "low_dim": {
154
+ "core_class": null,
155
+ "core_kwargs": {},
156
+ "obs_randomizer_class": null,
157
+ "obs_randomizer_kwargs": {}
158
+ },
159
+ "rgb": {
160
+ "core_class": "VisualCore",
161
+ "core_kwargs": {
162
+ "backbone_class": "ResNet18Conv",
163
+ "backbone_kwargs": {
164
+ "pretrained": false,
165
+ "input_coord_conv": false
166
+ },
167
+ "pool_class": "SpatialSoftmax",
168
+ "pool_kwargs": {
169
+ "num_kp": 32,
170
+ "learnable_temperature": false,
171
+ "temperature": 1.0,
172
+ "noise_std": 0.0
173
+ },
174
+ "feature_dimension": 64
175
+ },
176
+ "obs_randomizer_class": null,
177
+ "obs_randomizer_kwargs": {}
178
+ },
179
+ "depth": {
180
+ "core_class": "VisualCore",
181
+ "core_kwargs": {},
182
+ "obs_randomizer_class": null,
183
+ "obs_randomizer_kwargs": {}
184
+ },
185
+ "scan": {
186
+ "core_class": "ScanCore",
187
+ "core_kwargs": {},
188
+ "obs_randomizer_class": null,
189
+ "obs_randomizer_kwargs": {}
190
+ }
191
+ }
192
+ },
193
+ "meta": {
194
+ "hp_base_config_file": null,
195
+ "hp_keys": [],
196
+ "hp_values": []
197
+ }
198
+ }
vac_pre_vis_wrench_state/dataset_summary.json ADDED
@@ -0,0 +1,2032 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "path": "/workspace/training/vac_io_ablation/runs/pipe_io_ablation/vac_pre_vis_wrench_state/vac_pre_vis_wrench_state.hdf5",
3
+ "total_frames": 145712,
4
+ "train_episodes": 172,
5
+ "valid_episodes": 30,
6
+ "train_source_episodes": [
7
+ 0,
8
+ 1,
9
+ 2,
10
+ 3,
11
+ 4,
12
+ 5,
13
+ 6,
14
+ 7,
15
+ 8,
16
+ 9,
17
+ 10,
18
+ 11,
19
+ 12,
20
+ 13,
21
+ 14,
22
+ 15,
23
+ 16,
24
+ 17,
25
+ 18,
26
+ 19,
27
+ 20,
28
+ 21,
29
+ 22,
30
+ 23,
31
+ 24,
32
+ 25,
33
+ 26,
34
+ 27,
35
+ 28,
36
+ 29,
37
+ 30,
38
+ 31,
39
+ 32,
40
+ 33,
41
+ 34,
42
+ 35,
43
+ 36,
44
+ 37,
45
+ 38,
46
+ 39,
47
+ 40,
48
+ 41,
49
+ 42,
50
+ 43,
51
+ 44,
52
+ 45,
53
+ 46,
54
+ 47,
55
+ 48,
56
+ 49,
57
+ 50,
58
+ 51,
59
+ 52,
60
+ 53,
61
+ 54,
62
+ 55,
63
+ 56,
64
+ 57,
65
+ 58,
66
+ 59,
67
+ 60,
68
+ 61,
69
+ 62,
70
+ 63,
71
+ 64,
72
+ 65,
73
+ 66,
74
+ 67,
75
+ 68,
76
+ 69,
77
+ 70,
78
+ 71,
79
+ 72,
80
+ 73,
81
+ 74,
82
+ 75,
83
+ 76,
84
+ 77,
85
+ 78,
86
+ 79,
87
+ 80,
88
+ 81,
89
+ 82,
90
+ 83,
91
+ 84,
92
+ 85,
93
+ 86,
94
+ 87,
95
+ 88,
96
+ 89,
97
+ 90,
98
+ 91,
99
+ 92,
100
+ 93,
101
+ 94,
102
+ 95,
103
+ 96,
104
+ 97,
105
+ 98,
106
+ 99,
107
+ 100,
108
+ 101,
109
+ 102,
110
+ 103,
111
+ 104,
112
+ 105,
113
+ 106,
114
+ 107,
115
+ 108,
116
+ 109,
117
+ 110,
118
+ 111,
119
+ 112,
120
+ 113,
121
+ 114,
122
+ 115,
123
+ 116,
124
+ 117,
125
+ 118,
126
+ 119,
127
+ 120,
128
+ 121,
129
+ 122,
130
+ 123,
131
+ 124,
132
+ 125,
133
+ 126,
134
+ 127,
135
+ 128,
136
+ 129,
137
+ 130,
138
+ 131,
139
+ 132,
140
+ 133,
141
+ 134,
142
+ 135,
143
+ 136,
144
+ 137,
145
+ 138,
146
+ 139,
147
+ 140,
148
+ 141,
149
+ 142,
150
+ 143,
151
+ 144,
152
+ 145,
153
+ 146,
154
+ 147,
155
+ 148,
156
+ 149,
157
+ 150,
158
+ 151,
159
+ 152,
160
+ 153,
161
+ 154,
162
+ 155,
163
+ 156,
164
+ 157,
165
+ 158,
166
+ 159,
167
+ 160,
168
+ 161,
169
+ 162,
170
+ 163,
171
+ 164,
172
+ 165,
173
+ 166,
174
+ 167,
175
+ 168,
176
+ 169,
177
+ 170,
178
+ 171
179
+ ],
180
+ "valid_source_episodes": [
181
+ 172,
182
+ 173,
183
+ 174,
184
+ 175,
185
+ 176,
186
+ 177,
187
+ 178,
188
+ 179,
189
+ 180,
190
+ 181,
191
+ 182,
192
+ 183,
193
+ 184,
194
+ 185,
195
+ 186,
196
+ 187,
197
+ 188,
198
+ 189,
199
+ 190,
200
+ 191,
201
+ 192,
202
+ 193,
203
+ 194,
204
+ 195,
205
+ 196,
206
+ 197,
207
+ 198,
208
+ 199,
209
+ 200,
210
+ 201
211
+ ],
212
+ "episodes": [
213
+ {
214
+ "hdf5_key": "ep_00000",
215
+ "source_episode_index": 0,
216
+ "split": "train",
217
+ "num_samples": 663,
218
+ "action_dim": 9,
219
+ "obs_mode": "vision_wrench_state",
220
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
221
+ },
222
+ {
223
+ "hdf5_key": "ep_00001",
224
+ "source_episode_index": 1,
225
+ "split": "train",
226
+ "num_samples": 706,
227
+ "action_dim": 9,
228
+ "obs_mode": "vision_wrench_state",
229
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
230
+ },
231
+ {
232
+ "hdf5_key": "ep_00002",
233
+ "source_episode_index": 2,
234
+ "split": "train",
235
+ "num_samples": 1006,
236
+ "action_dim": 9,
237
+ "obs_mode": "vision_wrench_state",
238
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
239
+ },
240
+ {
241
+ "hdf5_key": "ep_00003",
242
+ "source_episode_index": 3,
243
+ "split": "train",
244
+ "num_samples": 709,
245
+ "action_dim": 9,
246
+ "obs_mode": "vision_wrench_state",
247
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
248
+ },
249
+ {
250
+ "hdf5_key": "ep_00004",
251
+ "source_episode_index": 4,
252
+ "split": "train",
253
+ "num_samples": 816,
254
+ "action_dim": 9,
255
+ "obs_mode": "vision_wrench_state",
256
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
257
+ },
258
+ {
259
+ "hdf5_key": "ep_00005",
260
+ "source_episode_index": 5,
261
+ "split": "train",
262
+ "num_samples": 882,
263
+ "action_dim": 9,
264
+ "obs_mode": "vision_wrench_state",
265
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
266
+ },
267
+ {
268
+ "hdf5_key": "ep_00006",
269
+ "source_episode_index": 6,
270
+ "split": "train",
271
+ "num_samples": 699,
272
+ "action_dim": 9,
273
+ "obs_mode": "vision_wrench_state",
274
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
275
+ },
276
+ {
277
+ "hdf5_key": "ep_00007",
278
+ "source_episode_index": 7,
279
+ "split": "train",
280
+ "num_samples": 851,
281
+ "action_dim": 9,
282
+ "obs_mode": "vision_wrench_state",
283
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
284
+ },
285
+ {
286
+ "hdf5_key": "ep_00008",
287
+ "source_episode_index": 8,
288
+ "split": "train",
289
+ "num_samples": 728,
290
+ "action_dim": 9,
291
+ "obs_mode": "vision_wrench_state",
292
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
293
+ },
294
+ {
295
+ "hdf5_key": "ep_00009",
296
+ "source_episode_index": 9,
297
+ "split": "train",
298
+ "num_samples": 770,
299
+ "action_dim": 9,
300
+ "obs_mode": "vision_wrench_state",
301
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
302
+ },
303
+ {
304
+ "hdf5_key": "ep_00010",
305
+ "source_episode_index": 10,
306
+ "split": "train",
307
+ "num_samples": 821,
308
+ "action_dim": 9,
309
+ "obs_mode": "vision_wrench_state",
310
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
311
+ },
312
+ {
313
+ "hdf5_key": "ep_00011",
314
+ "source_episode_index": 11,
315
+ "split": "train",
316
+ "num_samples": 561,
317
+ "action_dim": 9,
318
+ "obs_mode": "vision_wrench_state",
319
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
320
+ },
321
+ {
322
+ "hdf5_key": "ep_00012",
323
+ "source_episode_index": 12,
324
+ "split": "train",
325
+ "num_samples": 517,
326
+ "action_dim": 9,
327
+ "obs_mode": "vision_wrench_state",
328
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
329
+ },
330
+ {
331
+ "hdf5_key": "ep_00013",
332
+ "source_episode_index": 13,
333
+ "split": "train",
334
+ "num_samples": 690,
335
+ "action_dim": 9,
336
+ "obs_mode": "vision_wrench_state",
337
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
338
+ },
339
+ {
340
+ "hdf5_key": "ep_00014",
341
+ "source_episode_index": 14,
342
+ "split": "train",
343
+ "num_samples": 559,
344
+ "action_dim": 9,
345
+ "obs_mode": "vision_wrench_state",
346
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
347
+ },
348
+ {
349
+ "hdf5_key": "ep_00015",
350
+ "source_episode_index": 15,
351
+ "split": "train",
352
+ "num_samples": 889,
353
+ "action_dim": 9,
354
+ "obs_mode": "vision_wrench_state",
355
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
356
+ },
357
+ {
358
+ "hdf5_key": "ep_00016",
359
+ "source_episode_index": 16,
360
+ "split": "train",
361
+ "num_samples": 619,
362
+ "action_dim": 9,
363
+ "obs_mode": "vision_wrench_state",
364
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
365
+ },
366
+ {
367
+ "hdf5_key": "ep_00017",
368
+ "source_episode_index": 17,
369
+ "split": "train",
370
+ "num_samples": 640,
371
+ "action_dim": 9,
372
+ "obs_mode": "vision_wrench_state",
373
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
374
+ },
375
+ {
376
+ "hdf5_key": "ep_00018",
377
+ "source_episode_index": 18,
378
+ "split": "train",
379
+ "num_samples": 612,
380
+ "action_dim": 9,
381
+ "obs_mode": "vision_wrench_state",
382
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
383
+ },
384
+ {
385
+ "hdf5_key": "ep_00019",
386
+ "source_episode_index": 19,
387
+ "split": "train",
388
+ "num_samples": 702,
389
+ "action_dim": 9,
390
+ "obs_mode": "vision_wrench_state",
391
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
392
+ },
393
+ {
394
+ "hdf5_key": "ep_00020",
395
+ "source_episode_index": 20,
396
+ "split": "train",
397
+ "num_samples": 645,
398
+ "action_dim": 9,
399
+ "obs_mode": "vision_wrench_state",
400
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
401
+ },
402
+ {
403
+ "hdf5_key": "ep_00021",
404
+ "source_episode_index": 21,
405
+ "split": "train",
406
+ "num_samples": 751,
407
+ "action_dim": 9,
408
+ "obs_mode": "vision_wrench_state",
409
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
410
+ },
411
+ {
412
+ "hdf5_key": "ep_00022",
413
+ "source_episode_index": 22,
414
+ "split": "train",
415
+ "num_samples": 810,
416
+ "action_dim": 9,
417
+ "obs_mode": "vision_wrench_state",
418
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
419
+ },
420
+ {
421
+ "hdf5_key": "ep_00023",
422
+ "source_episode_index": 23,
423
+ "split": "train",
424
+ "num_samples": 734,
425
+ "action_dim": 9,
426
+ "obs_mode": "vision_wrench_state",
427
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
428
+ },
429
+ {
430
+ "hdf5_key": "ep_00024",
431
+ "source_episode_index": 24,
432
+ "split": "train",
433
+ "num_samples": 637,
434
+ "action_dim": 9,
435
+ "obs_mode": "vision_wrench_state",
436
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
437
+ },
438
+ {
439
+ "hdf5_key": "ep_00025",
440
+ "source_episode_index": 25,
441
+ "split": "train",
442
+ "num_samples": 640,
443
+ "action_dim": 9,
444
+ "obs_mode": "vision_wrench_state",
445
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
446
+ },
447
+ {
448
+ "hdf5_key": "ep_00026",
449
+ "source_episode_index": 26,
450
+ "split": "train",
451
+ "num_samples": 957,
452
+ "action_dim": 9,
453
+ "obs_mode": "vision_wrench_state",
454
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
455
+ },
456
+ {
457
+ "hdf5_key": "ep_00027",
458
+ "source_episode_index": 27,
459
+ "split": "train",
460
+ "num_samples": 617,
461
+ "action_dim": 9,
462
+ "obs_mode": "vision_wrench_state",
463
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
464
+ },
465
+ {
466
+ "hdf5_key": "ep_00028",
467
+ "source_episode_index": 28,
468
+ "split": "train",
469
+ "num_samples": 772,
470
+ "action_dim": 9,
471
+ "obs_mode": "vision_wrench_state",
472
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
473
+ },
474
+ {
475
+ "hdf5_key": "ep_00029",
476
+ "source_episode_index": 29,
477
+ "split": "train",
478
+ "num_samples": 616,
479
+ "action_dim": 9,
480
+ "obs_mode": "vision_wrench_state",
481
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
482
+ },
483
+ {
484
+ "hdf5_key": "ep_00030",
485
+ "source_episode_index": 30,
486
+ "split": "train",
487
+ "num_samples": 732,
488
+ "action_dim": 9,
489
+ "obs_mode": "vision_wrench_state",
490
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
491
+ },
492
+ {
493
+ "hdf5_key": "ep_00031",
494
+ "source_episode_index": 31,
495
+ "split": "train",
496
+ "num_samples": 761,
497
+ "action_dim": 9,
498
+ "obs_mode": "vision_wrench_state",
499
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
500
+ },
501
+ {
502
+ "hdf5_key": "ep_00032",
503
+ "source_episode_index": 32,
504
+ "split": "train",
505
+ "num_samples": 804,
506
+ "action_dim": 9,
507
+ "obs_mode": "vision_wrench_state",
508
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
509
+ },
510
+ {
511
+ "hdf5_key": "ep_00033",
512
+ "source_episode_index": 33,
513
+ "split": "train",
514
+ "num_samples": 735,
515
+ "action_dim": 9,
516
+ "obs_mode": "vision_wrench_state",
517
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
518
+ },
519
+ {
520
+ "hdf5_key": "ep_00034",
521
+ "source_episode_index": 34,
522
+ "split": "train",
523
+ "num_samples": 721,
524
+ "action_dim": 9,
525
+ "obs_mode": "vision_wrench_state",
526
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
527
+ },
528
+ {
529
+ "hdf5_key": "ep_00035",
530
+ "source_episode_index": 35,
531
+ "split": "train",
532
+ "num_samples": 612,
533
+ "action_dim": 9,
534
+ "obs_mode": "vision_wrench_state",
535
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
536
+ },
537
+ {
538
+ "hdf5_key": "ep_00036",
539
+ "source_episode_index": 36,
540
+ "split": "train",
541
+ "num_samples": 685,
542
+ "action_dim": 9,
543
+ "obs_mode": "vision_wrench_state",
544
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
545
+ },
546
+ {
547
+ "hdf5_key": "ep_00037",
548
+ "source_episode_index": 37,
549
+ "split": "train",
550
+ "num_samples": 685,
551
+ "action_dim": 9,
552
+ "obs_mode": "vision_wrench_state",
553
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
554
+ },
555
+ {
556
+ "hdf5_key": "ep_00038",
557
+ "source_episode_index": 38,
558
+ "split": "train",
559
+ "num_samples": 663,
560
+ "action_dim": 9,
561
+ "obs_mode": "vision_wrench_state",
562
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
563
+ },
564
+ {
565
+ "hdf5_key": "ep_00039",
566
+ "source_episode_index": 39,
567
+ "split": "train",
568
+ "num_samples": 855,
569
+ "action_dim": 9,
570
+ "obs_mode": "vision_wrench_state",
571
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
572
+ },
573
+ {
574
+ "hdf5_key": "ep_00040",
575
+ "source_episode_index": 40,
576
+ "split": "train",
577
+ "num_samples": 810,
578
+ "action_dim": 9,
579
+ "obs_mode": "vision_wrench_state",
580
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
581
+ },
582
+ {
583
+ "hdf5_key": "ep_00041",
584
+ "source_episode_index": 41,
585
+ "split": "train",
586
+ "num_samples": 655,
587
+ "action_dim": 9,
588
+ "obs_mode": "vision_wrench_state",
589
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
590
+ },
591
+ {
592
+ "hdf5_key": "ep_00042",
593
+ "source_episode_index": 42,
594
+ "split": "train",
595
+ "num_samples": 946,
596
+ "action_dim": 9,
597
+ "obs_mode": "vision_wrench_state",
598
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
599
+ },
600
+ {
601
+ "hdf5_key": "ep_00043",
602
+ "source_episode_index": 43,
603
+ "split": "train",
604
+ "num_samples": 703,
605
+ "action_dim": 9,
606
+ "obs_mode": "vision_wrench_state",
607
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
608
+ },
609
+ {
610
+ "hdf5_key": "ep_00044",
611
+ "source_episode_index": 44,
612
+ "split": "train",
613
+ "num_samples": 606,
614
+ "action_dim": 9,
615
+ "obs_mode": "vision_wrench_state",
616
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
617
+ },
618
+ {
619
+ "hdf5_key": "ep_00045",
620
+ "source_episode_index": 45,
621
+ "split": "train",
622
+ "num_samples": 914,
623
+ "action_dim": 9,
624
+ "obs_mode": "vision_wrench_state",
625
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
626
+ },
627
+ {
628
+ "hdf5_key": "ep_00046",
629
+ "source_episode_index": 46,
630
+ "split": "train",
631
+ "num_samples": 949,
632
+ "action_dim": 9,
633
+ "obs_mode": "vision_wrench_state",
634
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
635
+ },
636
+ {
637
+ "hdf5_key": "ep_00047",
638
+ "source_episode_index": 47,
639
+ "split": "train",
640
+ "num_samples": 818,
641
+ "action_dim": 9,
642
+ "obs_mode": "vision_wrench_state",
643
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
644
+ },
645
+ {
646
+ "hdf5_key": "ep_00048",
647
+ "source_episode_index": 48,
648
+ "split": "train",
649
+ "num_samples": 913,
650
+ "action_dim": 9,
651
+ "obs_mode": "vision_wrench_state",
652
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
653
+ },
654
+ {
655
+ "hdf5_key": "ep_00049",
656
+ "source_episode_index": 49,
657
+ "split": "train",
658
+ "num_samples": 904,
659
+ "action_dim": 9,
660
+ "obs_mode": "vision_wrench_state",
661
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
662
+ },
663
+ {
664
+ "hdf5_key": "ep_00050",
665
+ "source_episode_index": 50,
666
+ "split": "train",
667
+ "num_samples": 923,
668
+ "action_dim": 9,
669
+ "obs_mode": "vision_wrench_state",
670
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
671
+ },
672
+ {
673
+ "hdf5_key": "ep_00051",
674
+ "source_episode_index": 51,
675
+ "split": "train",
676
+ "num_samples": 824,
677
+ "action_dim": 9,
678
+ "obs_mode": "vision_wrench_state",
679
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
680
+ },
681
+ {
682
+ "hdf5_key": "ep_00052",
683
+ "source_episode_index": 52,
684
+ "split": "train",
685
+ "num_samples": 777,
686
+ "action_dim": 9,
687
+ "obs_mode": "vision_wrench_state",
688
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
689
+ },
690
+ {
691
+ "hdf5_key": "ep_00053",
692
+ "source_episode_index": 53,
693
+ "split": "train",
694
+ "num_samples": 714,
695
+ "action_dim": 9,
696
+ "obs_mode": "vision_wrench_state",
697
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
698
+ },
699
+ {
700
+ "hdf5_key": "ep_00054",
701
+ "source_episode_index": 54,
702
+ "split": "train",
703
+ "num_samples": 771,
704
+ "action_dim": 9,
705
+ "obs_mode": "vision_wrench_state",
706
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
707
+ },
708
+ {
709
+ "hdf5_key": "ep_00055",
710
+ "source_episode_index": 55,
711
+ "split": "train",
712
+ "num_samples": 684,
713
+ "action_dim": 9,
714
+ "obs_mode": "vision_wrench_state",
715
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
716
+ },
717
+ {
718
+ "hdf5_key": "ep_00056",
719
+ "source_episode_index": 56,
720
+ "split": "train",
721
+ "num_samples": 766,
722
+ "action_dim": 9,
723
+ "obs_mode": "vision_wrench_state",
724
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
725
+ },
726
+ {
727
+ "hdf5_key": "ep_00057",
728
+ "source_episode_index": 57,
729
+ "split": "train",
730
+ "num_samples": 890,
731
+ "action_dim": 9,
732
+ "obs_mode": "vision_wrench_state",
733
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
734
+ },
735
+ {
736
+ "hdf5_key": "ep_00058",
737
+ "source_episode_index": 58,
738
+ "split": "train",
739
+ "num_samples": 774,
740
+ "action_dim": 9,
741
+ "obs_mode": "vision_wrench_state",
742
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
743
+ },
744
+ {
745
+ "hdf5_key": "ep_00059",
746
+ "source_episode_index": 59,
747
+ "split": "train",
748
+ "num_samples": 781,
749
+ "action_dim": 9,
750
+ "obs_mode": "vision_wrench_state",
751
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
752
+ },
753
+ {
754
+ "hdf5_key": "ep_00060",
755
+ "source_episode_index": 60,
756
+ "split": "train",
757
+ "num_samples": 1093,
758
+ "action_dim": 9,
759
+ "obs_mode": "vision_wrench_state",
760
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
761
+ },
762
+ {
763
+ "hdf5_key": "ep_00061",
764
+ "source_episode_index": 61,
765
+ "split": "train",
766
+ "num_samples": 900,
767
+ "action_dim": 9,
768
+ "obs_mode": "vision_wrench_state",
769
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
770
+ },
771
+ {
772
+ "hdf5_key": "ep_00062",
773
+ "source_episode_index": 62,
774
+ "split": "train",
775
+ "num_samples": 814,
776
+ "action_dim": 9,
777
+ "obs_mode": "vision_wrench_state",
778
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
779
+ },
780
+ {
781
+ "hdf5_key": "ep_00063",
782
+ "source_episode_index": 63,
783
+ "split": "train",
784
+ "num_samples": 1108,
785
+ "action_dim": 9,
786
+ "obs_mode": "vision_wrench_state",
787
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
788
+ },
789
+ {
790
+ "hdf5_key": "ep_00064",
791
+ "source_episode_index": 64,
792
+ "split": "train",
793
+ "num_samples": 816,
794
+ "action_dim": 9,
795
+ "obs_mode": "vision_wrench_state",
796
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
797
+ },
798
+ {
799
+ "hdf5_key": "ep_00065",
800
+ "source_episode_index": 65,
801
+ "split": "train",
802
+ "num_samples": 963,
803
+ "action_dim": 9,
804
+ "obs_mode": "vision_wrench_state",
805
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
806
+ },
807
+ {
808
+ "hdf5_key": "ep_00066",
809
+ "source_episode_index": 66,
810
+ "split": "train",
811
+ "num_samples": 895,
812
+ "action_dim": 9,
813
+ "obs_mode": "vision_wrench_state",
814
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
815
+ },
816
+ {
817
+ "hdf5_key": "ep_00067",
818
+ "source_episode_index": 67,
819
+ "split": "train",
820
+ "num_samples": 572,
821
+ "action_dim": 9,
822
+ "obs_mode": "vision_wrench_state",
823
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
824
+ },
825
+ {
826
+ "hdf5_key": "ep_00068",
827
+ "source_episode_index": 68,
828
+ "split": "train",
829
+ "num_samples": 648,
830
+ "action_dim": 9,
831
+ "obs_mode": "vision_wrench_state",
832
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
833
+ },
834
+ {
835
+ "hdf5_key": "ep_00069",
836
+ "source_episode_index": 69,
837
+ "split": "train",
838
+ "num_samples": 772,
839
+ "action_dim": 9,
840
+ "obs_mode": "vision_wrench_state",
841
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
842
+ },
843
+ {
844
+ "hdf5_key": "ep_00070",
845
+ "source_episode_index": 70,
846
+ "split": "train",
847
+ "num_samples": 702,
848
+ "action_dim": 9,
849
+ "obs_mode": "vision_wrench_state",
850
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
851
+ },
852
+ {
853
+ "hdf5_key": "ep_00071",
854
+ "source_episode_index": 71,
855
+ "split": "train",
856
+ "num_samples": 786,
857
+ "action_dim": 9,
858
+ "obs_mode": "vision_wrench_state",
859
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
860
+ },
861
+ {
862
+ "hdf5_key": "ep_00072",
863
+ "source_episode_index": 72,
864
+ "split": "train",
865
+ "num_samples": 843,
866
+ "action_dim": 9,
867
+ "obs_mode": "vision_wrench_state",
868
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
869
+ },
870
+ {
871
+ "hdf5_key": "ep_00073",
872
+ "source_episode_index": 73,
873
+ "split": "train",
874
+ "num_samples": 767,
875
+ "action_dim": 9,
876
+ "obs_mode": "vision_wrench_state",
877
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
878
+ },
879
+ {
880
+ "hdf5_key": "ep_00074",
881
+ "source_episode_index": 74,
882
+ "split": "train",
883
+ "num_samples": 764,
884
+ "action_dim": 9,
885
+ "obs_mode": "vision_wrench_state",
886
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
887
+ },
888
+ {
889
+ "hdf5_key": "ep_00075",
890
+ "source_episode_index": 75,
891
+ "split": "train",
892
+ "num_samples": 750,
893
+ "action_dim": 9,
894
+ "obs_mode": "vision_wrench_state",
895
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
896
+ },
897
+ {
898
+ "hdf5_key": "ep_00076",
899
+ "source_episode_index": 76,
900
+ "split": "train",
901
+ "num_samples": 652,
902
+ "action_dim": 9,
903
+ "obs_mode": "vision_wrench_state",
904
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
905
+ },
906
+ {
907
+ "hdf5_key": "ep_00077",
908
+ "source_episode_index": 77,
909
+ "split": "train",
910
+ "num_samples": 812,
911
+ "action_dim": 9,
912
+ "obs_mode": "vision_wrench_state",
913
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
914
+ },
915
+ {
916
+ "hdf5_key": "ep_00078",
917
+ "source_episode_index": 78,
918
+ "split": "train",
919
+ "num_samples": 722,
920
+ "action_dim": 9,
921
+ "obs_mode": "vision_wrench_state",
922
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
923
+ },
924
+ {
925
+ "hdf5_key": "ep_00079",
926
+ "source_episode_index": 79,
927
+ "split": "train",
928
+ "num_samples": 723,
929
+ "action_dim": 9,
930
+ "obs_mode": "vision_wrench_state",
931
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
932
+ },
933
+ {
934
+ "hdf5_key": "ep_00080",
935
+ "source_episode_index": 80,
936
+ "split": "train",
937
+ "num_samples": 765,
938
+ "action_dim": 9,
939
+ "obs_mode": "vision_wrench_state",
940
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
941
+ },
942
+ {
943
+ "hdf5_key": "ep_00081",
944
+ "source_episode_index": 81,
945
+ "split": "train",
946
+ "num_samples": 981,
947
+ "action_dim": 9,
948
+ "obs_mode": "vision_wrench_state",
949
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
950
+ },
951
+ {
952
+ "hdf5_key": "ep_00082",
953
+ "source_episode_index": 82,
954
+ "split": "train",
955
+ "num_samples": 959,
956
+ "action_dim": 9,
957
+ "obs_mode": "vision_wrench_state",
958
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
959
+ },
960
+ {
961
+ "hdf5_key": "ep_00083",
962
+ "source_episode_index": 83,
963
+ "split": "train",
964
+ "num_samples": 655,
965
+ "action_dim": 9,
966
+ "obs_mode": "vision_wrench_state",
967
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
968
+ },
969
+ {
970
+ "hdf5_key": "ep_00084",
971
+ "source_episode_index": 84,
972
+ "split": "train",
973
+ "num_samples": 746,
974
+ "action_dim": 9,
975
+ "obs_mode": "vision_wrench_state",
976
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
977
+ },
978
+ {
979
+ "hdf5_key": "ep_00085",
980
+ "source_episode_index": 85,
981
+ "split": "train",
982
+ "num_samples": 660,
983
+ "action_dim": 9,
984
+ "obs_mode": "vision_wrench_state",
985
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
986
+ },
987
+ {
988
+ "hdf5_key": "ep_00086",
989
+ "source_episode_index": 86,
990
+ "split": "train",
991
+ "num_samples": 617,
992
+ "action_dim": 9,
993
+ "obs_mode": "vision_wrench_state",
994
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
995
+ },
996
+ {
997
+ "hdf5_key": "ep_00087",
998
+ "source_episode_index": 87,
999
+ "split": "train",
1000
+ "num_samples": 819,
1001
+ "action_dim": 9,
1002
+ "obs_mode": "vision_wrench_state",
1003
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1004
+ },
1005
+ {
1006
+ "hdf5_key": "ep_00088",
1007
+ "source_episode_index": 88,
1008
+ "split": "train",
1009
+ "num_samples": 696,
1010
+ "action_dim": 9,
1011
+ "obs_mode": "vision_wrench_state",
1012
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1013
+ },
1014
+ {
1015
+ "hdf5_key": "ep_00089",
1016
+ "source_episode_index": 89,
1017
+ "split": "train",
1018
+ "num_samples": 855,
1019
+ "action_dim": 9,
1020
+ "obs_mode": "vision_wrench_state",
1021
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1022
+ },
1023
+ {
1024
+ "hdf5_key": "ep_00090",
1025
+ "source_episode_index": 90,
1026
+ "split": "train",
1027
+ "num_samples": 678,
1028
+ "action_dim": 9,
1029
+ "obs_mode": "vision_wrench_state",
1030
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1031
+ },
1032
+ {
1033
+ "hdf5_key": "ep_00091",
1034
+ "source_episode_index": 91,
1035
+ "split": "train",
1036
+ "num_samples": 711,
1037
+ "action_dim": 9,
1038
+ "obs_mode": "vision_wrench_state",
1039
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1040
+ },
1041
+ {
1042
+ "hdf5_key": "ep_00092",
1043
+ "source_episode_index": 92,
1044
+ "split": "train",
1045
+ "num_samples": 763,
1046
+ "action_dim": 9,
1047
+ "obs_mode": "vision_wrench_state",
1048
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1049
+ },
1050
+ {
1051
+ "hdf5_key": "ep_00093",
1052
+ "source_episode_index": 93,
1053
+ "split": "train",
1054
+ "num_samples": 723,
1055
+ "action_dim": 9,
1056
+ "obs_mode": "vision_wrench_state",
1057
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1058
+ },
1059
+ {
1060
+ "hdf5_key": "ep_00094",
1061
+ "source_episode_index": 94,
1062
+ "split": "train",
1063
+ "num_samples": 723,
1064
+ "action_dim": 9,
1065
+ "obs_mode": "vision_wrench_state",
1066
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1067
+ },
1068
+ {
1069
+ "hdf5_key": "ep_00095",
1070
+ "source_episode_index": 95,
1071
+ "split": "train",
1072
+ "num_samples": 865,
1073
+ "action_dim": 9,
1074
+ "obs_mode": "vision_wrench_state",
1075
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1076
+ },
1077
+ {
1078
+ "hdf5_key": "ep_00096",
1079
+ "source_episode_index": 96,
1080
+ "split": "train",
1081
+ "num_samples": 495,
1082
+ "action_dim": 9,
1083
+ "obs_mode": "vision_wrench_state",
1084
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1085
+ },
1086
+ {
1087
+ "hdf5_key": "ep_00097",
1088
+ "source_episode_index": 97,
1089
+ "split": "train",
1090
+ "num_samples": 451,
1091
+ "action_dim": 9,
1092
+ "obs_mode": "vision_wrench_state",
1093
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1094
+ },
1095
+ {
1096
+ "hdf5_key": "ep_00098",
1097
+ "source_episode_index": 98,
1098
+ "split": "train",
1099
+ "num_samples": 690,
1100
+ "action_dim": 9,
1101
+ "obs_mode": "vision_wrench_state",
1102
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1103
+ },
1104
+ {
1105
+ "hdf5_key": "ep_00099",
1106
+ "source_episode_index": 99,
1107
+ "split": "train",
1108
+ "num_samples": 690,
1109
+ "action_dim": 9,
1110
+ "obs_mode": "vision_wrench_state",
1111
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1112
+ },
1113
+ {
1114
+ "hdf5_key": "ep_00100",
1115
+ "source_episode_index": 100,
1116
+ "split": "train",
1117
+ "num_samples": 615,
1118
+ "action_dim": 9,
1119
+ "obs_mode": "vision_wrench_state",
1120
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1121
+ },
1122
+ {
1123
+ "hdf5_key": "ep_00101",
1124
+ "source_episode_index": 101,
1125
+ "split": "train",
1126
+ "num_samples": 662,
1127
+ "action_dim": 9,
1128
+ "obs_mode": "vision_wrench_state",
1129
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1130
+ },
1131
+ {
1132
+ "hdf5_key": "ep_00102",
1133
+ "source_episode_index": 102,
1134
+ "split": "train",
1135
+ "num_samples": 614,
1136
+ "action_dim": 9,
1137
+ "obs_mode": "vision_wrench_state",
1138
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1139
+ },
1140
+ {
1141
+ "hdf5_key": "ep_00103",
1142
+ "source_episode_index": 103,
1143
+ "split": "train",
1144
+ "num_samples": 730,
1145
+ "action_dim": 9,
1146
+ "obs_mode": "vision_wrench_state",
1147
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1148
+ },
1149
+ {
1150
+ "hdf5_key": "ep_00104",
1151
+ "source_episode_index": 104,
1152
+ "split": "train",
1153
+ "num_samples": 683,
1154
+ "action_dim": 9,
1155
+ "obs_mode": "vision_wrench_state",
1156
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1157
+ },
1158
+ {
1159
+ "hdf5_key": "ep_00105",
1160
+ "source_episode_index": 105,
1161
+ "split": "train",
1162
+ "num_samples": 602,
1163
+ "action_dim": 9,
1164
+ "obs_mode": "vision_wrench_state",
1165
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1166
+ },
1167
+ {
1168
+ "hdf5_key": "ep_00106",
1169
+ "source_episode_index": 106,
1170
+ "split": "train",
1171
+ "num_samples": 652,
1172
+ "action_dim": 9,
1173
+ "obs_mode": "vision_wrench_state",
1174
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1175
+ },
1176
+ {
1177
+ "hdf5_key": "ep_00107",
1178
+ "source_episode_index": 107,
1179
+ "split": "train",
1180
+ "num_samples": 722,
1181
+ "action_dim": 9,
1182
+ "obs_mode": "vision_wrench_state",
1183
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1184
+ },
1185
+ {
1186
+ "hdf5_key": "ep_00108",
1187
+ "source_episode_index": 108,
1188
+ "split": "train",
1189
+ "num_samples": 605,
1190
+ "action_dim": 9,
1191
+ "obs_mode": "vision_wrench_state",
1192
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1193
+ },
1194
+ {
1195
+ "hdf5_key": "ep_00109",
1196
+ "source_episode_index": 109,
1197
+ "split": "train",
1198
+ "num_samples": 944,
1199
+ "action_dim": 9,
1200
+ "obs_mode": "vision_wrench_state",
1201
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1202
+ },
1203
+ {
1204
+ "hdf5_key": "ep_00110",
1205
+ "source_episode_index": 110,
1206
+ "split": "train",
1207
+ "num_samples": 661,
1208
+ "action_dim": 9,
1209
+ "obs_mode": "vision_wrench_state",
1210
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1211
+ },
1212
+ {
1213
+ "hdf5_key": "ep_00111",
1214
+ "source_episode_index": 111,
1215
+ "split": "train",
1216
+ "num_samples": 644,
1217
+ "action_dim": 9,
1218
+ "obs_mode": "vision_wrench_state",
1219
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1220
+ },
1221
+ {
1222
+ "hdf5_key": "ep_00112",
1223
+ "source_episode_index": 112,
1224
+ "split": "train",
1225
+ "num_samples": 614,
1226
+ "action_dim": 9,
1227
+ "obs_mode": "vision_wrench_state",
1228
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1229
+ },
1230
+ {
1231
+ "hdf5_key": "ep_00113",
1232
+ "source_episode_index": 113,
1233
+ "split": "train",
1234
+ "num_samples": 548,
1235
+ "action_dim": 9,
1236
+ "obs_mode": "vision_wrench_state",
1237
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1238
+ },
1239
+ {
1240
+ "hdf5_key": "ep_00114",
1241
+ "source_episode_index": 114,
1242
+ "split": "train",
1243
+ "num_samples": 568,
1244
+ "action_dim": 9,
1245
+ "obs_mode": "vision_wrench_state",
1246
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1247
+ },
1248
+ {
1249
+ "hdf5_key": "ep_00115",
1250
+ "source_episode_index": 115,
1251
+ "split": "train",
1252
+ "num_samples": 730,
1253
+ "action_dim": 9,
1254
+ "obs_mode": "vision_wrench_state",
1255
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1256
+ },
1257
+ {
1258
+ "hdf5_key": "ep_00116",
1259
+ "source_episode_index": 116,
1260
+ "split": "train",
1261
+ "num_samples": 481,
1262
+ "action_dim": 9,
1263
+ "obs_mode": "vision_wrench_state",
1264
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1265
+ },
1266
+ {
1267
+ "hdf5_key": "ep_00117",
1268
+ "source_episode_index": 117,
1269
+ "split": "train",
1270
+ "num_samples": 687,
1271
+ "action_dim": 9,
1272
+ "obs_mode": "vision_wrench_state",
1273
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1274
+ },
1275
+ {
1276
+ "hdf5_key": "ep_00118",
1277
+ "source_episode_index": 118,
1278
+ "split": "train",
1279
+ "num_samples": 618,
1280
+ "action_dim": 9,
1281
+ "obs_mode": "vision_wrench_state",
1282
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1283
+ },
1284
+ {
1285
+ "hdf5_key": "ep_00119",
1286
+ "source_episode_index": 119,
1287
+ "split": "train",
1288
+ "num_samples": 768,
1289
+ "action_dim": 9,
1290
+ "obs_mode": "vision_wrench_state",
1291
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1292
+ },
1293
+ {
1294
+ "hdf5_key": "ep_00120",
1295
+ "source_episode_index": 120,
1296
+ "split": "train",
1297
+ "num_samples": 734,
1298
+ "action_dim": 9,
1299
+ "obs_mode": "vision_wrench_state",
1300
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1301
+ },
1302
+ {
1303
+ "hdf5_key": "ep_00121",
1304
+ "source_episode_index": 121,
1305
+ "split": "train",
1306
+ "num_samples": 604,
1307
+ "action_dim": 9,
1308
+ "obs_mode": "vision_wrench_state",
1309
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1310
+ },
1311
+ {
1312
+ "hdf5_key": "ep_00122",
1313
+ "source_episode_index": 122,
1314
+ "split": "train",
1315
+ "num_samples": 610,
1316
+ "action_dim": 9,
1317
+ "obs_mode": "vision_wrench_state",
1318
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1319
+ },
1320
+ {
1321
+ "hdf5_key": "ep_00123",
1322
+ "source_episode_index": 123,
1323
+ "split": "train",
1324
+ "num_samples": 602,
1325
+ "action_dim": 9,
1326
+ "obs_mode": "vision_wrench_state",
1327
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1328
+ },
1329
+ {
1330
+ "hdf5_key": "ep_00124",
1331
+ "source_episode_index": 124,
1332
+ "split": "train",
1333
+ "num_samples": 630,
1334
+ "action_dim": 9,
1335
+ "obs_mode": "vision_wrench_state",
1336
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1337
+ },
1338
+ {
1339
+ "hdf5_key": "ep_00125",
1340
+ "source_episode_index": 125,
1341
+ "split": "train",
1342
+ "num_samples": 621,
1343
+ "action_dim": 9,
1344
+ "obs_mode": "vision_wrench_state",
1345
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1346
+ },
1347
+ {
1348
+ "hdf5_key": "ep_00126",
1349
+ "source_episode_index": 126,
1350
+ "split": "train",
1351
+ "num_samples": 600,
1352
+ "action_dim": 9,
1353
+ "obs_mode": "vision_wrench_state",
1354
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1355
+ },
1356
+ {
1357
+ "hdf5_key": "ep_00127",
1358
+ "source_episode_index": 127,
1359
+ "split": "train",
1360
+ "num_samples": 478,
1361
+ "action_dim": 9,
1362
+ "obs_mode": "vision_wrench_state",
1363
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1364
+ },
1365
+ {
1366
+ "hdf5_key": "ep_00128",
1367
+ "source_episode_index": 128,
1368
+ "split": "train",
1369
+ "num_samples": 606,
1370
+ "action_dim": 9,
1371
+ "obs_mode": "vision_wrench_state",
1372
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1373
+ },
1374
+ {
1375
+ "hdf5_key": "ep_00129",
1376
+ "source_episode_index": 129,
1377
+ "split": "train",
1378
+ "num_samples": 666,
1379
+ "action_dim": 9,
1380
+ "obs_mode": "vision_wrench_state",
1381
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1382
+ },
1383
+ {
1384
+ "hdf5_key": "ep_00130",
1385
+ "source_episode_index": 130,
1386
+ "split": "train",
1387
+ "num_samples": 655,
1388
+ "action_dim": 9,
1389
+ "obs_mode": "vision_wrench_state",
1390
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1391
+ },
1392
+ {
1393
+ "hdf5_key": "ep_00131",
1394
+ "source_episode_index": 131,
1395
+ "split": "train",
1396
+ "num_samples": 588,
1397
+ "action_dim": 9,
1398
+ "obs_mode": "vision_wrench_state",
1399
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1400
+ },
1401
+ {
1402
+ "hdf5_key": "ep_00132",
1403
+ "source_episode_index": 132,
1404
+ "split": "train",
1405
+ "num_samples": 806,
1406
+ "action_dim": 9,
1407
+ "obs_mode": "vision_wrench_state",
1408
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1409
+ },
1410
+ {
1411
+ "hdf5_key": "ep_00133",
1412
+ "source_episode_index": 133,
1413
+ "split": "train",
1414
+ "num_samples": 586,
1415
+ "action_dim": 9,
1416
+ "obs_mode": "vision_wrench_state",
1417
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1418
+ },
1419
+ {
1420
+ "hdf5_key": "ep_00134",
1421
+ "source_episode_index": 134,
1422
+ "split": "train",
1423
+ "num_samples": 714,
1424
+ "action_dim": 9,
1425
+ "obs_mode": "vision_wrench_state",
1426
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1427
+ },
1428
+ {
1429
+ "hdf5_key": "ep_00135",
1430
+ "source_episode_index": 135,
1431
+ "split": "train",
1432
+ "num_samples": 574,
1433
+ "action_dim": 9,
1434
+ "obs_mode": "vision_wrench_state",
1435
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1436
+ },
1437
+ {
1438
+ "hdf5_key": "ep_00136",
1439
+ "source_episode_index": 136,
1440
+ "split": "train",
1441
+ "num_samples": 577,
1442
+ "action_dim": 9,
1443
+ "obs_mode": "vision_wrench_state",
1444
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1445
+ },
1446
+ {
1447
+ "hdf5_key": "ep_00137",
1448
+ "source_episode_index": 137,
1449
+ "split": "train",
1450
+ "num_samples": 663,
1451
+ "action_dim": 9,
1452
+ "obs_mode": "vision_wrench_state",
1453
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1454
+ },
1455
+ {
1456
+ "hdf5_key": "ep_00138",
1457
+ "source_episode_index": 138,
1458
+ "split": "train",
1459
+ "num_samples": 720,
1460
+ "action_dim": 9,
1461
+ "obs_mode": "vision_wrench_state",
1462
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1463
+ },
1464
+ {
1465
+ "hdf5_key": "ep_00139",
1466
+ "source_episode_index": 139,
1467
+ "split": "train",
1468
+ "num_samples": 640,
1469
+ "action_dim": 9,
1470
+ "obs_mode": "vision_wrench_state",
1471
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1472
+ },
1473
+ {
1474
+ "hdf5_key": "ep_00140",
1475
+ "source_episode_index": 140,
1476
+ "split": "train",
1477
+ "num_samples": 544,
1478
+ "action_dim": 9,
1479
+ "obs_mode": "vision_wrench_state",
1480
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1481
+ },
1482
+ {
1483
+ "hdf5_key": "ep_00141",
1484
+ "source_episode_index": 141,
1485
+ "split": "train",
1486
+ "num_samples": 539,
1487
+ "action_dim": 9,
1488
+ "obs_mode": "vision_wrench_state",
1489
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1490
+ },
1491
+ {
1492
+ "hdf5_key": "ep_00142",
1493
+ "source_episode_index": 142,
1494
+ "split": "train",
1495
+ "num_samples": 592,
1496
+ "action_dim": 9,
1497
+ "obs_mode": "vision_wrench_state",
1498
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1499
+ },
1500
+ {
1501
+ "hdf5_key": "ep_00143",
1502
+ "source_episode_index": 143,
1503
+ "split": "train",
1504
+ "num_samples": 625,
1505
+ "action_dim": 9,
1506
+ "obs_mode": "vision_wrench_state",
1507
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1508
+ },
1509
+ {
1510
+ "hdf5_key": "ep_00144",
1511
+ "source_episode_index": 144,
1512
+ "split": "train",
1513
+ "num_samples": 684,
1514
+ "action_dim": 9,
1515
+ "obs_mode": "vision_wrench_state",
1516
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1517
+ },
1518
+ {
1519
+ "hdf5_key": "ep_00145",
1520
+ "source_episode_index": 145,
1521
+ "split": "train",
1522
+ "num_samples": 559,
1523
+ "action_dim": 9,
1524
+ "obs_mode": "vision_wrench_state",
1525
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1526
+ },
1527
+ {
1528
+ "hdf5_key": "ep_00146",
1529
+ "source_episode_index": 146,
1530
+ "split": "train",
1531
+ "num_samples": 677,
1532
+ "action_dim": 9,
1533
+ "obs_mode": "vision_wrench_state",
1534
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1535
+ },
1536
+ {
1537
+ "hdf5_key": "ep_00147",
1538
+ "source_episode_index": 147,
1539
+ "split": "train",
1540
+ "num_samples": 846,
1541
+ "action_dim": 9,
1542
+ "obs_mode": "vision_wrench_state",
1543
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1544
+ },
1545
+ {
1546
+ "hdf5_key": "ep_00148",
1547
+ "source_episode_index": 148,
1548
+ "split": "train",
1549
+ "num_samples": 615,
1550
+ "action_dim": 9,
1551
+ "obs_mode": "vision_wrench_state",
1552
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1553
+ },
1554
+ {
1555
+ "hdf5_key": "ep_00149",
1556
+ "source_episode_index": 149,
1557
+ "split": "train",
1558
+ "num_samples": 517,
1559
+ "action_dim": 9,
1560
+ "obs_mode": "vision_wrench_state",
1561
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1562
+ },
1563
+ {
1564
+ "hdf5_key": "ep_00150",
1565
+ "source_episode_index": 150,
1566
+ "split": "train",
1567
+ "num_samples": 817,
1568
+ "action_dim": 9,
1569
+ "obs_mode": "vision_wrench_state",
1570
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1571
+ },
1572
+ {
1573
+ "hdf5_key": "ep_00151",
1574
+ "source_episode_index": 151,
1575
+ "split": "train",
1576
+ "num_samples": 814,
1577
+ "action_dim": 9,
1578
+ "obs_mode": "vision_wrench_state",
1579
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1580
+ },
1581
+ {
1582
+ "hdf5_key": "ep_00152",
1583
+ "source_episode_index": 152,
1584
+ "split": "train",
1585
+ "num_samples": 828,
1586
+ "action_dim": 9,
1587
+ "obs_mode": "vision_wrench_state",
1588
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1589
+ },
1590
+ {
1591
+ "hdf5_key": "ep_00153",
1592
+ "source_episode_index": 153,
1593
+ "split": "train",
1594
+ "num_samples": 736,
1595
+ "action_dim": 9,
1596
+ "obs_mode": "vision_wrench_state",
1597
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1598
+ },
1599
+ {
1600
+ "hdf5_key": "ep_00154",
1601
+ "source_episode_index": 154,
1602
+ "split": "train",
1603
+ "num_samples": 860,
1604
+ "action_dim": 9,
1605
+ "obs_mode": "vision_wrench_state",
1606
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1607
+ },
1608
+ {
1609
+ "hdf5_key": "ep_00155",
1610
+ "source_episode_index": 155,
1611
+ "split": "train",
1612
+ "num_samples": 786,
1613
+ "action_dim": 9,
1614
+ "obs_mode": "vision_wrench_state",
1615
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1616
+ },
1617
+ {
1618
+ "hdf5_key": "ep_00156",
1619
+ "source_episode_index": 156,
1620
+ "split": "train",
1621
+ "num_samples": 725,
1622
+ "action_dim": 9,
1623
+ "obs_mode": "vision_wrench_state",
1624
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1625
+ },
1626
+ {
1627
+ "hdf5_key": "ep_00157",
1628
+ "source_episode_index": 157,
1629
+ "split": "train",
1630
+ "num_samples": 834,
1631
+ "action_dim": 9,
1632
+ "obs_mode": "vision_wrench_state",
1633
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1634
+ },
1635
+ {
1636
+ "hdf5_key": "ep_00158",
1637
+ "source_episode_index": 158,
1638
+ "split": "train",
1639
+ "num_samples": 652,
1640
+ "action_dim": 9,
1641
+ "obs_mode": "vision_wrench_state",
1642
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1643
+ },
1644
+ {
1645
+ "hdf5_key": "ep_00159",
1646
+ "source_episode_index": 159,
1647
+ "split": "train",
1648
+ "num_samples": 699,
1649
+ "action_dim": 9,
1650
+ "obs_mode": "vision_wrench_state",
1651
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1652
+ },
1653
+ {
1654
+ "hdf5_key": "ep_00160",
1655
+ "source_episode_index": 160,
1656
+ "split": "train",
1657
+ "num_samples": 812,
1658
+ "action_dim": 9,
1659
+ "obs_mode": "vision_wrench_state",
1660
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1661
+ },
1662
+ {
1663
+ "hdf5_key": "ep_00161",
1664
+ "source_episode_index": 161,
1665
+ "split": "train",
1666
+ "num_samples": 782,
1667
+ "action_dim": 9,
1668
+ "obs_mode": "vision_wrench_state",
1669
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1670
+ },
1671
+ {
1672
+ "hdf5_key": "ep_00162",
1673
+ "source_episode_index": 162,
1674
+ "split": "train",
1675
+ "num_samples": 681,
1676
+ "action_dim": 9,
1677
+ "obs_mode": "vision_wrench_state",
1678
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1679
+ },
1680
+ {
1681
+ "hdf5_key": "ep_00163",
1682
+ "source_episode_index": 163,
1683
+ "split": "train",
1684
+ "num_samples": 742,
1685
+ "action_dim": 9,
1686
+ "obs_mode": "vision_wrench_state",
1687
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1688
+ },
1689
+ {
1690
+ "hdf5_key": "ep_00164",
1691
+ "source_episode_index": 164,
1692
+ "split": "train",
1693
+ "num_samples": 766,
1694
+ "action_dim": 9,
1695
+ "obs_mode": "vision_wrench_state",
1696
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1697
+ },
1698
+ {
1699
+ "hdf5_key": "ep_00165",
1700
+ "source_episode_index": 165,
1701
+ "split": "train",
1702
+ "num_samples": 821,
1703
+ "action_dim": 9,
1704
+ "obs_mode": "vision_wrench_state",
1705
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1706
+ },
1707
+ {
1708
+ "hdf5_key": "ep_00166",
1709
+ "source_episode_index": 166,
1710
+ "split": "train",
1711
+ "num_samples": 1028,
1712
+ "action_dim": 9,
1713
+ "obs_mode": "vision_wrench_state",
1714
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1715
+ },
1716
+ {
1717
+ "hdf5_key": "ep_00167",
1718
+ "source_episode_index": 167,
1719
+ "split": "train",
1720
+ "num_samples": 720,
1721
+ "action_dim": 9,
1722
+ "obs_mode": "vision_wrench_state",
1723
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1724
+ },
1725
+ {
1726
+ "hdf5_key": "ep_00168",
1727
+ "source_episode_index": 168,
1728
+ "split": "train",
1729
+ "num_samples": 773,
1730
+ "action_dim": 9,
1731
+ "obs_mode": "vision_wrench_state",
1732
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1733
+ },
1734
+ {
1735
+ "hdf5_key": "ep_00169",
1736
+ "source_episode_index": 169,
1737
+ "split": "train",
1738
+ "num_samples": 690,
1739
+ "action_dim": 9,
1740
+ "obs_mode": "vision_wrench_state",
1741
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1742
+ },
1743
+ {
1744
+ "hdf5_key": "ep_00170",
1745
+ "source_episode_index": 170,
1746
+ "split": "train",
1747
+ "num_samples": 756,
1748
+ "action_dim": 9,
1749
+ "obs_mode": "vision_wrench_state",
1750
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1751
+ },
1752
+ {
1753
+ "hdf5_key": "ep_00171",
1754
+ "source_episode_index": 171,
1755
+ "split": "train",
1756
+ "num_samples": 742,
1757
+ "action_dim": 9,
1758
+ "obs_mode": "vision_wrench_state",
1759
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1760
+ },
1761
+ {
1762
+ "hdf5_key": "ep_00172",
1763
+ "source_episode_index": 172,
1764
+ "split": "valid",
1765
+ "num_samples": 689,
1766
+ "action_dim": 9,
1767
+ "obs_mode": "vision_wrench_state",
1768
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1769
+ },
1770
+ {
1771
+ "hdf5_key": "ep_00173",
1772
+ "source_episode_index": 173,
1773
+ "split": "valid",
1774
+ "num_samples": 639,
1775
+ "action_dim": 9,
1776
+ "obs_mode": "vision_wrench_state",
1777
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1778
+ },
1779
+ {
1780
+ "hdf5_key": "ep_00174",
1781
+ "source_episode_index": 174,
1782
+ "split": "valid",
1783
+ "num_samples": 694,
1784
+ "action_dim": 9,
1785
+ "obs_mode": "vision_wrench_state",
1786
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1787
+ },
1788
+ {
1789
+ "hdf5_key": "ep_00175",
1790
+ "source_episode_index": 175,
1791
+ "split": "valid",
1792
+ "num_samples": 532,
1793
+ "action_dim": 9,
1794
+ "obs_mode": "vision_wrench_state",
1795
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1796
+ },
1797
+ {
1798
+ "hdf5_key": "ep_00176",
1799
+ "source_episode_index": 176,
1800
+ "split": "valid",
1801
+ "num_samples": 665,
1802
+ "action_dim": 9,
1803
+ "obs_mode": "vision_wrench_state",
1804
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1805
+ },
1806
+ {
1807
+ "hdf5_key": "ep_00177",
1808
+ "source_episode_index": 177,
1809
+ "split": "valid",
1810
+ "num_samples": 771,
1811
+ "action_dim": 9,
1812
+ "obs_mode": "vision_wrench_state",
1813
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1814
+ },
1815
+ {
1816
+ "hdf5_key": "ep_00178",
1817
+ "source_episode_index": 178,
1818
+ "split": "valid",
1819
+ "num_samples": 602,
1820
+ "action_dim": 9,
1821
+ "obs_mode": "vision_wrench_state",
1822
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1823
+ },
1824
+ {
1825
+ "hdf5_key": "ep_00179",
1826
+ "source_episode_index": 179,
1827
+ "split": "valid",
1828
+ "num_samples": 700,
1829
+ "action_dim": 9,
1830
+ "obs_mode": "vision_wrench_state",
1831
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1832
+ },
1833
+ {
1834
+ "hdf5_key": "ep_00180",
1835
+ "source_episode_index": 180,
1836
+ "split": "valid",
1837
+ "num_samples": 880,
1838
+ "action_dim": 9,
1839
+ "obs_mode": "vision_wrench_state",
1840
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1841
+ },
1842
+ {
1843
+ "hdf5_key": "ep_00181",
1844
+ "source_episode_index": 181,
1845
+ "split": "valid",
1846
+ "num_samples": 522,
1847
+ "action_dim": 9,
1848
+ "obs_mode": "vision_wrench_state",
1849
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1850
+ },
1851
+ {
1852
+ "hdf5_key": "ep_00182",
1853
+ "source_episode_index": 182,
1854
+ "split": "valid",
1855
+ "num_samples": 697,
1856
+ "action_dim": 9,
1857
+ "obs_mode": "vision_wrench_state",
1858
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1859
+ },
1860
+ {
1861
+ "hdf5_key": "ep_00183",
1862
+ "source_episode_index": 183,
1863
+ "split": "valid",
1864
+ "num_samples": 621,
1865
+ "action_dim": 9,
1866
+ "obs_mode": "vision_wrench_state",
1867
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1868
+ },
1869
+ {
1870
+ "hdf5_key": "ep_00184",
1871
+ "source_episode_index": 184,
1872
+ "split": "valid",
1873
+ "num_samples": 710,
1874
+ "action_dim": 9,
1875
+ "obs_mode": "vision_wrench_state",
1876
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1877
+ },
1878
+ {
1879
+ "hdf5_key": "ep_00185",
1880
+ "source_episode_index": 185,
1881
+ "split": "valid",
1882
+ "num_samples": 774,
1883
+ "action_dim": 9,
1884
+ "obs_mode": "vision_wrench_state",
1885
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1886
+ },
1887
+ {
1888
+ "hdf5_key": "ep_00186",
1889
+ "source_episode_index": 186,
1890
+ "split": "valid",
1891
+ "num_samples": 682,
1892
+ "action_dim": 9,
1893
+ "obs_mode": "vision_wrench_state",
1894
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1895
+ },
1896
+ {
1897
+ "hdf5_key": "ep_00187",
1898
+ "source_episode_index": 187,
1899
+ "split": "valid",
1900
+ "num_samples": 715,
1901
+ "action_dim": 9,
1902
+ "obs_mode": "vision_wrench_state",
1903
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1904
+ },
1905
+ {
1906
+ "hdf5_key": "ep_00188",
1907
+ "source_episode_index": 188,
1908
+ "split": "valid",
1909
+ "num_samples": 798,
1910
+ "action_dim": 9,
1911
+ "obs_mode": "vision_wrench_state",
1912
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1913
+ },
1914
+ {
1915
+ "hdf5_key": "ep_00189",
1916
+ "source_episode_index": 189,
1917
+ "split": "valid",
1918
+ "num_samples": 786,
1919
+ "action_dim": 9,
1920
+ "obs_mode": "vision_wrench_state",
1921
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1922
+ },
1923
+ {
1924
+ "hdf5_key": "ep_00190",
1925
+ "source_episode_index": 190,
1926
+ "split": "valid",
1927
+ "num_samples": 759,
1928
+ "action_dim": 9,
1929
+ "obs_mode": "vision_wrench_state",
1930
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1931
+ },
1932
+ {
1933
+ "hdf5_key": "ep_00191",
1934
+ "source_episode_index": 191,
1935
+ "split": "valid",
1936
+ "num_samples": 812,
1937
+ "action_dim": 9,
1938
+ "obs_mode": "vision_wrench_state",
1939
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1940
+ },
1941
+ {
1942
+ "hdf5_key": "ep_00192",
1943
+ "source_episode_index": 192,
1944
+ "split": "valid",
1945
+ "num_samples": 803,
1946
+ "action_dim": 9,
1947
+ "obs_mode": "vision_wrench_state",
1948
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1949
+ },
1950
+ {
1951
+ "hdf5_key": "ep_00193",
1952
+ "source_episode_index": 193,
1953
+ "split": "valid",
1954
+ "num_samples": 780,
1955
+ "action_dim": 9,
1956
+ "obs_mode": "vision_wrench_state",
1957
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1958
+ },
1959
+ {
1960
+ "hdf5_key": "ep_00194",
1961
+ "source_episode_index": 194,
1962
+ "split": "valid",
1963
+ "num_samples": 853,
1964
+ "action_dim": 9,
1965
+ "obs_mode": "vision_wrench_state",
1966
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1967
+ },
1968
+ {
1969
+ "hdf5_key": "ep_00195",
1970
+ "source_episode_index": 195,
1971
+ "split": "valid",
1972
+ "num_samples": 710,
1973
+ "action_dim": 9,
1974
+ "obs_mode": "vision_wrench_state",
1975
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1976
+ },
1977
+ {
1978
+ "hdf5_key": "ep_00196",
1979
+ "source_episode_index": 196,
1980
+ "split": "valid",
1981
+ "num_samples": 855,
1982
+ "action_dim": 9,
1983
+ "obs_mode": "vision_wrench_state",
1984
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1985
+ },
1986
+ {
1987
+ "hdf5_key": "ep_00197",
1988
+ "source_episode_index": 197,
1989
+ "split": "valid",
1990
+ "num_samples": 707,
1991
+ "action_dim": 9,
1992
+ "obs_mode": "vision_wrench_state",
1993
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
1994
+ },
1995
+ {
1996
+ "hdf5_key": "ep_00198",
1997
+ "source_episode_index": 198,
1998
+ "split": "valid",
1999
+ "num_samples": 557,
2000
+ "action_dim": 9,
2001
+ "obs_mode": "vision_wrench_state",
2002
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
2003
+ },
2004
+ {
2005
+ "hdf5_key": "ep_00199",
2006
+ "source_episode_index": 199,
2007
+ "split": "valid",
2008
+ "num_samples": 522,
2009
+ "action_dim": 9,
2010
+ "obs_mode": "vision_wrench_state",
2011
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
2012
+ },
2013
+ {
2014
+ "hdf5_key": "ep_00200",
2015
+ "source_episode_index": 200,
2016
+ "split": "valid",
2017
+ "num_samples": 581,
2018
+ "action_dim": 9,
2019
+ "obs_mode": "vision_wrench_state",
2020
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
2021
+ },
2022
+ {
2023
+ "hdf5_key": "ep_00201",
2024
+ "source_episode_index": 201,
2025
+ "split": "valid",
2026
+ "num_samples": 574,
2027
+ "action_dim": 9,
2028
+ "obs_mode": "vision_wrench_state",
2029
+ "action_mode": "pre_usercmd_stiffness_zeta_hand_binary"
2030
+ }
2031
+ ]
2032
+ }
vac_pre_vis_wrench_state/last.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6829f73d431908d6b9f8601af09efb386a53e739a9a820f3a90c9d229f0082c2
3
+ size 1260694301