mimirmimir commited on
Commit
b65522c
·
verified ·
1 Parent(s): 88ba06a

Upload DataProcessorPipeline

Browse files
policy_preprocessor.json ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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.images.base_camera": {
42
+ "type": "VISUAL",
43
+ "shape": [
44
+ 3,
45
+ 224,
46
+ 224
47
+ ]
48
+ },
49
+ "observation.state": {
50
+ "type": "STATE",
51
+ "shape": [
52
+ 9
53
+ ]
54
+ },
55
+ "action": {
56
+ "type": "ACTION",
57
+ "shape": [
58
+ 8
59
+ ]
60
+ }
61
+ },
62
+ "norm_map": {
63
+ "VISUAL": "IDENTITY",
64
+ "STATE": "MEAN_STD",
65
+ "ACTION": "MEAN_STD"
66
+ }
67
+ },
68
+ "state_file": "policy_preprocessor_step_5_normalizer_processor.safetensors"
69
+ }
70
+ ]
71
+ }
policy_preprocessor_step_5_normalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a88996fd56e561726fc6ab1dd3f5ecc63165baf133d73a75be4f705d3e52cf9
3
+ size 3392