libero_object/dataset_manifest_shard_0000.json
31.8 KB · 1314 lines · json Raw
1 {
2 "schema_version": "exact_window_v1",
3 "suite": "libero_object",
4 "source_dataset": "/disk/rl/data/LIBERO_LeRobot_v3/libero_object",
5 "image_size": 256,
6 "chunk_length": 16,
7 "camera_mode": "concat_view",
8 "sample_stride": 1,
9 "fps": 20.0,
10 "latent_shape": [
11 5,
12 48,
13 12,
14 20
15 ],
16 "vae_path": "/disk/rl/psm_wma/cosmos-framework/examples/checkpoints/wan22_vae/Wan2.2_VAE.pth",
17 "vae_encode_contract": {
18 "compute_dtype": "torch.bfloat16",
19 "encode_exact_durations": [
20 17,
21 61,
22 73
23 ],
24 "encode_chunk_frames": {
25 "256": 68,
26 "480": 24,
27 "720": 8,
28 "768": 8
29 }
30 },
31 "script_revision": "330f001",
32 "task_index": null,
33 "episode_count": 91,
34 "window_count": 11699,
35 "episodes": [
36 {
37 "episode_index": 0,
38 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000000.pt",
39 "window_count": 127,
40 "image_size": 256,
41 "target_h": 192,
42 "target_w": 320,
43 "latent_shape": [
44 5,
45 48,
46 12,
47 20
48 ]
49 },
50 {
51 "episode_index": 5,
52 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000005.pt",
53 "window_count": 116,
54 "image_size": 256,
55 "target_h": 192,
56 "target_w": 320,
57 "latent_shape": [
58 5,
59 48,
60 12,
61 20
62 ]
63 },
64 {
65 "episode_index": 10,
66 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000010.pt",
67 "window_count": 127,
68 "image_size": 256,
69 "target_h": 192,
70 "target_w": 320,
71 "latent_shape": [
72 5,
73 48,
74 12,
75 20
76 ]
77 },
78 {
79 "episode_index": 15,
80 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000015.pt",
81 "window_count": 99,
82 "image_size": 256,
83 "target_h": 192,
84 "target_w": 320,
85 "latent_shape": [
86 5,
87 48,
88 12,
89 20
90 ]
91 },
92 {
93 "episode_index": 20,
94 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000020.pt",
95 "window_count": 140,
96 "image_size": 256,
97 "target_h": 192,
98 "target_w": 320,
99 "latent_shape": [
100 5,
101 48,
102 12,
103 20
104 ]
105 },
106 {
107 "episode_index": 25,
108 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000025.pt",
109 "window_count": 120,
110 "image_size": 256,
111 "target_h": 192,
112 "target_w": 320,
113 "latent_shape": [
114 5,
115 48,
116 12,
117 20
118 ]
119 },
120 {
121 "episode_index": 30,
122 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000030.pt",
123 "window_count": 109,
124 "image_size": 256,
125 "target_h": 192,
126 "target_w": 320,
127 "latent_shape": [
128 5,
129 48,
130 12,
131 20
132 ]
133 },
134 {
135 "episode_index": 35,
136 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000035.pt",
137 "window_count": 152,
138 "image_size": 256,
139 "target_h": 192,
140 "target_w": 320,
141 "latent_shape": [
142 5,
143 48,
144 12,
145 20
146 ]
147 },
148 {
149 "episode_index": 40,
150 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000040.pt",
151 "window_count": 147,
152 "image_size": 256,
153 "target_h": 192,
154 "target_w": 320,
155 "latent_shape": [
156 5,
157 48,
158 12,
159 20
160 ]
161 },
162 {
163 "episode_index": 45,
164 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000045.pt",
165 "window_count": 153,
166 "image_size": 256,
167 "target_h": 192,
168 "target_w": 320,
169 "latent_shape": [
170 5,
171 48,
172 12,
173 20
174 ]
175 },
176 {
177 "episode_index": 50,
178 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000050.pt",
179 "window_count": 138,
180 "image_size": 256,
181 "target_h": 192,
182 "target_w": 320,
183 "latent_shape": [
184 5,
185 48,
186 12,
187 20
188 ]
189 },
190 {
191 "episode_index": 55,
192 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000055.pt",
193 "window_count": 151,
194 "image_size": 256,
195 "target_h": 192,
196 "target_w": 320,
197 "latent_shape": [
198 5,
199 48,
200 12,
201 20
202 ]
203 },
204 {
205 "episode_index": 60,
206 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000060.pt",
207 "window_count": 112,
208 "image_size": 256,
209 "target_h": 192,
210 "target_w": 320,
211 "latent_shape": [
212 5,
213 48,
214 12,
215 20
216 ]
217 },
218 {
219 "episode_index": 65,
220 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000065.pt",
221 "window_count": 135,
222 "image_size": 256,
223 "target_h": 192,
224 "target_w": 320,
225 "latent_shape": [
226 5,
227 48,
228 12,
229 20
230 ]
231 },
232 {
233 "episode_index": 70,
234 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000070.pt",
235 "window_count": 137,
236 "image_size": 256,
237 "target_h": 192,
238 "target_w": 320,
239 "latent_shape": [
240 5,
241 48,
242 12,
243 20
244 ]
245 },
246 {
247 "episode_index": 75,
248 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000075.pt",
249 "window_count": 112,
250 "image_size": 256,
251 "target_h": 192,
252 "target_w": 320,
253 "latent_shape": [
254 5,
255 48,
256 12,
257 20
258 ]
259 },
260 {
261 "episode_index": 80,
262 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000080.pt",
263 "window_count": 127,
264 "image_size": 256,
265 "target_h": 192,
266 "target_w": 320,
267 "latent_shape": [
268 5,
269 48,
270 12,
271 20
272 ]
273 },
274 {
275 "episode_index": 85,
276 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000085.pt",
277 "window_count": 155,
278 "image_size": 256,
279 "target_h": 192,
280 "target_w": 320,
281 "latent_shape": [
282 5,
283 48,
284 12,
285 20
286 ]
287 },
288 {
289 "episode_index": 90,
290 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000090.pt",
291 "window_count": 110,
292 "image_size": 256,
293 "target_h": 192,
294 "target_w": 320,
295 "latent_shape": [
296 5,
297 48,
298 12,
299 20
300 ]
301 },
302 {
303 "episode_index": 95,
304 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000095.pt",
305 "window_count": 108,
306 "image_size": 256,
307 "target_h": 192,
308 "target_w": 320,
309 "latent_shape": [
310 5,
311 48,
312 12,
313 20
314 ]
315 },
316 {
317 "episode_index": 100,
318 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000100.pt",
319 "window_count": 138,
320 "image_size": 256,
321 "target_h": 192,
322 "target_w": 320,
323 "latent_shape": [
324 5,
325 48,
326 12,
327 20
328 ]
329 },
330 {
331 "episode_index": 105,
332 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000105.pt",
333 "window_count": 122,
334 "image_size": 256,
335 "target_h": 192,
336 "target_w": 320,
337 "latent_shape": [
338 5,
339 48,
340 12,
341 20
342 ]
343 },
344 {
345 "episode_index": 110,
346 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000110.pt",
347 "window_count": 134,
348 "image_size": 256,
349 "target_h": 192,
350 "target_w": 320,
351 "latent_shape": [
352 5,
353 48,
354 12,
355 20
356 ]
357 },
358 {
359 "episode_index": 115,
360 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000115.pt",
361 "window_count": 113,
362 "image_size": 256,
363 "target_h": 192,
364 "target_w": 320,
365 "latent_shape": [
366 5,
367 48,
368 12,
369 20
370 ]
371 },
372 {
373 "episode_index": 120,
374 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000120.pt",
375 "window_count": 141,
376 "image_size": 256,
377 "target_h": 192,
378 "target_w": 320,
379 "latent_shape": [
380 5,
381 48,
382 12,
383 20
384 ]
385 },
386 {
387 "episode_index": 125,
388 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000125.pt",
389 "window_count": 111,
390 "image_size": 256,
391 "target_h": 192,
392 "target_w": 320,
393 "latent_shape": [
394 5,
395 48,
396 12,
397 20
398 ]
399 },
400 {
401 "episode_index": 130,
402 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000130.pt",
403 "window_count": 134,
404 "image_size": 256,
405 "target_h": 192,
406 "target_w": 320,
407 "latent_shape": [
408 5,
409 48,
410 12,
411 20
412 ]
413 },
414 {
415 "episode_index": 135,
416 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000135.pt",
417 "window_count": 140,
418 "image_size": 256,
419 "target_h": 192,
420 "target_w": 320,
421 "latent_shape": [
422 5,
423 48,
424 12,
425 20
426 ]
427 },
428 {
429 "episode_index": 140,
430 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000140.pt",
431 "window_count": 122,
432 "image_size": 256,
433 "target_h": 192,
434 "target_w": 320,
435 "latent_shape": [
436 5,
437 48,
438 12,
439 20
440 ]
441 },
442 {
443 "episode_index": 145,
444 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000145.pt",
445 "window_count": 108,
446 "image_size": 256,
447 "target_h": 192,
448 "target_w": 320,
449 "latent_shape": [
450 5,
451 48,
452 12,
453 20
454 ]
455 },
456 {
457 "episode_index": 150,
458 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000150.pt",
459 "window_count": 134,
460 "image_size": 256,
461 "target_h": 192,
462 "target_w": 320,
463 "latent_shape": [
464 5,
465 48,
466 12,
467 20
468 ]
469 },
470 {
471 "episode_index": 155,
472 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000155.pt",
473 "window_count": 121,
474 "image_size": 256,
475 "target_h": 192,
476 "target_w": 320,
477 "latent_shape": [
478 5,
479 48,
480 12,
481 20
482 ]
483 },
484 {
485 "episode_index": 160,
486 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000160.pt",
487 "window_count": 129,
488 "image_size": 256,
489 "target_h": 192,
490 "target_w": 320,
491 "latent_shape": [
492 5,
493 48,
494 12,
495 20
496 ]
497 },
498 {
499 "episode_index": 165,
500 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000165.pt",
501 "window_count": 134,
502 "image_size": 256,
503 "target_h": 192,
504 "target_w": 320,
505 "latent_shape": [
506 5,
507 48,
508 12,
509 20
510 ]
511 },
512 {
513 "episode_index": 170,
514 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000170.pt",
515 "window_count": 144,
516 "image_size": 256,
517 "target_h": 192,
518 "target_w": 320,
519 "latent_shape": [
520 5,
521 48,
522 12,
523 20
524 ]
525 },
526 {
527 "episode_index": 175,
528 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000175.pt",
529 "window_count": 119,
530 "image_size": 256,
531 "target_h": 192,
532 "target_w": 320,
533 "latent_shape": [
534 5,
535 48,
536 12,
537 20
538 ]
539 },
540 {
541 "episode_index": 180,
542 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000180.pt",
543 "window_count": 123,
544 "image_size": 256,
545 "target_h": 192,
546 "target_w": 320,
547 "latent_shape": [
548 5,
549 48,
550 12,
551 20
552 ]
553 },
554 {
555 "episode_index": 185,
556 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000185.pt",
557 "window_count": 126,
558 "image_size": 256,
559 "target_h": 192,
560 "target_w": 320,
561 "latent_shape": [
562 5,
563 48,
564 12,
565 20
566 ]
567 },
568 {
569 "episode_index": 190,
570 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000190.pt",
571 "window_count": 161,
572 "image_size": 256,
573 "target_h": 192,
574 "target_w": 320,
575 "latent_shape": [
576 5,
577 48,
578 12,
579 20
580 ]
581 },
582 {
583 "episode_index": 195,
584 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000195.pt",
585 "window_count": 156,
586 "image_size": 256,
587 "target_h": 192,
588 "target_w": 320,
589 "latent_shape": [
590 5,
591 48,
592 12,
593 20
594 ]
595 },
596 {
597 "episode_index": 200,
598 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000200.pt",
599 "window_count": 109,
600 "image_size": 256,
601 "target_h": 192,
602 "target_w": 320,
603 "latent_shape": [
604 5,
605 48,
606 12,
607 20
608 ]
609 },
610 {
611 "episode_index": 205,
612 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000205.pt",
613 "window_count": 136,
614 "image_size": 256,
615 "target_h": 192,
616 "target_w": 320,
617 "latent_shape": [
618 5,
619 48,
620 12,
621 20
622 ]
623 },
624 {
625 "episode_index": 210,
626 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000210.pt",
627 "window_count": 102,
628 "image_size": 256,
629 "target_h": 192,
630 "target_w": 320,
631 "latent_shape": [
632 5,
633 48,
634 12,
635 20
636 ]
637 },
638 {
639 "episode_index": 215,
640 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000215.pt",
641 "window_count": 159,
642 "image_size": 256,
643 "target_h": 192,
644 "target_w": 320,
645 "latent_shape": [
646 5,
647 48,
648 12,
649 20
650 ]
651 },
652 {
653 "episode_index": 220,
654 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000220.pt",
655 "window_count": 130,
656 "image_size": 256,
657 "target_h": 192,
658 "target_w": 320,
659 "latent_shape": [
660 5,
661 48,
662 12,
663 20
664 ]
665 },
666 {
667 "episode_index": 225,
668 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000225.pt",
669 "window_count": 141,
670 "image_size": 256,
671 "target_h": 192,
672 "target_w": 320,
673 "latent_shape": [
674 5,
675 48,
676 12,
677 20
678 ]
679 },
680 {
681 "episode_index": 230,
682 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000230.pt",
683 "window_count": 167,
684 "image_size": 256,
685 "target_h": 192,
686 "target_w": 320,
687 "latent_shape": [
688 5,
689 48,
690 12,
691 20
692 ]
693 },
694 {
695 "episode_index": 235,
696 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000235.pt",
697 "window_count": 128,
698 "image_size": 256,
699 "target_h": 192,
700 "target_w": 320,
701 "latent_shape": [
702 5,
703 48,
704 12,
705 20
706 ]
707 },
708 {
709 "episode_index": 240,
710 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000240.pt",
711 "window_count": 150,
712 "image_size": 256,
713 "target_h": 192,
714 "target_w": 320,
715 "latent_shape": [
716 5,
717 48,
718 12,
719 20
720 ]
721 },
722 {
723 "episode_index": 245,
724 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000245.pt",
725 "window_count": 131,
726 "image_size": 256,
727 "target_h": 192,
728 "target_w": 320,
729 "latent_shape": [
730 5,
731 48,
732 12,
733 20
734 ]
735 },
736 {
737 "episode_index": 250,
738 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000250.pt",
739 "window_count": 119,
740 "image_size": 256,
741 "target_h": 192,
742 "target_w": 320,
743 "latent_shape": [
744 5,
745 48,
746 12,
747 20
748 ]
749 },
750 {
751 "episode_index": 255,
752 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000255.pt",
753 "window_count": 124,
754 "image_size": 256,
755 "target_h": 192,
756 "target_w": 320,
757 "latent_shape": [
758 5,
759 48,
760 12,
761 20
762 ]
763 },
764 {
765 "episode_index": 260,
766 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000260.pt",
767 "window_count": 146,
768 "image_size": 256,
769 "target_h": 192,
770 "target_w": 320,
771 "latent_shape": [
772 5,
773 48,
774 12,
775 20
776 ]
777 },
778 {
779 "episode_index": 265,
780 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000265.pt",
781 "window_count": 117,
782 "image_size": 256,
783 "target_h": 192,
784 "target_w": 320,
785 "latent_shape": [
786 5,
787 48,
788 12,
789 20
790 ]
791 },
792 {
793 "episode_index": 270,
794 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000270.pt",
795 "window_count": 119,
796 "image_size": 256,
797 "target_h": 192,
798 "target_w": 320,
799 "latent_shape": [
800 5,
801 48,
802 12,
803 20
804 ]
805 },
806 {
807 "episode_index": 275,
808 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000275.pt",
809 "window_count": 134,
810 "image_size": 256,
811 "target_h": 192,
812 "target_w": 320,
813 "latent_shape": [
814 5,
815 48,
816 12,
817 20
818 ]
819 },
820 {
821 "episode_index": 280,
822 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000280.pt",
823 "window_count": 144,
824 "image_size": 256,
825 "target_h": 192,
826 "target_w": 320,
827 "latent_shape": [
828 5,
829 48,
830 12,
831 20
832 ]
833 },
834 {
835 "episode_index": 285,
836 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000285.pt",
837 "window_count": 121,
838 "image_size": 256,
839 "target_h": 192,
840 "target_w": 320,
841 "latent_shape": [
842 5,
843 48,
844 12,
845 20
846 ]
847 },
848 {
849 "episode_index": 290,
850 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000290.pt",
851 "window_count": 142,
852 "image_size": 256,
853 "target_h": 192,
854 "target_w": 320,
855 "latent_shape": [
856 5,
857 48,
858 12,
859 20
860 ]
861 },
862 {
863 "episode_index": 295,
864 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000295.pt",
865 "window_count": 134,
866 "image_size": 256,
867 "target_h": 192,
868 "target_w": 320,
869 "latent_shape": [
870 5,
871 48,
872 12,
873 20
874 ]
875 },
876 {
877 "episode_index": 300,
878 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000300.pt",
879 "window_count": 110,
880 "image_size": 256,
881 "target_h": 192,
882 "target_w": 320,
883 "latent_shape": [
884 5,
885 48,
886 12,
887 20
888 ]
889 },
890 {
891 "episode_index": 305,
892 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000305.pt",
893 "window_count": 134,
894 "image_size": 256,
895 "target_h": 192,
896 "target_w": 320,
897 "latent_shape": [
898 5,
899 48,
900 12,
901 20
902 ]
903 },
904 {
905 "episode_index": 310,
906 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000310.pt",
907 "window_count": 133,
908 "image_size": 256,
909 "target_h": 192,
910 "target_w": 320,
911 "latent_shape": [
912 5,
913 48,
914 12,
915 20
916 ]
917 },
918 {
919 "episode_index": 315,
920 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000315.pt",
921 "window_count": 104,
922 "image_size": 256,
923 "target_h": 192,
924 "target_w": 320,
925 "latent_shape": [
926 5,
927 48,
928 12,
929 20
930 ]
931 },
932 {
933 "episode_index": 320,
934 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000320.pt",
935 "window_count": 143,
936 "image_size": 256,
937 "target_h": 192,
938 "target_w": 320,
939 "latent_shape": [
940 5,
941 48,
942 12,
943 20
944 ]
945 },
946 {
947 "episode_index": 325,
948 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000325.pt",
949 "window_count": 120,
950 "image_size": 256,
951 "target_h": 192,
952 "target_w": 320,
953 "latent_shape": [
954 5,
955 48,
956 12,
957 20
958 ]
959 },
960 {
961 "episode_index": 330,
962 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000330.pt",
963 "window_count": 120,
964 "image_size": 256,
965 "target_h": 192,
966 "target_w": 320,
967 "latent_shape": [
968 5,
969 48,
970 12,
971 20
972 ]
973 },
974 {
975 "episode_index": 335,
976 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000335.pt",
977 "window_count": 129,
978 "image_size": 256,
979 "target_h": 192,
980 "target_w": 320,
981 "latent_shape": [
982 5,
983 48,
984 12,
985 20
986 ]
987 },
988 {
989 "episode_index": 340,
990 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000340.pt",
991 "window_count": 130,
992 "image_size": 256,
993 "target_h": 192,
994 "target_w": 320,
995 "latent_shape": [
996 5,
997 48,
998 12,
999 20
1000 ]
1001 },
1002 {
1003 "episode_index": 345,
1004 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000345.pt",
1005 "window_count": 136,
1006 "image_size": 256,
1007 "target_h": 192,
1008 "target_w": 320,
1009 "latent_shape": [
1010 5,
1011 48,
1012 12,
1013 20
1014 ]
1015 },
1016 {
1017 "episode_index": 350,
1018 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000350.pt",
1019 "window_count": 107,
1020 "image_size": 256,
1021 "target_h": 192,
1022 "target_w": 320,
1023 "latent_shape": [
1024 5,
1025 48,
1026 12,
1027 20
1028 ]
1029 },
1030 {
1031 "episode_index": 355,
1032 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000355.pt",
1033 "window_count": 148,
1034 "image_size": 256,
1035 "target_h": 192,
1036 "target_w": 320,
1037 "latent_shape": [
1038 5,
1039 48,
1040 12,
1041 20
1042 ]
1043 },
1044 {
1045 "episode_index": 360,
1046 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000360.pt",
1047 "window_count": 104,
1048 "image_size": 256,
1049 "target_h": 192,
1050 "target_w": 320,
1051 "latent_shape": [
1052 5,
1053 48,
1054 12,
1055 20
1056 ]
1057 },
1058 {
1059 "episode_index": 365,
1060 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000365.pt",
1061 "window_count": 126,
1062 "image_size": 256,
1063 "target_h": 192,
1064 "target_w": 320,
1065 "latent_shape": [
1066 5,
1067 48,
1068 12,
1069 20
1070 ]
1071 },
1072 {
1073 "episode_index": 370,
1074 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000370.pt",
1075 "window_count": 185,
1076 "image_size": 256,
1077 "target_h": 192,
1078 "target_w": 320,
1079 "latent_shape": [
1080 5,
1081 48,
1082 12,
1083 20
1084 ]
1085 },
1086 {
1087 "episode_index": 375,
1088 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000375.pt",
1089 "window_count": 116,
1090 "image_size": 256,
1091 "target_h": 192,
1092 "target_w": 320,
1093 "latent_shape": [
1094 5,
1095 48,
1096 12,
1097 20
1098 ]
1099 },
1100 {
1101 "episode_index": 380,
1102 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000380.pt",
1103 "window_count": 135,
1104 "image_size": 256,
1105 "target_h": 192,
1106 "target_w": 320,
1107 "latent_shape": [
1108 5,
1109 48,
1110 12,
1111 20
1112 ]
1113 },
1114 {
1115 "episode_index": 385,
1116 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000385.pt",
1117 "window_count": 141,
1118 "image_size": 256,
1119 "target_h": 192,
1120 "target_w": 320,
1121 "latent_shape": [
1122 5,
1123 48,
1124 12,
1125 20
1126 ]
1127 },
1128 {
1129 "episode_index": 390,
1130 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000390.pt",
1131 "window_count": 105,
1132 "image_size": 256,
1133 "target_h": 192,
1134 "target_w": 320,
1135 "latent_shape": [
1136 5,
1137 48,
1138 12,
1139 20
1140 ]
1141 },
1142 {
1143 "episode_index": 395,
1144 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000395.pt",
1145 "window_count": 123,
1146 "image_size": 256,
1147 "target_h": 192,
1148 "target_w": 320,
1149 "latent_shape": [
1150 5,
1151 48,
1152 12,
1153 20
1154 ]
1155 },
1156 {
1157 "episode_index": 400,
1158 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000400.pt",
1159 "window_count": 117,
1160 "image_size": 256,
1161 "target_h": 192,
1162 "target_w": 320,
1163 "latent_shape": [
1164 5,
1165 48,
1166 12,
1167 20
1168 ]
1169 },
1170 {
1171 "episode_index": 405,
1172 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000405.pt",
1173 "window_count": 137,
1174 "image_size": 256,
1175 "target_h": 192,
1176 "target_w": 320,
1177 "latent_shape": [
1178 5,
1179 48,
1180 12,
1181 20
1182 ]
1183 },
1184 {
1185 "episode_index": 410,
1186 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000410.pt",
1187 "window_count": 125,
1188 "image_size": 256,
1189 "target_h": 192,
1190 "target_w": 320,
1191 "latent_shape": [
1192 5,
1193 48,
1194 12,
1195 20
1196 ]
1197 },
1198 {
1199 "episode_index": 415,
1200 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000415.pt",
1201 "window_count": 112,
1202 "image_size": 256,
1203 "target_h": 192,
1204 "target_w": 320,
1205 "latent_shape": [
1206 5,
1207 48,
1208 12,
1209 20
1210 ]
1211 },
1212 {
1213 "episode_index": 420,
1214 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000420.pt",
1215 "window_count": 110,
1216 "image_size": 256,
1217 "target_h": 192,
1218 "target_w": 320,
1219 "latent_shape": [
1220 5,
1221 48,
1222 12,
1223 20
1224 ]
1225 },
1226 {
1227 "episode_index": 425,
1228 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000425.pt",
1229 "window_count": 112,
1230 "image_size": 256,
1231 "target_h": 192,
1232 "target_w": 320,
1233 "latent_shape": [
1234 5,
1235 48,
1236 12,
1237 20
1238 ]
1239 },
1240 {
1241 "episode_index": 430,
1242 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000430.pt",
1243 "window_count": 133,
1244 "image_size": 256,
1245 "target_h": 192,
1246 "target_w": 320,
1247 "latent_shape": [
1248 5,
1249 48,
1250 12,
1251 20
1252 ]
1253 },
1254 {
1255 "episode_index": 435,
1256 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000435.pt",
1257 "window_count": 139,
1258 "image_size": 256,
1259 "target_h": 192,
1260 "target_w": 320,
1261 "latent_shape": [
1262 5,
1263 48,
1264 12,
1265 20
1266 ]
1267 },
1268 {
1269 "episode_index": 440,
1270 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000440.pt",
1271 "window_count": 102,
1272 "image_size": 256,
1273 "target_h": 192,
1274 "target_w": 320,
1275 "latent_shape": [
1276 5,
1277 48,
1278 12,
1279 20
1280 ]
1281 },
1282 {
1283 "episode_index": 445,
1284 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000445.pt",
1285 "window_count": 109,
1286 "image_size": 256,
1287 "target_h": 192,
1288 "target_w": 320,
1289 "latent_shape": [
1290 5,
1291 48,
1292 12,
1293 20
1294 ]
1295 },
1296 {
1297 "episode_index": 450,
1298 "episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000450.pt",
1299 "window_count": 116,
1300 "image_size": 256,
1301 "target_h": 192,
1302 "target_w": 320,
1303 "latent_shape": [
1304 5,
1305 48,
1306 12,
1307 20
1308 ]
1309 }
1310 ],
1311 "episode_shard": 0,
1312 "num_shards": 5
1313 }
1314