Hugging Face's logo

eunjuri
/
pi0_training_baseball_ft

Robotics
LeRobot
Safetensors
pi0
Model card Files
xet
 Community
eunjuri commited on
Commit
92ddd11
·
verified ·
1 Parent(s): d16c6b8

Upload DataProcessorPipeline

Browse files
Files changed (2) hide show
  1. policy_preprocessor.json +363 -0
  2. policy_preprocessor_step_5_normalizer_processor.safetensors +3 -0
policy_preprocessor.json ADDED
@@ -0,0 +1,363 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_preprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "rename_observations_processor",
6
+ "config": {
7
+ "rename_map": {}
8
+ }
9
+ },
10
+ {
11
+ "registry_name": "to_batch_processor",
12
+ "config": {}
13
+ },
14
+ {
15
+ "registry_name": "pi0_new_line_processor",
16
+ "config": {}
17
+ },
18
+ {
19
+ "registry_name": "tokenizer_processor",
20
+ "config": {
21
+ "max_length": 48,
22
+ "task_key": "task",
23
+ "padding_side": "right",
24
+ "padding": "max_length",
25
+ "truncation": true,
26
+ "tokenizer_name": "google/paligemma-3b-pt-224"
27
+ }
28
+ },
29
+ {
30
+ "registry_name": "device_processor",
31
+ "config": {
32
+ "device": "cuda",
33
+ "float_dtype": null
34
+ }
35
+ },
36
+ {
37
+ "registry_name": "normalizer_processor",
38
+ "config": {
39
+ "eps": 1e-08,
40
+ "features": {
41
+ "observation.state": {
42
+ "type": "STATE",
43
+ "shape": [
44
+ 26
45
+ ]
46
+ },
47
+ "observation.language.tokens": {
48
+ "type": "STATE",
49
+ "shape": [
50
+ 48
51
+ ]
52
+ },
53
+ "observation.language.attention_mask": {
54
+ "type": "STATE",
55
+ "shape": [
56
+ 48
57
+ ]
58
+ },
59
+ "observation.images.cam_left_high": {
60
+ "type": "VISUAL",
61
+ "shape": [
62
+ 3,
63
+ 480,
64
+ 848
65
+ ]
66
+ },
67
+ "observation.images.cam_third": {
68
+ "type": "VISUAL",
69
+ "shape": [
70
+ 3,
71
+ 480,
72
+ 640
73
+ ]
74
+ },
75
+ "observation.tactiles.left_tactile_little_finger_tip": {
76
+ "type": "VISUAL",
77
+ "shape": [
78
+ 3,
79
+ 3,
80
+ 3
81
+ ]
82
+ },
83
+ "observation.tactiles.left_tactile_little_finger_nail": {
84
+ "type": "VISUAL",
85
+ "shape": [
86
+ 3,
87
+ 12,
88
+ 8
89
+ ]
90
+ },
91
+ "observation.tactiles.left_tactile_little_finger_pad": {
92
+ "type": "VISUAL",
93
+ "shape": [
94
+ 3,
95
+ 10,
96
+ 8
97
+ ]
98
+ },
99
+ "observation.tactiles.left_tactile_ring_finger_tip": {
100
+ "type": "VISUAL",
101
+ "shape": [
102
+ 3,
103
+ 3,
104
+ 3
105
+ ]
106
+ },
107
+ "observation.tactiles.left_tactile_ring_finger_nail": {
108
+ "type": "VISUAL",
109
+ "shape": [
110
+ 3,
111
+ 12,
112
+ 8
113
+ ]
114
+ },
115
+ "observation.tactiles.left_tactile_ring_finger_pad": {
116
+ "type": "VISUAL",
117
+ "shape": [
118
+ 3,
119
+ 10,
120
+ 8
121
+ ]
122
+ },
123
+ "observation.tactiles.left_tactile_middle_finger_tip": {
124
+ "type": "VISUAL",
125
+ "shape": [
126
+ 3,
127
+ 3,
128
+ 3
129
+ ]
130
+ },
131
+ "observation.tactiles.left_tactile_middle_finger_nail": {
132
+ "type": "VISUAL",
133
+ "shape": [
134
+ 3,
135
+ 12,
136
+ 8
137
+ ]
138
+ },
139
+ "observation.tactiles.left_tactile_middle_finger_pad": {
140
+ "type": "VISUAL",
141
+ "shape": [
142
+ 3,
143
+ 10,
144
+ 8
145
+ ]
146
+ },
147
+ "observation.tactiles.left_tactile_index_finger_tip": {
148
+ "type": "VISUAL",
149
+ "shape": [
150
+ 3,
151
+ 3,
152
+ 3
153
+ ]
154
+ },
155
+ "observation.tactiles.left_tactile_index_finger_nail": {
156
+ "type": "VISUAL",
157
+ "shape": [
158
+ 3,
159
+ 12,
160
+ 8
161
+ ]
162
+ },
163
+ "observation.tactiles.left_tactile_index_finger_pad": {
164
+ "type": "VISUAL",
165
+ "shape": [
166
+ 3,
167
+ 10,
168
+ 8
169
+ ]
170
+ },
171
+ "observation.tactiles.left_tactile_thumb_tip": {
172
+ "type": "VISUAL",
173
+ "shape": [
174
+ 3,
175
+ 3,
176
+ 3
177
+ ]
178
+ },
179
+ "observation.tactiles.left_tactile_thumb_nail": {
180
+ "type": "VISUAL",
181
+ "shape": [
182
+ 3,
183
+ 12,
184
+ 8
185
+ ]
186
+ },
187
+ "observation.tactiles.left_tactile_thumb_middle": {
188
+ "type": "VISUAL",
189
+ "shape": [
190
+ 3,
191
+ 3,
192
+ 3
193
+ ]
194
+ },
195
+ "observation.tactiles.left_tactile_thumb_pad": {
196
+ "type": "VISUAL",
197
+ "shape": [
198
+ 3,
199
+ 12,
200
+ 8
201
+ ]
202
+ },
203
+ "observation.tactiles.left_tactile_palm": {
204
+ "type": "VISUAL",
205
+ "shape": [
206
+ 3,
207
+ 8,
208
+ 14
209
+ ]
210
+ },
211
+ "observation.tactiles.right_tactile_little_finger_tip": {
212
+ "type": "VISUAL",
213
+ "shape": [
214
+ 3,
215
+ 3,
216
+ 3
217
+ ]
218
+ },
219
+ "observation.tactiles.right_tactile_little_finger_nail": {
220
+ "type": "VISUAL",
221
+ "shape": [
222
+ 3,
223
+ 12,
224
+ 8
225
+ ]
226
+ },
227
+ "observation.tactiles.right_tactile_little_finger_pad": {
228
+ "type": "VISUAL",
229
+ "shape": [
230
+ 3,
231
+ 10,
232
+ 8
233
+ ]
234
+ },
235
+ "observation.tactiles.right_tactile_ring_finger_tip": {
236
+ "type": "VISUAL",
237
+ "shape": [
238
+ 3,
239
+ 3,
240
+ 3
241
+ ]
242
+ },
243
+ "observation.tactiles.right_tactile_ring_finger_nail": {
244
+ "type": "VISUAL",
245
+ "shape": [
246
+ 3,
247
+ 12,
248
+ 8
249
+ ]
250
+ },
251
+ "observation.tactiles.right_tactile_ring_finger_pad": {
252
+ "type": "VISUAL",
253
+ "shape": [
254
+ 3,
255
+ 10,
256
+ 8
257
+ ]
258
+ },
259
+ "observation.tactiles.right_tactile_middle_finger_tip": {
260
+ "type": "VISUAL",
261
+ "shape": [
262
+ 3,
263
+ 3,
264
+ 3
265
+ ]
266
+ },
267
+ "observation.tactiles.right_tactile_middle_finger_nail": {
268
+ "type": "VISUAL",
269
+ "shape": [
270
+ 3,
271
+ 12,
272
+ 8
273
+ ]
274
+ },
275
+ "observation.tactiles.right_tactile_middle_finger_pad": {
276
+ "type": "VISUAL",
277
+ "shape": [
278
+ 3,
279
+ 10,
280
+ 8
281
+ ]
282
+ },
283
+ "observation.tactiles.right_tactile_index_finger_tip": {
284
+ "type": "VISUAL",
285
+ "shape": [
286
+ 3,
287
+ 3,
288
+ 3
289
+ ]
290
+ },
291
+ "observation.tactiles.right_tactile_index_finger_nail": {
292
+ "type": "VISUAL",
293
+ "shape": [
294
+ 3,
295
+ 12,
296
+ 8
297
+ ]
298
+ },
299
+ "observation.tactiles.right_tactile_index_finger_pad": {
300
+ "type": "VISUAL",
301
+ "shape": [
302
+ 3,
303
+ 10,
304
+ 8
305
+ ]
306
+ },
307
+ "observation.tactiles.right_tactile_thumb_tip": {
308
+ "type": "VISUAL",
309
+ "shape": [
310
+ 3,
311
+ 3,
312
+ 3
313
+ ]
314
+ },
315
+ "observation.tactiles.right_tactile_thumb_nail": {
316
+ "type": "VISUAL",
317
+ "shape": [
318
+ 3,
319
+ 12,
320
+ 8
321
+ ]
322
+ },
323
+ "observation.tactiles.right_tactile_thumb_middle": {
324
+ "type": "VISUAL",
325
+ "shape": [
326
+ 3,
327
+ 3,
328
+ 3
329
+ ]
330
+ },
331
+ "observation.tactiles.right_tactile_thumb_pad": {
332
+ "type": "VISUAL",
333
+ "shape": [
334
+ 3,
335
+ 12,
336
+ 8
337
+ ]
338
+ },
339
+ "observation.tactiles.right_tactile_palm": {
340
+ "type": "VISUAL",
341
+ "shape": [
342
+ 3,
343
+ 8,
344
+ 14
345
+ ]
346
+ },
347
+ "action": {
348
+ "type": "ACTION",
349
+ "shape": [
350
+ 26
351
+ ]
352
+ }
353
+ },
354
+ "norm_map": {
355
+ "VISUAL": "IDENTITY",
356
+ "STATE": "MEAN_STD",
357
+ "ACTION": "MEAN_STD"
358
+ }
359
+ },
360
+ "state_file": "policy_preprocessor_step_5_normalizer_processor.safetensors"
361
+ }
362
+ ]
363
+ }
policy_preprocessor_step_5_normalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61dd806945cbbc91ead5cc2a07a0d41d5ad7a3675f8d3a5e8087ac36082d6ee8
3
+ size 28236