Hugging Face's logo

ELLLLLLLY
/
grasp_single_block_act_151

Safetensors
Model card Files
xet
 Community
ELLLLLLLY commited on
Commit
d854b8b
·
verified ·
1 Parent(s): a49c7ee

Add files using upload-large-folder tool

Browse files
Files changed (14) hide show
  1. checkpoints/050000/pretrained_model/config.json +71 -0
  2. checkpoints/050000/pretrained_model/model.safetensors +3 -0
  3. checkpoints/050000/pretrained_model/policy_postprocessor.json +32 -0
  4. checkpoints/050000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
  5. checkpoints/050000/pretrained_model/policy_preprocessor.json +64 -0
  6. checkpoints/050000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +3 -0
  7. checkpoints/050000/pretrained_model/train_config.json +203 -0
  8. checkpoints/070000/config.json +71 -0
  9. checkpoints/070000/model.safetensors +3 -0
  10. checkpoints/070000/policy_postprocessor.json +32 -0
  11. checkpoints/070000/policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
  12. checkpoints/070000/policy_preprocessor.json +64 -0
  13. checkpoints/070000/policy_preprocessor_step_3_normalizer_processor.safetensors +3 -0
  14. checkpoints/070000/train_config.json +339 -0
checkpoints/050000/pretrained_model/config.json ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "act",
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
+ 480,
16
+ 640
17
+ ]
18
+ },
19
+ "observation.images.side": {
20
+ "type": "VISUAL",
21
+ "shape": [
22
+ 3,
23
+ 480,
24
+ 640
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
+ "use_peft": false,
39
+ "push_to_hub": true,
40
+ "repo_id": "ELLLLLLLY/grasp_block_151",
41
+ "private": null,
42
+ "tags": null,
43
+ "license": null,
44
+ "pretrained_path": "/home/vipuser/lerobot/outputs/train/grasp_single_block_act_151/checkpoints/last/pretrained_model",
45
+ "chunk_size": 100,
46
+ "n_action_steps": 100,
47
+ "normalization_mapping": {
48
+ "VISUAL": "MEAN_STD",
49
+ "STATE": "MEAN_STD",
50
+ "ACTION": "MEAN_STD"
51
+ },
52
+ "vision_backbone": "resnet18",
53
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
54
+ "replace_final_stride_with_dilation": 0,
55
+ "pre_norm": false,
56
+ "dim_model": 512,
57
+ "n_heads": 8,
58
+ "dim_feedforward": 3200,
59
+ "feedforward_activation": "relu",
60
+ "n_encoder_layers": 4,
61
+ "n_decoder_layers": 1,
62
+ "use_vae": true,
63
+ "latent_dim": 32,
64
+ "n_vae_encoder_layers": 4,
65
+ "temporal_ensemble_coeff": null,
66
+ "dropout": 0.1,
67
+ "kl_weight": 10.0,
68
+ "optimizer_lr": 1e-05,
69
+ "optimizer_weight_decay": 0.0001,
70
+ "optimizer_lr_backbone": 1e-05
71
+ }
checkpoints/050000/pretrained_model/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:98b8abeb484cb2508a46ac45effce39b123f8ef769bfc1a1ad82219cc7ad6156
3
+ size 206699736
checkpoints/050000/pretrained_model/policy_postprocessor.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_postprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "unnormalizer_processor",
6
+ "config": {
7
+ "eps": 1e-08,
8
+ "features": {
9
+ "action": {
10
+ "type": "ACTION",
11
+ "shape": [
12
+ 6
13
+ ]
14
+ }
15
+ },
16
+ "norm_map": {
17
+ "VISUAL": "MEAN_STD",
18
+ "STATE": "MEAN_STD",
19
+ "ACTION": "MEAN_STD"
20
+ }
21
+ },
22
+ "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
23
+ },
24
+ {
25
+ "registry_name": "device_processor",
26
+ "config": {
27
+ "device": "cpu",
28
+ "float_dtype": null
29
+ }
30
+ }
31
+ ]
32
+ }
checkpoints/050000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1178190662fbd47b41c2b58cd1798f7554a47a720b8d9e8ddd8ded391121a9e6
3
+ size 7544
checkpoints/050000/pretrained_model/policy_preprocessor.json ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "device_processor",
16
+ "config": {
17
+ "device": "cuda",
18
+ "float_dtype": null
19
+ }
20
+ },
21
+ {
22
+ "registry_name": "normalizer_processor",
23
+ "config": {
24
+ "eps": 1e-08,
25
+ "features": {
26
+ "observation.state": {
27
+ "type": "STATE",
28
+ "shape": [
29
+ 6
30
+ ]
31
+ },
32
+ "observation.images.front": {
33
+ "type": "VISUAL",
34
+ "shape": [
35
+ 3,
36
+ 480,
37
+ 640
38
+ ]
39
+ },
40
+ "observation.images.side": {
41
+ "type": "VISUAL",
42
+ "shape": [
43
+ 3,
44
+ 480,
45
+ 640
46
+ ]
47
+ },
48
+ "action": {
49
+ "type": "ACTION",
50
+ "shape": [
51
+ 6
52
+ ]
53
+ }
54
+ },
55
+ "norm_map": {
56
+ "VISUAL": "MEAN_STD",
57
+ "STATE": "MEAN_STD",
58
+ "ACTION": "MEAN_STD"
59
+ }
60
+ },
61
+ "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
62
+ }
63
+ ]
64
+ }
checkpoints/050000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1178190662fbd47b41c2b58cd1798f7554a47a720b8d9e8ddd8ded391121a9e6
3
+ size 7544
checkpoints/050000/pretrained_model/train_config.json ADDED
@@ -0,0 +1,203 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": {
3
+ "repo_id": "local/grasp_block_151",
4
+ "root": "/home/vipuser/lerobot/datas/grasp_block_151",
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
+ },
82
+ "env": null,
83
+ "policy": {
84
+ "type": "act",
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
+ 480,
98
+ 640
99
+ ]
100
+ },
101
+ "observation.images.side": {
102
+ "type": "VISUAL",
103
+ "shape": [
104
+ 3,
105
+ 480,
106
+ 640
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
+ "use_peft": false,
121
+ "push_to_hub": true,
122
+ "repo_id": "ELLLLLLLY/grasp_block_151",
123
+ "private": null,
124
+ "tags": null,
125
+ "license": null,
126
+ "pretrained_path": "/home/vipuser/lerobot/outputs/train/grasp_single_block_act_151/checkpoints/last/pretrained_model",
127
+ "chunk_size": 100,
128
+ "n_action_steps": 100,
129
+ "normalization_mapping": {
130
+ "VISUAL": "MEAN_STD",
131
+ "STATE": "MEAN_STD",
132
+ "ACTION": "MEAN_STD"
133
+ },
134
+ "vision_backbone": "resnet18",
135
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
136
+ "replace_final_stride_with_dilation": 0,
137
+ "pre_norm": false,
138
+ "dim_model": 512,
139
+ "n_heads": 8,
140
+ "dim_feedforward": 3200,
141
+ "feedforward_activation": "relu",
142
+ "n_encoder_layers": 4,
143
+ "n_decoder_layers": 1,
144
+ "use_vae": true,
145
+ "latent_dim": 32,
146
+ "n_vae_encoder_layers": 4,
147
+ "temporal_ensemble_coeff": null,
148
+ "dropout": 0.1,
149
+ "kl_weight": 10.0,
150
+ "optimizer_lr": 1e-05,
151
+ "optimizer_weight_decay": 0.0001,
152
+ "optimizer_lr_backbone": 1e-05
153
+ },
154
+ "output_dir": "/home/vipuser/lerobot/outputs/train/grasp_single_block_act_151",
155
+ "job_name": "grasp single block",
156
+ "resume": true,
157
+ "seed": 1000,
158
+ "cudnn_deterministic": false,
159
+ "num_workers": 4,
160
+ "batch_size": 8,
161
+ "steps": 50000,
162
+ "eval_freq": 20000,
163
+ "log_freq": 100,
164
+ "tolerance_s": 0.0001,
165
+ "save_checkpoint": true,
166
+ "save_freq": 10000,
167
+ "use_policy_training_preset": true,
168
+ "optimizer": {
169
+ "type": "adamw",
170
+ "lr": 1e-05,
171
+ "weight_decay": 0.0001,
172
+ "grad_clip_norm": 10.0,
173
+ "betas": [
174
+ 0.9,
175
+ 0.999
176
+ ],
177
+ "eps": 1e-08
178
+ },
179
+ "scheduler": null,
180
+ "eval": {
181
+ "n_episodes": 50,
182
+ "batch_size": 50,
183
+ "use_async_envs": false
184
+ },
185
+ "wandb": {
186
+ "enable": true,
187
+ "disable_artifact": false,
188
+ "project": "lerobot",
189
+ "entity": "eleanorluo99-no",
190
+ "notes": null,
191
+ "run_id": "ckgkbxv1",
192
+ "mode": "online",
193
+ "add_tags": true
194
+ },
195
+ "peft": null,
196
+ "use_rabc": false,
197
+ "rabc_progress_path": null,
198
+ "rabc_kappa": 0.01,
199
+ "rabc_epsilon": 1e-06,
200
+ "rabc_head_mode": "sparse",
201
+ "rename_map": {},
202
+ "checkpoint_path": "/home/vipuser/lerobot/outputs/train/grasp_single_block_act_151/checkpoints/last"
203
+ }
checkpoints/070000/config.json ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "act",
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
+ 480,
16
+ 640
17
+ ]
18
+ },
19
+ "observation.images.side": {
20
+ "type": "VISUAL",
21
+ "shape": [
22
+ 3,
23
+ 480,
24
+ 640
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
+ "use_peft": false,
39
+ "push_to_hub": false,
40
+ "repo_id": null,
41
+ "private": null,
42
+ "tags": null,
43
+ "license": null,
44
+ "pretrained_path": "/home/vipuser/lerobot/outputs/train/grasp_block_151_act_holdout10_seed1000/checkpoints/last/pretrained_model",
45
+ "chunk_size": 100,
46
+ "n_action_steps": 100,
47
+ "normalization_mapping": {
48
+ "VISUAL": "MEAN_STD",
49
+ "STATE": "MEAN_STD",
50
+ "ACTION": "MEAN_STD"
51
+ },
52
+ "vision_backbone": "resnet18",
53
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
54
+ "replace_final_stride_with_dilation": 0,
55
+ "pre_norm": false,
56
+ "dim_model": 512,
57
+ "n_heads": 8,
58
+ "dim_feedforward": 3200,
59
+ "feedforward_activation": "relu",
60
+ "n_encoder_layers": 4,
61
+ "n_decoder_layers": 1,
62
+ "use_vae": true,
63
+ "latent_dim": 32,
64
+ "n_vae_encoder_layers": 4,
65
+ "temporal_ensemble_coeff": null,
66
+ "dropout": 0.1,
67
+ "kl_weight": 10.0,
68
+ "optimizer_lr": 1e-05,
69
+ "optimizer_weight_decay": 0.0001,
70
+ "optimizer_lr_backbone": 1e-05
71
+ }
checkpoints/070000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:80f72ed1bb9e05f1e04adbf5ddb5e2971ef1d72ba645fbfe24cb7b4ed30aa996
3
+ size 206699736
checkpoints/070000/policy_postprocessor.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_postprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "unnormalizer_processor",
6
+ "config": {
7
+ "eps": 1e-08,
8
+ "features": {
9
+ "action": {
10
+ "type": "ACTION",
11
+ "shape": [
12
+ 6
13
+ ]
14
+ }
15
+ },
16
+ "norm_map": {
17
+ "VISUAL": "MEAN_STD",
18
+ "STATE": "MEAN_STD",
19
+ "ACTION": "MEAN_STD"
20
+ }
21
+ },
22
+ "state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
23
+ },
24
+ {
25
+ "registry_name": "device_processor",
26
+ "config": {
27
+ "device": "cpu",
28
+ "float_dtype": null
29
+ }
30
+ }
31
+ ]
32
+ }
checkpoints/070000/policy_postprocessor_step_0_unnormalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2b74fd46170b993a4fd3398e4c5ca7c3e4411958930423889a1db7d2a3dc5383
3
+ size 7544
checkpoints/070000/policy_preprocessor.json ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "device_processor",
16
+ "config": {
17
+ "device": "cuda",
18
+ "float_dtype": null
19
+ }
20
+ },
21
+ {
22
+ "registry_name": "normalizer_processor",
23
+ "config": {
24
+ "eps": 1e-08,
25
+ "features": {
26
+ "observation.state": {
27
+ "type": "STATE",
28
+ "shape": [
29
+ 6
30
+ ]
31
+ },
32
+ "observation.images.front": {
33
+ "type": "VISUAL",
34
+ "shape": [
35
+ 3,
36
+ 480,
37
+ 640
38
+ ]
39
+ },
40
+ "observation.images.side": {
41
+ "type": "VISUAL",
42
+ "shape": [
43
+ 3,
44
+ 480,
45
+ 640
46
+ ]
47
+ },
48
+ "action": {
49
+ "type": "ACTION",
50
+ "shape": [
51
+ 6
52
+ ]
53
+ }
54
+ },
55
+ "norm_map": {
56
+ "VISUAL": "MEAN_STD",
57
+ "STATE": "MEAN_STD",
58
+ "ACTION": "MEAN_STD"
59
+ }
60
+ },
61
+ "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
62
+ }
63
+ ]
64
+ }
checkpoints/070000/policy_preprocessor_step_3_normalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2b74fd46170b993a4fd3398e4c5ca7c3e4411958930423889a1db7d2a3dc5383
3
+ size 7544
checkpoints/070000/train_config.json ADDED
@@ -0,0 +1,339 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": {
3
+ "repo_id": "local/grasp_block_151_split_seed1000",
4
+ "root": "/home/vipuser/lerobot/datas/grasp_block_151_split_seed1000",
5
+ "episodes": [
6
+ 0,
7
+ 1,
8
+ 2,
9
+ 3,
10
+ 4,
11
+ 5,
12
+ 6,
13
+ 7,
14
+ 8,
15
+ 9,
16
+ 10,
17
+ 11,
18
+ 12,
19
+ 13,
20
+ 14,
21
+ 15,
22
+ 17,
23
+ 18,
24
+ 19,
25
+ 20,
26
+ 21,
27
+ 22,
28
+ 23,
29
+ 24,
30
+ 26,
31
+ 27,
32
+ 28,
33
+ 29,
34
+ 30,
35
+ 31,
36
+ 32,
37
+ 34,
38
+ 35,
39
+ 36,
40
+ 37,
41
+ 38,
42
+ 39,
43
+ 40,
44
+ 41,
45
+ 43,
46
+ 44,
47
+ 45,
48
+ 46,
49
+ 47,
50
+ 48,
51
+ 49,
52
+ 50,
53
+ 52,
54
+ 53,
55
+ 54,
56
+ 55,
57
+ 57,
58
+ 58,
59
+ 59,
60
+ 60,
61
+ 62,
62
+ 63,
63
+ 64,
64
+ 65,
65
+ 66,
66
+ 67,
67
+ 68,
68
+ 69,
69
+ 70,
70
+ 71,
71
+ 72,
72
+ 73,
73
+ 74,
74
+ 75,
75
+ 76,
76
+ 77,
77
+ 78,
78
+ 79,
79
+ 80,
80
+ 81,
81
+ 82,
82
+ 83,
83
+ 84,
84
+ 85,
85
+ 86,
86
+ 87,
87
+ 88,
88
+ 89,
89
+ 91,
90
+ 92,
91
+ 94,
92
+ 95,
93
+ 96,
94
+ 97,
95
+ 98,
96
+ 99,
97
+ 101,
98
+ 102,
99
+ 103,
100
+ 104,
101
+ 105,
102
+ 106,
103
+ 107,
104
+ 108,
105
+ 110,
106
+ 112,
107
+ 113,
108
+ 114,
109
+ 115,
110
+ 116,
111
+ 117,
112
+ 118,
113
+ 120,
114
+ 121,
115
+ 122,
116
+ 123,
117
+ 124,
118
+ 126,
119
+ 127,
120
+ 128,
121
+ 129,
122
+ 130,
123
+ 131,
124
+ 132,
125
+ 133,
126
+ 134,
127
+ 135,
128
+ 137,
129
+ 138,
130
+ 139,
131
+ 140,
132
+ 141,
133
+ 142,
134
+ 143,
135
+ 144,
136
+ 145,
137
+ 146,
138
+ 147,
139
+ 148,
140
+ 149
141
+ ],
142
+ "image_transforms": {
143
+ "enable": false,
144
+ "max_num_transforms": 3,
145
+ "random_order": false,
146
+ "tfs": {
147
+ "brightness": {
148
+ "weight": 1.0,
149
+ "type": "ColorJitter",
150
+ "kwargs": {
151
+ "brightness": [
152
+ 0.8,
153
+ 1.2
154
+ ]
155
+ }
156
+ },
157
+ "contrast": {
158
+ "weight": 1.0,
159
+ "type": "ColorJitter",
160
+ "kwargs": {
161
+ "contrast": [
162
+ 0.8,
163
+ 1.2
164
+ ]
165
+ }
166
+ },
167
+ "saturation": {
168
+ "weight": 1.0,
169
+ "type": "ColorJitter",
170
+ "kwargs": {
171
+ "saturation": [
172
+ 0.5,
173
+ 1.5
174
+ ]
175
+ }
176
+ },
177
+ "hue": {
178
+ "weight": 1.0,
179
+ "type": "ColorJitter",
180
+ "kwargs": {
181
+ "hue": [
182
+ -0.05,
183
+ 0.05
184
+ ]
185
+ }
186
+ },
187
+ "sharpness": {
188
+ "weight": 1.0,
189
+ "type": "SharpnessJitter",
190
+ "kwargs": {
191
+ "sharpness": [
192
+ 0.5,
193
+ 1.5
194
+ ]
195
+ }
196
+ },
197
+ "affine": {
198
+ "weight": 1.0,
199
+ "type": "RandomAffine",
200
+ "kwargs": {
201
+ "degrees": [
202
+ -5.0,
203
+ 5.0
204
+ ],
205
+ "translate": [
206
+ 0.05,
207
+ 0.05
208
+ ]
209
+ }
210
+ }
211
+ }
212
+ },
213
+ "revision": null,
214
+ "use_imagenet_stats": true,
215
+ "video_backend": "torchcodec",
216
+ "streaming": false
217
+ },
218
+ "env": null,
219
+ "policy": {
220
+ "type": "act",
221
+ "n_obs_steps": 1,
222
+ "input_features": {
223
+ "observation.state": {
224
+ "type": "STATE",
225
+ "shape": [
226
+ 6
227
+ ]
228
+ },
229
+ "observation.images.front": {
230
+ "type": "VISUAL",
231
+ "shape": [
232
+ 3,
233
+ 480,
234
+ 640
235
+ ]
236
+ },
237
+ "observation.images.side": {
238
+ "type": "VISUAL",
239
+ "shape": [
240
+ 3,
241
+ 480,
242
+ 640
243
+ ]
244
+ }
245
+ },
246
+ "output_features": {
247
+ "action": {
248
+ "type": "ACTION",
249
+ "shape": [
250
+ 6
251
+ ]
252
+ }
253
+ },
254
+ "device": "cuda",
255
+ "use_amp": false,
256
+ "use_peft": false,
257
+ "push_to_hub": false,
258
+ "repo_id": null,
259
+ "private": null,
260
+ "tags": null,
261
+ "license": null,
262
+ "pretrained_path": "/home/vipuser/lerobot/outputs/train/grasp_block_151_act_holdout10_seed1000/checkpoints/last/pretrained_model",
263
+ "chunk_size": 100,
264
+ "n_action_steps": 100,
265
+ "normalization_mapping": {
266
+ "VISUAL": "MEAN_STD",
267
+ "STATE": "MEAN_STD",
268
+ "ACTION": "MEAN_STD"
269
+ },
270
+ "vision_backbone": "resnet18",
271
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
272
+ "replace_final_stride_with_dilation": 0,
273
+ "pre_norm": false,
274
+ "dim_model": 512,
275
+ "n_heads": 8,
276
+ "dim_feedforward": 3200,
277
+ "feedforward_activation": "relu",
278
+ "n_encoder_layers": 4,
279
+ "n_decoder_layers": 1,
280
+ "use_vae": true,
281
+ "latent_dim": 32,
282
+ "n_vae_encoder_layers": 4,
283
+ "temporal_ensemble_coeff": null,
284
+ "dropout": 0.1,
285
+ "kl_weight": 10.0,
286
+ "optimizer_lr": 1e-05,
287
+ "optimizer_weight_decay": 0.0001,
288
+ "optimizer_lr_backbone": 1e-05
289
+ },
290
+ "output_dir": "/home/vipuser/lerobot/outputs/train/grasp_block_151_act_holdout10_seed1000",
291
+ "job_name": "grasp_block_151_act_holdout10_seed1000",
292
+ "resume": true,
293
+ "seed": 1000,
294
+ "cudnn_deterministic": false,
295
+ "num_workers": 4,
296
+ "batch_size": 16,
297
+ "steps": 70000,
298
+ "eval_freq": 0,
299
+ "log_freq": 100,
300
+ "tolerance_s": 0.0001,
301
+ "save_checkpoint": true,
302
+ "save_freq": 5000,
303
+ "use_policy_training_preset": true,
304
+ "optimizer": {
305
+ "type": "adamw",
306
+ "lr": 1e-05,
307
+ "weight_decay": 0.0001,
308
+ "grad_clip_norm": 10.0,
309
+ "betas": [
310
+ 0.9,
311
+ 0.999
312
+ ],
313
+ "eps": 1e-08
314
+ },
315
+ "scheduler": null,
316
+ "eval": {
317
+ "n_episodes": 50,
318
+ "batch_size": 50,
319
+ "use_async_envs": false
320
+ },
321
+ "wandb": {
322
+ "enable": true,
323
+ "disable_artifact": true,
324
+ "project": "lerobot",
325
+ "entity": "eleanorluo99-no",
326
+ "notes": null,
327
+ "run_id": "f3bifs80",
328
+ "mode": "online",
329
+ "add_tags": true
330
+ },
331
+ "peft": null,
332
+ "use_rabc": false,
333
+ "rabc_progress_path": null,
334
+ "rabc_kappa": 0.01,
335
+ "rabc_epsilon": 1e-06,
336
+ "rabc_head_mode": "sparse",
337
+ "rename_map": {},
338
+ "checkpoint_path": "/home/vipuser/lerobot/outputs/train/grasp_block_151_act_holdout10_seed1000/checkpoints/last"
339
+ }