Hugging Face's logo

lbxa
/
so101_gr00t_rubix_stack

Safetensors
Model card Files
xet
Community
lbxa commited on
Commit
7488955
·
verified ·
1 Parent(s): a75b2d6

Delete pretrained_model

Browse files
Files changed (7) hide show
  1. pretrained_model/config.json +0 -88
  2. pretrained_model/model.safetensors +0 -3
  3. pretrained_model/policy_postprocessor.json +0 -20
  4. pretrained_model/policy_postprocessor_step_0_groot_action_unpack_unnormalize_v1.safetensors +0 -3
  5. pretrained_model/policy_preprocessor.json +0 -52
  6. pretrained_model/policy_preprocessor_step_2_groot_pack_inputs_v3.safetensors +0 -3
  7. pretrained_model/train_config.json +0 -217
pretrained_model/config.json DELETED
@@ -1,88 +0,0 @@
1
- {
2
- "type": "groot",
3
- "n_obs_steps": 1,
4
- "input_features": {
5
- "observation.state": {
6
- "type": "STATE",
7
- "shape": [
8
- 6
9
- ]
10
- },
11
- "observation.images.front": {
12
- "type": "VISUAL",
13
- "shape": [
14
- 3,
15
- 720,
16
- 1280
17
- ]
18
- },
19
- "observation.images.top": {
20
- "type": "VISUAL",
21
- "shape": [
22
- 3,
23
- 1080,
24
- 1920
25
- ]
26
- }
27
- },
28
- "output_features": {
29
- "action": {
30
- "type": "ACTION",
31
- "shape": [
32
- 6
33
- ]
34
- }
35
- },
36
- "device": "cuda",
37
- "use_amp": false,
38
- "push_to_hub": true,
39
- "repo_id": "lbxa/tiny",
40
- "private": null,
41
- "tags": null,
42
- "license": null,
43
- "pretrained_path": null,
44
- "chunk_size": 50,
45
- "n_action_steps": 50,
46
- "max_state_dim": 64,
47
- "max_action_dim": 32,
48
- "normalization_mapping": {
49
- "VISUAL": "IDENTITY",
50
- "STATE": "MEAN_STD",
51
- "ACTION": "MEAN_STD"
52
- },
53
- "image_size": [
54
- 224,
55
- 224
56
- ],
57
- "base_model_path": "nvidia/GR00T-N1.5-3B",
58
- "tokenizer_assets_repo": "lerobot/eagle2hg-processor-groot-n1p5",
59
- "embodiment_tag": "new_embodiment",
60
- "tune_llm": false,
61
- "tune_visual": false,
62
- "tune_projector": true,
63
- "tune_diffusion_model": true,
64
- "lora_rank": 0,
65
- "lora_alpha": 16,
66
- "lora_dropout": 0.1,
67
- "lora_full_model": false,
68
- "optimizer_lr": 0.0001,
69
- "optimizer_betas": [
70
- 0.95,
71
- 0.999
72
- ],
73
- "optimizer_eps": 1e-08,
74
- "optimizer_weight_decay": 1e-05,
75
- "warmup_ratio": 0.05,
76
- "use_bf16": true,
77
- "video_backend": "decord",
78
- "balance_dataset_weights": true,
79
- "balance_trajectory_weights": true,
80
- "dataset_paths": null,
81
- "output_dir": "./tmp/gr00t",
82
- "save_steps": 1000,
83
- "max_steps": 10000,
84
- "batch_size": 32,
85
- "dataloader_num_workers": 8,
86
- "report_to": "wandb",
87
- "resume": false
88
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
pretrained_model/model.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:b64dcacdacaa6728b7cb2b0cc9fd6bf7708617b1335cef5b553ccb906e8f85c6
3
- size 6964802952
 
 
 
 
pretrained_model/policy_postprocessor.json DELETED
@@ -1,20 +0,0 @@
1
- {
2
- "name": "policy_postprocessor",
3
- "steps": [
4
- {
5
- "registry_name": "groot_action_unpack_unnormalize_v1",
6
- "config": {
7
- "env_action_dim": 6,
8
- "normalize_min_max": true
9
- },
10
- "state_file": "policy_postprocessor_step_0_groot_action_unpack_unnormalize_v1.safetensors"
11
- },
12
- {
13
- "registry_name": "device_processor",
14
- "config": {
15
- "device": "cpu",
16
- "float_dtype": null
17
- }
18
- }
19
- ]
20
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
pretrained_model/policy_postprocessor_step_0_groot_action_unpack_unnormalize_v1.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:b2f9c682187ab9b47ce47e091cb44cc0e1ea37e0d35303cbd2ee92b304f19cf8
3
- size 8400
 
 
 
 
pretrained_model/policy_preprocessor.json DELETED
@@ -1,52 +0,0 @@
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_pack_inputs_v3",
16
- "config": {
17
- "state_horizon": 1,
18
- "action_horizon": 16,
19
- "max_state_dim": 64,
20
- "max_action_dim": 32,
21
- "language_key": "task",
22
- "formalize_language": false,
23
- "embodiment_tag": "new_embodiment",
24
- "embodiment_mapping": {
25
- "new_embodiment": 31,
26
- "oxe_droid": 17,
27
- "agibot_genie1": 26,
28
- "gr1": 24,
29
- "so100": 2,
30
- "unitree_g1": 3
31
- },
32
- "normalize_min_max": true
33
- },
34
- "state_file": "policy_preprocessor_step_2_groot_pack_inputs_v3.safetensors"
35
- },
36
- {
37
- "registry_name": "groot_eagle_encode_v3",
38
- "config": {}
39
- },
40
- {
41
- "registry_name": "groot_eagle_collate_v3",
42
- "config": {}
43
- },
44
- {
45
- "registry_name": "device_processor",
46
- "config": {
47
- "device": "cuda",
48
- "float_dtype": null
49
- }
50
- }
51
- ]
52
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
pretrained_model/policy_preprocessor_step_2_groot_pack_inputs_v3.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:b2f9c682187ab9b47ce47e091cb44cc0e1ea37e0d35303cbd2ee92b304f19cf8
3
- size 8400
 
 
 
 
pretrained_model/train_config.json DELETED
@@ -1,217 +0,0 @@
1
- {
2
- "dataset": {
3
- "repo_id": "lbxa/rubix_stack",
4
- "root": null,
5
- "episodes": null,
6
- "image_transforms": {
7
- "enable": false,
8
- "max_num_transforms": 3,
9
- "random_order": false,
10
- "tfs": {
11
- "brightness": {
12
- "weight": 1.0,
13
- "type": "ColorJitter",
14
- "kwargs": {
15
- "brightness": [
16
- 0.8,
17
- 1.2
18
- ]
19
- }
20
- },
21
- "contrast": {
22
- "weight": 1.0,
23
- "type": "ColorJitter",
24
- "kwargs": {
25
- "contrast": [
26
- 0.8,
27
- 1.2
28
- ]
29
- }
30
- },
31
- "saturation": {
32
- "weight": 1.0,
33
- "type": "ColorJitter",
34
- "kwargs": {
35
- "saturation": [
36
- 0.5,
37
- 1.5
38
- ]
39
- }
40
- },
41
- "hue": {
42
- "weight": 1.0,
43
- "type": "ColorJitter",
44
- "kwargs": {
45
- "hue": [
46
- -0.05,
47
- 0.05
48
- ]
49
- }
50
- },
51
- "sharpness": {
52
- "weight": 1.0,
53
- "type": "SharpnessJitter",
54
- "kwargs": {
55
- "sharpness": [
56
- 0.5,
57
- 1.5
58
- ]
59
- }
60
- },
61
- "affine": {
62
- "weight": 1.0,
63
- "type": "RandomAffine",
64
- "kwargs": {
65
- "degrees": [
66
- -5.0,
67
- 5.0
68
- ],
69
- "translate": [
70
- 0.05,
71
- 0.05
72
- ]
73
- }
74
- }
75
- }
76
- },
77
- "revision": null,
78
- "use_imagenet_stats": true,
79
- "video_backend": "pyav",
80
- "streaming": false
81
- },
82
- "env": null,
83
- "policy": {
84
- "type": "groot",
85
- "n_obs_steps": 1,
86
- "input_features": {
87
- "observation.state": {
88
- "type": "STATE",
89
- "shape": [
90
- 6
91
- ]
92
- },
93
- "observation.images.front": {
94
- "type": "VISUAL",
95
- "shape": [
96
- 3,
97
- 720,
98
- 1280
99
- ]
100
- },
101
- "observation.images.top": {
102
- "type": "VISUAL",
103
- "shape": [
104
- 3,
105
- 1080,
106
- 1920
107
- ]
108
- }
109
- },
110
- "output_features": {
111
- "action": {
112
- "type": "ACTION",
113
- "shape": [
114
- 6
115
- ]
116
- }
117
- },
118
- "device": "cuda",
119
- "use_amp": false,
120
- "push_to_hub": true,
121
- "repo_id": "lbxa/tiny",
122
- "private": null,
123
- "tags": null,
124
- "license": null,
125
- "pretrained_path": null,
126
- "chunk_size": 50,
127
- "n_action_steps": 50,
128
- "max_state_dim": 64,
129
- "max_action_dim": 32,
130
- "normalization_mapping": {
131
- "VISUAL": "IDENTITY",
132
- "STATE": "MEAN_STD",
133
- "ACTION": "MEAN_STD"
134
- },
135
- "image_size": [
136
- 224,
137
- 224
138
- ],
139
- "base_model_path": "nvidia/GR00T-N1.5-3B",
140
- "tokenizer_assets_repo": "lerobot/eagle2hg-processor-groot-n1p5",
141
- "embodiment_tag": "new_embodiment",
142
- "tune_llm": false,
143
- "tune_visual": false,
144
- "tune_projector": true,
145
- "tune_diffusion_model": true,
146
- "lora_rank": 0,
147
- "lora_alpha": 16,
148
- "lora_dropout": 0.1,
149
- "lora_full_model": false,
150
- "optimizer_lr": 0.0001,
151
- "optimizer_betas": [
152
- 0.95,
153
- 0.999
154
- ],
155
- "optimizer_eps": 1e-08,
156
- "optimizer_weight_decay": 1e-05,
157
- "warmup_ratio": 0.05,
158
- "use_bf16": true,
159
- "video_backend": "decord",
160
- "balance_dataset_weights": true,
161
- "balance_trajectory_weights": true,
162
- "dataset_paths": null,
163
- "output_dir": "./tmp/gr00t",
164
- "save_steps": 1000,
165
- "max_steps": 10000,
166
- "batch_size": 32,
167
- "dataloader_num_workers": 8,
168
- "report_to": "wandb",
169
- "resume": false
170
- },
171
- "output_dir": "/outputs/train/so101_gr00t_rubix_stack_2025-11-25",
172
- "job_name": "so101_gr00t_rubix_stack_2025-11-25",
173
- "resume": false,
174
- "seed": 1000,
175
- "num_workers": 4,
176
- "batch_size": 8,
177
- "steps": 15000,
178
- "eval_freq": 2000,
179
- "log_freq": 100,
180
- "save_checkpoint": true,
181
- "save_freq": 2000,
182
- "use_policy_training_preset": true,
183
- "optimizer": {
184
- "type": "adamw",
185
- "lr": 0.0001,
186
- "weight_decay": 1e-05,
187
- "grad_clip_norm": 10.0,
188
- "betas": [
189
- 0.95,
190
- 0.999
191
- ],
192
- "eps": 1e-08
193
- },
194
- "scheduler": {
195
- "type": "cosine_decay_with_warmup",
196
- "num_warmup_steps": 500,
197
- "num_decay_steps": 10000,
198
- "peak_lr": 0.0001,
199
- "decay_lr": 1e-05
200
- },
201
- "eval": {
202
- "n_episodes": 50,
203
- "batch_size": 50,
204
- "use_async_envs": false
205
- },
206
- "wandb": {
207
- "enable": true,
208
- "disable_artifact": false,
209
- "project": "lerobot",
210
- "entity": null,
211
- "notes": null,
212
- "run_id": "57ca17kq",
213
- "mode": null
214
- },
215
- "checkpoint_path": null,
216
- "rename_map": {}
217
- }