Robotics
LeRobot
Safetensors
groot
raymondlo84-nvidia commited on
Commit
596fa69
·
verified ·
1 Parent(s): 883fd8d

Upload DataProcessorPipeline

Browse files
policy_preprocessor.json ADDED
@@ -0,0 +1,77 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "groot_n1_7_pack_inputs_v1",
16
+ "config": {
17
+ "state_horizon": 1,
18
+ "action_horizon": 40,
19
+ "valid_action_horizon": 40,
20
+ "video_horizon": null,
21
+ "max_state_dim": 132,
22
+ "max_action_dim": 132,
23
+ "language_key": "task",
24
+ "formalize_language": true,
25
+ "embodiment_tag": "new_embodiment",
26
+ "embodiment_mapping": {
27
+ "oxe_droid_relative_eef_relative_joint": 24,
28
+ "xdof_relative_eef_relative_joint": 27,
29
+ "xdof_relative_eef_relative_joint_subtask": 27,
30
+ "real_g1_relative_eef_relative_joints": 25,
31
+ "real_r1_pro_sharpa_relative_eef": 26,
32
+ "real_r1_pro_sharpa_relative_eef_human": 26,
33
+ "real_r1_pro_sharpa_relative_eef_maxinsights": 26,
34
+ "real_r1_pro_sharpa_relative_eef_mecka": 26,
35
+ "unitree_g1_full_body_with_waist_height_nav_cmd": 25,
36
+ "simpler_env_google": 0,
37
+ "simpler_env_widowx": 1,
38
+ "libero_sim": 2,
39
+ "new_embodiment": 10
40
+ },
41
+ "normalize_min_max": true,
42
+ "clip_outliers": true,
43
+ "use_percentiles": false,
44
+ "video_modality_keys": null,
45
+ "raw_stats": null,
46
+ "modality_config": null
47
+ },
48
+ "state_file": "policy_preprocessor_step_2_groot_n1_7_pack_inputs_v1.safetensors"
49
+ },
50
+ {
51
+ "registry_name": "groot_n1_7_vlm_encode_v1",
52
+ "config": {
53
+ "model_name": "nvidia/Cosmos-Reason2-2B",
54
+ "image_crop_size": [
55
+ 230,
56
+ 230
57
+ ],
58
+ "image_target_size": [
59
+ 256,
60
+ 256
61
+ ],
62
+ "shortest_image_edge": null,
63
+ "crop_fraction": null,
64
+ "use_albumentations": false,
65
+ "letter_box_transform": false,
66
+ "device": "cuda"
67
+ }
68
+ },
69
+ {
70
+ "registry_name": "device_processor",
71
+ "config": {
72
+ "device": "cuda",
73
+ "float_dtype": null
74
+ }
75
+ }
76
+ ]
77
+ }
policy_preprocessor_step_2_groot_n1_7_pack_inputs_v1.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a58870c13c27d075cec932fdfd557824b033bb3324a456266e197abb924f14c
3
+ size 7312