Robotics
LeRobot
Safetensors
diffusion
112550099NYCU commited on
Commit
7ff435b
·
verified ·
1 Parent(s): acc270d

Delete old training run folder training_runs/act_dining_cleanup_v2_2_20260609_033312

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/pretrained_model/config.json +0 -92
  2. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/pretrained_model/model.safetensors +0 -3
  3. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/pretrained_model/policy_postprocessor.json +0 -32
  4. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +0 -3
  5. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/pretrained_model/policy_preprocessor.json +0 -64
  6. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +0 -3
  7. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/pretrained_model/train_config.json +0 -226
  8. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/training_state/optimizer_param_groups.json +0 -233
  9. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/training_state/optimizer_state.safetensors +0 -3
  10. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/training_state/rng_state.safetensors +0 -3
  11. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/training_state/scheduler_state.json +0 -15
  12. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/training_state/training_step.json +0 -3
  13. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/pretrained_model/config.json +0 -92
  14. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/pretrained_model/model.safetensors +0 -3
  15. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/pretrained_model/policy_postprocessor.json +0 -32
  16. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +0 -3
  17. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/pretrained_model/policy_preprocessor.json +0 -64
  18. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +0 -3
  19. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/pretrained_model/train_config.json +0 -226
  20. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/training_state/optimizer_param_groups.json +0 -233
  21. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/training_state/optimizer_state.safetensors +0 -3
  22. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/training_state/rng_state.safetensors +0 -3
  23. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/training_state/scheduler_state.json +0 -15
  24. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/training_state/training_step.json +0 -3
  25. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/pretrained_model/config.json +0 -92
  26. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/pretrained_model/model.safetensors +0 -3
  27. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/pretrained_model/policy_postprocessor.json +0 -32
  28. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +0 -3
  29. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/pretrained_model/policy_preprocessor.json +0 -64
  30. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +0 -3
  31. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/pretrained_model/train_config.json +0 -226
  32. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/training_state/optimizer_param_groups.json +0 -233
  33. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/training_state/optimizer_state.safetensors +0 -3
  34. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/training_state/rng_state.safetensors +0 -3
  35. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/training_state/scheduler_state.json +0 -15
  36. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/training_state/training_step.json +0 -3
  37. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/pretrained_model/config.json +0 -92
  38. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/pretrained_model/model.safetensors +0 -3
  39. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/pretrained_model/policy_postprocessor.json +0 -32
  40. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +0 -3
  41. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/pretrained_model/policy_preprocessor.json +0 -64
  42. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +0 -3
  43. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/pretrained_model/train_config.json +0 -226
  44. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/training_state/optimizer_param_groups.json +0 -233
  45. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/training_state/optimizer_state.safetensors +0 -3
  46. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/training_state/rng_state.safetensors +0 -3
  47. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/training_state/scheduler_state.json +0 -15
  48. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/training_state/training_step.json +0 -3
  49. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/100000/pretrained_model/config.json +0 -92
  50. training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/100000/pretrained_model/model.safetensors +0 -3
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/pretrained_model/config.json DELETED
@@ -1,92 +0,0 @@
1
- {
2
- "type": "diffusion",
3
- "n_obs_steps": 2,
4
- "input_features": {
5
- "observation.state": {
6
- "type": "STATE",
7
- "shape": [
8
- 9
9
- ]
10
- },
11
- "observation.images.wrist": {
12
- "type": "VISUAL",
13
- "shape": [
14
- 3,
15
- 480,
16
- 640
17
- ]
18
- },
19
- "observation.images.front": {
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
- 8
33
- ]
34
- }
35
- },
36
- "device": "cuda",
37
- "use_amp": false,
38
- "use_peft": false,
39
- "push_to_hub": true,
40
- "repo_id": "AI-Final/advanced-diffusion-v3",
41
- "private": null,
42
- "tags": null,
43
- "license": null,
44
- "pretrained_path": null,
45
- "horizon": 16,
46
- "n_action_steps": 8,
47
- "normalization_mapping": {
48
- "VISUAL": "MEAN_STD",
49
- "STATE": "MIN_MAX",
50
- "ACTION": "MIN_MAX"
51
- },
52
- "drop_n_last_frames": 7,
53
- "vision_backbone": "resnet18",
54
- "resize_shape": null,
55
- "crop_ratio": 1.0,
56
- "crop_shape": null,
57
- "crop_is_random": true,
58
- "pretrained_backbone_weights": null,
59
- "use_group_norm": true,
60
- "spatial_softmax_num_keypoints": 32,
61
- "use_separate_rgb_encoder_per_camera": false,
62
- "down_dims": [
63
- 512,
64
- 1024,
65
- 2048
66
- ],
67
- "kernel_size": 5,
68
- "n_groups": 8,
69
- "diffusion_step_embed_dim": 128,
70
- "use_film_scale_modulation": true,
71
- "noise_scheduler_type": "DDPM",
72
- "num_train_timesteps": 100,
73
- "beta_schedule": "squaredcos_cap_v2",
74
- "beta_start": 0.0001,
75
- "beta_end": 0.02,
76
- "prediction_type": "epsilon",
77
- "clip_sample": true,
78
- "clip_sample_range": 1.0,
79
- "num_inference_steps": null,
80
- "compile_model": false,
81
- "compile_mode": "reduce-overhead",
82
- "do_mask_loss_for_padding": false,
83
- "optimizer_lr": 0.0001,
84
- "optimizer_betas": [
85
- 0.95,
86
- 0.999
87
- ],
88
- "optimizer_eps": 1e-08,
89
- "optimizer_weight_decay": 1e-06,
90
- "scheduler_name": "cosine",
91
- "scheduler_warmup_steps": 500
92
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/pretrained_model/model.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:bd803a592fc7659e0ccefa374fb48a9cfb9c5e3802e22316ac8606b6048dc64e
3
- size 1067235520
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/pretrained_model/policy_postprocessor.json DELETED
@@ -1,32 +0,0 @@
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
- 8
13
- ]
14
- }
15
- },
16
- "norm_map": {
17
- "VISUAL": "MEAN_STD",
18
- "STATE": "MIN_MAX",
19
- "ACTION": "MIN_MAX"
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
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:86b34463feeba35ffc7fce28f390a6fdd072dd2d8036f7a62d3dc0de5cc7a3f6
3
- size 7756
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/pretrained_model/policy_preprocessor.json DELETED
@@ -1,64 +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": "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
- 9
30
- ]
31
- },
32
- "observation.images.wrist": {
33
- "type": "VISUAL",
34
- "shape": [
35
- 3,
36
- 480,
37
- 640
38
- ]
39
- },
40
- "observation.images.front": {
41
- "type": "VISUAL",
42
- "shape": [
43
- 3,
44
- 480,
45
- 640
46
- ]
47
- },
48
- "action": {
49
- "type": "ACTION",
50
- "shape": [
51
- 8
52
- ]
53
- }
54
- },
55
- "norm_map": {
56
- "VISUAL": "MEAN_STD",
57
- "STATE": "MIN_MAX",
58
- "ACTION": "MIN_MAX"
59
- }
60
- },
61
- "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
62
- }
63
- ]
64
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:86b34463feeba35ffc7fce28f390a6fdd072dd2d8036f7a62d3dc0de5cc7a3f6
3
- size 7756
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/pretrained_model/train_config.json DELETED
@@ -1,226 +0,0 @@
1
- {
2
- "dataset": {
3
- "repo_id": "AI-Final/AI-aiCapstoneData-lerobot-advanced-v2-2",
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": "main",
78
- "use_imagenet_stats": true,
79
- "video_backend": "torchcodec",
80
- "streaming": false
81
- },
82
- "env": null,
83
- "policy": {
84
- "type": "diffusion",
85
- "n_obs_steps": 2,
86
- "input_features": {
87
- "observation.state": {
88
- "type": "STATE",
89
- "shape": [
90
- 9
91
- ]
92
- },
93
- "observation.images.wrist": {
94
- "type": "VISUAL",
95
- "shape": [
96
- 3,
97
- 480,
98
- 640
99
- ]
100
- },
101
- "observation.images.front": {
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
- 8
115
- ]
116
- }
117
- },
118
- "device": "cuda",
119
- "use_amp": false,
120
- "use_peft": false,
121
- "push_to_hub": true,
122
- "repo_id": "AI-Final/advanced-diffusion-v3",
123
- "private": null,
124
- "tags": null,
125
- "license": null,
126
- "pretrained_path": null,
127
- "horizon": 16,
128
- "n_action_steps": 8,
129
- "normalization_mapping": {
130
- "VISUAL": "MEAN_STD",
131
- "STATE": "MIN_MAX",
132
- "ACTION": "MIN_MAX"
133
- },
134
- "drop_n_last_frames": 7,
135
- "vision_backbone": "resnet18",
136
- "resize_shape": null,
137
- "crop_ratio": 1.0,
138
- "crop_shape": null,
139
- "crop_is_random": true,
140
- "pretrained_backbone_weights": null,
141
- "use_group_norm": true,
142
- "spatial_softmax_num_keypoints": 32,
143
- "use_separate_rgb_encoder_per_camera": false,
144
- "down_dims": [
145
- 512,
146
- 1024,
147
- 2048
148
- ],
149
- "kernel_size": 5,
150
- "n_groups": 8,
151
- "diffusion_step_embed_dim": 128,
152
- "use_film_scale_modulation": true,
153
- "noise_scheduler_type": "DDPM",
154
- "num_train_timesteps": 100,
155
- "beta_schedule": "squaredcos_cap_v2",
156
- "beta_start": 0.0001,
157
- "beta_end": 0.02,
158
- "prediction_type": "epsilon",
159
- "clip_sample": true,
160
- "clip_sample_range": 1.0,
161
- "num_inference_steps": null,
162
- "compile_model": false,
163
- "compile_mode": "reduce-overhead",
164
- "do_mask_loss_for_padding": false,
165
- "optimizer_lr": 0.0001,
166
- "optimizer_betas": [
167
- 0.95,
168
- 0.999
169
- ],
170
- "optimizer_eps": 1e-08,
171
- "optimizer_weight_decay": 1e-06,
172
- "scheduler_name": "cosine",
173
- "scheduler_warmup_steps": 500
174
- },
175
- "output_dir": "outputs/train/act_dining_cleanup_v2_2_20260609_033312",
176
- "job_name": "act_dining_cleanup_v2_2_20260609_033312",
177
- "resume": false,
178
- "seed": 1000,
179
- "num_workers": 4,
180
- "batch_size": 16,
181
- "steps": 200000,
182
- "eval_freq": 0,
183
- "log_freq": 200,
184
- "tolerance_s": 0.0001,
185
- "save_checkpoint": true,
186
- "save_freq": 20000,
187
- "use_policy_training_preset": true,
188
- "optimizer": {
189
- "type": "adam",
190
- "lr": 0.0001,
191
- "weight_decay": 1e-06,
192
- "grad_clip_norm": 10.0,
193
- "betas": [
194
- 0.95,
195
- 0.999
196
- ],
197
- "eps": 1e-08
198
- },
199
- "scheduler": {
200
- "type": "diffuser",
201
- "num_warmup_steps": 500,
202
- "name": "cosine"
203
- },
204
- "eval": {
205
- "n_episodes": 50,
206
- "batch_size": 50,
207
- "use_async_envs": false
208
- },
209
- "wandb": {
210
- "enable": false,
211
- "disable_artifact": false,
212
- "project": "lerobot",
213
- "entity": null,
214
- "notes": null,
215
- "run_id": null,
216
- "mode": null
217
- },
218
- "peft": null,
219
- "use_rabc": false,
220
- "rabc_progress_path": null,
221
- "rabc_kappa": 0.01,
222
- "rabc_epsilon": 1e-06,
223
- "rabc_head_mode": "sparse",
224
- "rename_map": {},
225
- "checkpoint_path": null
226
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/training_state/optimizer_param_groups.json DELETED
@@ -1,233 +0,0 @@
1
- [
2
- {
3
- "lr": 9.766112029998846e-05,
4
- "betas": [
5
- 0.95,
6
- 0.999
7
- ],
8
- "eps": 1e-08,
9
- "weight_decay": 1e-06,
10
- "amsgrad": false,
11
- "maximize": false,
12
- "foreach": null,
13
- "capturable": false,
14
- "differentiable": false,
15
- "fused": null,
16
- "decoupled_weight_decay": false,
17
- "initial_lr": 0.0001,
18
- "params": [
19
- 0,
20
- 1,
21
- 2,
22
- 3,
23
- 4,
24
- 5,
25
- 6,
26
- 7,
27
- 8,
28
- 9,
29
- 10,
30
- 11,
31
- 12,
32
- 13,
33
- 14,
34
- 15,
35
- 16,
36
- 17,
37
- 18,
38
- 19,
39
- 20,
40
- 21,
41
- 22,
42
- 23,
43
- 24,
44
- 25,
45
- 26,
46
- 27,
47
- 28,
48
- 29,
49
- 30,
50
- 31,
51
- 32,
52
- 33,
53
- 34,
54
- 35,
55
- 36,
56
- 37,
57
- 38,
58
- 39,
59
- 40,
60
- 41,
61
- 42,
62
- 43,
63
- 44,
64
- 45,
65
- 46,
66
- 47,
67
- 48,
68
- 49,
69
- 50,
70
- 51,
71
- 52,
72
- 53,
73
- 54,
74
- 55,
75
- 56,
76
- 57,
77
- 58,
78
- 59,
79
- 60,
80
- 61,
81
- 62,
82
- 63,
83
- 64,
84
- 65,
85
- 66,
86
- 67,
87
- 68,
88
- 69,
89
- 70,
90
- 71,
91
- 72,
92
- 73,
93
- 74,
94
- 75,
95
- 76,
96
- 77,
97
- 78,
98
- 79,
99
- 80,
100
- 81,
101
- 82,
102
- 83,
103
- 84,
104
- 85,
105
- 86,
106
- 87,
107
- 88,
108
- 89,
109
- 90,
110
- 91,
111
- 92,
112
- 93,
113
- 94,
114
- 95,
115
- 96,
116
- 97,
117
- 98,
118
- 99,
119
- 100,
120
- 101,
121
- 102,
122
- 103,
123
- 104,
124
- 105,
125
- 106,
126
- 107,
127
- 108,
128
- 109,
129
- 110,
130
- 111,
131
- 112,
132
- 113,
133
- 114,
134
- 115,
135
- 116,
136
- 117,
137
- 118,
138
- 119,
139
- 120,
140
- 121,
141
- 122,
142
- 123,
143
- 124,
144
- 125,
145
- 126,
146
- 127,
147
- 128,
148
- 129,
149
- 130,
150
- 131,
151
- 132,
152
- 133,
153
- 134,
154
- 135,
155
- 136,
156
- 137,
157
- 138,
158
- 139,
159
- 140,
160
- 141,
161
- 142,
162
- 143,
163
- 144,
164
- 145,
165
- 146,
166
- 147,
167
- 148,
168
- 149,
169
- 150,
170
- 151,
171
- 152,
172
- 153,
173
- 154,
174
- 155,
175
- 156,
176
- 157,
177
- 158,
178
- 159,
179
- 160,
180
- 161,
181
- 162,
182
- 163,
183
- 164,
184
- 165,
185
- 166,
186
- 167,
187
- 168,
188
- 169,
189
- 170,
190
- 171,
191
- 172,
192
- 173,
193
- 174,
194
- 175,
195
- 176,
196
- 177,
197
- 178,
198
- 179,
199
- 180,
200
- 181,
201
- 182,
202
- 183,
203
- 184,
204
- 185,
205
- 186,
206
- 187,
207
- 188,
208
- 189,
209
- 190,
210
- 191,
211
- 192,
212
- 193,
213
- 194,
214
- 195,
215
- 196,
216
- 197,
217
- 198,
218
- 199,
219
- 200,
220
- 201,
221
- 202,
222
- 203,
223
- 204,
224
- 205,
225
- 206,
226
- 207,
227
- 208,
228
- 209,
229
- 210,
230
- 211
231
- ]
232
- }
233
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/training_state/optimizer_state.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:bdb7d2c2983f37ff3a2d12902fcc6a9ef8ce531d949ec203cd2a7ffc83709fac
3
- size 2134472696
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/training_state/rng_state.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:30b4bc2eb89c1acd2146d668da03bcea46ff027bf7ddcc8115bc613179bbf9ab
3
- size 15708
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/training_state/scheduler_state.json DELETED
@@ -1,15 +0,0 @@
1
- {
2
- "base_lrs": [
3
- 0.0001
4
- ],
5
- "last_epoch": 20000,
6
- "_step_count": 20001,
7
- "_is_initial": false,
8
- "_get_lr_called_within_step": false,
9
- "_last_lr": [
10
- 9.766112029998846e-05
11
- ],
12
- "lr_lambdas": [
13
- null
14
- ]
15
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/020000/training_state/training_step.json DELETED
@@ -1,3 +0,0 @@
1
- {
2
- "step": 20000
3
- }
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/pretrained_model/config.json DELETED
@@ -1,92 +0,0 @@
1
- {
2
- "type": "diffusion",
3
- "n_obs_steps": 2,
4
- "input_features": {
5
- "observation.state": {
6
- "type": "STATE",
7
- "shape": [
8
- 9
9
- ]
10
- },
11
- "observation.images.wrist": {
12
- "type": "VISUAL",
13
- "shape": [
14
- 3,
15
- 480,
16
- 640
17
- ]
18
- },
19
- "observation.images.front": {
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
- 8
33
- ]
34
- }
35
- },
36
- "device": "cuda",
37
- "use_amp": false,
38
- "use_peft": false,
39
- "push_to_hub": true,
40
- "repo_id": "AI-Final/advanced-diffusion-v3",
41
- "private": null,
42
- "tags": null,
43
- "license": null,
44
- "pretrained_path": null,
45
- "horizon": 16,
46
- "n_action_steps": 8,
47
- "normalization_mapping": {
48
- "VISUAL": "MEAN_STD",
49
- "STATE": "MIN_MAX",
50
- "ACTION": "MIN_MAX"
51
- },
52
- "drop_n_last_frames": 7,
53
- "vision_backbone": "resnet18",
54
- "resize_shape": null,
55
- "crop_ratio": 1.0,
56
- "crop_shape": null,
57
- "crop_is_random": true,
58
- "pretrained_backbone_weights": null,
59
- "use_group_norm": true,
60
- "spatial_softmax_num_keypoints": 32,
61
- "use_separate_rgb_encoder_per_camera": false,
62
- "down_dims": [
63
- 512,
64
- 1024,
65
- 2048
66
- ],
67
- "kernel_size": 5,
68
- "n_groups": 8,
69
- "diffusion_step_embed_dim": 128,
70
- "use_film_scale_modulation": true,
71
- "noise_scheduler_type": "DDPM",
72
- "num_train_timesteps": 100,
73
- "beta_schedule": "squaredcos_cap_v2",
74
- "beta_start": 0.0001,
75
- "beta_end": 0.02,
76
- "prediction_type": "epsilon",
77
- "clip_sample": true,
78
- "clip_sample_range": 1.0,
79
- "num_inference_steps": null,
80
- "compile_model": false,
81
- "compile_mode": "reduce-overhead",
82
- "do_mask_loss_for_padding": false,
83
- "optimizer_lr": 0.0001,
84
- "optimizer_betas": [
85
- 0.95,
86
- 0.999
87
- ],
88
- "optimizer_eps": 1e-08,
89
- "optimizer_weight_decay": 1e-06,
90
- "scheduler_name": "cosine",
91
- "scheduler_warmup_steps": 500
92
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/pretrained_model/model.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:7c2714f8fb74baa779a3f0b0d1db254110f3ea0d8f322d0385a79bf6f656f992
3
- size 1067235520
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/pretrained_model/policy_postprocessor.json DELETED
@@ -1,32 +0,0 @@
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
- 8
13
- ]
14
- }
15
- },
16
- "norm_map": {
17
- "VISUAL": "MEAN_STD",
18
- "STATE": "MIN_MAX",
19
- "ACTION": "MIN_MAX"
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
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:86b34463feeba35ffc7fce28f390a6fdd072dd2d8036f7a62d3dc0de5cc7a3f6
3
- size 7756
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/pretrained_model/policy_preprocessor.json DELETED
@@ -1,64 +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": "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
- 9
30
- ]
31
- },
32
- "observation.images.wrist": {
33
- "type": "VISUAL",
34
- "shape": [
35
- 3,
36
- 480,
37
- 640
38
- ]
39
- },
40
- "observation.images.front": {
41
- "type": "VISUAL",
42
- "shape": [
43
- 3,
44
- 480,
45
- 640
46
- ]
47
- },
48
- "action": {
49
- "type": "ACTION",
50
- "shape": [
51
- 8
52
- ]
53
- }
54
- },
55
- "norm_map": {
56
- "VISUAL": "MEAN_STD",
57
- "STATE": "MIN_MAX",
58
- "ACTION": "MIN_MAX"
59
- }
60
- },
61
- "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
62
- }
63
- ]
64
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:86b34463feeba35ffc7fce28f390a6fdd072dd2d8036f7a62d3dc0de5cc7a3f6
3
- size 7756
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/pretrained_model/train_config.json DELETED
@@ -1,226 +0,0 @@
1
- {
2
- "dataset": {
3
- "repo_id": "AI-Final/AI-aiCapstoneData-lerobot-advanced-v2-2",
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": "main",
78
- "use_imagenet_stats": true,
79
- "video_backend": "torchcodec",
80
- "streaming": false
81
- },
82
- "env": null,
83
- "policy": {
84
- "type": "diffusion",
85
- "n_obs_steps": 2,
86
- "input_features": {
87
- "observation.state": {
88
- "type": "STATE",
89
- "shape": [
90
- 9
91
- ]
92
- },
93
- "observation.images.wrist": {
94
- "type": "VISUAL",
95
- "shape": [
96
- 3,
97
- 480,
98
- 640
99
- ]
100
- },
101
- "observation.images.front": {
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
- 8
115
- ]
116
- }
117
- },
118
- "device": "cuda",
119
- "use_amp": false,
120
- "use_peft": false,
121
- "push_to_hub": true,
122
- "repo_id": "AI-Final/advanced-diffusion-v3",
123
- "private": null,
124
- "tags": null,
125
- "license": null,
126
- "pretrained_path": null,
127
- "horizon": 16,
128
- "n_action_steps": 8,
129
- "normalization_mapping": {
130
- "VISUAL": "MEAN_STD",
131
- "STATE": "MIN_MAX",
132
- "ACTION": "MIN_MAX"
133
- },
134
- "drop_n_last_frames": 7,
135
- "vision_backbone": "resnet18",
136
- "resize_shape": null,
137
- "crop_ratio": 1.0,
138
- "crop_shape": null,
139
- "crop_is_random": true,
140
- "pretrained_backbone_weights": null,
141
- "use_group_norm": true,
142
- "spatial_softmax_num_keypoints": 32,
143
- "use_separate_rgb_encoder_per_camera": false,
144
- "down_dims": [
145
- 512,
146
- 1024,
147
- 2048
148
- ],
149
- "kernel_size": 5,
150
- "n_groups": 8,
151
- "diffusion_step_embed_dim": 128,
152
- "use_film_scale_modulation": true,
153
- "noise_scheduler_type": "DDPM",
154
- "num_train_timesteps": 100,
155
- "beta_schedule": "squaredcos_cap_v2",
156
- "beta_start": 0.0001,
157
- "beta_end": 0.02,
158
- "prediction_type": "epsilon",
159
- "clip_sample": true,
160
- "clip_sample_range": 1.0,
161
- "num_inference_steps": null,
162
- "compile_model": false,
163
- "compile_mode": "reduce-overhead",
164
- "do_mask_loss_for_padding": false,
165
- "optimizer_lr": 0.0001,
166
- "optimizer_betas": [
167
- 0.95,
168
- 0.999
169
- ],
170
- "optimizer_eps": 1e-08,
171
- "optimizer_weight_decay": 1e-06,
172
- "scheduler_name": "cosine",
173
- "scheduler_warmup_steps": 500
174
- },
175
- "output_dir": "outputs/train/act_dining_cleanup_v2_2_20260609_033312",
176
- "job_name": "act_dining_cleanup_v2_2_20260609_033312",
177
- "resume": false,
178
- "seed": 1000,
179
- "num_workers": 4,
180
- "batch_size": 16,
181
- "steps": 200000,
182
- "eval_freq": 0,
183
- "log_freq": 200,
184
- "tolerance_s": 0.0001,
185
- "save_checkpoint": true,
186
- "save_freq": 20000,
187
- "use_policy_training_preset": true,
188
- "optimizer": {
189
- "type": "adam",
190
- "lr": 0.0001,
191
- "weight_decay": 1e-06,
192
- "grad_clip_norm": 10.0,
193
- "betas": [
194
- 0.95,
195
- 0.999
196
- ],
197
- "eps": 1e-08
198
- },
199
- "scheduler": {
200
- "type": "diffuser",
201
- "num_warmup_steps": 500,
202
- "name": "cosine"
203
- },
204
- "eval": {
205
- "n_episodes": 50,
206
- "batch_size": 50,
207
- "use_async_envs": false
208
- },
209
- "wandb": {
210
- "enable": false,
211
- "disable_artifact": false,
212
- "project": "lerobot",
213
- "entity": null,
214
- "notes": null,
215
- "run_id": null,
216
- "mode": null
217
- },
218
- "peft": null,
219
- "use_rabc": false,
220
- "rabc_progress_path": null,
221
- "rabc_kappa": 0.01,
222
- "rabc_epsilon": 1e-06,
223
- "rabc_head_mode": "sparse",
224
- "rename_map": {},
225
- "checkpoint_path": null
226
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/training_state/optimizer_param_groups.json DELETED
@@ -1,233 +0,0 @@
1
- [
2
- {
3
- "lr": 9.063516700762651e-05,
4
- "betas": [
5
- 0.95,
6
- 0.999
7
- ],
8
- "eps": 1e-08,
9
- "weight_decay": 1e-06,
10
- "amsgrad": false,
11
- "maximize": false,
12
- "foreach": null,
13
- "capturable": false,
14
- "differentiable": false,
15
- "fused": null,
16
- "decoupled_weight_decay": false,
17
- "initial_lr": 0.0001,
18
- "params": [
19
- 0,
20
- 1,
21
- 2,
22
- 3,
23
- 4,
24
- 5,
25
- 6,
26
- 7,
27
- 8,
28
- 9,
29
- 10,
30
- 11,
31
- 12,
32
- 13,
33
- 14,
34
- 15,
35
- 16,
36
- 17,
37
- 18,
38
- 19,
39
- 20,
40
- 21,
41
- 22,
42
- 23,
43
- 24,
44
- 25,
45
- 26,
46
- 27,
47
- 28,
48
- 29,
49
- 30,
50
- 31,
51
- 32,
52
- 33,
53
- 34,
54
- 35,
55
- 36,
56
- 37,
57
- 38,
58
- 39,
59
- 40,
60
- 41,
61
- 42,
62
- 43,
63
- 44,
64
- 45,
65
- 46,
66
- 47,
67
- 48,
68
- 49,
69
- 50,
70
- 51,
71
- 52,
72
- 53,
73
- 54,
74
- 55,
75
- 56,
76
- 57,
77
- 58,
78
- 59,
79
- 60,
80
- 61,
81
- 62,
82
- 63,
83
- 64,
84
- 65,
85
- 66,
86
- 67,
87
- 68,
88
- 69,
89
- 70,
90
- 71,
91
- 72,
92
- 73,
93
- 74,
94
- 75,
95
- 76,
96
- 77,
97
- 78,
98
- 79,
99
- 80,
100
- 81,
101
- 82,
102
- 83,
103
- 84,
104
- 85,
105
- 86,
106
- 87,
107
- 88,
108
- 89,
109
- 90,
110
- 91,
111
- 92,
112
- 93,
113
- 94,
114
- 95,
115
- 96,
116
- 97,
117
- 98,
118
- 99,
119
- 100,
120
- 101,
121
- 102,
122
- 103,
123
- 104,
124
- 105,
125
- 106,
126
- 107,
127
- 108,
128
- 109,
129
- 110,
130
- 111,
131
- 112,
132
- 113,
133
- 114,
134
- 115,
135
- 116,
136
- 117,
137
- 118,
138
- 119,
139
- 120,
140
- 121,
141
- 122,
142
- 123,
143
- 124,
144
- 125,
145
- 126,
146
- 127,
147
- 128,
148
- 129,
149
- 130,
150
- 131,
151
- 132,
152
- 133,
153
- 134,
154
- 135,
155
- 136,
156
- 137,
157
- 138,
158
- 139,
159
- 140,
160
- 141,
161
- 142,
162
- 143,
163
- 144,
164
- 145,
165
- 146,
166
- 147,
167
- 148,
168
- 149,
169
- 150,
170
- 151,
171
- 152,
172
- 153,
173
- 154,
174
- 155,
175
- 156,
176
- 157,
177
- 158,
178
- 159,
179
- 160,
180
- 161,
181
- 162,
182
- 163,
183
- 164,
184
- 165,
185
- 166,
186
- 167,
187
- 168,
188
- 169,
189
- 170,
190
- 171,
191
- 172,
192
- 173,
193
- 174,
194
- 175,
195
- 176,
196
- 177,
197
- 178,
198
- 179,
199
- 180,
200
- 181,
201
- 182,
202
- 183,
203
- 184,
204
- 185,
205
- 186,
206
- 187,
207
- 188,
208
- 189,
209
- 190,
210
- 191,
211
- 192,
212
- 193,
213
- 194,
214
- 195,
215
- 196,
216
- 197,
217
- 198,
218
- 199,
219
- 200,
220
- 201,
221
- 202,
222
- 203,
223
- 204,
224
- 205,
225
- 206,
226
- 207,
227
- 208,
228
- 209,
229
- 210,
230
- 211
231
- ]
232
- }
233
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/training_state/optimizer_state.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:4359855b7a4e66e97cb8fbd3822cc3f4e673f4705ff35b8fe648c9a05a406378
3
- size 2134472696
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/training_state/rng_state.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:7bad751b7bced848c3ca3fe3195d6eac016a574e45f2ce0bff567cf2c0a49c47
3
- size 15708
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/training_state/scheduler_state.json DELETED
@@ -1,15 +0,0 @@
1
- {
2
- "base_lrs": [
3
- 0.0001
4
- ],
5
- "last_epoch": 40000,
6
- "_step_count": 40001,
7
- "_is_initial": false,
8
- "_get_lr_called_within_step": false,
9
- "_last_lr": [
10
- 9.063516700762651e-05
11
- ],
12
- "lr_lambdas": [
13
- null
14
- ]
15
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/040000/training_state/training_step.json DELETED
@@ -1,3 +0,0 @@
1
- {
2
- "step": 40000
3
- }
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/pretrained_model/config.json DELETED
@@ -1,92 +0,0 @@
1
- {
2
- "type": "diffusion",
3
- "n_obs_steps": 2,
4
- "input_features": {
5
- "observation.state": {
6
- "type": "STATE",
7
- "shape": [
8
- 9
9
- ]
10
- },
11
- "observation.images.wrist": {
12
- "type": "VISUAL",
13
- "shape": [
14
- 3,
15
- 480,
16
- 640
17
- ]
18
- },
19
- "observation.images.front": {
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
- 8
33
- ]
34
- }
35
- },
36
- "device": "cuda",
37
- "use_amp": false,
38
- "use_peft": false,
39
- "push_to_hub": true,
40
- "repo_id": "AI-Final/advanced-diffusion-v3",
41
- "private": null,
42
- "tags": null,
43
- "license": null,
44
- "pretrained_path": null,
45
- "horizon": 16,
46
- "n_action_steps": 8,
47
- "normalization_mapping": {
48
- "VISUAL": "MEAN_STD",
49
- "STATE": "MIN_MAX",
50
- "ACTION": "MIN_MAX"
51
- },
52
- "drop_n_last_frames": 7,
53
- "vision_backbone": "resnet18",
54
- "resize_shape": null,
55
- "crop_ratio": 1.0,
56
- "crop_shape": null,
57
- "crop_is_random": true,
58
- "pretrained_backbone_weights": null,
59
- "use_group_norm": true,
60
- "spatial_softmax_num_keypoints": 32,
61
- "use_separate_rgb_encoder_per_camera": false,
62
- "down_dims": [
63
- 512,
64
- 1024,
65
- 2048
66
- ],
67
- "kernel_size": 5,
68
- "n_groups": 8,
69
- "diffusion_step_embed_dim": 128,
70
- "use_film_scale_modulation": true,
71
- "noise_scheduler_type": "DDPM",
72
- "num_train_timesteps": 100,
73
- "beta_schedule": "squaredcos_cap_v2",
74
- "beta_start": 0.0001,
75
- "beta_end": 0.02,
76
- "prediction_type": "epsilon",
77
- "clip_sample": true,
78
- "clip_sample_range": 1.0,
79
- "num_inference_steps": null,
80
- "compile_model": false,
81
- "compile_mode": "reduce-overhead",
82
- "do_mask_loss_for_padding": false,
83
- "optimizer_lr": 0.0001,
84
- "optimizer_betas": [
85
- 0.95,
86
- 0.999
87
- ],
88
- "optimizer_eps": 1e-08,
89
- "optimizer_weight_decay": 1e-06,
90
- "scheduler_name": "cosine",
91
- "scheduler_warmup_steps": 500
92
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/pretrained_model/model.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:8b902fe2492341a138b2da57e93cd3890763ddeca25f50f15fbfbb7bb287034f
3
- size 1067235520
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/pretrained_model/policy_postprocessor.json DELETED
@@ -1,32 +0,0 @@
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
- 8
13
- ]
14
- }
15
- },
16
- "norm_map": {
17
- "VISUAL": "MEAN_STD",
18
- "STATE": "MIN_MAX",
19
- "ACTION": "MIN_MAX"
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
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:86b34463feeba35ffc7fce28f390a6fdd072dd2d8036f7a62d3dc0de5cc7a3f6
3
- size 7756
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/pretrained_model/policy_preprocessor.json DELETED
@@ -1,64 +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": "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
- 9
30
- ]
31
- },
32
- "observation.images.wrist": {
33
- "type": "VISUAL",
34
- "shape": [
35
- 3,
36
- 480,
37
- 640
38
- ]
39
- },
40
- "observation.images.front": {
41
- "type": "VISUAL",
42
- "shape": [
43
- 3,
44
- 480,
45
- 640
46
- ]
47
- },
48
- "action": {
49
- "type": "ACTION",
50
- "shape": [
51
- 8
52
- ]
53
- }
54
- },
55
- "norm_map": {
56
- "VISUAL": "MEAN_STD",
57
- "STATE": "MIN_MAX",
58
- "ACTION": "MIN_MAX"
59
- }
60
- },
61
- "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
62
- }
63
- ]
64
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:86b34463feeba35ffc7fce28f390a6fdd072dd2d8036f7a62d3dc0de5cc7a3f6
3
- size 7756
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/pretrained_model/train_config.json DELETED
@@ -1,226 +0,0 @@
1
- {
2
- "dataset": {
3
- "repo_id": "AI-Final/AI-aiCapstoneData-lerobot-advanced-v2-2",
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": "main",
78
- "use_imagenet_stats": true,
79
- "video_backend": "torchcodec",
80
- "streaming": false
81
- },
82
- "env": null,
83
- "policy": {
84
- "type": "diffusion",
85
- "n_obs_steps": 2,
86
- "input_features": {
87
- "observation.state": {
88
- "type": "STATE",
89
- "shape": [
90
- 9
91
- ]
92
- },
93
- "observation.images.wrist": {
94
- "type": "VISUAL",
95
- "shape": [
96
- 3,
97
- 480,
98
- 640
99
- ]
100
- },
101
- "observation.images.front": {
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
- 8
115
- ]
116
- }
117
- },
118
- "device": "cuda",
119
- "use_amp": false,
120
- "use_peft": false,
121
- "push_to_hub": true,
122
- "repo_id": "AI-Final/advanced-diffusion-v3",
123
- "private": null,
124
- "tags": null,
125
- "license": null,
126
- "pretrained_path": null,
127
- "horizon": 16,
128
- "n_action_steps": 8,
129
- "normalization_mapping": {
130
- "VISUAL": "MEAN_STD",
131
- "STATE": "MIN_MAX",
132
- "ACTION": "MIN_MAX"
133
- },
134
- "drop_n_last_frames": 7,
135
- "vision_backbone": "resnet18",
136
- "resize_shape": null,
137
- "crop_ratio": 1.0,
138
- "crop_shape": null,
139
- "crop_is_random": true,
140
- "pretrained_backbone_weights": null,
141
- "use_group_norm": true,
142
- "spatial_softmax_num_keypoints": 32,
143
- "use_separate_rgb_encoder_per_camera": false,
144
- "down_dims": [
145
- 512,
146
- 1024,
147
- 2048
148
- ],
149
- "kernel_size": 5,
150
- "n_groups": 8,
151
- "diffusion_step_embed_dim": 128,
152
- "use_film_scale_modulation": true,
153
- "noise_scheduler_type": "DDPM",
154
- "num_train_timesteps": 100,
155
- "beta_schedule": "squaredcos_cap_v2",
156
- "beta_start": 0.0001,
157
- "beta_end": 0.02,
158
- "prediction_type": "epsilon",
159
- "clip_sample": true,
160
- "clip_sample_range": 1.0,
161
- "num_inference_steps": null,
162
- "compile_model": false,
163
- "compile_mode": "reduce-overhead",
164
- "do_mask_loss_for_padding": false,
165
- "optimizer_lr": 0.0001,
166
- "optimizer_betas": [
167
- 0.95,
168
- 0.999
169
- ],
170
- "optimizer_eps": 1e-08,
171
- "optimizer_weight_decay": 1e-06,
172
- "scheduler_name": "cosine",
173
- "scheduler_warmup_steps": 500
174
- },
175
- "output_dir": "outputs/train/act_dining_cleanup_v2_2_20260609_033312",
176
- "job_name": "act_dining_cleanup_v2_2_20260609_033312",
177
- "resume": false,
178
- "seed": 1000,
179
- "num_workers": 4,
180
- "batch_size": 16,
181
- "steps": 200000,
182
- "eval_freq": 0,
183
- "log_freq": 200,
184
- "tolerance_s": 0.0001,
185
- "save_checkpoint": true,
186
- "save_freq": 20000,
187
- "use_policy_training_preset": true,
188
- "optimizer": {
189
- "type": "adam",
190
- "lr": 0.0001,
191
- "weight_decay": 1e-06,
192
- "grad_clip_norm": 10.0,
193
- "betas": [
194
- 0.95,
195
- 0.999
196
- ],
197
- "eps": 1e-08
198
- },
199
- "scheduler": {
200
- "type": "diffuser",
201
- "num_warmup_steps": 500,
202
- "name": "cosine"
203
- },
204
- "eval": {
205
- "n_episodes": 50,
206
- "batch_size": 50,
207
- "use_async_envs": false
208
- },
209
- "wandb": {
210
- "enable": false,
211
- "disable_artifact": false,
212
- "project": "lerobot",
213
- "entity": null,
214
- "notes": null,
215
- "run_id": null,
216
- "mode": null
217
- },
218
- "peft": null,
219
- "use_rabc": false,
220
- "rabc_progress_path": null,
221
- "rabc_kappa": 0.01,
222
- "rabc_epsilon": 1e-06,
223
- "rabc_head_mode": "sparse",
224
- "rename_map": {},
225
- "checkpoint_path": null
226
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/training_state/optimizer_param_groups.json DELETED
@@ -1,233 +0,0 @@
1
- [
2
- {
3
- "lr": 7.961176263324901e-05,
4
- "betas": [
5
- 0.95,
6
- 0.999
7
- ],
8
- "eps": 1e-08,
9
- "weight_decay": 1e-06,
10
- "amsgrad": false,
11
- "maximize": false,
12
- "foreach": null,
13
- "capturable": false,
14
- "differentiable": false,
15
- "fused": null,
16
- "decoupled_weight_decay": false,
17
- "initial_lr": 0.0001,
18
- "params": [
19
- 0,
20
- 1,
21
- 2,
22
- 3,
23
- 4,
24
- 5,
25
- 6,
26
- 7,
27
- 8,
28
- 9,
29
- 10,
30
- 11,
31
- 12,
32
- 13,
33
- 14,
34
- 15,
35
- 16,
36
- 17,
37
- 18,
38
- 19,
39
- 20,
40
- 21,
41
- 22,
42
- 23,
43
- 24,
44
- 25,
45
- 26,
46
- 27,
47
- 28,
48
- 29,
49
- 30,
50
- 31,
51
- 32,
52
- 33,
53
- 34,
54
- 35,
55
- 36,
56
- 37,
57
- 38,
58
- 39,
59
- 40,
60
- 41,
61
- 42,
62
- 43,
63
- 44,
64
- 45,
65
- 46,
66
- 47,
67
- 48,
68
- 49,
69
- 50,
70
- 51,
71
- 52,
72
- 53,
73
- 54,
74
- 55,
75
- 56,
76
- 57,
77
- 58,
78
- 59,
79
- 60,
80
- 61,
81
- 62,
82
- 63,
83
- 64,
84
- 65,
85
- 66,
86
- 67,
87
- 68,
88
- 69,
89
- 70,
90
- 71,
91
- 72,
92
- 73,
93
- 74,
94
- 75,
95
- 76,
96
- 77,
97
- 78,
98
- 79,
99
- 80,
100
- 81,
101
- 82,
102
- 83,
103
- 84,
104
- 85,
105
- 86,
106
- 87,
107
- 88,
108
- 89,
109
- 90,
110
- 91,
111
- 92,
112
- 93,
113
- 94,
114
- 95,
115
- 96,
116
- 97,
117
- 98,
118
- 99,
119
- 100,
120
- 101,
121
- 102,
122
- 103,
123
- 104,
124
- 105,
125
- 106,
126
- 107,
127
- 108,
128
- 109,
129
- 110,
130
- 111,
131
- 112,
132
- 113,
133
- 114,
134
- 115,
135
- 116,
136
- 117,
137
- 118,
138
- 119,
139
- 120,
140
- 121,
141
- 122,
142
- 123,
143
- 124,
144
- 125,
145
- 126,
146
- 127,
147
- 128,
148
- 129,
149
- 130,
150
- 131,
151
- 132,
152
- 133,
153
- 134,
154
- 135,
155
- 136,
156
- 137,
157
- 138,
158
- 139,
159
- 140,
160
- 141,
161
- 142,
162
- 143,
163
- 144,
164
- 145,
165
- 146,
166
- 147,
167
- 148,
168
- 149,
169
- 150,
170
- 151,
171
- 152,
172
- 153,
173
- 154,
174
- 155,
175
- 156,
176
- 157,
177
- 158,
178
- 159,
179
- 160,
180
- 161,
181
- 162,
182
- 163,
183
- 164,
184
- 165,
185
- 166,
186
- 167,
187
- 168,
188
- 169,
189
- 170,
190
- 171,
191
- 172,
192
- 173,
193
- 174,
194
- 175,
195
- 176,
196
- 177,
197
- 178,
198
- 179,
199
- 180,
200
- 181,
201
- 182,
202
- 183,
203
- 184,
204
- 185,
205
- 186,
206
- 187,
207
- 188,
208
- 189,
209
- 190,
210
- 191,
211
- 192,
212
- 193,
213
- 194,
214
- 195,
215
- 196,
216
- 197,
217
- 198,
218
- 199,
219
- 200,
220
- 201,
221
- 202,
222
- 203,
223
- 204,
224
- 205,
225
- 206,
226
- 207,
227
- 208,
228
- 209,
229
- 210,
230
- 211
231
- ]
232
- }
233
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/training_state/optimizer_state.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:f577c82a44368f4bc601f80d6322e1898f301493d67800739dae27cb4b494cef
3
- size 2134472696
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/training_state/rng_state.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:916a0ec657a5a06edc9c7101136ea8f89c986d5e830428767ff56720f6c98399
3
- size 15708
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/training_state/scheduler_state.json DELETED
@@ -1,15 +0,0 @@
1
- {
2
- "base_lrs": [
3
- 0.0001
4
- ],
5
- "last_epoch": 60000,
6
- "_step_count": 60001,
7
- "_is_initial": false,
8
- "_get_lr_called_within_step": false,
9
- "_last_lr": [
10
- 7.961176263324901e-05
11
- ],
12
- "lr_lambdas": [
13
- null
14
- ]
15
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/060000/training_state/training_step.json DELETED
@@ -1,3 +0,0 @@
1
- {
2
- "step": 60000
3
- }
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/pretrained_model/config.json DELETED
@@ -1,92 +0,0 @@
1
- {
2
- "type": "diffusion",
3
- "n_obs_steps": 2,
4
- "input_features": {
5
- "observation.state": {
6
- "type": "STATE",
7
- "shape": [
8
- 9
9
- ]
10
- },
11
- "observation.images.wrist": {
12
- "type": "VISUAL",
13
- "shape": [
14
- 3,
15
- 480,
16
- 640
17
- ]
18
- },
19
- "observation.images.front": {
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
- 8
33
- ]
34
- }
35
- },
36
- "device": "cuda",
37
- "use_amp": false,
38
- "use_peft": false,
39
- "push_to_hub": true,
40
- "repo_id": "AI-Final/advanced-diffusion-v3",
41
- "private": null,
42
- "tags": null,
43
- "license": null,
44
- "pretrained_path": null,
45
- "horizon": 16,
46
- "n_action_steps": 8,
47
- "normalization_mapping": {
48
- "VISUAL": "MEAN_STD",
49
- "STATE": "MIN_MAX",
50
- "ACTION": "MIN_MAX"
51
- },
52
- "drop_n_last_frames": 7,
53
- "vision_backbone": "resnet18",
54
- "resize_shape": null,
55
- "crop_ratio": 1.0,
56
- "crop_shape": null,
57
- "crop_is_random": true,
58
- "pretrained_backbone_weights": null,
59
- "use_group_norm": true,
60
- "spatial_softmax_num_keypoints": 32,
61
- "use_separate_rgb_encoder_per_camera": false,
62
- "down_dims": [
63
- 512,
64
- 1024,
65
- 2048
66
- ],
67
- "kernel_size": 5,
68
- "n_groups": 8,
69
- "diffusion_step_embed_dim": 128,
70
- "use_film_scale_modulation": true,
71
- "noise_scheduler_type": "DDPM",
72
- "num_train_timesteps": 100,
73
- "beta_schedule": "squaredcos_cap_v2",
74
- "beta_start": 0.0001,
75
- "beta_end": 0.02,
76
- "prediction_type": "epsilon",
77
- "clip_sample": true,
78
- "clip_sample_range": 1.0,
79
- "num_inference_steps": null,
80
- "compile_model": false,
81
- "compile_mode": "reduce-overhead",
82
- "do_mask_loss_for_padding": false,
83
- "optimizer_lr": 0.0001,
84
- "optimizer_betas": [
85
- 0.95,
86
- 0.999
87
- ],
88
- "optimizer_eps": 1e-08,
89
- "optimizer_weight_decay": 1e-06,
90
- "scheduler_name": "cosine",
91
- "scheduler_warmup_steps": 500
92
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/pretrained_model/model.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:c9c4b8693cabaabe0e2346cd48026bb7f06782efa085a2e55495466f1cd6736c
3
- size 1067235520
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/pretrained_model/policy_postprocessor.json DELETED
@@ -1,32 +0,0 @@
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
- 8
13
- ]
14
- }
15
- },
16
- "norm_map": {
17
- "VISUAL": "MEAN_STD",
18
- "STATE": "MIN_MAX",
19
- "ACTION": "MIN_MAX"
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
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:86b34463feeba35ffc7fce28f390a6fdd072dd2d8036f7a62d3dc0de5cc7a3f6
3
- size 7756
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/pretrained_model/policy_preprocessor.json DELETED
@@ -1,64 +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": "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
- 9
30
- ]
31
- },
32
- "observation.images.wrist": {
33
- "type": "VISUAL",
34
- "shape": [
35
- 3,
36
- 480,
37
- 640
38
- ]
39
- },
40
- "observation.images.front": {
41
- "type": "VISUAL",
42
- "shape": [
43
- 3,
44
- 480,
45
- 640
46
- ]
47
- },
48
- "action": {
49
- "type": "ACTION",
50
- "shape": [
51
- 8
52
- ]
53
- }
54
- },
55
- "norm_map": {
56
- "VISUAL": "MEAN_STD",
57
- "STATE": "MIN_MAX",
58
- "ACTION": "MIN_MAX"
59
- }
60
- },
61
- "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
62
- }
63
- ]
64
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:86b34463feeba35ffc7fce28f390a6fdd072dd2d8036f7a62d3dc0de5cc7a3f6
3
- size 7756
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/pretrained_model/train_config.json DELETED
@@ -1,226 +0,0 @@
1
- {
2
- "dataset": {
3
- "repo_id": "AI-Final/AI-aiCapstoneData-lerobot-advanced-v2-2",
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": "main",
78
- "use_imagenet_stats": true,
79
- "video_backend": "torchcodec",
80
- "streaming": false
81
- },
82
- "env": null,
83
- "policy": {
84
- "type": "diffusion",
85
- "n_obs_steps": 2,
86
- "input_features": {
87
- "observation.state": {
88
- "type": "STATE",
89
- "shape": [
90
- 9
91
- ]
92
- },
93
- "observation.images.wrist": {
94
- "type": "VISUAL",
95
- "shape": [
96
- 3,
97
- 480,
98
- 640
99
- ]
100
- },
101
- "observation.images.front": {
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
- 8
115
- ]
116
- }
117
- },
118
- "device": "cuda",
119
- "use_amp": false,
120
- "use_peft": false,
121
- "push_to_hub": true,
122
- "repo_id": "AI-Final/advanced-diffusion-v3",
123
- "private": null,
124
- "tags": null,
125
- "license": null,
126
- "pretrained_path": null,
127
- "horizon": 16,
128
- "n_action_steps": 8,
129
- "normalization_mapping": {
130
- "VISUAL": "MEAN_STD",
131
- "STATE": "MIN_MAX",
132
- "ACTION": "MIN_MAX"
133
- },
134
- "drop_n_last_frames": 7,
135
- "vision_backbone": "resnet18",
136
- "resize_shape": null,
137
- "crop_ratio": 1.0,
138
- "crop_shape": null,
139
- "crop_is_random": true,
140
- "pretrained_backbone_weights": null,
141
- "use_group_norm": true,
142
- "spatial_softmax_num_keypoints": 32,
143
- "use_separate_rgb_encoder_per_camera": false,
144
- "down_dims": [
145
- 512,
146
- 1024,
147
- 2048
148
- ],
149
- "kernel_size": 5,
150
- "n_groups": 8,
151
- "diffusion_step_embed_dim": 128,
152
- "use_film_scale_modulation": true,
153
- "noise_scheduler_type": "DDPM",
154
- "num_train_timesteps": 100,
155
- "beta_schedule": "squaredcos_cap_v2",
156
- "beta_start": 0.0001,
157
- "beta_end": 0.02,
158
- "prediction_type": "epsilon",
159
- "clip_sample": true,
160
- "clip_sample_range": 1.0,
161
- "num_inference_steps": null,
162
- "compile_model": false,
163
- "compile_mode": "reduce-overhead",
164
- "do_mask_loss_for_padding": false,
165
- "optimizer_lr": 0.0001,
166
- "optimizer_betas": [
167
- 0.95,
168
- 0.999
169
- ],
170
- "optimizer_eps": 1e-08,
171
- "optimizer_weight_decay": 1e-06,
172
- "scheduler_name": "cosine",
173
- "scheduler_warmup_steps": 500
174
- },
175
- "output_dir": "outputs/train/act_dining_cleanup_v2_2_20260609_033312",
176
- "job_name": "act_dining_cleanup_v2_2_20260609_033312",
177
- "resume": false,
178
- "seed": 1000,
179
- "num_workers": 4,
180
- "batch_size": 16,
181
- "steps": 200000,
182
- "eval_freq": 0,
183
- "log_freq": 200,
184
- "tolerance_s": 0.0001,
185
- "save_checkpoint": true,
186
- "save_freq": 20000,
187
- "use_policy_training_preset": true,
188
- "optimizer": {
189
- "type": "adam",
190
- "lr": 0.0001,
191
- "weight_decay": 1e-06,
192
- "grad_clip_norm": 10.0,
193
- "betas": [
194
- 0.95,
195
- 0.999
196
- ],
197
- "eps": 1e-08
198
- },
199
- "scheduler": {
200
- "type": "diffuser",
201
- "num_warmup_steps": 500,
202
- "name": "cosine"
203
- },
204
- "eval": {
205
- "n_episodes": 50,
206
- "batch_size": 50,
207
- "use_async_envs": false
208
- },
209
- "wandb": {
210
- "enable": false,
211
- "disable_artifact": false,
212
- "project": "lerobot",
213
- "entity": null,
214
- "notes": null,
215
- "run_id": null,
216
- "mode": null
217
- },
218
- "peft": null,
219
- "use_rabc": false,
220
- "rabc_progress_path": null,
221
- "rabc_kappa": 0.01,
222
- "rabc_epsilon": 1e-06,
223
- "rabc_head_mode": "sparse",
224
- "rename_map": {},
225
- "checkpoint_path": null
226
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/training_state/optimizer_param_groups.json DELETED
@@ -1,233 +0,0 @@
1
- [
2
- {
3
- "lr": 6.567532550188907e-05,
4
- "betas": [
5
- 0.95,
6
- 0.999
7
- ],
8
- "eps": 1e-08,
9
- "weight_decay": 1e-06,
10
- "amsgrad": false,
11
- "maximize": false,
12
- "foreach": null,
13
- "capturable": false,
14
- "differentiable": false,
15
- "fused": null,
16
- "decoupled_weight_decay": false,
17
- "initial_lr": 0.0001,
18
- "params": [
19
- 0,
20
- 1,
21
- 2,
22
- 3,
23
- 4,
24
- 5,
25
- 6,
26
- 7,
27
- 8,
28
- 9,
29
- 10,
30
- 11,
31
- 12,
32
- 13,
33
- 14,
34
- 15,
35
- 16,
36
- 17,
37
- 18,
38
- 19,
39
- 20,
40
- 21,
41
- 22,
42
- 23,
43
- 24,
44
- 25,
45
- 26,
46
- 27,
47
- 28,
48
- 29,
49
- 30,
50
- 31,
51
- 32,
52
- 33,
53
- 34,
54
- 35,
55
- 36,
56
- 37,
57
- 38,
58
- 39,
59
- 40,
60
- 41,
61
- 42,
62
- 43,
63
- 44,
64
- 45,
65
- 46,
66
- 47,
67
- 48,
68
- 49,
69
- 50,
70
- 51,
71
- 52,
72
- 53,
73
- 54,
74
- 55,
75
- 56,
76
- 57,
77
- 58,
78
- 59,
79
- 60,
80
- 61,
81
- 62,
82
- 63,
83
- 64,
84
- 65,
85
- 66,
86
- 67,
87
- 68,
88
- 69,
89
- 70,
90
- 71,
91
- 72,
92
- 73,
93
- 74,
94
- 75,
95
- 76,
96
- 77,
97
- 78,
98
- 79,
99
- 80,
100
- 81,
101
- 82,
102
- 83,
103
- 84,
104
- 85,
105
- 86,
106
- 87,
107
- 88,
108
- 89,
109
- 90,
110
- 91,
111
- 92,
112
- 93,
113
- 94,
114
- 95,
115
- 96,
116
- 97,
117
- 98,
118
- 99,
119
- 100,
120
- 101,
121
- 102,
122
- 103,
123
- 104,
124
- 105,
125
- 106,
126
- 107,
127
- 108,
128
- 109,
129
- 110,
130
- 111,
131
- 112,
132
- 113,
133
- 114,
134
- 115,
135
- 116,
136
- 117,
137
- 118,
138
- 119,
139
- 120,
140
- 121,
141
- 122,
142
- 123,
143
- 124,
144
- 125,
145
- 126,
146
- 127,
147
- 128,
148
- 129,
149
- 130,
150
- 131,
151
- 132,
152
- 133,
153
- 134,
154
- 135,
155
- 136,
156
- 137,
157
- 138,
158
- 139,
159
- 140,
160
- 141,
161
- 142,
162
- 143,
163
- 144,
164
- 145,
165
- 146,
166
- 147,
167
- 148,
168
- 149,
169
- 150,
170
- 151,
171
- 152,
172
- 153,
173
- 154,
174
- 155,
175
- 156,
176
- 157,
177
- 158,
178
- 159,
179
- 160,
180
- 161,
181
- 162,
182
- 163,
183
- 164,
184
- 165,
185
- 166,
186
- 167,
187
- 168,
188
- 169,
189
- 170,
190
- 171,
191
- 172,
192
- 173,
193
- 174,
194
- 175,
195
- 176,
196
- 177,
197
- 178,
198
- 179,
199
- 180,
200
- 181,
201
- 182,
202
- 183,
203
- 184,
204
- 185,
205
- 186,
206
- 187,
207
- 188,
208
- 189,
209
- 190,
210
- 191,
211
- 192,
212
- 193,
213
- 194,
214
- 195,
215
- 196,
216
- 197,
217
- 198,
218
- 199,
219
- 200,
220
- 201,
221
- 202,
222
- 203,
223
- 204,
224
- 205,
225
- 206,
226
- 207,
227
- 208,
228
- 209,
229
- 210,
230
- 211
231
- ]
232
- }
233
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/training_state/optimizer_state.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:d1d7ba6269dc157908c48dcd8189860d2a56bb481ccd26135ab50be49054fae9
3
- size 2134472696
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/training_state/rng_state.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:ae7f82adb79c890e0a941471aa22f03f0a029151dae62d5266360131ffb49ba5
3
- size 15708
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/training_state/scheduler_state.json DELETED
@@ -1,15 +0,0 @@
1
- {
2
- "base_lrs": [
3
- 0.0001
4
- ],
5
- "last_epoch": 80000,
6
- "_step_count": 80001,
7
- "_is_initial": false,
8
- "_get_lr_called_within_step": false,
9
- "_last_lr": [
10
- 6.567532550188907e-05
11
- ],
12
- "lr_lambdas": [
13
- null
14
- ]
15
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/080000/training_state/training_step.json DELETED
@@ -1,3 +0,0 @@
1
- {
2
- "step": 80000
3
- }
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/100000/pretrained_model/config.json DELETED
@@ -1,92 +0,0 @@
1
- {
2
- "type": "diffusion",
3
- "n_obs_steps": 2,
4
- "input_features": {
5
- "observation.state": {
6
- "type": "STATE",
7
- "shape": [
8
- 9
9
- ]
10
- },
11
- "observation.images.wrist": {
12
- "type": "VISUAL",
13
- "shape": [
14
- 3,
15
- 480,
16
- 640
17
- ]
18
- },
19
- "observation.images.front": {
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
- 8
33
- ]
34
- }
35
- },
36
- "device": "cuda",
37
- "use_amp": false,
38
- "use_peft": false,
39
- "push_to_hub": true,
40
- "repo_id": "AI-Final/advanced-diffusion-v3",
41
- "private": null,
42
- "tags": null,
43
- "license": null,
44
- "pretrained_path": null,
45
- "horizon": 16,
46
- "n_action_steps": 8,
47
- "normalization_mapping": {
48
- "VISUAL": "MEAN_STD",
49
- "STATE": "MIN_MAX",
50
- "ACTION": "MIN_MAX"
51
- },
52
- "drop_n_last_frames": 7,
53
- "vision_backbone": "resnet18",
54
- "resize_shape": null,
55
- "crop_ratio": 1.0,
56
- "crop_shape": null,
57
- "crop_is_random": true,
58
- "pretrained_backbone_weights": null,
59
- "use_group_norm": true,
60
- "spatial_softmax_num_keypoints": 32,
61
- "use_separate_rgb_encoder_per_camera": false,
62
- "down_dims": [
63
- 512,
64
- 1024,
65
- 2048
66
- ],
67
- "kernel_size": 5,
68
- "n_groups": 8,
69
- "diffusion_step_embed_dim": 128,
70
- "use_film_scale_modulation": true,
71
- "noise_scheduler_type": "DDPM",
72
- "num_train_timesteps": 100,
73
- "beta_schedule": "squaredcos_cap_v2",
74
- "beta_start": 0.0001,
75
- "beta_end": 0.02,
76
- "prediction_type": "epsilon",
77
- "clip_sample": true,
78
- "clip_sample_range": 1.0,
79
- "num_inference_steps": null,
80
- "compile_model": false,
81
- "compile_mode": "reduce-overhead",
82
- "do_mask_loss_for_padding": false,
83
- "optimizer_lr": 0.0001,
84
- "optimizer_betas": [
85
- 0.95,
86
- 0.999
87
- ],
88
- "optimizer_eps": 1e-08,
89
- "optimizer_weight_decay": 1e-06,
90
- "scheduler_name": "cosine",
91
- "scheduler_warmup_steps": 500
92
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
training_runs/act_dining_cleanup_v2_2_20260609_033312/checkpoints/100000/pretrained_model/model.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:a4dd721ad4cb9f6d3822c0feeb83a1cfe1094bce59f60fa58979dc9800d67138
3
- size 1067235520