Hugging Face's logo

rlearning2026romsdrums
/
diffusion_unet_checkpoints

Safetensors
Model card Files
xet
 Community
rguntz commited on
Commit
5d1b7dc
·
verified ·
1 Parent(s): 5f9c349

Delete checkpoints/diffusion_unet_dino

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