benmayeux commited on
Commit
36298f1
·
verified ·
1 Parent(s): a3a3214

Upload DataProcessorPipeline

Browse files
policy_preprocessor.json ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_preprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "image_crop_resize_processor",
6
+ "config": {
7
+ "crop_params_dict": null,
8
+ "resize_size": [
9
+ 224,
10
+ 224
11
+ ]
12
+ }
13
+ },
14
+ {
15
+ "registry_name": "rename_observations_processor",
16
+ "config": {
17
+ "rename_map": {}
18
+ }
19
+ },
20
+ {
21
+ "registry_name": "to_batch_processor",
22
+ "config": {}
23
+ },
24
+ {
25
+ "registry_name": "groot_pack_inputs_v3",
26
+ "config": {
27
+ "state_horizon": 1,
28
+ "action_horizon": 16,
29
+ "max_state_dim": 64,
30
+ "max_action_dim": 32,
31
+ "language_key": "task",
32
+ "formalize_language": false,
33
+ "embodiment_tag": "new_embodiment",
34
+ "embodiment_mapping": {
35
+ "new_embodiment": 31,
36
+ "oxe_droid": 17,
37
+ "agibot_genie1": 26,
38
+ "gr1": 24,
39
+ "so100": 2,
40
+ "unitree_g1": 3
41
+ },
42
+ "normalize_min_max": true
43
+ },
44
+ "state_file": "policy_preprocessor_step_3_groot_pack_inputs_v3.safetensors"
45
+ },
46
+ {
47
+ "registry_name": "groot_eagle_encode_v3",
48
+ "config": {}
49
+ },
50
+ {
51
+ "registry_name": "groot_eagle_collate_v3",
52
+ "config": {}
53
+ },
54
+ {
55
+ "registry_name": "device_processor",
56
+ "config": {
57
+ "device": "cuda",
58
+ "float_dtype": null
59
+ }
60
+ }
61
+ ]
62
+ }
policy_preprocessor_step_3_groot_pack_inputs_v3.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:18505bffb344713b80edfe4d0595a63a2401dc1f486459f04024d3bac275a29c
3
+ size 8568