Hugging Face's logo

ywu67
/
b1k-task0000

Safetensors
Model card Files
xet
Community
ywu67 commited on
Commit
6b3f724
·
verified ·
1 Parent(s): 919b058

Clean root directory - remove train_config.json

Browse files
Files changed (1) hide show
  1. train_config.json +0 -262
train_config.json DELETED
@@ -1,262 +0,0 @@
1
- {
2
- "dataset": {
3
- "repo_id": "ywu67/2025-challenge-demos-task0000",
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": "torchcodec",
80
- "streaming": false,
81
- "tolerance_s": 0.0001
82
- },
83
- "env": null,
84
- "policy": {
85
- "type": "act",
86
- "n_obs_steps": 1,
87
- "input_features": {
88
- "observation.images.rgb.left_wrist": {
89
- "type": "VISUAL",
90
- "shape": [
91
- 480,
92
- 480,
93
- 3
94
- ]
95
- },
96
- "observation.images.rgb.right_wrist": {
97
- "type": "VISUAL",
98
- "shape": [
99
- 480,
100
- 480,
101
- 3
102
- ]
103
- },
104
- "observation.images.rgb.head": {
105
- "type": "VISUAL",
106
- "shape": [
107
- 720,
108
- 720,
109
- 3
110
- ]
111
- },
112
- "observation.images.depth.left_wrist": {
113
- "type": "VISUAL",
114
- "shape": [
115
- 480,
116
- 480,
117
- 3
118
- ]
119
- },
120
- "observation.images.depth.right_wrist": {
121
- "type": "VISUAL",
122
- "shape": [
123
- 480,
124
- 480,
125
- 3
126
- ]
127
- },
128
- "observation.images.depth.head": {
129
- "type": "VISUAL",
130
- "shape": [
131
- 720,
132
- 720,
133
- 3
134
- ]
135
- },
136
- "observation.images.seg_instance_id.left_wrist": {
137
- "type": "VISUAL",
138
- "shape": [
139
- 480,
140
- 480,
141
- 3
142
- ]
143
- },
144
- "observation.images.seg_instance_id.right_wrist": {
145
- "type": "VISUAL",
146
- "shape": [
147
- 480,
148
- 480,
149
- 3
150
- ]
151
- },
152
- "observation.images.seg_instance_id.head": {
153
- "type": "VISUAL",
154
- "shape": [
155
- 720,
156
- 720,
157
- 3
158
- ]
159
- },
160
- "observation.cam_rel_poses": {
161
- "type": "STATE",
162
- "shape": [
163
- 21
164
- ]
165
- },
166
- "observation.state": {
167
- "type": "STATE",
168
- "shape": [
169
- 256
170
- ]
171
- },
172
- "observation.task_info": {
173
- "type": "STATE",
174
- "shape": [
175
- 46
176
- ]
177
- }
178
- },
179
- "output_features": {
180
- "action": {
181
- "type": "ACTION",
182
- "shape": [
183
- 23
184
- ]
185
- }
186
- },
187
- "device": "cuda",
188
- "use_amp": true,
189
- "push_to_hub": true,
190
- "repo_id": "ywu67/b1k-task0000",
191
- "private": null,
192
- "tags": null,
193
- "license": null,
194
- "pretrained_path": null,
195
- "chunk_size": 100,
196
- "n_action_steps": 100,
197
- "normalization_mapping": {
198
- "VISUAL": "MEAN_STD",
199
- "STATE": "MEAN_STD",
200
- "ACTION": "MEAN_STD"
201
- },
202
- "vision_backbone": "resnet18",
203
- "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
204
- "replace_final_stride_with_dilation": false,
205
- "pre_norm": false,
206
- "dim_model": 512,
207
- "n_heads": 8,
208
- "dim_feedforward": 3200,
209
- "feedforward_activation": "relu",
210
- "n_encoder_layers": 4,
211
- "n_decoder_layers": 1,
212
- "use_vae": true,
213
- "latent_dim": 32,
214
- "n_vae_encoder_layers": 4,
215
- "temporal_ensemble_coeff": null,
216
- "dropout": 0.1,
217
- "kl_weight": 10.0,
218
- "optimizer_lr": 1e-05,
219
- "optimizer_weight_decay": 0.0001,
220
- "optimizer_lr_backbone": 1e-05
221
- },
222
- "output_dir": "outputs/task_0000_training",
223
- "job_name": "act",
224
- "resume": false,
225
- "seed": 1000,
226
- "num_workers": 12,
227
- "batch_size": 16,
228
- "steps": 100000,
229
- "eval_freq": 20000,
230
- "log_freq": 200,
231
- "save_checkpoint": true,
232
- "save_freq": 5000,
233
- "use_policy_training_preset": true,
234
- "optimizer": {
235
- "type": "adamw",
236
- "lr": 1e-05,
237
- "weight_decay": 0.0001,
238
- "grad_clip_norm": 10.0,
239
- "betas": [
240
- 0.9,
241
- 0.999
242
- ],
243
- "eps": 1e-08
244
- },
245
- "scheduler": null,
246
- "eval": {
247
- "n_episodes": 50,
248
- "batch_size": 50,
249
- "use_async_envs": false
250
- },
251
- "wandb": {
252
- "enable": true,
253
- "disable_artifact": false,
254
- "project": "lerobot-2025-challenge",
255
- "entity": null,
256
- "notes": "Training with torchcodec, batch_size=16, num_workers=12",
257
- "run_id": "130gsib5",
258
- "mode": null
259
- },
260
- "checkpoint_path": null,
261
- "rename_map": {}
262
- }