Hugging Face's logo

rlearning2026romsdrums
/
diffusion_unet_checkpoints

Safetensors
Model card Files
xet
 Community
rguntz commited on
Commit
fd14929
·
verified ·
1 Parent(s): 8b36842

Upload folder using huggingface_hub

Browse files
Files changed (22) hide show
  1. .gitattributes +1 -0
  2. checkpoints/generalized_eval1/checkpoints/002000/pretrained_model/config.json +92 -0
  3. checkpoints/generalized_eval1/checkpoints/002000/pretrained_model/model.safetensors +3 -0
  4. checkpoints/generalized_eval1/checkpoints/002000/pretrained_model/policy_postprocessor.json +32 -0
  5. checkpoints/generalized_eval1/checkpoints/002000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors +3 -0
  6. checkpoints/generalized_eval1/checkpoints/002000/pretrained_model/policy_preprocessor.json +56 -0
  7. checkpoints/generalized_eval1/checkpoints/002000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors +3 -0
  8. checkpoints/generalized_eval1/checkpoints/002000/pretrained_model/train_config.json +302 -0
  9. checkpoints/generalized_eval1/checkpoints/002000/training_state/optimizer_param_groups.json +396 -0
  10. checkpoints/generalized_eval1/checkpoints/002000/training_state/optimizer_state.safetensors +3 -0
  11. checkpoints/generalized_eval1/checkpoints/002000/training_state/rng_state.safetensors +3 -0
  12. checkpoints/generalized_eval1/checkpoints/002000/training_state/scheduler_state.json +15 -0
  13. checkpoints/generalized_eval1/checkpoints/002000/training_state/training_step.json +3 -0
  14. checkpoints/generalized_eval1/wandb/debug-internal.log +95 -0
  15. checkpoints/generalized_eval1/wandb/debug.log +19 -0
  16. checkpoints/generalized_eval1/wandb/run-20260521_122331-91fmyb38/files/output.log +126 -0
  17. checkpoints/generalized_eval1/wandb/run-20260521_122331-91fmyb38/files/requirements.txt +114 -0
  18. checkpoints/generalized_eval1/wandb/run-20260521_122331-91fmyb38/files/wandb-metadata.json +60 -0
  19. checkpoints/generalized_eval1/wandb/run-20260521_122331-91fmyb38/logs/debug-core.log +7 -0
  20. checkpoints/generalized_eval1/wandb/run-20260521_122331-91fmyb38/logs/debug-internal.log +95 -0
  21. checkpoints/generalized_eval1/wandb/run-20260521_122331-91fmyb38/logs/debug.log +19 -0
  22. checkpoints/generalized_eval1/wandb/run-20260521_122331-91fmyb38/run-91fmyb38.wandb +3 -0
.gitattributes CHANGED
@@ -52,3 +52,4 @@ checkpoints/dagger/diffusion_dagger_iter3_25000steps_20260521_015658/wandb/run-2
52
  checkpoints/diffusion_base_dataset_dino/wandb/run-20260520_132058-etjgt65i/run-etjgt65i.wandb filter=lfs diff=lfs merge=lfs -text
53
  checkpoints/diffusion_base_dataset_523_dino/wandb/run-20260520_132058-etjgt65i/run-etjgt65i.wandb filter=lfs diff=lfs merge=lfs -text
54
  checkpoints/dinov3-final/wandb/run-20260519_235029-pu35iy7k/run-pu35iy7k.wandb filter=lfs diff=lfs merge=lfs -text
 
 
52
  checkpoints/diffusion_base_dataset_dino/wandb/run-20260520_132058-etjgt65i/run-etjgt65i.wandb filter=lfs diff=lfs merge=lfs -text
53
  checkpoints/diffusion_base_dataset_523_dino/wandb/run-20260520_132058-etjgt65i/run-etjgt65i.wandb filter=lfs diff=lfs merge=lfs -text
54
  checkpoints/dinov3-final/wandb/run-20260519_235029-pu35iy7k/run-pu35iy7k.wandb filter=lfs diff=lfs merge=lfs -text
55
+ checkpoints/generalized_eval1/wandb/run-20260521_122331-91fmyb38/run-91fmyb38.wandb filter=lfs diff=lfs merge=lfs -text
checkpoints/generalized_eval1/checkpoints/002000/pretrained_model/config.json ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "/home/shadeform/Desktop/robot_learning/diffusion_780_new_dataset_dino/checkpoints/diffusion_780_new_dataset_dino/checkpoints/400000/pretrained_model",
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
+ "vit_pool_type": "spatial_softmax",
60
+ "vit_feature_dim": 512,
61
+ "use_separate_rgb_encoder_per_camera": true,
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": 100,
80
+ "compile_model": false,
81
+ "compile_mode": "reduce-overhead",
82
+ "do_mask_loss_for_padding": false,
83
+ "optimizer_lr": 5e-06,
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": 100
92
+ }
checkpoints/generalized_eval1/checkpoints/002000/pretrained_model/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:56ab2428b461d3b048de451b52f559f052d2387a9703c78b558d1147f255151c
3
+ size 1095367712
checkpoints/generalized_eval1/checkpoints/002000/pretrained_model/policy_postprocessor.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "policy_postprocessor",
3
+ "steps": [
4
+ {
5
+ "registry_name": "unnormalizer_processor",
6
+ "config": {
7
+ "eps": 1e-08,
8
+ "features": {
9
+ "action": {
10
+ "type": "ACTION",
11
+ "shape": [
12
+ 6
13
+ ]
14
+ }
15
+ },
16
+ "norm_map": {
17
+ "VISUAL": "MEAN_STD",
18
+ "STATE": "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/generalized_eval1/checkpoints/002000/pretrained_model/policy_postprocessor_step_0_unnormalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f3d380169b508a971c4d5acf43be6d884af56ce0afe1112d643afcf915cc2bb
3
+ size 6560
checkpoints/generalized_eval1/checkpoints/002000/pretrained_model/policy_preprocessor.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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/generalized_eval1/checkpoints/002000/pretrained_model/policy_preprocessor_step_3_normalizer_processor.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f3d380169b508a971c4d5acf43be6d884af56ce0afe1112d643afcf915cc2bb
3
+ size 6560
checkpoints/generalized_eval1/checkpoints/002000/pretrained_model/train_config.json ADDED
@@ -0,0 +1,302 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": {
3
+ "repo_id": "local/act_collected_episodes",
4
+ "root": "/home/shadeform/Desktop/robot_learning/random_pick_new_plus_new300_dark",
5
+ "episodes": null,
6
+ "image_transforms": {
7
+ "enable": false,
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.0,
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.0,
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": "/home/shadeform/Desktop/robot_learning/diffusion_780_new_dataset_dino/checkpoints/diffusion_780_new_dataset_dino/checkpoints/400000/pretrained_model",
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
+ "vit_pool_type": "spatial_softmax",
217
+ "vit_feature_dim": 512,
218
+ "use_separate_rgb_encoder_per_camera": true,
219
+ "down_dims": [
220
+ 512,
221
+ 1024,
222
+ 2048
223
+ ],
224
+ "kernel_size": 5,
225
+ "n_groups": 8,
226
+ "diffusion_step_embed_dim": 128,
227
+ "use_film_scale_modulation": true,
228
+ "noise_scheduler_type": "DDPM",
229
+ "num_train_timesteps": 100,
230
+ "beta_schedule": "squaredcos_cap_v2",
231
+ "beta_start": 0.0001,
232
+ "beta_end": 0.02,
233
+ "prediction_type": "epsilon",
234
+ "clip_sample": true,
235
+ "clip_sample_range": 1.0,
236
+ "num_inference_steps": 100,
237
+ "compile_model": false,
238
+ "compile_mode": "reduce-overhead",
239
+ "do_mask_loss_for_padding": false,
240
+ "optimizer_lr": 5e-06,
241
+ "optimizer_betas": [
242
+ 0.95,
243
+ 0.999
244
+ ],
245
+ "optimizer_eps": 1e-08,
246
+ "optimizer_weight_decay": 1e-06,
247
+ "scheduler_name": "cosine",
248
+ "scheduler_warmup_steps": 100
249
+ },
250
+ "reward_model": null,
251
+ "output_dir": "/home/shadeform/Desktop/robot_learning/outputs/train/diffusion_fine_tune_generalized_1",
252
+ "job_name": "diffusion_fine_tune_generalized_1",
253
+ "resume": false,
254
+ "seed": 1000,
255
+ "cudnn_deterministic": false,
256
+ "num_workers": 8,
257
+ "batch_size": 32,
258
+ "prefetch_factor": 4,
259
+ "persistent_workers": true,
260
+ "steps": 25000,
261
+ "eval_freq": 0,
262
+ "log_freq": 20,
263
+ "tolerance_s": 0.0001,
264
+ "save_checkpoint": true,
265
+ "save_freq": 2000,
266
+ "use_policy_training_preset": true,
267
+ "optimizer": {
268
+ "type": "adam",
269
+ "lr": 5e-06,
270
+ "weight_decay": 1e-06,
271
+ "grad_clip_norm": 10.0,
272
+ "betas": [
273
+ 0.95,
274
+ 0.999
275
+ ],
276
+ "eps": 1e-08
277
+ },
278
+ "scheduler": {
279
+ "type": "diffuser",
280
+ "num_warmup_steps": 100,
281
+ "name": "cosine"
282
+ },
283
+ "eval": {
284
+ "n_episodes": 50,
285
+ "batch_size": 19,
286
+ "use_async_envs": true
287
+ },
288
+ "wandb": {
289
+ "enable": true,
290
+ "disable_artifact": false,
291
+ "project": "lerobot",
292
+ "entity": null,
293
+ "notes": null,
294
+ "run_id": "91fmyb38",
295
+ "mode": null,
296
+ "add_tags": true
297
+ },
298
+ "peft": null,
299
+ "sample_weighting": null,
300
+ "rename_map": {},
301
+ "checkpoint_path": null
302
+ }
checkpoints/generalized_eval1/checkpoints/002000/training_state/optimizer_param_groups.json ADDED
@@ -0,0 +1,396 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ {
3
+ "lr": 4.928511280455169e-06,
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": 5e-06,
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/generalized_eval1/checkpoints/002000/training_state/optimizer_state.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:277e6fd4208ece2511235625de24a47745bf21d48886c3479ee2123a01f9b6f9
3
+ size 2014217744
checkpoints/generalized_eval1/checkpoints/002000/training_state/rng_state.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd22954f10666b932ba0940f086e97c4213a28d4c5ea3a34728e9c4654d3c938
3
+ size 15708
checkpoints/generalized_eval1/checkpoints/002000/training_state/scheduler_state.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "base_lrs": [
3
+ 5e-06
4
+ ],
5
+ "last_epoch": 2000,
6
+ "_step_count": 2001,
7
+ "_is_initial": false,
8
+ "_get_lr_called_within_step": false,
9
+ "_last_lr": [
10
+ 4.928511280455169e-06
11
+ ],
12
+ "lr_lambdas": [
13
+ null
14
+ ]
15
+ }
checkpoints/generalized_eval1/checkpoints/002000/training_state/training_step.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "step": 2000
3
+ }
checkpoints/generalized_eval1/wandb/debug-internal.log ADDED
@@ -0,0 +1,95 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-05-21T12:23:31.235249938Z","level":"INFO","msg":"wandb-core"}
2
+ {"time":"2026-05-21T12:23:31.235495117Z","level":"INFO","msg":"stream: starting","core version":"0.27.0"}
3
+ {"time":"2026-05-21T12:23:31.436106922Z","level":"INFO","msg":"stream: created new stream","id":"91fmyb38"}
4
+ {"time":"2026-05-21T12:23:31.436203377Z","level":"INFO","msg":"handler: started"}
5
+ {"time":"2026-05-21T12:23:31.436309017Z","level":"INFO","msg":"stream: started"}
6
+ {"time":"2026-05-21T12:23:31.436354926Z","level":"INFO","msg":"writer: started","stream_id":"91fmyb38"}
7
+ {"time":"2026-05-21T12:23:31.436376138Z","level":"INFO","msg":"sender: started"}
8
+ {"time":"2026-05-21T12:23:32.297492478Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":0,"console_lines":1}
9
+ {"time":"2026-05-21T12:23:32.837113Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
10
+ {"time":"2026-05-21T12:23:47.297546752Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":0,"events_lines":1,"console_offset":1,"console_lines":17,"uploaded_len":2}
11
+ {"time":"2026-05-21T12:23:47.515947561Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
12
+ {"time":"2026-05-21T12:24:02.298262523Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":0,"history_lines":3,"events_offset":1,"events_lines":2,"console_offset":17,"console_lines":4}
13
+ {"time":"2026-05-21T12:24:02.561305727Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
14
+ {"time":"2026-05-21T12:24:17.298235128Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":3,"history_lines":3,"events_offset":3,"events_lines":2,"console_offset":20,"console_lines":4}
15
+ {"time":"2026-05-21T12:24:17.936748721Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
16
+ {"time":"2026-05-21T12:24:32.298300854Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":6,"history_lines":2,"events_offset":5,"events_lines":2,"console_offset":23,"console_lines":3}
17
+ {"time":"2026-05-21T12:24:32.699635088Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
18
+ {"time":"2026-05-21T12:24:47.298617002Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":8,"history_lines":3,"events_offset":7,"events_lines":2,"console_offset":25,"console_lines":4}
19
+ {"time":"2026-05-21T12:24:48.265336142Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
20
+ {"time":"2026-05-21T12:25:02.298222417Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":11,"history_lines":3,"events_offset":9,"events_lines":2,"console_offset":28,"console_lines":3}
21
+ {"time":"2026-05-21T12:25:03.341512826Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
22
+ {"time":"2026-05-21T12:25:17.297761366Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":14,"history_lines":2,"events_offset":11,"events_lines":2,"console_offset":31,"console_lines":3}
23
+ {"time":"2026-05-21T12:25:17.60161801Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
24
+ {"time":"2026-05-21T12:25:32.297669007Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":16,"history_lines":3,"events_offset":13,"events_lines":2,"console_offset":33,"console_lines":4}
25
+ {"time":"2026-05-21T12:25:32.507496784Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
26
+ {"time":"2026-05-21T12:25:47.297895545Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":19,"history_lines":2,"events_offset":15,"events_lines":2,"console_offset":36,"console_lines":3}
27
+ {"time":"2026-05-21T12:25:48.245571706Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
28
+ {"time":"2026-05-21T12:26:02.298042403Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":21,"history_lines":3,"events_offset":17,"events_lines":2,"console_offset":38,"console_lines":4}
29
+ {"time":"2026-05-21T12:26:02.552618844Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
30
+ {"time":"2026-05-21T12:26:17.298031263Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":24,"history_lines":2,"events_offset":19,"events_lines":2,"console_offset":41,"console_lines":3}
31
+ {"time":"2026-05-21T12:26:18.196392246Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
32
+ {"time":"2026-05-21T12:26:32.299060379Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":26,"history_lines":3,"events_offset":21,"events_lines":2,"console_offset":43,"console_lines":4}
33
+ {"time":"2026-05-21T12:26:32.967831752Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
34
+ {"time":"2026-05-21T12:26:47.298303969Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":29,"history_lines":3,"events_offset":23,"events_lines":2,"console_offset":46,"console_lines":3}
35
+ {"time":"2026-05-21T12:26:47.83934363Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
36
+ {"time":"2026-05-21T12:27:02.298402147Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":32,"history_lines":2,"events_offset":25,"events_lines":2,"console_offset":49,"console_lines":3}
37
+ {"time":"2026-05-21T12:27:02.919167995Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
38
+ {"time":"2026-05-21T12:27:17.29808968Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":34,"history_lines":2,"events_offset":27,"events_lines":2,"console_offset":51,"console_lines":3}
39
+ {"time":"2026-05-21T12:27:17.810947902Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
40
+ {"time":"2026-05-21T12:27:32.298013663Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":36,"history_lines":3,"events_offset":29,"events_lines":2,"console_offset":53,"console_lines":4}
41
+ {"time":"2026-05-21T12:27:33.419987472Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
42
+ {"time":"2026-05-21T12:27:47.298428951Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":39,"history_lines":2,"events_offset":31,"events_lines":2,"console_offset":56,"console_lines":3}
43
+ {"time":"2026-05-21T12:27:47.487975401Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
44
+ {"time":"2026-05-21T12:28:02.297700362Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":41,"history_lines":3,"events_offset":33,"events_lines":2,"console_offset":58,"console_lines":4}
45
+ {"time":"2026-05-21T12:28:02.53586069Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
46
+ {"time":"2026-05-21T12:28:17.298434159Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":44,"history_lines":2,"events_offset":35,"events_lines":2,"console_offset":61,"console_lines":3}
47
+ {"time":"2026-05-21T12:28:17.678643338Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
48
+ {"time":"2026-05-21T12:28:32.2985949Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":46,"history_lines":3,"events_offset":37,"events_lines":2,"console_offset":63,"console_lines":4}
49
+ {"time":"2026-05-21T12:28:32.502397298Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
50
+ {"time":"2026-05-21T12:28:47.297947735Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":49,"history_lines":2,"events_offset":39,"events_lines":2,"console_offset":66,"console_lines":3}
51
+ {"time":"2026-05-21T12:28:48.007234263Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
52
+ {"time":"2026-05-21T12:29:02.297700728Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":51,"history_lines":3,"events_offset":41,"events_lines":2,"console_offset":68,"console_lines":4}
53
+ {"time":"2026-05-21T12:29:02.728084572Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
54
+ {"time":"2026-05-21T12:29:17.298431882Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":54,"history_lines":3,"events_offset":43,"events_lines":2,"console_offset":71,"console_lines":4}
55
+ {"time":"2026-05-21T12:29:17.682900167Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
56
+ {"time":"2026-05-21T12:29:32.297791442Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":57,"history_lines":2,"events_offset":45,"events_lines":2,"console_offset":74,"console_lines":3}
57
+ {"time":"2026-05-21T12:29:32.716602354Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
58
+ {"time":"2026-05-21T12:29:47.297755622Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":59,"history_lines":3,"events_offset":47,"events_lines":2,"console_offset":76,"console_lines":3}
59
+ {"time":"2026-05-21T12:29:48.158392751Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
60
+ {"time":"2026-05-21T12:30:02.298385353Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":62,"history_lines":2,"events_offset":49,"events_lines":2,"console_offset":79,"console_lines":3}
61
+ {"time":"2026-05-21T12:30:02.643764568Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
62
+ {"time":"2026-05-21T12:30:17.298434031Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":64,"history_lines":3,"events_offset":51,"events_lines":2,"console_offset":81,"console_lines":4}
63
+ {"time":"2026-05-21T12:30:18.505473171Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
64
+ {"time":"2026-05-21T12:30:32.297993966Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":67,"history_lines":2,"events_offset":53,"events_lines":2,"console_offset":84,"console_lines":3}
65
+ {"time":"2026-05-21T12:30:32.851250185Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
66
+ {"time":"2026-05-21T12:30:47.298561389Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":69,"history_lines":3,"events_offset":55,"events_lines":2,"console_offset":86,"console_lines":3}
67
+ {"time":"2026-05-21T12:30:48.842829385Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
68
+ {"time":"2026-05-21T12:31:02.298526077Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":72,"history_lines":2,"events_offset":57,"events_lines":2,"console_offset":89,"console_lines":3}
69
+ {"time":"2026-05-21T12:31:02.585431307Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
70
+ {"time":"2026-05-21T12:31:17.298519151Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":74,"history_lines":3,"events_offset":59,"events_lines":2,"console_offset":91,"console_lines":4}
71
+ {"time":"2026-05-21T12:31:18.029621131Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
72
+ {"time":"2026-05-21T12:31:32.29801427Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":77,"history_lines":3,"events_offset":61,"events_lines":2,"console_offset":94,"console_lines":3}
73
+ {"time":"2026-05-21T12:31:32.951656172Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
74
+ {"time":"2026-05-21T12:31:47.297600579Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":80,"history_lines":2,"events_offset":63,"events_lines":2,"console_offset":97,"console_lines":3}
75
+ {"time":"2026-05-21T12:31:48.325261241Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
76
+ {"time":"2026-05-21T12:32:02.298369593Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":82,"history_lines":3,"events_offset":65,"events_lines":2,"console_offset":99,"console_lines":4}
77
+ {"time":"2026-05-21T12:32:02.719801212Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
78
+ {"time":"2026-05-21T12:32:17.298178577Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":85,"history_lines":3,"events_offset":67,"events_lines":2,"console_offset":102,"console_lines":3}
79
+ {"time":"2026-05-21T12:32:17.829373922Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
80
+ {"time":"2026-05-21T12:32:32.297731002Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":88,"history_lines":2,"events_offset":69,"events_lines":2,"console_offset":105,"console_lines":3}
81
+ {"time":"2026-05-21T12:32:32.868550719Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
82
+ {"time":"2026-05-21T12:32:47.297785787Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":90,"history_lines":3,"events_offset":71,"events_lines":2,"console_offset":107,"console_lines":4}
83
+ {"time":"2026-05-21T12:32:47.827099892Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
84
+ {"time":"2026-05-21T12:33:02.298392242Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":93,"history_lines":2,"events_offset":73,"events_lines":2,"console_offset":110,"console_lines":3}
85
+ {"time":"2026-05-21T12:33:02.965528874Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
86
+ {"time":"2026-05-21T12:33:17.298256374Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":95,"history_lines":3,"events_offset":75,"events_lines":2,"console_offset":112,"console_lines":3}
87
+ {"time":"2026-05-21T12:33:18.042320109Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
88
+ {"time":"2026-05-21T12:33:32.298320573Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":98,"history_lines":1,"events_offset":77,"events_lines":2,"console_offset":115,"console_lines":3}
89
+ {"time":"2026-05-21T12:33:33.095702573Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
90
+ {"time":"2026-05-21T12:33:47.298433704Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":99,"history_lines":4,"events_offset":79,"events_lines":2,"console_offset":117,"console_lines":5}
91
+ {"time":"2026-05-21T12:33:47.826432477Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
92
+ {"time":"2026-05-21T12:34:02.298252019Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":103,"history_lines":2,"events_offset":81,"events_lines":2,"console_offset":121,"console_lines":3}
93
+ {"time":"2026-05-21T12:34:02.741382701Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
94
+ {"time":"2026-05-21T12:34:17.298278155Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":105,"history_lines":2,"events_offset":83,"events_lines":2,"console_offset":123,"console_lines":3}
95
+ {"time":"2026-05-21T12:34:17.983055483Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
checkpoints/generalized_eval1/wandb/debug.log ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-05-21 12:23:31,014 INFO MainThread:1860986 [wandb_setup.py:_flush():81] Current SDK version is 0.27.0
2
+ 2026-05-21 12:23:31,015 INFO MainThread:1860986 [wandb_setup.py:_flush():81] Configure stats pid to 1860986
3
+ 2026-05-21 12:23:31,015 INFO MainThread:1860986 [wandb_setup.py:_flush():81] Loading settings from environment variables
4
+ 2026-05-21 12:23:31,015 INFO MainThread:1860986 [wandb_init.py:setup_run_log_directory():723] Logging user logs to /home/shadeform/Desktop/robot_learning/outputs/train/diffusion_fine_tune_generalized_1/wandb/run-20260521_122331-91fmyb38/logs/debug.log
5
+ 2026-05-21 12:23:31,015 INFO MainThread:1860986 [wandb_init.py:setup_run_log_directory():724] Logging internal logs to /home/shadeform/Desktop/robot_learning/outputs/train/diffusion_fine_tune_generalized_1/wandb/run-20260521_122331-91fmyb38/logs/debug-internal.log
6
+ 2026-05-21 12:23:31,015 INFO MainThread:1860986 [wandb_init.py:init():851] calling init triggers
7
+ 2026-05-21 12:23:31,015 INFO MainThread:1860986 [wandb_init.py:init():856] wandb.init called with sweep_config: {}
8
+ config: {'dataset': {'repo_id': 'local/act_collected_episodes', 'root': '/home/shadeform/Desktop/robot_learning/random_pick_new_plus_new300_dark', 'episodes': None, 'image_transforms': {'enable': False, 'max_num_transforms': 3, 'random_order': False, 'same_cloth_color': False, 'tfs': {'color_jitter': {'weight': 1.5, 'type': 'ColorJitter', 'kwargs': {'brightness': [0.7, 1.3], 'contrast': [0.7, 1.3], 'saturation': [0.4, 1.6], 'hue': [-0.5, 0.5]}}, 'grayscale': {'weight': 0.5, 'type': 'RandomGrayscale', 'kwargs': {'p': 1.0}}, 'shadow': {'weight': 0.0, 'type': 'RandomShadow', 'kwargs': {'num_shadows': [1, 2], 'intensity': [0.4, 0.7], 'blur_sigma': [4.0, 20.0]}}, 'highlight': {'weight': 0.0, 'type': 'RandomHighlight', 'kwargs': {'num_highlights': [0, 1], 'intensity': [0.2, 0.5], 'sigma': [20.0, 60.0]}}, 'gamma': {'weight': 0.7, 'type': 'RandomGamma', 'kwargs': {'gamma': [0.6, 1.4]}}, 'blur': {'weight': 0.0, 'type': 'GaussianBlur', 'kwargs': {'kernel_size': [3, 3], 'sigma': [0.1, 1.5]}}, 'jpeg': {'weight': 0.0, 'type': 'RandomJPEG', 'kwargs': {'quality': [60, 95]}}, 'crop': {'weight': 0.5, 'type': 'RandomCropPreserveSize', 'kwargs': {'scale': [0.97, 1.0], 'ratio': [0.99, 1.01]}}, 'affine': {'weight': 0.0, 'type': 'RandomAffine', 'kwargs': {'degrees': [-5.0, 5.0], 'translate': [0.05, 0.05]}}, 'sharpness': {'weight': 1.0, 'type': 'SharpnessJitter', 'kwargs': {'sharpness': [0.5, 1.5]}}}}, 'revision': None, 'use_imagenet_stats': True, 'video_backend': 'pyav', 'return_uint8': False, 'streaming': False}, 'env': None, 'policy': {'type': 'diffusion', 'n_obs_steps': 2, 'input_features': {'observation.state': {'type': <FeatureType.STATE: 'STATE'>, 'shape': [6]}, 'observation.images.front': {'type': <FeatureType.VISUAL: 'VISUAL'>, 'shape': [3, 480, 640]}}, 'output_features': {'action': {'type': <FeatureType.ACTION: 'ACTION'>, 'shape': [6]}}, 'device': 'cuda', 'use_amp': False, 'use_peft': False, 'push_to_hub': False, 'repo_id': None, 'private': None, 'tags': None, 'license': None, 'pretrained_path': '/home/shadeform/Desktop/robot_learning/diffusion_780_new_dataset_dino/checkpoints/diffusion_780_new_dataset_dino/checkpoints/400000/pretrained_model', 'horizon': 16, 'n_action_steps': 8, 'normalization_mapping': {'VISUAL': <NormalizationMode.MEAN_STD: 'MEAN_STD'>, 'STATE': <NormalizationMode.MIN_MAX: 'MIN_MAX'>, 'ACTION': <NormalizationMode.MIN_MAX: 'MIN_MAX'>}, 'drop_n_last_frames': 7, 'vision_backbone_type': 'dinov2', 'vision_encoder_name': 'facebook/dinov2-small', 'freeze_vision_encoder': True, 'vision_backbone': 'resnet18', 'resize_shape': [224, 224], 'crop_ratio': 1.0, 'crop_shape': None, 'crop_is_random': True, 'pretrained_backbone_weights': 'ResNet18_Weights.IMAGENET1K_V1', 'use_group_norm': False, 'spatial_softmax_num_keypoints': 32, 'vit_pool_type': 'spatial_softmax', 'vit_feature_dim': 512, 'use_separate_rgb_encoder_per_camera': True, 'down_dims': [512, 1024, 2048], 'kernel_size': 5, 'n_groups': 8, 'diffusion_step_embed_dim': 128, 'use_film_scale_modulation': True, 'noise_scheduler_type': 'DDPM', 'num_train_timesteps': 100, 'beta_schedule': 'squaredcos_cap_v2', 'beta_start': 0.0001, 'beta_end': 0.02, 'prediction_type': 'epsilon', 'clip_sample': True, 'clip_sample_range': 1.0, 'num_inference_steps': 100, 'compile_model': False, 'compile_mode': 'reduce-overhead', 'do_mask_loss_for_padding': False, 'optimizer_lr': 5e-06, 'optimizer_betas': [0.95, 0.999], 'optimizer_eps': 1e-08, 'optimizer_weight_decay': 1e-06, 'scheduler_name': 'cosine', 'scheduler_warmup_steps': 100}, 'reward_model': None, 'output_dir': '/home/shadeform/Desktop/robot_learning/outputs/train/diffusion_fine_tune_generalized_1', 'job_name': 'diffusion_fine_tune_generalized_1', 'resume': False, 'seed': 1000, 'cudnn_deterministic': False, 'num_workers': 8, 'batch_size': 32, 'prefetch_factor': 4, 'persistent_workers': True, 'steps': 25000, 'eval_freq': 0, 'log_freq': 20, 'tolerance_s': 0.0001, 'save_checkpoint': True, 'save_freq': 2000, 'use_policy_training_preset': True, 'optimizer': {'type': 'adam', 'lr': 5e-06, 'weight_decay': 1e-06, 'grad_clip_norm': 10.0, 'betas': [0.95, 0.999], 'eps': 1e-08}, 'scheduler': {'type': 'diffuser', 'num_warmup_steps': 100, 'name': 'cosine'}, 'eval': {'n_episodes': 50, 'batch_size': 19, 'use_async_envs': True}, 'wandb': {'enable': True, 'disable_artifact': False, 'project': 'lerobot', 'entity': None, 'notes': None, 'run_id': None, 'mode': None, 'add_tags': True}, 'peft': None, 'sample_weighting': None, 'rename_map': {}, 'checkpoint_path': None, '_wandb': {}}
9
+ 2026-05-21 12:23:31,015 INFO MainThread:1860986 [wandb_init.py:init():899] starting backend
10
+ 2026-05-21 12:23:31,230 INFO MainThread:1860986 [wandb_init.py:init():914] sending inform_init request
11
+ 2026-05-21 12:23:31,436 INFO MainThread:1860986 [wandb_init.py:init():919] backend started and connected
12
+ 2026-05-21 12:23:31,439 INFO MainThread:1860986 [wandb_init.py:init():989] updated telemetry
13
+ 2026-05-21 12:23:31,443 INFO MainThread:1860986 [wandb_init.py:init():1012] communicating run to backend with 90.0 second timeout
14
+ 2026-05-21 12:23:32,240 INFO MainThread:1860986 [wandb_init.py:init():1057] starting run threads in backend
15
+ 2026-05-21 12:23:32,295 INFO MainThread:1860986 [wandb_run.py:_console_start():2509] atexit reg
16
+ 2026-05-21 12:23:32,295 INFO MainThread:1860986 [wandb_run.py:_redirect():2359] redirect: wrap_raw
17
+ 2026-05-21 12:23:32,295 INFO MainThread:1860986 [wandb_run.py:_redirect():2428] Wrapping output streams.
18
+ 2026-05-21 12:23:32,295 INFO MainThread:1860986 [wandb_run.py:_redirect():2451] Redirects installed.
19
+ 2026-05-21 12:23:32,296 INFO MainThread:1860986 [wandb_init.py:init():1095] run started, returning control to user process
checkpoints/generalized_eval1/wandb/run-20260521_122331-91fmyb38/files/output.log ADDED
@@ -0,0 +1,126 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ INFO 2026-05-21 12:23:32 db_utils.py:121 Logs will be synced with wandb.
2
+ INFO 2026-05-21 12:23:32 db_utils.py:122 Track this run --> https://wandb.ai/romainguntz-eth-z-rich/lerobot/runs/91fmyb38
3
+ INFO 2026-05-21 12:23:32 ot_train.py:236 Creating dataset
4
+ INFO 2026-05-21 12:23:32 eo_utils.py:110 Using video codec: libsvtav1
5
+ INFO 2026-05-21 12:23:32 ot_train.py:270 Creating policy
6
+ Loading weights: 100%|███████████████████████████████████| 223/223 [00:00<00:00, 6950.25it/s]
7
+ Loading weights from local directory
8
+ INFO 2026-05-21 12:23:34 ot_train.py:347 Creating optimizer and scheduler
9
+ INFO 2026-05-21 12:23:34 ot_train.py:374 Output dir: /home/shadeform/Desktop/robot_learning/outputs/train/diffusion_fine_tune_generalized_1
10
+ INFO 2026-05-21 12:23:34 ot_train.py:381 cfg.steps=25000 (25K)
11
+ INFO 2026-05-21 12:23:34 ot_train.py:382 dataset.num_frames=254140 (254K)
12
+ INFO 2026-05-21 12:23:34 ot_train.py:383 dataset.num_episodes=405
13
+ INFO 2026-05-21 12:23:34 ot_train.py:386 Effective batch size: 32 x 1 = 32
14
+ INFO 2026-05-21 12:23:34 ot_train.py:387 num_learnable_params=251772134 (252M)
15
+ INFO 2026-05-21 12:23:34 ot_train.py:388 num_total_params=273828710 (274M)
16
+ Training: 0%| | 0/25000 [00:00<?, ?step/s]INFO 2026-05-21 12:23:34 ot_train.py:453 Start offline training on a fixed dataset, with effective batch size: 32
17
+ Training: 0%| | 19/25000 [00:07<1:22:17, 5.06step/s]INFO 2026-05-21 12:23:42 ot_train.py:488 step:20 smpl:640 ep:1 epch:0.00 loss:0.015 grdn:0.536 lr:5.2e-07 updt_s:0.184 data_s:0.218
18
+ Training: 0%| | 39/25000 [00:12<1:22:44, 5.03step/s]INFO 2026-05-21 12:23:48 ot_train.py:488 step:40 smpl:1K ep:2 epch:0.01 loss:0.019 grdn:0.602 lr:1.5e-06 updt_s:0.050 data_s:0.250
19
+ Training: 0%| | 59/25000 [00:19<2:07:49, 3.25step/s]INFO 2026-05-21 12:23:54 ot_train.py:488 step:60 smpl:2K ep:3 epch:0.01 loss:0.013 grdn:0.457 lr:2.5e-06 updt_s:0.051 data_s:0.217
20
+ Training: 0%| | 79/25000 [00:23<1:12:36, 5.72step/s]INFO 2026-05-21 12:23:59 ot_train.py:488 step:80 smpl:3K ep:4 epch:0.01 loss:0.015 grdn:0.488 lr:3.5e-06 updt_s:0.051 data_s:0.240
21
+ Training: 0%|▏ | 99/25000 [00:30<1:47:03, 3.88step/s]INFO 2026-05-21 12:24:05 ot_train.py:488 step:100 smpl:3K ep:5 epch:0.01 loss:0.015 grdn:0.471 lr:4.5e-06 updt_s:0.052 data_s:0.211
22
+ Training: 0%|▏ | 119/25000 [00:35<1:24:17, 4.92step/s]INFO 2026-05-21 12:24:11 ot_train.py:488 step:120 smpl:4K ep:6 epch:0.02 loss:0.011 grdn:0.389 lr:5.0e-06 updt_s:0.054 data_s:0.265
23
+ Training: 1%|▏ | 139/25000 [00:41<1:59:45, 3.46step/s]INFO 2026-05-21 12:24:16 ot_train.py:488 step:140 smpl:4K ep:7 epch:0.02 loss:0.011 grdn:0.428 lr:5.0e-06 updt_s:0.057 data_s:0.187
24
+ Training: 1%|▏ | 159/25000 [00:47<1:54:08, 3.63step/s]INFO 2026-05-21 12:24:22 ot_train.py:488 step:160 smpl:5K ep:8 epch:0.02 loss:0.012 grdn:0.427 lr:5.0e-06 updt_s:0.050 data_s:0.250
25
+ Training: 1%|▎ | 179/25000 [00:52<1:14:21, 5.56step/s]INFO 2026-05-21 12:24:27 ot_train.py:488 step:180 smpl:6K ep:9 epch:0.02 loss:0.010 grdn:0.385 lr:5.0e-06 updt_s:0.051 data_s:0.208
26
+ Training: 1%|▎ | 199/25000 [00:59<2:04:51, 3.31step/s]INFO 2026-05-21 12:24:34 ot_train.py:488 step:200 smpl:6K ep:10 epch:0.03 loss:0.012 grdn:0.407 lr:5.0e-06 updt_s:0.051 data_s:0.274
27
+ Training: 1%|▎ | 219/25000 [01:04<1:19:56, 5.17step/s]INFO 2026-05-21 12:24:39 ot_train.py:488 step:220 smpl:7K ep:11 epch:0.03 loss:0.012 grdn:0.412 lr:5.0e-06 updt_s:0.049 data_s:0.218
28
+ Training: 1%|▎ | 239/25000 [01:11<2:01:43, 3.39step/s]INFO 2026-05-21 12:24:45 ot_train.py:488 step:240 smpl:8K ep:12 epch:0.03 loss:0.011 grdn:0.345 lr:5.0e-06 updt_s:0.051 data_s:0.256
29
+ Training: 1%|▍ | 259/25000 [01:15<1:10:38, 5.84step/s]INFO 2026-05-21 12:24:50 ot_train.py:488 step:260 smpl:8K ep:13 epch:0.03 loss:0.011 grdn:0.353 lr:5.0e-06 updt_s:0.052 data_s:0.202
30
+ Training: 1%|▍ | 279/25000 [01:21<2:01:42, 3.39step/s]INFO 2026-05-21 12:24:56 ot_train.py:488 step:280 smpl:9K ep:14 epch:0.04 loss:0.009 grdn:0.317 lr:5.0e-06 updt_s:0.051 data_s:0.234
31
+ Training: 1%|▍ | 299/25000 [01:26<1:16:20, 5.39step/s]INFO 2026-05-21 12:25:01 ot_train.py:488 step:300 smpl:10K ep:15 epch:0.04 loss:0.012 grdn:0.380 lr:5.0e-06 updt_s:0.049 data_s:0.227
32
+ Training: 1%|▍ | 319/25000 [01:33<1:59:44, 3.44step/s]INFO 2026-05-21 12:25:08 ot_train.py:488 step:320 smpl:10K ep:16 epch:0.04 loss:0.010 grdn:0.304 lr:5.0e-06 updt_s:0.050 data_s:0.253
33
+ Training: 1%|▌ | 339/25000 [01:38<1:32:59, 4.42step/s]INFO 2026-05-21 12:25:13 ot_train.py:488 step:340 smpl:11K ep:17 epch:0.04 loss:0.008 grdn:0.292 lr:5.0e-06 updt_s:0.051 data_s:0.223
34
+ Training: 1%|▌ | 359/25000 [01:44<1:37:48, 4.20step/s]INFO 2026-05-21 12:25:19 ot_train.py:488 step:360 smpl:12K ep:18 epch:0.05 loss:0.010 grdn:0.321 lr:5.0e-06 updt_s:0.052 data_s:0.226
35
+ Training: 2%|▌ | 379/25000 [01:49<1:30:35, 4.53step/s]INFO 2026-05-21 12:25:24 ot_train.py:488 step:380 smpl:12K ep:19 epch:0.05 loss:0.008 grdn:0.266 lr:5.0e-06 updt_s:0.050 data_s:0.220
36
+ Training: 2%|▌ | 399/25000 [01:56<2:00:56, 3.39step/s]INFO 2026-05-21 12:25:30 ot_train.py:488 step:400 smpl:13K ep:20 epch:0.05 loss:0.008 grdn:0.298 lr:5.0e-06 updt_s:0.051 data_s:0.269
37
+ Training: 2%|▋ | 419/25000 [02:01<1:13:13, 5.60step/s]INFO 2026-05-21 12:25:36 ot_train.py:488 step:420 smpl:13K ep:21 epch:0.05 loss:0.009 grdn:0.285 lr:5.0e-06 updt_s:0.051 data_s:0.205
38
+ Training: 2%|▋ | 439/25000 [02:08<2:29:15, 2.74step/s]INFO 2026-05-21 12:25:43 ot_train.py:488 step:440 smpl:14K ep:22 epch:0.06 loss:0.011 grdn:0.331 lr:5.0e-06 updt_s:0.050 data_s:0.311
39
+ Training: 2%|▋ | 459/25000 [02:13<1:17:18, 5.29step/s]INFO 2026-05-21 12:25:48 ot_train.py:488 step:460 smpl:15K ep:23 epch:0.06 loss:0.009 grdn:0.255 lr:5.0e-06 updt_s:0.049 data_s:0.198
40
+ Training: 2%|▋ | 479/25000 [02:20<1:58:41, 3.44step/s]INFO 2026-05-21 12:25:54 ot_train.py:488 step:480 smpl:15K ep:24 epch:0.06 loss:0.009 grdn:0.267 lr:5.0e-06 updt_s:0.050 data_s:0.276
41
+ Training: 2%|▊ | 499/25000 [02:24<1:07:06, 6.09step/s]INFO 2026-05-21 12:25:59 ot_train.py:488 step:500 smpl:16K ep:25 epch:0.06 loss:0.010 grdn:0.275 lr:5.0e-06 updt_s:0.049 data_s:0.177
42
+ Training: 2%|▊ | 519/25000 [02:31<2:07:23, 3.20step/s]INFO 2026-05-21 12:26:06 ot_train.py:488 step:520 smpl:17K ep:27 epch:0.07 loss:0.008 grdn:0.268 lr:5.0e-06 updt_s:0.052 data_s:0.282
43
+ Training: 2%|▊ | 539/25000 [02:36<1:20:24, 5.07step/s]INFO 2026-05-21 12:26:10 ot_train.py:488 step:540 smpl:17K ep:28 epch:0.07 loss:0.008 grdn:0.265 lr:5.0e-06 updt_s:0.052 data_s:0.190
44
+ Training: 2%|▊ | 559/25000 [02:43<2:19:22, 2.92step/s]INFO 2026-05-21 12:26:17 ot_train.py:488 step:560 smpl:18K ep:29 epch:0.07 loss:0.009 grdn:0.287 lr:5.0e-06 updt_s:0.051 data_s:0.286
45
+ Training: 2%|▉ | 579/25000 [02:47<1:19:05, 5.15step/s]INFO 2026-05-21 12:26:22 ot_train.py:488 step:580 smpl:19K ep:30 epch:0.07 loss:0.010 grdn:0.269 lr:5.0e-06 updt_s:0.052 data_s:0.185
46
+ Training: 2%|▉ | 599/25000 [02:54<2:22:55, 2.85step/s]INFO 2026-05-21 12:26:29 ot_train.py:488 step:600 smpl:19K ep:31 epch:0.08 loss:0.007 grdn:0.266 lr:5.0e-06 updt_s:0.052 data_s:0.307
47
+ Training: 2%|▉ | 619/25000 [03:00<1:28:38, 4.58step/s]INFO 2026-05-21 12:26:34 ot_train.py:488 step:620 smpl:20K ep:32 epch:0.08 loss:0.008 grdn:0.259 lr:5.0e-06 updt_s:0.052 data_s:0.201
48
+ Training: 3%|▉ | 639/25000 [03:06<1:51:00, 3.66step/s]INFO 2026-05-21 12:26:41 ot_train.py:488 step:640 smpl:20K ep:33 epch:0.08 loss:0.008 grdn:0.253 lr:5.0e-06 updt_s:0.050 data_s:0.269
49
+ Training: 3%|█ | 659/25000 [03:12<1:38:51, 4.10step/s]INFO 2026-05-21 12:26:46 ot_train.py:488 step:660 smpl:21K ep:34 epch:0.08 loss:0.009 grdn:0.291 lr:5.0e-06 updt_s:0.051 data_s:0.231
50
+ Training: 3%|█ | 679/25000 [03:19<2:13:46, 3.03step/s]INFO 2026-05-21 12:26:53 ot_train.py:488 step:680 smpl:22K ep:35 epch:0.09 loss:0.008 grdn:0.261 lr:5.0e-06 updt_s:0.051 data_s:0.303
51
+ Training: 3%|█ | 699/25000 [03:24<1:15:21, 5.37step/s]INFO 2026-05-21 12:26:58 ot_train.py:488 step:700 smpl:22K ep:36 epch:0.09 loss:0.009 grdn:0.290 lr:5.0e-06 updt_s:0.053 data_s:0.202
52
+ Training: 3%|█ | 719/25000 [03:31<2:04:22, 3.25step/s]INFO 2026-05-21 12:27:05 ot_train.py:488 step:720 smpl:23K ep:37 epch:0.09 loss:0.008 grdn:0.252 lr:5.0e-06 updt_s:0.051 data_s:0.291
53
+ Training: 3%|█ | 739/25000 [03:36<1:23:40, 4.83step/s]INFO 2026-05-21 12:27:11 ot_train.py:488 step:740 smpl:24K ep:38 epch:0.09 loss:0.006 grdn:0.229 lr:5.0e-06 updt_s:0.052 data_s:0.219
54
+ Training: 3%|█▏ | 759/25000 [03:43<2:13:42, 3.02step/s]INFO 2026-05-21 12:27:18 ot_train.py:488 step:760 smpl:24K ep:39 epch:0.10 loss:0.007 grdn:0.212 lr:5.0e-06 updt_s:0.051 data_s:0.292
55
+ Training: 3%|█▏ | 779/25000 [03:48<1:13:52, 5.46step/s]INFO 2026-05-21 12:27:23 ot_train.py:488 step:780 smpl:25K ep:40 epch:0.10 loss:0.008 grdn:0.245 lr:5.0e-06 updt_s:0.054 data_s:0.211
56
+ Training: 3%|█▏ | 799/25000 [03:55<2:05:38, 3.21step/s]INFO 2026-05-21 12:27:30 ot_train.py:488 step:800 smpl:26K ep:41 epch:0.10 loss:0.008 grdn:0.225 lr:5.0e-06 updt_s:0.050 data_s:0.297
57
+ Training: 3%|█▏ | 819/25000 [04:01<1:22:18, 4.90step/s]INFO 2026-05-21 12:27:36 ot_train.py:488 step:820 smpl:26K ep:42 epch:0.10 loss:0.008 grdn:0.263 lr:5.0e-06 updt_s:0.053 data_s:0.234
58
+ Training: 3%|█▎ | 839/25000 [04:08<2:09:55, 3.10step/s]INFO 2026-05-21 12:27:42 ot_train.py:488 step:840 smpl:27K ep:43 epch:0.11 loss:0.006 grdn:0.219 lr:5.0e-06 updt_s:0.052 data_s:0.279
59
+ Training: 3%|█▎ | 859/25000 [04:13<1:25:44, 4.69step/s]INFO 2026-05-21 12:27:48 ot_train.py:488 step:860 smpl:28K ep:44 epch:0.11 loss:0.009 grdn:0.283 lr:5.0e-06 updt_s:0.050 data_s:0.234
60
+ Training: 4%|█▎ | 879/25000 [04:20<1:57:08, 3.43step/s]INFO 2026-05-21 12:27:54 ot_train.py:488 step:880 smpl:28K ep:45 epch:0.11 loss:0.009 grdn:0.280 lr:5.0e-06 updt_s:0.052 data_s:0.262
61
+ Training: 4%|█▎ | 899/25000 [04:25<1:15:10, 5.34step/s]INFO 2026-05-21 12:27:59 ot_train.py:488 step:900 smpl:29K ep:46 epch:0.11 loss:0.006 grdn:0.236 lr:5.0e-06 updt_s:0.051 data_s:0.194
62
+ Training: 4%|█▍ | 919/25000 [04:32<2:10:02, 3.09step/s]INFO 2026-05-21 12:28:06 ot_train.py:488 step:920 smpl:29K ep:47 epch:0.12 loss:0.008 grdn:0.270 lr:5.0e-06 updt_s:0.053 data_s:0.308
63
+ Training: 4%|█▍ | 939/25000 [04:37<1:15:47, 5.29step/s]INFO 2026-05-21 12:28:11 ot_train.py:488 step:940 smpl:30K ep:48 epch:0.12 loss:0.007 grdn:0.246 lr:5.0e-06 updt_s:0.052 data_s:0.199
64
+ Training: 4%|█▍ | 959/25000 [04:44<2:25:35, 2.75step/s]INFO 2026-05-21 12:28:18 ot_train.py:488 step:960 smpl:31K ep:49 epch:0.12 loss:0.009 grdn:0.279 lr:5.0e-06 updt_s:0.054 data_s:0.291
65
+ Training: 4%|█▍ | 979/25000 [04:49<1:12:47, 5.50step/s]INFO 2026-05-21 12:28:24 ot_train.py:488 step:980 smpl:31K ep:50 epch:0.12 loss:0.008 grdn:0.251 lr:5.0e-06 updt_s:0.051 data_s:0.208
66
+ Training: 4%|█▌ | 999/25000 [04:56<2:10:25, 3.07step/s]INFO 2026-05-21 12:28:31 ot_train.py:488 step:1K smpl:32K ep:51 epch:0.13 loss:0.008 grdn:0.245 lr:5.0e-06 updt_s:0.052 data_s:0.295
67
+ Training: 4%|█▌ | 1019/25000 [05:01<1:12:46, 5.49step/s]INFO 2026-05-21 12:28:35 ot_train.py:488 step:1K smpl:33K ep:52 epch:0.13 loss:0.008 grdn:0.297 lr:5.0e-06 updt_s:0.056 data_s:0.184
68
+ Training: 4%|█▌ | 1039/25000 [05:07<2:17:51, 2.90step/s]INFO 2026-05-21 12:28:42 ot_train.py:488 step:1K smpl:33K ep:53 epch:0.13 loss:0.007 grdn:0.229 lr:5.0e-06 updt_s:0.051 data_s:0.286
69
+ Training: 4%|█▌ | 1059/25000 [05:12<1:16:32, 5.21step/s]INFO 2026-05-21 12:28:47 ot_train.py:488 step:1K smpl:34K ep:54 epch:0.13 loss:0.007 grdn:0.269 lr:5.0e-06 updt_s:0.054 data_s:0.187
70
+ Training: 4%|█▌ | 1079/25000 [05:20<2:11:19, 3.04step/s]INFO 2026-05-21 12:28:54 ot_train.py:488 step:1K smpl:35K ep:55 epch:0.14 loss:0.009 grdn:0.272 lr:5.0e-06 updt_s:0.052 data_s:0.311
71
+ Training: 4%|█▋ | 1099/25000 [05:24<1:08:55, 5.78step/s]INFO 2026-05-21 12:28:59 ot_train.py:488 step:1K smpl:35K ep:56 epch:0.14 loss:0.008 grdn:0.272 lr:5.0e-06 updt_s:0.050 data_s:0.188
72
+ Training: 4%|█▋ | 1119/25000 [05:31<2:05:51, 3.16step/s]INFO 2026-05-21 12:29:06 ot_train.py:488 step:1K smpl:36K ep:57 epch:0.14 loss:0.008 grdn:0.298 lr:5.0e-06 updt_s:0.055 data_s:0.287
73
+ Training: 5%|█▋ | 1139/25000 [05:36<1:09:03, 5.76step/s]INFO 2026-05-21 12:29:10 ot_train.py:488 step:1K smpl:36K ep:58 epch:0.14 loss:0.007 grdn:0.227 lr:5.0e-06 updt_s:0.057 data_s:0.165
74
+ Training: 5%|█▋ | 1159/25000 [05:42<2:06:29, 3.14step/s]INFO 2026-05-21 12:29:17 ot_train.py:488 step:1K smpl:37K ep:59 epch:0.15 loss:0.010 grdn:0.284 lr:5.0e-06 updt_s:0.049 data_s:0.267
75
+ Training: 5%|█▋ | 1179/25000 [05:47<1:12:40, 5.46step/s]INFO 2026-05-21 12:29:22 ot_train.py:488 step:1K smpl:38K ep:60 epch:0.15 loss:0.007 grdn:0.221 lr:5.0e-06 updt_s:0.054 data_s:0.196
76
+ Training: 5%|█▊ | 1199/25000 [05:54<2:27:35, 2.69step/s]INFO 2026-05-21 12:29:29 ot_train.py:488 step:1K smpl:38K ep:61 epch:0.15 loss:0.007 grdn:0.258 lr:5.0e-06 updt_s:0.051 data_s:0.306
77
+ Training: 5%|█▊ | 1219/25000 [05:59<1:08:13, 5.81step/s]INFO 2026-05-21 12:29:33 ot_train.py:488 step:1K smpl:39K ep:62 epch:0.15 loss:0.007 grdn:0.245 lr:5.0e-06 updt_s:0.051 data_s:0.181
78
+ Training: 5%|█▊ | 1239/25000 [06:06<2:01:53, 3.25step/s]INFO 2026-05-21 12:29:40 ot_train.py:488 step:1K smpl:40K ep:63 epch:0.16 loss:0.006 grdn:0.221 lr:5.0e-06 updt_s:0.050 data_s:0.286
79
+ Training: 5%|█▊ | 1259/25000 [06:11<1:20:03, 4.94step/s]INFO 2026-05-21 12:29:45 ot_train.py:488 step:1K smpl:40K ep:64 epch:0.16 loss:0.007 grdn:0.262 lr:5.0e-06 updt_s:0.051 data_s:0.206
80
+ Training: 5%|█▉ | 1279/25000 [06:17<2:16:10, 2.90step/s]INFO 2026-05-21 12:29:52 ot_train.py:488 step:1K smpl:41K ep:65 epch:0.16 loss:0.007 grdn:0.260 lr:5.0e-06 updt_s:0.051 data_s:0.283
81
+ Training: 5%|█▉ | 1299/25000 [06:22<1:14:42, 5.29step/s]INFO 2026-05-21 12:29:57 ot_train.py:488 step:1K smpl:42K ep:66 epch:0.16 loss:0.008 grdn:0.251 lr:5.0e-06 updt_s:0.052 data_s:0.181
82
+ Training: 5%|█▉ | 1319/25000 [06:29<2:12:31, 2.98step/s]INFO 2026-05-21 12:30:04 ot_train.py:488 step:1K smpl:42K ep:67 epch:0.17 loss:0.008 grdn:0.268 lr:5.0e-06 updt_s:0.051 data_s:0.311
83
+ Training: 5%|█▉ | 1339/25000 [06:34<1:07:23, 5.85step/s]INFO 2026-05-21 12:30:09 ot_train.py:488 step:1K smpl:43K ep:68 epch:0.17 loss:0.007 grdn:0.265 lr:5.0e-06 updt_s:0.051 data_s:0.185
84
+ Training: 5%|██ | 1359/25000 [06:41<2:06:49, 3.11step/s]INFO 2026-05-21 12:30:16 ot_train.py:488 step:1K smpl:44K ep:69 epch:0.17 loss:0.008 grdn:0.294 lr:5.0e-06 updt_s:0.050 data_s:0.295
85
+ Training: 6%|██ | 1379/25000 [06:46<1:17:00, 5.11step/s]INFO 2026-05-21 12:30:21 ot_train.py:488 step:1K smpl:44K ep:70 epch:0.17 loss:0.008 grdn:0.291 lr:5.0e-06 updt_s:0.056 data_s:0.198
86
+ Training: 6%|██ | 1399/25000 [06:53<2:03:49, 3.18step/s]INFO 2026-05-21 12:30:28 ot_train.py:488 step:1K smpl:45K ep:71 epch:0.18 loss:0.007 grdn:0.254 lr:5.0e-06 updt_s:0.053 data_s:0.292
87
+ Training: 6%|██ | 1419/25000 [06:58<1:12:13, 5.44step/s]INFO 2026-05-21 12:30:32 ot_train.py:488 step:1K smpl:45K ep:72 epch:0.18 loss:0.008 grdn:0.249 lr:5.0e-06 updt_s:0.050 data_s:0.188
88
+ Training: 6%|██▏ | 1439/25000 [07:05<2:20:06, 2.80step/s]INFO 2026-05-21 12:30:40 ot_train.py:488 step:1K smpl:46K ep:73 epch:0.18 loss:0.008 grdn:0.296 lr:5.0e-06 updt_s:0.052 data_s:0.314
89
+ Training: 6%|██▏ | 1459/25000 [07:10<1:16:25, 5.13step/s]INFO 2026-05-21 12:30:45 ot_train.py:488 step:1K smpl:47K ep:74 epch:0.18 loss:0.007 grdn:0.265 lr:5.0e-06 updt_s:0.053 data_s:0.205
90
+ Training: 6%|██▏ | 1479/25000 [07:17<2:06:47, 3.09step/s]INFO 2026-05-21 12:30:52 ot_train.py:488 step:1K smpl:47K ep:75 epch:0.19 loss:0.007 grdn:0.255 lr:5.0e-06 updt_s:0.055 data_s:0.290
91
+ Training: 6%|██▏ | 1499/25000 [07:22<1:10:14, 5.58step/s]INFO 2026-05-21 12:30:57 ot_train.py:488 step:2K smpl:48K ep:76 epch:0.19 loss:0.007 grdn:0.300 lr:5.0e-06 updt_s:0.052 data_s:0.186
92
+ Training: 6%|██▏ | 1519/25000 [07:29<2:16:49, 2.86step/s]INFO 2026-05-21 12:31:04 ot_train.py:488 step:2K smpl:49K ep:78 epch:0.19 loss:0.007 grdn:0.271 lr:5.0e-06 updt_s:0.054 data_s:0.318
93
+ Training: 6%|██▎ | 1539/25000 [07:34<1:17:51, 5.02step/s]INFO 2026-05-21 12:31:09 ot_train.py:488 step:2K smpl:49K ep:79 epch:0.19 loss:0.006 grdn:0.207 lr:5.0e-06 updt_s:0.054 data_s:0.188
94
+ Training: 6%|██▎ | 1559/25000 [07:41<1:59:04, 3.28step/s]INFO 2026-05-21 12:31:15 ot_train.py:488 step:2K smpl:50K ep:80 epch:0.20 loss:0.008 grdn:0.259 lr:5.0e-06 updt_s:0.050 data_s:0.258
95
+ Training: 6%|██▎ | 1579/25000 [07:45<1:14:31, 5.24step/s]INFO 2026-05-21 12:31:20 ot_train.py:488 step:2K smpl:51K ep:81 epch:0.20 loss:0.006 grdn:0.230 lr:5.0e-06 updt_s:0.053 data_s:0.193
96
+ Training: 6%|██▎ | 1599/25000 [07:52<1:57:24, 3.32step/s]INFO 2026-05-21 12:31:27 ot_train.py:488 step:2K smpl:51K ep:82 epch:0.20 loss:0.008 grdn:0.255 lr:5.0e-06 updt_s:0.051 data_s:0.277
97
+ Training: 6%|██▍ | 1619/25000 [07:57<1:06:57, 5.82step/s]INFO 2026-05-21 12:31:31 ot_train.py:488 step:2K smpl:52K ep:83 epch:0.20 loss:0.008 grdn:0.271 lr:5.0e-06 updt_s:0.051 data_s:0.172
98
+ Training: 7%|██▍ | 1639/25000 [08:03<1:48:46, 3.58step/s]INFO 2026-05-21 12:31:38 ot_train.py:488 step:2K smpl:52K ep:84 epch:0.21 loss:0.008 grdn:0.270 lr:5.0e-06 updt_s:0.052 data_s:0.287
99
+ Training: 7%|██▍ | 1659/25000 [08:08<1:10:46, 5.50step/s]INFO 2026-05-21 12:31:43 ot_train.py:488 step:2K smpl:53K ep:85 epch:0.21 loss:0.006 grdn:0.249 lr:5.0e-06 updt_s:0.050 data_s:0.197
100
+ Training: 7%|██▍ | 1679/25000 [08:15<2:00:15, 3.23step/s]INFO 2026-05-21 12:31:50 ot_train.py:488 step:2K smpl:54K ep:86 epch:0.21 loss:0.006 grdn:0.246 lr:5.0e-06 updt_s:0.050 data_s:0.288
101
+ Training: 7%|██▌ | 1699/25000 [08:19<1:01:50, 6.28step/s]INFO 2026-05-21 12:31:54 ot_train.py:488 step:2K smpl:54K ep:87 epch:0.21 loss:0.007 grdn:0.267 lr:4.9e-06 updt_s:0.049 data_s:0.163
102
+ Training: 7%|██▌ | 1719/25000 [08:26<2:00:52, 3.21step/s]INFO 2026-05-21 12:32:01 ot_train.py:488 step:2K smpl:55K ep:88 epch:0.22 loss:0.008 grdn:0.261 lr:4.9e-06 updt_s:0.049 data_s:0.282
103
+ Training: 7%|██▌ | 1739/25000 [08:31<1:01:23, 6.32step/s]INFO 2026-05-21 12:32:05 ot_train.py:488 step:2K smpl:56K ep:89 epch:0.22 loss:0.007 grdn:0.254 lr:4.9e-06 updt_s:0.049 data_s:0.180
104
+ Training: 7%|██▌ | 1759/25000 [08:37<2:04:00, 3.12step/s]INFO 2026-05-21 12:32:12 ot_train.py:488 step:2K smpl:56K ep:90 epch:0.22 loss:0.010 grdn:0.278 lr:4.9e-06 updt_s:0.049 data_s:0.280
105
+ Training: 7%|██▋ | 1779/25000 [08:42<1:06:57, 5.78step/s]INFO 2026-05-21 12:32:16 ot_train.py:488 step:2K smpl:57K ep:91 epch:0.22 loss:0.007 grdn:0.266 lr:4.9e-06 updt_s:0.050 data_s:0.172
106
+ Training: 7%|██▋ | 1799/25000 [08:48<1:54:36, 3.37step/s]INFO 2026-05-21 12:32:23 ot_train.py:488 step:2K smpl:58K ep:92 epch:0.23 loss:0.007 grdn:0.249 lr:4.9e-06 updt_s:0.051 data_s:0.292
107
+ Training: 7%|██▋ | 1819/25000 [08:54<1:13:34, 5.25step/s]INFO 2026-05-21 12:32:28 ot_train.py:488 step:2K smpl:58K ep:93 epch:0.23 loss:0.007 grdn:0.245 lr:4.9e-06 updt_s:0.054 data_s:0.207
108
+ Training: 7%|██▋ | 1839/25000 [09:00<1:57:03, 3.30step/s]INFO 2026-05-21 12:32:35 ot_train.py:488 step:2K smpl:59K ep:94 epch:0.23 loss:0.006 grdn:0.221 lr:4.9e-06 updt_s:0.053 data_s:0.265
109
+ Training: 7%|██▊ | 1859/25000 [09:05<1:07:35, 5.71step/s]INFO 2026-05-21 12:32:39 ot_train.py:488 step:2K smpl:60K ep:95 epch:0.23 loss:0.009 grdn:0.290 lr:4.9e-06 updt_s:0.055 data_s:0.185
110
+ Training: 8%|██▊ | 1879/25000 [09:12<1:57:14, 3.29step/s]INFO 2026-05-21 12:32:46 ot_train.py:488 step:2K smpl:60K ep:96 epch:0.24 loss:0.006 grdn:0.228 lr:4.9e-06 updt_s:0.053 data_s:0.296
111
+ Training: 8%|██▊ | 1899/25000 [09:16<1:09:54, 5.51step/s]INFO 2026-05-21 12:32:51 ot_train.py:488 step:2K smpl:61K ep:97 epch:0.24 loss:0.008 grdn:0.253 lr:4.9e-06 updt_s:0.052 data_s:0.178
112
+ Training: 8%|██▊ | 1919/25000 [09:23<2:00:52, 3.18step/s]INFO 2026-05-21 12:32:58 ot_train.py:488 step:2K smpl:61K ep:98 epch:0.24 loss:0.007 grdn:0.234 lr:4.9e-06 updt_s:0.055 data_s:0.295
113
+ Training: 8%|██▊ | 1939/25000 [09:29<1:12:56, 5.27step/s]INFO 2026-05-21 12:33:03 ot_train.py:488 step:2K smpl:62K ep:99 epch:0.24 loss:0.008 grdn:0.293 lr:4.9e-06 updt_s:0.055 data_s:0.201
114
+ Training: 8%|██▉ | 1959/25000 [09:36<2:13:43, 2.87step/s]INFO 2026-05-21 12:33:10 ot_train.py:488 step:2K smpl:63K ep:100 epch:0.25 loss:0.006 grdn:0.221 lr:4.9e-06 updt_s:0.053 data_s:0.293
115
+ Training: 8%|██▉ | 1979/25000 [09:41<1:11:53, 5.34step/s]INFO 2026-05-21 12:33:15 ot_train.py:488 step:2K smpl:63K ep:101 epch:0.25 loss:0.007 grdn:0.252 lr:4.9e-06 updt_s:0.051 data_s:0.199
116
+ Training: 8%|██▉ | 1999/25000 [09:47<1:58:17, 3.24step/s]INFO 2026-05-21 12:33:22 ot_train.py:488 step:2K smpl:64K ep:102 epch:0.25 loss:0.008 grdn:0.251 lr:4.9e-06 updt_s:0.052 data_s:0.285
117
+ INFO 2026-05-21 12:33:22 ot_train.py:502 Checkpoint policy after step 2000
118
+ Training: 8%|███▏ | 2019/25000 [09:58<44:50, 8.54step/s]INFO 2026-05-21 12:33:32 ot_train.py:488 step:2K smpl:65K ep:103 epch:0.25 loss:0.007 grdn:0.230 lr:4.9e-06 updt_s:0.049 data_s:0.003
119
+ Training: 8%|███▏ | 2039/25000 [10:00<37:20, 10.25step/s]INFO 2026-05-21 12:33:34 ot_train.py:488 step:2K smpl:65K ep:104 epch:0.26 loss:0.006 grdn:0.237 lr:4.9e-06 updt_s:0.050 data_s:0.044
120
+ Training: 8%|███ | 2060/25000 [10:07<1:51:11, 3.44step/s]INFO 2026-05-21 12:33:41 ot_train.py:488 step:2K smpl:66K ep:105 epch:0.26 loss:0.008 grdn:0.282 lr:4.9e-06 updt_s:0.050 data_s:0.294
121
+ Training: 8%|███ | 2079/25000 [10:12<1:25:20, 4.48step/s]INFO 2026-05-21 12:33:46 ot_train.py:488 step:2K smpl:67K ep:106 epch:0.26 loss:0.007 grdn:0.245 lr:4.9e-06 updt_s:0.050 data_s:0.204
122
+ Training: 8%|███ | 2099/25000 [10:19<2:38:17, 2.41step/s]INFO 2026-05-21 12:33:53 ot_train.py:488 step:2K smpl:67K ep:107 epch:0.26 loss:0.008 grdn:0.282 lr:4.9e-06 updt_s:0.052 data_s:0.297
123
+ Training: 8%|███▏ | 2119/25000 [10:24<1:30:12, 4.23step/s]INFO 2026-05-21 12:33:58 ot_train.py:488 step:2K smpl:68K ep:108 epch:0.27 loss:0.006 grdn:0.223 lr:4.9e-06 updt_s:0.050 data_s:0.200
124
+ Training: 9%|███▏ | 2139/25000 [10:31<2:46:11, 2.29step/s]INFO 2026-05-21 12:34:05 ot_train.py:488 step:2K smpl:68K ep:109 epch:0.27 loss:0.008 grdn:0.247 lr:4.9e-06 updt_s:0.053 data_s:0.300
125
+ Training: 9%|███▏ | 2159/25000 [10:36<1:31:04, 4.18step/s]INFO 2026-05-21 12:34:10 ot_train.py:488 step:2K smpl:69K ep:110 epch:0.27 loss:0.006 grdn:0.231 lr:4.9e-06 updt_s:0.050 data_s:0.184
126
+ Training: 9%|███▏ | 2177/25000 [10:40<1:06:32, 5.72step/s]
checkpoints/generalized_eval1/wandb/run-20260521_122331-91fmyb38/files/requirements.txt ADDED
@@ -0,0 +1,114 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ lerobot==0.5.2
2
+ pydantic_core==2.46.4
3
+ fsspec==2026.2.0
4
+ MarkupSafe==3.0.3
5
+ opencv-python==4.13.0.92
6
+ setuptools==80.10.2
7
+ sentry-sdk==2.60.0
8
+ idna==3.13
9
+ pandas==2.3.3
10
+ packaging==25.0
11
+ h11==0.16.0
12
+ nvidia-cufft-cu12==11.3.3.83
13
+ pillow==12.2.0
14
+ yarl==1.24.2
15
+ toml==0.10.2
16
+ typing-inspect==0.9.0
17
+ triton==3.6.0
18
+ Jinja2==3.1.6
19
+ networkx==3.6.1
20
+ Farama-Notifications==0.0.6
21
+ aiohappyeyeballs==2.6.2
22
+ cloudpickle==3.1.2
23
+ draccus==0.10.0
24
+ pyyaml-include==1.4.1
25
+ aiohttp==3.13.5
26
+ markdown-it-py==4.0.0
27
+ termcolor==3.3.0
28
+ wandb==0.27.0
29
+ shellingham==1.5.4
30
+ PyYAML==6.0.3
31
+ smmap==5.0.3
32
+ nvidia-cusparselt-cu12==0.7.1
33
+ charset-normalizer==3.4.7
34
+ multiprocess==0.70.19
35
+ tzdata==2026.2
36
+ typer==0.25.1
37
+ mypy_extensions==1.1.0
38
+ multidict==6.7.1
39
+ torch==2.10.0
40
+ Flask==3.1.3
41
+ nvidia-cusparse-cu12==12.5.8.93
42
+ GitPython==3.1.50
43
+ mdurl==0.1.2
44
+ blinker==1.9.0
45
+ itsdangerous==2.2.0
46
+ datasets==4.8.5
47
+ httpcore==1.0.9
48
+ jsonlines==4.0.0
49
+ nvidia-cusolver-cu12==11.7.3.90
50
+ cmake==4.1.3
51
+ propcache==0.5.2
52
+ Werkzeug==3.1.8
53
+ typing_extensions==4.15.0
54
+ platformdirs==4.9.6
55
+ psutil==7.2.2
56
+ dill==0.4.1
57
+ lerobot==0.5.2
58
+ xxhash==3.7.0
59
+ gymnasium==1.3.0
60
+ nvidia-cublas-cu12==12.8.4.1
61
+ opencv-python-headless==4.13.0.92
62
+ mergedeep==1.3.4
63
+ pyarrow==24.0.0
64
+ annotated-types==0.7.0
65
+ anyio==4.13.0
66
+ cuda-pathfinder==1.5.4
67
+ six==1.17.0
68
+ nvidia-cuda-runtime-cu12==12.8.90
69
+ click==8.3.3
70
+ tqdm==4.67.3
71
+ annotated-doc==0.0.4
72
+ frozenlist==1.8.0
73
+ av==15.1.0
74
+ pydantic==2.13.4
75
+ torchcodec==0.10.0
76
+ zipp==4.1.0
77
+ mpmath==1.3.0
78
+ pytz==2026.2
79
+ importlib_metadata==9.0.0
80
+ huggingface_hub==1.13.0
81
+ hf-xet==1.4.3
82
+ attrs==26.1.0
83
+ nvidia-nvjitlink-cu12==12.8.93
84
+ sympy==1.14.0
85
+ nvidia-cufile-cu12==1.13.1.3
86
+ nvidia-cudnn-cu12==9.10.2.21
87
+ protobuf==7.35.0
88
+ diffusers==0.35.2
89
+ cuda-bindings==12.9.4
90
+ nvidia-cuda-cupti-cu12==12.8.90
91
+ tokenizers==0.22.2
92
+ urllib3==2.6.3
93
+ aiosignal==1.4.0
94
+ torchvision==0.25.0
95
+ httpx==0.28.1
96
+ requests==2.33.1
97
+ certifi==2026.4.22
98
+ rich==15.0.0
99
+ python-dateutil==2.9.0.post0
100
+ nvidia-nccl-cu12==2.27.5
101
+ accelerate==1.13.0
102
+ numpy==2.2.6
103
+ nvidia-nvshmem-cu12==3.4.5
104
+ nvidia-curand-cu12==10.3.9.90
105
+ gitdb==4.0.12
106
+ regex==2026.5.9
107
+ filelock==3.29.0
108
+ Pygments==2.20.0
109
+ transformers==5.9.0
110
+ safetensors==0.7.0
111
+ einops==0.8.2
112
+ nvidia-nvtx-cu12==12.8.90
113
+ nvidia-cuda-nvrtc-cu12==12.8.93
114
+ typing-inspection==0.4.2
checkpoints/generalized_eval1/wandb/run-20260521_122331-91fmyb38/files/wandb-metadata.json ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-6.8.0-90-generic-x86_64-with-glibc2.35",
3
+ "python": "CPython 3.12.13",
4
+ "startedAt": "2026-05-21T12:23:31.013952Z",
5
+ "args": [
6
+ "--policy.path=/home/shadeform/Desktop/robot_learning/diffusion_780_new_dataset_dino/checkpoints/diffusion_780_new_dataset_dino/checkpoints/400000/pretrained_model/",
7
+ "--dataset.repo_id=local/act_collected_episodes",
8
+ "--dataset.root=/home/shadeform/Desktop/robot_learning/random_pick_new_plus_new300_dark",
9
+ "--dataset.video_backend=pyav",
10
+ "--policy.device=cuda",
11
+ "--policy.push_to_hub=false",
12
+ "--policy.freeze_vision_encoder=true",
13
+ "--policy.optimizer_lr=5e-6",
14
+ "--policy.scheduler_warmup_steps=100",
15
+ "--optimizer.grad_clip_norm=1.0",
16
+ "--output_dir=/home/shadeform/Desktop/robot_learning/outputs/train/diffusion_fine_tune_generalized_1",
17
+ "--job_name=diffusion_fine_tune_generalized_1",
18
+ "--batch_size=32",
19
+ "--steps=25000",
20
+ "--num_workers=8",
21
+ "--log_freq=20",
22
+ "--save_freq=2000",
23
+ "--eval_freq=0",
24
+ "--wandb.enable=True"
25
+ ],
26
+ "program": "/home/shadeform/Desktop/robot_learning/lerobot/.venv/bin/lerobot-train",
27
+ "codePath": ".venv/bin/lerobot-train",
28
+ "codePathLocal": ".venv/bin/lerobot-train",
29
+ "git": {
30
+ "remote": "https://github.com/michavol/lerobot.git",
31
+ "commit": "46c24e328ea25e6e78e8776b8c64753b80a95cde"
32
+ },
33
+ "root": "/home/shadeform/Desktop/robot_learning/outputs/train/diffusion_fine_tune_generalized_1",
34
+ "host": "brev-baxch8nuw",
35
+ "executable": "/home/shadeform/Desktop/robot_learning/lerobot/.venv/bin/python3",
36
+ "cpu_count": 28,
37
+ "cpu_count_logical": 28,
38
+ "gpu": "NVIDIA H100 PCIe",
39
+ "gpu_count": 1,
40
+ "disk": {
41
+ "/": {
42
+ "total": "103865303040",
43
+ "used": "84695650304"
44
+ }
45
+ },
46
+ "memory": {
47
+ "total": "190128123904"
48
+ },
49
+ "gpu_nvidia": [
50
+ {
51
+ "name": "NVIDIA H100 PCIe",
52
+ "memoryTotal": "85520809984",
53
+ "cudaCores": 14592,
54
+ "architecture": "Hopper",
55
+ "uuid": "GPU-ce315b98-20ff-34fd-307b-fe05646f5913"
56
+ }
57
+ ],
58
+ "cudaVersion": "12.8",
59
+ "writerId": "6atm3zfsyrs3tmoqrmduk3nxzsgz25wn"
60
+ }
checkpoints/generalized_eval1/wandb/run-20260521_122331-91fmyb38/logs/debug-core.log ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {"time":"2026-05-21T12:23:31.044825009Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpqns9f98t/port-1860986.txt","pid":1860986,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
2
+ {"time":"2026-05-21T12:23:31.045404791Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-1860986-1861061-4241581904/socket","Net":"unix"}}
3
+ {"time":"2026-05-21T12:23:31.045475998Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":1860986}
4
+ {"time":"2026-05-21T12:23:31.230098117Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
5
+ {"time":"2026-05-21T12:23:31.23514456Z","level":"INFO","msg":"handleInformInit: received","streamId":"91fmyb38","id":"1(@)"}
6
+ {"time":"2026-05-21T12:23:31.436313984Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"91fmyb38","id":"1(@)"}
7
+ {"time":"2026-05-21T12:23:37.300872552Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"aypwcwl487hu"}
checkpoints/generalized_eval1/wandb/run-20260521_122331-91fmyb38/logs/debug-internal.log ADDED
@@ -0,0 +1,95 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-05-21T12:23:31.235249938Z","level":"INFO","msg":"wandb-core"}
2
+ {"time":"2026-05-21T12:23:31.235495117Z","level":"INFO","msg":"stream: starting","core version":"0.27.0"}
3
+ {"time":"2026-05-21T12:23:31.436106922Z","level":"INFO","msg":"stream: created new stream","id":"91fmyb38"}
4
+ {"time":"2026-05-21T12:23:31.436203377Z","level":"INFO","msg":"handler: started"}
5
+ {"time":"2026-05-21T12:23:31.436309017Z","level":"INFO","msg":"stream: started"}
6
+ {"time":"2026-05-21T12:23:31.436354926Z","level":"INFO","msg":"writer: started","stream_id":"91fmyb38"}
7
+ {"time":"2026-05-21T12:23:31.436376138Z","level":"INFO","msg":"sender: started"}
8
+ {"time":"2026-05-21T12:23:32.297492478Z","level":"INFO","msg":"filestream: sending request","total_files":1,"console_offset":0,"console_lines":1}
9
+ {"time":"2026-05-21T12:23:32.837113Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
10
+ {"time":"2026-05-21T12:23:47.297546752Z","level":"INFO","msg":"filestream: sending request","total_files":2,"events_offset":0,"events_lines":1,"console_offset":1,"console_lines":17,"uploaded_len":2}
11
+ {"time":"2026-05-21T12:23:47.515947561Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
12
+ {"time":"2026-05-21T12:24:02.298262523Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":0,"history_lines":3,"events_offset":1,"events_lines":2,"console_offset":17,"console_lines":4}
13
+ {"time":"2026-05-21T12:24:02.561305727Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
14
+ {"time":"2026-05-21T12:24:17.298235128Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":3,"history_lines":3,"events_offset":3,"events_lines":2,"console_offset":20,"console_lines":4}
15
+ {"time":"2026-05-21T12:24:17.936748721Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
16
+ {"time":"2026-05-21T12:24:32.298300854Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":6,"history_lines":2,"events_offset":5,"events_lines":2,"console_offset":23,"console_lines":3}
17
+ {"time":"2026-05-21T12:24:32.699635088Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
18
+ {"time":"2026-05-21T12:24:47.298617002Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":8,"history_lines":3,"events_offset":7,"events_lines":2,"console_offset":25,"console_lines":4}
19
+ {"time":"2026-05-21T12:24:48.265336142Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
20
+ {"time":"2026-05-21T12:25:02.298222417Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":11,"history_lines":3,"events_offset":9,"events_lines":2,"console_offset":28,"console_lines":3}
21
+ {"time":"2026-05-21T12:25:03.341512826Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
22
+ {"time":"2026-05-21T12:25:17.297761366Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":14,"history_lines":2,"events_offset":11,"events_lines":2,"console_offset":31,"console_lines":3}
23
+ {"time":"2026-05-21T12:25:17.60161801Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
24
+ {"time":"2026-05-21T12:25:32.297669007Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":16,"history_lines":3,"events_offset":13,"events_lines":2,"console_offset":33,"console_lines":4}
25
+ {"time":"2026-05-21T12:25:32.507496784Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
26
+ {"time":"2026-05-21T12:25:47.297895545Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":19,"history_lines":2,"events_offset":15,"events_lines":2,"console_offset":36,"console_lines":3}
27
+ {"time":"2026-05-21T12:25:48.245571706Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
28
+ {"time":"2026-05-21T12:26:02.298042403Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":21,"history_lines":3,"events_offset":17,"events_lines":2,"console_offset":38,"console_lines":4}
29
+ {"time":"2026-05-21T12:26:02.552618844Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
30
+ {"time":"2026-05-21T12:26:17.298031263Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":24,"history_lines":2,"events_offset":19,"events_lines":2,"console_offset":41,"console_lines":3}
31
+ {"time":"2026-05-21T12:26:18.196392246Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
32
+ {"time":"2026-05-21T12:26:32.299060379Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":26,"history_lines":3,"events_offset":21,"events_lines":2,"console_offset":43,"console_lines":4}
33
+ {"time":"2026-05-21T12:26:32.967831752Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
34
+ {"time":"2026-05-21T12:26:47.298303969Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":29,"history_lines":3,"events_offset":23,"events_lines":2,"console_offset":46,"console_lines":3}
35
+ {"time":"2026-05-21T12:26:47.83934363Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
36
+ {"time":"2026-05-21T12:27:02.298402147Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":32,"history_lines":2,"events_offset":25,"events_lines":2,"console_offset":49,"console_lines":3}
37
+ {"time":"2026-05-21T12:27:02.919167995Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
38
+ {"time":"2026-05-21T12:27:17.29808968Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":34,"history_lines":2,"events_offset":27,"events_lines":2,"console_offset":51,"console_lines":3}
39
+ {"time":"2026-05-21T12:27:17.810947902Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
40
+ {"time":"2026-05-21T12:27:32.298013663Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":36,"history_lines":3,"events_offset":29,"events_lines":2,"console_offset":53,"console_lines":4}
41
+ {"time":"2026-05-21T12:27:33.419987472Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
42
+ {"time":"2026-05-21T12:27:47.298428951Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":39,"history_lines":2,"events_offset":31,"events_lines":2,"console_offset":56,"console_lines":3}
43
+ {"time":"2026-05-21T12:27:47.487975401Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
44
+ {"time":"2026-05-21T12:28:02.297700362Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":41,"history_lines":3,"events_offset":33,"events_lines":2,"console_offset":58,"console_lines":4}
45
+ {"time":"2026-05-21T12:28:02.53586069Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
46
+ {"time":"2026-05-21T12:28:17.298434159Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":44,"history_lines":2,"events_offset":35,"events_lines":2,"console_offset":61,"console_lines":3}
47
+ {"time":"2026-05-21T12:28:17.678643338Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
48
+ {"time":"2026-05-21T12:28:32.2985949Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":46,"history_lines":3,"events_offset":37,"events_lines":2,"console_offset":63,"console_lines":4}
49
+ {"time":"2026-05-21T12:28:32.502397298Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
50
+ {"time":"2026-05-21T12:28:47.297947735Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":49,"history_lines":2,"events_offset":39,"events_lines":2,"console_offset":66,"console_lines":3}
51
+ {"time":"2026-05-21T12:28:48.007234263Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
52
+ {"time":"2026-05-21T12:29:02.297700728Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":51,"history_lines":3,"events_offset":41,"events_lines":2,"console_offset":68,"console_lines":4}
53
+ {"time":"2026-05-21T12:29:02.728084572Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
54
+ {"time":"2026-05-21T12:29:17.298431882Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":54,"history_lines":3,"events_offset":43,"events_lines":2,"console_offset":71,"console_lines":4}
55
+ {"time":"2026-05-21T12:29:17.682900167Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
56
+ {"time":"2026-05-21T12:29:32.297791442Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":57,"history_lines":2,"events_offset":45,"events_lines":2,"console_offset":74,"console_lines":3}
57
+ {"time":"2026-05-21T12:29:32.716602354Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
58
+ {"time":"2026-05-21T12:29:47.297755622Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":59,"history_lines":3,"events_offset":47,"events_lines":2,"console_offset":76,"console_lines":3}
59
+ {"time":"2026-05-21T12:29:48.158392751Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
60
+ {"time":"2026-05-21T12:30:02.298385353Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":62,"history_lines":2,"events_offset":49,"events_lines":2,"console_offset":79,"console_lines":3}
61
+ {"time":"2026-05-21T12:30:02.643764568Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
62
+ {"time":"2026-05-21T12:30:17.298434031Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":64,"history_lines":3,"events_offset":51,"events_lines":2,"console_offset":81,"console_lines":4}
63
+ {"time":"2026-05-21T12:30:18.505473171Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
64
+ {"time":"2026-05-21T12:30:32.297993966Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":67,"history_lines":2,"events_offset":53,"events_lines":2,"console_offset":84,"console_lines":3}
65
+ {"time":"2026-05-21T12:30:32.851250185Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
66
+ {"time":"2026-05-21T12:30:47.298561389Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":69,"history_lines":3,"events_offset":55,"events_lines":2,"console_offset":86,"console_lines":3}
67
+ {"time":"2026-05-21T12:30:48.842829385Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
68
+ {"time":"2026-05-21T12:31:02.298526077Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":72,"history_lines":2,"events_offset":57,"events_lines":2,"console_offset":89,"console_lines":3}
69
+ {"time":"2026-05-21T12:31:02.585431307Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
70
+ {"time":"2026-05-21T12:31:17.298519151Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":74,"history_lines":3,"events_offset":59,"events_lines":2,"console_offset":91,"console_lines":4}
71
+ {"time":"2026-05-21T12:31:18.029621131Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
72
+ {"time":"2026-05-21T12:31:32.29801427Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":77,"history_lines":3,"events_offset":61,"events_lines":2,"console_offset":94,"console_lines":3}
73
+ {"time":"2026-05-21T12:31:32.951656172Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
74
+ {"time":"2026-05-21T12:31:47.297600579Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":80,"history_lines":2,"events_offset":63,"events_lines":2,"console_offset":97,"console_lines":3}
75
+ {"time":"2026-05-21T12:31:48.325261241Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
76
+ {"time":"2026-05-21T12:32:02.298369593Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":82,"history_lines":3,"events_offset":65,"events_lines":2,"console_offset":99,"console_lines":4}
77
+ {"time":"2026-05-21T12:32:02.719801212Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
78
+ {"time":"2026-05-21T12:32:17.298178577Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":85,"history_lines":3,"events_offset":67,"events_lines":2,"console_offset":102,"console_lines":3}
79
+ {"time":"2026-05-21T12:32:17.829373922Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
80
+ {"time":"2026-05-21T12:32:32.297731002Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":88,"history_lines":2,"events_offset":69,"events_lines":2,"console_offset":105,"console_lines":3}
81
+ {"time":"2026-05-21T12:32:32.868550719Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
82
+ {"time":"2026-05-21T12:32:47.297785787Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":90,"history_lines":3,"events_offset":71,"events_lines":2,"console_offset":107,"console_lines":4}
83
+ {"time":"2026-05-21T12:32:47.827099892Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
84
+ {"time":"2026-05-21T12:33:02.298392242Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":93,"history_lines":2,"events_offset":73,"events_lines":2,"console_offset":110,"console_lines":3}
85
+ {"time":"2026-05-21T12:33:02.965528874Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
86
+ {"time":"2026-05-21T12:33:17.298256374Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":95,"history_lines":3,"events_offset":75,"events_lines":2,"console_offset":112,"console_lines":3}
87
+ {"time":"2026-05-21T12:33:18.042320109Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
88
+ {"time":"2026-05-21T12:33:32.298320573Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":98,"history_lines":1,"events_offset":77,"events_lines":2,"console_offset":115,"console_lines":3}
89
+ {"time":"2026-05-21T12:33:33.095702573Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
90
+ {"time":"2026-05-21T12:33:47.298433704Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":99,"history_lines":4,"events_offset":79,"events_lines":2,"console_offset":117,"console_lines":5}
91
+ {"time":"2026-05-21T12:33:47.826432477Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
92
+ {"time":"2026-05-21T12:34:02.298252019Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":103,"history_lines":2,"events_offset":81,"events_lines":2,"console_offset":121,"console_lines":3}
93
+ {"time":"2026-05-21T12:34:02.741382701Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
94
+ {"time":"2026-05-21T12:34:17.298278155Z","level":"INFO","msg":"filestream: sending request","total_files":4,"history_offset":105,"history_lines":2,"events_offset":83,"events_lines":2,"console_offset":123,"console_lines":3}
95
+ {"time":"2026-05-21T12:34:17.983055483Z","level":"INFO","msg":"filestream: request sent","status":"200 OK"}
checkpoints/generalized_eval1/wandb/run-20260521_122331-91fmyb38/logs/debug.log ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-05-21 12:23:31,014 INFO MainThread:1860986 [wandb_setup.py:_flush():81] Current SDK version is 0.27.0
2
+ 2026-05-21 12:23:31,015 INFO MainThread:1860986 [wandb_setup.py:_flush():81] Configure stats pid to 1860986
3
+ 2026-05-21 12:23:31,015 INFO MainThread:1860986 [wandb_setup.py:_flush():81] Loading settings from environment variables
4
+ 2026-05-21 12:23:31,015 INFO MainThread:1860986 [wandb_init.py:setup_run_log_directory():723] Logging user logs to /home/shadeform/Desktop/robot_learning/outputs/train/diffusion_fine_tune_generalized_1/wandb/run-20260521_122331-91fmyb38/logs/debug.log
5
+ 2026-05-21 12:23:31,015 INFO MainThread:1860986 [wandb_init.py:setup_run_log_directory():724] Logging internal logs to /home/shadeform/Desktop/robot_learning/outputs/train/diffusion_fine_tune_generalized_1/wandb/run-20260521_122331-91fmyb38/logs/debug-internal.log
6
+ 2026-05-21 12:23:31,015 INFO MainThread:1860986 [wandb_init.py:init():851] calling init triggers
7
+ 2026-05-21 12:23:31,015 INFO MainThread:1860986 [wandb_init.py:init():856] wandb.init called with sweep_config: {}
8
+ config: {'dataset': {'repo_id': 'local/act_collected_episodes', 'root': '/home/shadeform/Desktop/robot_learning/random_pick_new_plus_new300_dark', 'episodes': None, 'image_transforms': {'enable': False, 'max_num_transforms': 3, 'random_order': False, 'same_cloth_color': False, 'tfs': {'color_jitter': {'weight': 1.5, 'type': 'ColorJitter', 'kwargs': {'brightness': [0.7, 1.3], 'contrast': [0.7, 1.3], 'saturation': [0.4, 1.6], 'hue': [-0.5, 0.5]}}, 'grayscale': {'weight': 0.5, 'type': 'RandomGrayscale', 'kwargs': {'p': 1.0}}, 'shadow': {'weight': 0.0, 'type': 'RandomShadow', 'kwargs': {'num_shadows': [1, 2], 'intensity': [0.4, 0.7], 'blur_sigma': [4.0, 20.0]}}, 'highlight': {'weight': 0.0, 'type': 'RandomHighlight', 'kwargs': {'num_highlights': [0, 1], 'intensity': [0.2, 0.5], 'sigma': [20.0, 60.0]}}, 'gamma': {'weight': 0.7, 'type': 'RandomGamma', 'kwargs': {'gamma': [0.6, 1.4]}}, 'blur': {'weight': 0.0, 'type': 'GaussianBlur', 'kwargs': {'kernel_size': [3, 3], 'sigma': [0.1, 1.5]}}, 'jpeg': {'weight': 0.0, 'type': 'RandomJPEG', 'kwargs': {'quality': [60, 95]}}, 'crop': {'weight': 0.5, 'type': 'RandomCropPreserveSize', 'kwargs': {'scale': [0.97, 1.0], 'ratio': [0.99, 1.01]}}, 'affine': {'weight': 0.0, 'type': 'RandomAffine', 'kwargs': {'degrees': [-5.0, 5.0], 'translate': [0.05, 0.05]}}, 'sharpness': {'weight': 1.0, 'type': 'SharpnessJitter', 'kwargs': {'sharpness': [0.5, 1.5]}}}}, 'revision': None, 'use_imagenet_stats': True, 'video_backend': 'pyav', 'return_uint8': False, 'streaming': False}, 'env': None, 'policy': {'type': 'diffusion', 'n_obs_steps': 2, 'input_features': {'observation.state': {'type': <FeatureType.STATE: 'STATE'>, 'shape': [6]}, 'observation.images.front': {'type': <FeatureType.VISUAL: 'VISUAL'>, 'shape': [3, 480, 640]}}, 'output_features': {'action': {'type': <FeatureType.ACTION: 'ACTION'>, 'shape': [6]}}, 'device': 'cuda', 'use_amp': False, 'use_peft': False, 'push_to_hub': False, 'repo_id': None, 'private': None, 'tags': None, 'license': None, 'pretrained_path': '/home/shadeform/Desktop/robot_learning/diffusion_780_new_dataset_dino/checkpoints/diffusion_780_new_dataset_dino/checkpoints/400000/pretrained_model', 'horizon': 16, 'n_action_steps': 8, 'normalization_mapping': {'VISUAL': <NormalizationMode.MEAN_STD: 'MEAN_STD'>, 'STATE': <NormalizationMode.MIN_MAX: 'MIN_MAX'>, 'ACTION': <NormalizationMode.MIN_MAX: 'MIN_MAX'>}, 'drop_n_last_frames': 7, 'vision_backbone_type': 'dinov2', 'vision_encoder_name': 'facebook/dinov2-small', 'freeze_vision_encoder': True, 'vision_backbone': 'resnet18', 'resize_shape': [224, 224], 'crop_ratio': 1.0, 'crop_shape': None, 'crop_is_random': True, 'pretrained_backbone_weights': 'ResNet18_Weights.IMAGENET1K_V1', 'use_group_norm': False, 'spatial_softmax_num_keypoints': 32, 'vit_pool_type': 'spatial_softmax', 'vit_feature_dim': 512, 'use_separate_rgb_encoder_per_camera': True, 'down_dims': [512, 1024, 2048], 'kernel_size': 5, 'n_groups': 8, 'diffusion_step_embed_dim': 128, 'use_film_scale_modulation': True, 'noise_scheduler_type': 'DDPM', 'num_train_timesteps': 100, 'beta_schedule': 'squaredcos_cap_v2', 'beta_start': 0.0001, 'beta_end': 0.02, 'prediction_type': 'epsilon', 'clip_sample': True, 'clip_sample_range': 1.0, 'num_inference_steps': 100, 'compile_model': False, 'compile_mode': 'reduce-overhead', 'do_mask_loss_for_padding': False, 'optimizer_lr': 5e-06, 'optimizer_betas': [0.95, 0.999], 'optimizer_eps': 1e-08, 'optimizer_weight_decay': 1e-06, 'scheduler_name': 'cosine', 'scheduler_warmup_steps': 100}, 'reward_model': None, 'output_dir': '/home/shadeform/Desktop/robot_learning/outputs/train/diffusion_fine_tune_generalized_1', 'job_name': 'diffusion_fine_tune_generalized_1', 'resume': False, 'seed': 1000, 'cudnn_deterministic': False, 'num_workers': 8, 'batch_size': 32, 'prefetch_factor': 4, 'persistent_workers': True, 'steps': 25000, 'eval_freq': 0, 'log_freq': 20, 'tolerance_s': 0.0001, 'save_checkpoint': True, 'save_freq': 2000, 'use_policy_training_preset': True, 'optimizer': {'type': 'adam', 'lr': 5e-06, 'weight_decay': 1e-06, 'grad_clip_norm': 10.0, 'betas': [0.95, 0.999], 'eps': 1e-08}, 'scheduler': {'type': 'diffuser', 'num_warmup_steps': 100, 'name': 'cosine'}, 'eval': {'n_episodes': 50, 'batch_size': 19, 'use_async_envs': True}, 'wandb': {'enable': True, 'disable_artifact': False, 'project': 'lerobot', 'entity': None, 'notes': None, 'run_id': None, 'mode': None, 'add_tags': True}, 'peft': None, 'sample_weighting': None, 'rename_map': {}, 'checkpoint_path': None, '_wandb': {}}
9
+ 2026-05-21 12:23:31,015 INFO MainThread:1860986 [wandb_init.py:init():899] starting backend
10
+ 2026-05-21 12:23:31,230 INFO MainThread:1860986 [wandb_init.py:init():914] sending inform_init request
11
+ 2026-05-21 12:23:31,436 INFO MainThread:1860986 [wandb_init.py:init():919] backend started and connected
12
+ 2026-05-21 12:23:31,439 INFO MainThread:1860986 [wandb_init.py:init():989] updated telemetry
13
+ 2026-05-21 12:23:31,443 INFO MainThread:1860986 [wandb_init.py:init():1012] communicating run to backend with 90.0 second timeout
14
+ 2026-05-21 12:23:32,240 INFO MainThread:1860986 [wandb_init.py:init():1057] starting run threads in backend
15
+ 2026-05-21 12:23:32,295 INFO MainThread:1860986 [wandb_run.py:_console_start():2509] atexit reg
16
+ 2026-05-21 12:23:32,295 INFO MainThread:1860986 [wandb_run.py:_redirect():2359] redirect: wrap_raw
17
+ 2026-05-21 12:23:32,295 INFO MainThread:1860986 [wandb_run.py:_redirect():2428] Wrapping output streams.
18
+ 2026-05-21 12:23:32,295 INFO MainThread:1860986 [wandb_run.py:_redirect():2451] Redirects installed.
19
+ 2026-05-21 12:23:32,296 INFO MainThread:1860986 [wandb_init.py:init():1095] run started, returning control to user process
checkpoints/generalized_eval1/wandb/run-20260521_122331-91fmyb38/run-91fmyb38.wandb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce30f7b63c5b47060ed22606530ca87ba79d35f99884517990349d1dfd33d0c2
3
+ size 294912