matCercola18 commited on
Commit
48ec395
·
verified ·
1 Parent(s): 7a915b6

Upload DataProcessorPipeline

Browse files
policy_preprocessor.json ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_preprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "rename_observations_processor",
6
+ "config": {
7
+ "rename_map": {
8
+ "observation.images.image_0": "observation.images.top"
9
+ }
10
+ }
11
+ },
12
+ {
13
+ "registry_name": "to_batch_processor",
14
+ "config": {}
15
+ },
16
+ {
17
+ "registry_name": "pi0_new_line_processor",
18
+ "config": {}
19
+ },
20
+ {
21
+ "registry_name": "tokenizer_processor",
22
+ "config": {
23
+ "max_length": 72,
24
+ "task_key": "task",
25
+ "padding_side": "right",
26
+ "padding": "max_length",
27
+ "truncation": true,
28
+ "tokenizer_name": "google/paligemma-3b-pt-224"
29
+ }
30
+ },
31
+ {
32
+ "registry_name": "device_processor",
33
+ "config": {
34
+ "device": "cuda",
35
+ "float_dtype": null
36
+ }
37
+ },
38
+ {
39
+ "registry_name": "normalizer_processor",
40
+ "config": {
41
+ "eps": 1e-08,
42
+ "features": {
43
+ "observation.images.top": {
44
+ "type": "VISUAL",
45
+ "shape": [
46
+ 3,
47
+ 224,
48
+ 224
49
+ ]
50
+ },
51
+ "observation.state": {
52
+ "type": "STATE",
53
+ "shape": [
54
+ 7
55
+ ]
56
+ },
57
+ "action": {
58
+ "type": "ACTION",
59
+ "shape": [
60
+ 7
61
+ ]
62
+ }
63
+ },
64
+ "norm_map": {
65
+ "VISUAL": "IDENTITY",
66
+ "STATE": "IDENTITY",
67
+ "ACTION": "IDENTITY"
68
+ }
69
+ },
70
+ "state_file": "policy_preprocessor_step_5_normalizer_processor.safetensors"
71
+ }
72
+ ]
73
+ }
policy_preprocessor_step_5_normalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4edc8af3c926be6c6c8768dafdd39a91e7bad7ea6c905039ace6dc4ed59908f6
3
+ size 4828