Grigorij commited on
Commit
c2f7fec
·
verified ·
1 Parent(s): 28cfcc2

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": "normalizer_processor",
16
+ "config": {
17
+ "eps": 1e-08,
18
+ "features": {
19
+ "observation.state": {
20
+ "type": "STATE",
21
+ "shape": [
22
+ 3
23
+ ]
24
+ },
25
+ "observation.images.main": {
26
+ "type": "VISUAL",
27
+ "shape": [
28
+ 3,
29
+ 480,
30
+ 640
31
+ ]
32
+ },
33
+ "action": {
34
+ "type": "ACTION",
35
+ "shape": [
36
+ 4
37
+ ]
38
+ }
39
+ },
40
+ "norm_map": {
41
+ "VISUAL": "IDENTITY",
42
+ "STATE": "QUANTILES",
43
+ "ACTION": "QUANTILES"
44
+ }
45
+ },
46
+ "state_file": "policy_preprocessor_step_2_normalizer_processor.safetensors"
47
+ },
48
+ {
49
+ "registry_name": "pi05_prepare_state_tokenizer_processor_step",
50
+ "config": {}
51
+ },
52
+ {
53
+ "registry_name": "tokenizer_processor",
54
+ "config": {
55
+ "max_length": 200,
56
+ "task_key": "task",
57
+ "padding_side": "right",
58
+ "padding": "max_length",
59
+ "truncation": true,
60
+ "tokenizer_name": "google/paligemma-3b-pt-224"
61
+ }
62
+ },
63
+ {
64
+ "registry_name": "device_processor",
65
+ "config": {
66
+ "device": "cuda",
67
+ "float_dtype": null
68
+ }
69
+ }
70
+ ]
71
+ }
policy_preprocessor_step_2_normalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13d11d2684b4d3dcebf9960be0c89b4d64d7123265018482ddf5411df7e04a4b
3
+ size 6372