Hugging Face's logo

rlearning2026romsdrums
/
diffusion_unet_checkpoints

Safetensors
Model card Files
xet
 Community
alpant commited on
Commit
04a372f
·
verified ·
1 Parent(s): 285bfef

Delete diffusion_unet_dino_350_augment

Browse files
Files changed (24) hide show
  1. diffusion_unet_dino_350_augment/checkpoints/100000/pretrained_model/config.json +0 -90
  2. diffusion_unet_dino_350_augment/checkpoints/100000/pretrained_model/model.safetensors +0 -3
  3. diffusion_unet_dino_350_augment/checkpoints/100000/pretrained_model/policy_postprocessor.json +0 -32
  4. diffusion_unet_dino_350_augment/checkpoints/100000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +0 -3
  5. diffusion_unet_dino_350_augment/checkpoints/100000/pretrained_model/policy_preprocessor.json +0 -56
  6. diffusion_unet_dino_350_augment/checkpoints/100000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +0 -3
  7. diffusion_unet_dino_350_augment/checkpoints/100000/pretrained_model/train_config.json +0 -300
  8. diffusion_unet_dino_350_augment/checkpoints/100000/training_state/optimizer_param_groups.json +0 -396
  9. diffusion_unet_dino_350_augment/checkpoints/100000/training_state/optimizer_state.safetensors +0 -3
  10. diffusion_unet_dino_350_augment/checkpoints/100000/training_state/rng_state.safetensors +0 -3
  11. diffusion_unet_dino_350_augment/checkpoints/100000/training_state/scheduler_state.json +0 -15
  12. diffusion_unet_dino_350_augment/checkpoints/100000/training_state/training_step.json +0 -3
  13. diffusion_unet_dino_350_augment/checkpoints/200000/pretrained_model/config.json +0 -90
  14. diffusion_unet_dino_350_augment/checkpoints/200000/pretrained_model/model.safetensors +0 -3
  15. diffusion_unet_dino_350_augment/checkpoints/200000/pretrained_model/policy_postprocessor.json +0 -32
  16. diffusion_unet_dino_350_augment/checkpoints/200000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +0 -3
  17. diffusion_unet_dino_350_augment/checkpoints/200000/pretrained_model/policy_preprocessor.json +0 -56
  18. diffusion_unet_dino_350_augment/checkpoints/200000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +0 -3
  19. diffusion_unet_dino_350_augment/checkpoints/200000/pretrained_model/train_config.json +0 -300
  20. diffusion_unet_dino_350_augment/checkpoints/200000/training_state/optimizer_param_groups.json +0 -396
  21. diffusion_unet_dino_350_augment/checkpoints/200000/training_state/optimizer_state.safetensors +0 -3
  22. diffusion_unet_dino_350_augment/checkpoints/200000/training_state/rng_state.safetensors +0 -3
  23. diffusion_unet_dino_350_augment/checkpoints/200000/training_state/scheduler_state.json +0 -15
  24. diffusion_unet_dino_350_augment/checkpoints/200000/training_state/training_step.json +0 -3
diffusion_unet_dino_350_augment/checkpoints/100000/pretrained_model/config.json DELETED
@@ -1,90 +0,0 @@
1
- {
2
- "type": "diffusion",
3
- "n_obs_steps": 2,
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
- },
20
- "output_features": {
21
- "action": {
22
- "type": "ACTION",
23
- "shape": [
24
- 6
25
- ]
26
- }
27
- },
28
- "device": "cuda",
29
- "use_amp": false,
30
- "use_peft": false,
31
- "push_to_hub": false,
32
- "repo_id": null,
33
- "private": null,
34
- "tags": null,
35
- "license": null,
36
- "pretrained_path": null,
37
- "horizon": 16,
38
- "n_action_steps": 8,
39
- "normalization_mapping": {
40
- "VISUAL": "MEAN_STD",
41
- "STATE": "MIN_MAX",
42
- "ACTION": "MIN_MAX"
43
- },
44
- "drop_n_last_frames": 7,
45
- "vision_backbone_type": "dinov2",
46
- "vision_encoder_name": "facebook/dinov2-small",
47
- "freeze_vision_encoder": true,
48
- "vision_backbone": "resnet18",
49
- "resize_shape": [
50
- 224,
51
- 224
52
- ],
53
- "crop_ratio": 1.0,
54
- "crop_shape": null,
55
- "crop_is_random": true,
56
- "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
57
- "use_group_norm": false,
58
- "spatial_softmax_num_keypoints": 32,
59
- "use_separate_rgb_encoder_per_camera": true,
60
- "down_dims": [
61
- 512,
62
- 1024,
63
- 2048
64
- ],
65
- "kernel_size": 5,
66
- "n_groups": 8,
67
- "diffusion_step_embed_dim": 128,
68
- "use_film_scale_modulation": true,
69
- "noise_scheduler_type": "DDPM",
70
- "num_train_timesteps": 100,
71
- "beta_schedule": "squaredcos_cap_v2",
72
- "beta_start": 0.0001,
73
- "beta_end": 0.02,
74
- "prediction_type": "epsilon",
75
- "clip_sample": true,
76
- "clip_sample_range": 1.0,
77
- "num_inference_steps": 100,
78
- "compile_model": false,
79
- "compile_mode": "reduce-overhead",
80
- "do_mask_loss_for_padding": false,
81
- "optimizer_lr": 0.0001,
82
- "optimizer_betas": [
83
- 0.95,
84
- 0.999
85
- ],
86
- "optimizer_eps": 1e-08,
87
- "optimizer_weight_decay": 1e-06,
88
- "scheduler_name": "cosine",
89
- "scheduler_warmup_steps": 500
90
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/100000/pretrained_model/model.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:699d3a0d9ba2384d88ebe7ec22a4d4567082ec4daccb4a16d1634267af6b03eb
3
- size 1095367712
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/100000/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
- 6
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
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/100000/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:f23b22ea05e4b3f2a72f520a60ef30fb248bf15c7b59132b9c3e5e0f3eabcc1c
3
- size 6560
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/100000/pretrained_model/policy_preprocessor.json DELETED
@@ -1,56 +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
- 6
30
- ]
31
- },
32
- "observation.images.front": {
33
- "type": "VISUAL",
34
- "shape": [
35
- 3,
36
- 480,
37
- 640
38
- ]
39
- },
40
- "action": {
41
- "type": "ACTION",
42
- "shape": [
43
- 6
44
- ]
45
- }
46
- },
47
- "norm_map": {
48
- "VISUAL": "MEAN_STD",
49
- "STATE": "MIN_MAX",
50
- "ACTION": "MIN_MAX"
51
- }
52
- },
53
- "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
54
- }
55
- ]
56
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/100000/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:f23b22ea05e4b3f2a72f520a60ef30fb248bf15c7b59132b9c3e5e0f3eabcc1c
3
- size 6560
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/100000/pretrained_model/train_config.json DELETED
@@ -1,300 +0,0 @@
1
- {
2
- "dataset": {
3
- "repo_id": "local/aggregated_new_dataset_532",
4
- "root": "/home/shadeform/rl_project/lerobot/../aggregated_new_dataset_532",
5
- "episodes": null,
6
- "image_transforms": {
7
- "enable": true,
8
- "max_num_transforms": 3,
9
- "random_order": false,
10
- "same_cloth_color": false,
11
- "tfs": {
12
- "color_jitter": {
13
- "weight": 1.5,
14
- "type": "ColorJitter",
15
- "kwargs": {
16
- "brightness": [
17
- 0.7,
18
- 1.3
19
- ],
20
- "contrast": [
21
- 0.7,
22
- 1.3
23
- ],
24
- "saturation": [
25
- 0.4,
26
- 1.6
27
- ],
28
- "hue": [
29
- -0.5,
30
- 0.5
31
- ]
32
- }
33
- },
34
- "grayscale": {
35
- "weight": 0.5,
36
- "type": "RandomGrayscale",
37
- "kwargs": {
38
- "p": 1.0
39
- }
40
- },
41
- "shadow": {
42
- "weight": 0.8,
43
- "type": "RandomShadow",
44
- "kwargs": {
45
- "num_shadows": [
46
- 1,
47
- 2
48
- ],
49
- "intensity": [
50
- 0.4,
51
- 0.7
52
- ],
53
- "blur_sigma": [
54
- 4.0,
55
- 20.0
56
- ]
57
- }
58
- },
59
- "highlight": {
60
- "weight": 0.5,
61
- "type": "RandomHighlight",
62
- "kwargs": {
63
- "num_highlights": [
64
- 0,
65
- 1
66
- ],
67
- "intensity": [
68
- 0.2,
69
- 0.5
70
- ],
71
- "sigma": [
72
- 20.0,
73
- 60.0
74
- ]
75
- }
76
- },
77
- "gamma": {
78
- "weight": 0.7,
79
- "type": "RandomGamma",
80
- "kwargs": {
81
- "gamma": [
82
- 0.6,
83
- 1.4
84
- ]
85
- }
86
- },
87
- "blur": {
88
- "weight": 0.0,
89
- "type": "GaussianBlur",
90
- "kwargs": {
91
- "kernel_size": [
92
- 3,
93
- 3
94
- ],
95
- "sigma": [
96
- 0.1,
97
- 1.5
98
- ]
99
- }
100
- },
101
- "jpeg": {
102
- "weight": 0.0,
103
- "type": "RandomJPEG",
104
- "kwargs": {
105
- "quality": [
106
- 60,
107
- 95
108
- ]
109
- }
110
- },
111
- "crop": {
112
- "weight": 0.5,
113
- "type": "RandomCropPreserveSize",
114
- "kwargs": {
115
- "scale": [
116
- 0.97,
117
- 1.0
118
- ],
119
- "ratio": [
120
- 0.99,
121
- 1.01
122
- ]
123
- }
124
- },
125
- "affine": {
126
- "weight": 0.0,
127
- "type": "RandomAffine",
128
- "kwargs": {
129
- "degrees": [
130
- -5.0,
131
- 5.0
132
- ],
133
- "translate": [
134
- 0.05,
135
- 0.05
136
- ]
137
- }
138
- },
139
- "sharpness": {
140
- "weight": 1.0,
141
- "type": "SharpnessJitter",
142
- "kwargs": {
143
- "sharpness": [
144
- 0.5,
145
- 1.5
146
- ]
147
- }
148
- }
149
- }
150
- },
151
- "revision": null,
152
- "use_imagenet_stats": true,
153
- "video_backend": "pyav",
154
- "return_uint8": false,
155
- "streaming": false
156
- },
157
- "env": null,
158
- "policy": {
159
- "type": "diffusion",
160
- "n_obs_steps": 2,
161
- "input_features": {
162
- "observation.state": {
163
- "type": "STATE",
164
- "shape": [
165
- 6
166
- ]
167
- },
168
- "observation.images.front": {
169
- "type": "VISUAL",
170
- "shape": [
171
- 3,
172
- 480,
173
- 640
174
- ]
175
- }
176
- },
177
- "output_features": {
178
- "action": {
179
- "type": "ACTION",
180
- "shape": [
181
- 6
182
- ]
183
- }
184
- },
185
- "device": "cuda",
186
- "use_amp": false,
187
- "use_peft": false,
188
- "push_to_hub": false,
189
- "repo_id": null,
190
- "private": null,
191
- "tags": null,
192
- "license": null,
193
- "pretrained_path": null,
194
- "horizon": 16,
195
- "n_action_steps": 8,
196
- "normalization_mapping": {
197
- "VISUAL": "MEAN_STD",
198
- "STATE": "MIN_MAX",
199
- "ACTION": "MIN_MAX"
200
- },
201
- "drop_n_last_frames": 7,
202
- "vision_backbone_type": "dinov2",
203
- "vision_encoder_name": "facebook/dinov2-small",
204
- "freeze_vision_encoder": true,
205
- "vision_backbone": "resnet18",
206
- "resize_shape": [
207
- 224,
208
- 224
209
- ],
210
- "crop_ratio": 1.0,
211
- "crop_shape": null,
212
- "crop_is_random": true,
213
- "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
214
- "use_group_norm": false,
215
- "spatial_softmax_num_keypoints": 32,
216
- "use_separate_rgb_encoder_per_camera": true,
217
- "down_dims": [
218
- 512,
219
- 1024,
220
- 2048
221
- ],
222
- "kernel_size": 5,
223
- "n_groups": 8,
224
- "diffusion_step_embed_dim": 128,
225
- "use_film_scale_modulation": true,
226
- "noise_scheduler_type": "DDPM",
227
- "num_train_timesteps": 100,
228
- "beta_schedule": "squaredcos_cap_v2",
229
- "beta_start": 0.0001,
230
- "beta_end": 0.02,
231
- "prediction_type": "epsilon",
232
- "clip_sample": true,
233
- "clip_sample_range": 1.0,
234
- "num_inference_steps": 100,
235
- "compile_model": false,
236
- "compile_mode": "reduce-overhead",
237
- "do_mask_loss_for_padding": false,
238
- "optimizer_lr": 0.0001,
239
- "optimizer_betas": [
240
- 0.95,
241
- 0.999
242
- ],
243
- "optimizer_eps": 1e-08,
244
- "optimizer_weight_decay": 1e-06,
245
- "scheduler_name": "cosine",
246
- "scheduler_warmup_steps": 500
247
- },
248
- "reward_model": null,
249
- "output_dir": "/home/shadeform/rl_project/outputs/train/diffusion_laptop_20260518_183541",
250
- "job_name": "diffusion_laptop_20260518_183541",
251
- "resume": false,
252
- "seed": 1000,
253
- "cudnn_deterministic": false,
254
- "num_workers": 16,
255
- "batch_size": 32,
256
- "prefetch_factor": 4,
257
- "persistent_workers": true,
258
- "steps": 200000,
259
- "eval_freq": 0,
260
- "log_freq": 20,
261
- "tolerance_s": 0.0001,
262
- "save_checkpoint": true,
263
- "save_freq": 100000,
264
- "use_policy_training_preset": true,
265
- "optimizer": {
266
- "type": "adam",
267
- "lr": 0.0001,
268
- "weight_decay": 1e-06,
269
- "grad_clip_norm": 10.0,
270
- "betas": [
271
- 0.95,
272
- 0.999
273
- ],
274
- "eps": 1e-08
275
- },
276
- "scheduler": {
277
- "type": "diffuser",
278
- "num_warmup_steps": 500,
279
- "name": "cosine"
280
- },
281
- "eval": {
282
- "n_episodes": 50,
283
- "batch_size": 11,
284
- "use_async_envs": true
285
- },
286
- "wandb": {
287
- "enable": true,
288
- "disable_artifact": false,
289
- "project": "lerobot",
290
- "entity": null,
291
- "notes": null,
292
- "run_id": "qy0stoak",
293
- "mode": null,
294
- "add_tags": true
295
- },
296
- "peft": null,
297
- "sample_weighting": null,
298
- "rename_map": {},
299
- "checkpoint_path": null
300
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/100000/training_state/optimizer_param_groups.json DELETED
@@ -1,396 +0,0 @@
1
- [
2
- {
3
- "lr": 5.0196841136498776e-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
- 212,
232
- 213,
233
- 214,
234
- 215,
235
- 216,
236
- 217,
237
- 218,
238
- 219,
239
- 220,
240
- 221,
241
- 222,
242
- 223,
243
- 224,
244
- 225,
245
- 226,
246
- 227,
247
- 228,
248
- 229,
249
- 230,
250
- 231,
251
- 232,
252
- 233,
253
- 234,
254
- 235,
255
- 236,
256
- 237,
257
- 238,
258
- 239,
259
- 240,
260
- 241,
261
- 242,
262
- 243,
263
- 244,
264
- 245,
265
- 246,
266
- 247,
267
- 248,
268
- 249,
269
- 250,
270
- 251,
271
- 252,
272
- 253,
273
- 254,
274
- 255,
275
- 256,
276
- 257,
277
- 258,
278
- 259,
279
- 260,
280
- 261,
281
- 262,
282
- 263,
283
- 264,
284
- 265,
285
- 266,
286
- 267,
287
- 268,
288
- 269,
289
- 270,
290
- 271,
291
- 272,
292
- 273,
293
- 274,
294
- 275,
295
- 276,
296
- 277,
297
- 278,
298
- 279,
299
- 280,
300
- 281,
301
- 282,
302
- 283,
303
- 284,
304
- 285,
305
- 286,
306
- 287,
307
- 288,
308
- 289,
309
- 290,
310
- 291,
311
- 292,
312
- 293,
313
- 294,
314
- 295,
315
- 296,
316
- 297,
317
- 298,
318
- 299,
319
- 300,
320
- 301,
321
- 302,
322
- 303,
323
- 304,
324
- 305,
325
- 306,
326
- 307,
327
- 308,
328
- 309,
329
- 310,
330
- 311,
331
- 312,
332
- 313,
333
- 314,
334
- 315,
335
- 316,
336
- 317,
337
- 318,
338
- 319,
339
- 320,
340
- 321,
341
- 322,
342
- 323,
343
- 324,
344
- 325,
345
- 326,
346
- 327,
347
- 328,
348
- 329,
349
- 330,
350
- 331,
351
- 332,
352
- 333,
353
- 334,
354
- 335,
355
- 336,
356
- 337,
357
- 338,
358
- 339,
359
- 340,
360
- 341,
361
- 342,
362
- 343,
363
- 344,
364
- 345,
365
- 346,
366
- 347,
367
- 348,
368
- 349,
369
- 350,
370
- 351,
371
- 352,
372
- 353,
373
- 354,
374
- 355,
375
- 356,
376
- 357,
377
- 358,
378
- 359,
379
- 360,
380
- 361,
381
- 362,
382
- 363,
383
- 364,
384
- 365,
385
- 366,
386
- 367,
387
- 368,
388
- 369,
389
- 370,
390
- 371,
391
- 372,
392
- 373,
393
- 374
394
- ]
395
- }
396
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/100000/training_state/optimizer_state.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:9881578187b341bf96fd3589c3e1f96f32ebd295f9565e9536b3070e9fe4806d
3
- size 2014217744
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/100000/training_state/rng_state.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:a35119affd48626f70ad0ad8d0b06543865e56a13aa1775da4924c4dd0fdbeb4
3
- size 15708
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/100000/training_state/scheduler_state.json DELETED
@@ -1,15 +0,0 @@
1
- {
2
- "base_lrs": [
3
- 0.0001
4
- ],
5
- "last_epoch": 100000,
6
- "_step_count": 100001,
7
- "_is_initial": false,
8
- "_get_lr_called_within_step": false,
9
- "_last_lr": [
10
- 5.0196841136498776e-05
11
- ],
12
- "lr_lambdas": [
13
- null
14
- ]
15
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/100000/training_state/training_step.json DELETED
@@ -1,3 +0,0 @@
1
- {
2
- "step": 100000
3
- }
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/200000/pretrained_model/config.json DELETED
@@ -1,90 +0,0 @@
1
- {
2
- "type": "diffusion",
3
- "n_obs_steps": 2,
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
- },
20
- "output_features": {
21
- "action": {
22
- "type": "ACTION",
23
- "shape": [
24
- 6
25
- ]
26
- }
27
- },
28
- "device": "cuda",
29
- "use_amp": false,
30
- "use_peft": false,
31
- "push_to_hub": false,
32
- "repo_id": null,
33
- "private": null,
34
- "tags": null,
35
- "license": null,
36
- "pretrained_path": null,
37
- "horizon": 16,
38
- "n_action_steps": 8,
39
- "normalization_mapping": {
40
- "VISUAL": "MEAN_STD",
41
- "STATE": "MIN_MAX",
42
- "ACTION": "MIN_MAX"
43
- },
44
- "drop_n_last_frames": 7,
45
- "vision_backbone_type": "dinov2",
46
- "vision_encoder_name": "facebook/dinov2-small",
47
- "freeze_vision_encoder": true,
48
- "vision_backbone": "resnet18",
49
- "resize_shape": [
50
- 224,
51
- 224
52
- ],
53
- "crop_ratio": 1.0,
54
- "crop_shape": null,
55
- "crop_is_random": true,
56
- "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
57
- "use_group_norm": false,
58
- "spatial_softmax_num_keypoints": 32,
59
- "use_separate_rgb_encoder_per_camera": true,
60
- "down_dims": [
61
- 512,
62
- 1024,
63
- 2048
64
- ],
65
- "kernel_size": 5,
66
- "n_groups": 8,
67
- "diffusion_step_embed_dim": 128,
68
- "use_film_scale_modulation": true,
69
- "noise_scheduler_type": "DDPM",
70
- "num_train_timesteps": 100,
71
- "beta_schedule": "squaredcos_cap_v2",
72
- "beta_start": 0.0001,
73
- "beta_end": 0.02,
74
- "prediction_type": "epsilon",
75
- "clip_sample": true,
76
- "clip_sample_range": 1.0,
77
- "num_inference_steps": 100,
78
- "compile_model": false,
79
- "compile_mode": "reduce-overhead",
80
- "do_mask_loss_for_padding": false,
81
- "optimizer_lr": 0.0001,
82
- "optimizer_betas": [
83
- 0.95,
84
- 0.999
85
- ],
86
- "optimizer_eps": 1e-08,
87
- "optimizer_weight_decay": 1e-06,
88
- "scheduler_name": "cosine",
89
- "scheduler_warmup_steps": 500
90
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/200000/pretrained_model/model.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:a8b50ffd241953154e665f010749b298e87f0ffdbf3db11d0830d601b65474e8
3
- size 1095367712
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/200000/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
- 6
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
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/200000/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:f23b22ea05e4b3f2a72f520a60ef30fb248bf15c7b59132b9c3e5e0f3eabcc1c
3
- size 6560
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/200000/pretrained_model/policy_preprocessor.json DELETED
@@ -1,56 +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
- 6
30
- ]
31
- },
32
- "observation.images.front": {
33
- "type": "VISUAL",
34
- "shape": [
35
- 3,
36
- 480,
37
- 640
38
- ]
39
- },
40
- "action": {
41
- "type": "ACTION",
42
- "shape": [
43
- 6
44
- ]
45
- }
46
- },
47
- "norm_map": {
48
- "VISUAL": "MEAN_STD",
49
- "STATE": "MIN_MAX",
50
- "ACTION": "MIN_MAX"
51
- }
52
- },
53
- "state_file": "policy_preprocessor_step_3_normalizer_processor.safetensors"
54
- }
55
- ]
56
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/200000/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:f23b22ea05e4b3f2a72f520a60ef30fb248bf15c7b59132b9c3e5e0f3eabcc1c
3
- size 6560
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/200000/pretrained_model/train_config.json DELETED
@@ -1,300 +0,0 @@
1
- {
2
- "dataset": {
3
- "repo_id": "local/aggregated_new_dataset_532",
4
- "root": "/home/shadeform/rl_project/lerobot/../aggregated_new_dataset_532",
5
- "episodes": null,
6
- "image_transforms": {
7
- "enable": true,
8
- "max_num_transforms": 3,
9
- "random_order": false,
10
- "same_cloth_color": false,
11
- "tfs": {
12
- "color_jitter": {
13
- "weight": 1.5,
14
- "type": "ColorJitter",
15
- "kwargs": {
16
- "brightness": [
17
- 0.7,
18
- 1.3
19
- ],
20
- "contrast": [
21
- 0.7,
22
- 1.3
23
- ],
24
- "saturation": [
25
- 0.4,
26
- 1.6
27
- ],
28
- "hue": [
29
- -0.5,
30
- 0.5
31
- ]
32
- }
33
- },
34
- "grayscale": {
35
- "weight": 0.5,
36
- "type": "RandomGrayscale",
37
- "kwargs": {
38
- "p": 1.0
39
- }
40
- },
41
- "shadow": {
42
- "weight": 0.8,
43
- "type": "RandomShadow",
44
- "kwargs": {
45
- "num_shadows": [
46
- 1,
47
- 2
48
- ],
49
- "intensity": [
50
- 0.4,
51
- 0.7
52
- ],
53
- "blur_sigma": [
54
- 4.0,
55
- 20.0
56
- ]
57
- }
58
- },
59
- "highlight": {
60
- "weight": 0.5,
61
- "type": "RandomHighlight",
62
- "kwargs": {
63
- "num_highlights": [
64
- 0,
65
- 1
66
- ],
67
- "intensity": [
68
- 0.2,
69
- 0.5
70
- ],
71
- "sigma": [
72
- 20.0,
73
- 60.0
74
- ]
75
- }
76
- },
77
- "gamma": {
78
- "weight": 0.7,
79
- "type": "RandomGamma",
80
- "kwargs": {
81
- "gamma": [
82
- 0.6,
83
- 1.4
84
- ]
85
- }
86
- },
87
- "blur": {
88
- "weight": 0.0,
89
- "type": "GaussianBlur",
90
- "kwargs": {
91
- "kernel_size": [
92
- 3,
93
- 3
94
- ],
95
- "sigma": [
96
- 0.1,
97
- 1.5
98
- ]
99
- }
100
- },
101
- "jpeg": {
102
- "weight": 0.0,
103
- "type": "RandomJPEG",
104
- "kwargs": {
105
- "quality": [
106
- 60,
107
- 95
108
- ]
109
- }
110
- },
111
- "crop": {
112
- "weight": 0.5,
113
- "type": "RandomCropPreserveSize",
114
- "kwargs": {
115
- "scale": [
116
- 0.97,
117
- 1.0
118
- ],
119
- "ratio": [
120
- 0.99,
121
- 1.01
122
- ]
123
- }
124
- },
125
- "affine": {
126
- "weight": 0.0,
127
- "type": "RandomAffine",
128
- "kwargs": {
129
- "degrees": [
130
- -5.0,
131
- 5.0
132
- ],
133
- "translate": [
134
- 0.05,
135
- 0.05
136
- ]
137
- }
138
- },
139
- "sharpness": {
140
- "weight": 1.0,
141
- "type": "SharpnessJitter",
142
- "kwargs": {
143
- "sharpness": [
144
- 0.5,
145
- 1.5
146
- ]
147
- }
148
- }
149
- }
150
- },
151
- "revision": null,
152
- "use_imagenet_stats": true,
153
- "video_backend": "pyav",
154
- "return_uint8": false,
155
- "streaming": false
156
- },
157
- "env": null,
158
- "policy": {
159
- "type": "diffusion",
160
- "n_obs_steps": 2,
161
- "input_features": {
162
- "observation.state": {
163
- "type": "STATE",
164
- "shape": [
165
- 6
166
- ]
167
- },
168
- "observation.images.front": {
169
- "type": "VISUAL",
170
- "shape": [
171
- 3,
172
- 480,
173
- 640
174
- ]
175
- }
176
- },
177
- "output_features": {
178
- "action": {
179
- "type": "ACTION",
180
- "shape": [
181
- 6
182
- ]
183
- }
184
- },
185
- "device": "cuda",
186
- "use_amp": false,
187
- "use_peft": false,
188
- "push_to_hub": false,
189
- "repo_id": null,
190
- "private": null,
191
- "tags": null,
192
- "license": null,
193
- "pretrained_path": null,
194
- "horizon": 16,
195
- "n_action_steps": 8,
196
- "normalization_mapping": {
197
- "VISUAL": "MEAN_STD",
198
- "STATE": "MIN_MAX",
199
- "ACTION": "MIN_MAX"
200
- },
201
- "drop_n_last_frames": 7,
202
- "vision_backbone_type": "dinov2",
203
- "vision_encoder_name": "facebook/dinov2-small",
204
- "freeze_vision_encoder": true,
205
- "vision_backbone": "resnet18",
206
- "resize_shape": [
207
- 224,
208
- 224
209
- ],
210
- "crop_ratio": 1.0,
211
- "crop_shape": null,
212
- "crop_is_random": true,
213
- "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
214
- "use_group_norm": false,
215
- "spatial_softmax_num_keypoints": 32,
216
- "use_separate_rgb_encoder_per_camera": true,
217
- "down_dims": [
218
- 512,
219
- 1024,
220
- 2048
221
- ],
222
- "kernel_size": 5,
223
- "n_groups": 8,
224
- "diffusion_step_embed_dim": 128,
225
- "use_film_scale_modulation": true,
226
- "noise_scheduler_type": "DDPM",
227
- "num_train_timesteps": 100,
228
- "beta_schedule": "squaredcos_cap_v2",
229
- "beta_start": 0.0001,
230
- "beta_end": 0.02,
231
- "prediction_type": "epsilon",
232
- "clip_sample": true,
233
- "clip_sample_range": 1.0,
234
- "num_inference_steps": 100,
235
- "compile_model": false,
236
- "compile_mode": "reduce-overhead",
237
- "do_mask_loss_for_padding": false,
238
- "optimizer_lr": 0.0001,
239
- "optimizer_betas": [
240
- 0.95,
241
- 0.999
242
- ],
243
- "optimizer_eps": 1e-08,
244
- "optimizer_weight_decay": 1e-06,
245
- "scheduler_name": "cosine",
246
- "scheduler_warmup_steps": 500
247
- },
248
- "reward_model": null,
249
- "output_dir": "/home/shadeform/rl_project/outputs/train/diffusion_laptop_20260518_183541",
250
- "job_name": "diffusion_laptop_20260518_183541",
251
- "resume": false,
252
- "seed": 1000,
253
- "cudnn_deterministic": false,
254
- "num_workers": 16,
255
- "batch_size": 32,
256
- "prefetch_factor": 4,
257
- "persistent_workers": true,
258
- "steps": 200000,
259
- "eval_freq": 0,
260
- "log_freq": 20,
261
- "tolerance_s": 0.0001,
262
- "save_checkpoint": true,
263
- "save_freq": 100000,
264
- "use_policy_training_preset": true,
265
- "optimizer": {
266
- "type": "adam",
267
- "lr": 0.0001,
268
- "weight_decay": 1e-06,
269
- "grad_clip_norm": 10.0,
270
- "betas": [
271
- 0.95,
272
- 0.999
273
- ],
274
- "eps": 1e-08
275
- },
276
- "scheduler": {
277
- "type": "diffuser",
278
- "num_warmup_steps": 500,
279
- "name": "cosine"
280
- },
281
- "eval": {
282
- "n_episodes": 50,
283
- "batch_size": 11,
284
- "use_async_envs": true
285
- },
286
- "wandb": {
287
- "enable": true,
288
- "disable_artifact": false,
289
- "project": "lerobot",
290
- "entity": null,
291
- "notes": null,
292
- "run_id": "qy0stoak",
293
- "mode": null,
294
- "add_tags": true
295
- },
296
- "peft": null,
297
- "sample_weighting": null,
298
- "rename_map": {},
299
- "checkpoint_path": null
300
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/200000/training_state/optimizer_param_groups.json DELETED
@@ -1,396 +0,0 @@
1
- [
2
- {
3
- "lr": 0.0,
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
- 212,
232
- 213,
233
- 214,
234
- 215,
235
- 216,
236
- 217,
237
- 218,
238
- 219,
239
- 220,
240
- 221,
241
- 222,
242
- 223,
243
- 224,
244
- 225,
245
- 226,
246
- 227,
247
- 228,
248
- 229,
249
- 230,
250
- 231,
251
- 232,
252
- 233,
253
- 234,
254
- 235,
255
- 236,
256
- 237,
257
- 238,
258
- 239,
259
- 240,
260
- 241,
261
- 242,
262
- 243,
263
- 244,
264
- 245,
265
- 246,
266
- 247,
267
- 248,
268
- 249,
269
- 250,
270
- 251,
271
- 252,
272
- 253,
273
- 254,
274
- 255,
275
- 256,
276
- 257,
277
- 258,
278
- 259,
279
- 260,
280
- 261,
281
- 262,
282
- 263,
283
- 264,
284
- 265,
285
- 266,
286
- 267,
287
- 268,
288
- 269,
289
- 270,
290
- 271,
291
- 272,
292
- 273,
293
- 274,
294
- 275,
295
- 276,
296
- 277,
297
- 278,
298
- 279,
299
- 280,
300
- 281,
301
- 282,
302
- 283,
303
- 284,
304
- 285,
305
- 286,
306
- 287,
307
- 288,
308
- 289,
309
- 290,
310
- 291,
311
- 292,
312
- 293,
313
- 294,
314
- 295,
315
- 296,
316
- 297,
317
- 298,
318
- 299,
319
- 300,
320
- 301,
321
- 302,
322
- 303,
323
- 304,
324
- 305,
325
- 306,
326
- 307,
327
- 308,
328
- 309,
329
- 310,
330
- 311,
331
- 312,
332
- 313,
333
- 314,
334
- 315,
335
- 316,
336
- 317,
337
- 318,
338
- 319,
339
- 320,
340
- 321,
341
- 322,
342
- 323,
343
- 324,
344
- 325,
345
- 326,
346
- 327,
347
- 328,
348
- 329,
349
- 330,
350
- 331,
351
- 332,
352
- 333,
353
- 334,
354
- 335,
355
- 336,
356
- 337,
357
- 338,
358
- 339,
359
- 340,
360
- 341,
361
- 342,
362
- 343,
363
- 344,
364
- 345,
365
- 346,
366
- 347,
367
- 348,
368
- 349,
369
- 350,
370
- 351,
371
- 352,
372
- 353,
373
- 354,
374
- 355,
375
- 356,
376
- 357,
377
- 358,
378
- 359,
379
- 360,
380
- 361,
381
- 362,
382
- 363,
383
- 364,
384
- 365,
385
- 366,
386
- 367,
387
- 368,
388
- 369,
389
- 370,
390
- 371,
391
- 372,
392
- 373,
393
- 374
394
- ]
395
- }
396
- ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/200000/training_state/optimizer_state.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:69136d956e58faacec80f3fd5ecc05d92bfadae6ce736b57fe7c17deef0feb03
3
- size 2014217744
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/200000/training_state/rng_state.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:ba13091283e99b0538ebd8af8d00e4f4b93f768c80457d4da441aac4ea75973a
3
- size 15708
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/200000/training_state/scheduler_state.json DELETED
@@ -1,15 +0,0 @@
1
- {
2
- "base_lrs": [
3
- 0.0001
4
- ],
5
- "last_epoch": 200000,
6
- "_step_count": 200001,
7
- "_is_initial": false,
8
- "_get_lr_called_within_step": false,
9
- "_last_lr": [
10
- 0.0
11
- ],
12
- "lr_lambdas": [
13
- null
14
- ]
15
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
diffusion_unet_dino_350_augment/checkpoints/200000/training_state/training_step.json DELETED
@@ -1,3 +0,0 @@
1
- {
2
- "step": 200000
3
- }