libero_object/dataset_manifest_shard_0003.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": 12249,`
35	`"episodes": [`
36	`{`
37	`"episode_index": 3,`
38	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000003.pt",`
39	`"window_count": 114,`
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": 8,`
52	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000008.pt",`
53	`"window_count": 136,`
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": 13,`
66	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000013.pt",`
67	`"window_count": 135,`
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": 18,`
80	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000018.pt",`
81	`"window_count": 137,`
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": 23,`
94	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000023.pt",`
95	`"window_count": 114,`
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": 28,`
108	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000028.pt",`
109	`"window_count": 113,`
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": 33,`
122	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000033.pt",`
123	`"window_count": 137,`
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": 38,`
136	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000038.pt",`
137	`"window_count": 142,`
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": 43,`
150	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000043.pt",`
151	`"window_count": 136,`
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": 48,`
164	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000048.pt",`
165	`"window_count": 135,`
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": 53,`
178	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000053.pt",`
179	`"window_count": 137,`
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": 58,`
192	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000058.pt",`
193	`"window_count": 109,`
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": 63,`
206	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000063.pt",`
207	`"window_count": 119,`
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": 68,`
220	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000068.pt",`
221	`"window_count": 137,`
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": 73,`
234	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000073.pt",`
235	`"window_count": 112,`
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": 78,`
248	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000078.pt",`
249	`"window_count": 159,`
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": 83,`
262	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000083.pt",`
263	`"window_count": 143,`
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": 88,`
276	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000088.pt",`
277	`"window_count": 123,`
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": 93,`
290	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000093.pt",`
291	`"window_count": 124,`
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": 98,`
304	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000098.pt",`
305	`"window_count": 138,`
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": 103,`
318	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000103.pt",`
319	`"window_count": 141,`
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": 108,`
332	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000108.pt",`
333	`"window_count": 144,`
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": 113,`
346	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000113.pt",`
347	`"window_count": 136,`
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": 118,`
360	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000118.pt",`
361	`"window_count": 141,`
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": 123,`
374	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000123.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": 128,`
388	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000128.pt",`
389	`"window_count": 122,`
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": 133,`
402	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000133.pt",`
403	`"window_count": 125,`
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": 138,`
416	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000138.pt",`
417	`"window_count": 138,`
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": 143,`
430	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000143.pt",`
431	`"window_count": 145,`
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": 148,`
444	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000148.pt",`
445	`"window_count": 132,`
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": 153,`
458	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000153.pt",`
459	`"window_count": 151,`
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": 158,`
472	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000158.pt",`
473	`"window_count": 146,`
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": 163,`
486	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000163.pt",`
487	`"window_count": 118,`
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": 168,`
500	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000168.pt",`
501	`"window_count": 148,`
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": 173,`
514	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000173.pt",`
515	`"window_count": 116,`
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": 178,`
528	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000178.pt",`
529	`"window_count": 129,`
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": 183,`
542	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000183.pt",`
543	`"window_count": 134,`
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": 188,`
556	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000188.pt",`
557	`"window_count": 141,`
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": 193,`
570	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000193.pt",`
571	`"window_count": 111,`
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": 198,`
584	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000198.pt",`
585	`"window_count": 146,`
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": 203,`
598	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000203.pt",`
599	`"window_count": 128,`
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": 208,`
612	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000208.pt",`
613	`"window_count": 117,`
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": 213,`
626	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000213.pt",`
627	`"window_count": 145,`
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": 218,`
640	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000218.pt",`
641	`"window_count": 133,`
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": 223,`
654	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000223.pt",`
655	`"window_count": 145,`
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": 228,`
668	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000228.pt",`
669	`"window_count": 120,`
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": 233,`
682	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000233.pt",`
683	`"window_count": 106,`
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": 238,`
696	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000238.pt",`
697	`"window_count": 126,`
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": 243,`
710	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000243.pt",`
711	`"window_count": 107,`
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": 248,`
724	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000248.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": 253,`
738	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000253.pt",`
739	`"window_count": 140,`
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": 258,`
752	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000258.pt",`
753	`"window_count": 115,`
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": 263,`
766	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000263.pt",`
767	`"window_count": 117,`
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": 268,`
780	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000268.pt",`
781	`"window_count": 144,`
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": 273,`
794	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000273.pt",`
795	`"window_count": 127,`
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": 278,`
808	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000278.pt",`
809	`"window_count": 115,`
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": 283,`
822	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000283.pt",`
823	`"window_count": 203,`
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": 288,`
836	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000288.pt",`
837	`"window_count": 134,`
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": 293,`
850	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000293.pt",`
851	`"window_count": 163,`
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": 298,`
864	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000298.pt",`
865	`"window_count": 180,`
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": 303,`
878	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000303.pt",`
879	`"window_count": 135,`
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": 308,`
892	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000308.pt",`
893	`"window_count": 148,`
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": 313,`
906	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000313.pt",`
907	`"window_count": 178,`
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": 318,`
920	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000318.pt",`
921	`"window_count": 154,`
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": 323,`
934	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000323.pt",`
935	`"window_count": 150,`
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": 328,`
948	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000328.pt",`
949	`"window_count": 158,`
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": 333,`
962	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000333.pt",`
963	`"window_count": 146,`
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": 338,`
976	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000338.pt",`
977	`"window_count": 148,`
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": 343,`
990	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000343.pt",`
991	`"window_count": 133,`
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": 348,`
1004	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000348.pt",`
1005	`"window_count": 108,`
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": 353,`
1018	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000353.pt",`
1019	`"window_count": 125,`
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": 358,`
1032	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000358.pt",`
1033	`"window_count": 108,`
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": 363,`
1046	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000363.pt",`
1047	`"window_count": 107,`
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": 368,`
1060	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000368.pt",`
1061	`"window_count": 118,`
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": 373,`
1074	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000373.pt",`
1075	`"window_count": 159,`
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": 378,`
1088	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000378.pt",`
1089	`"window_count": 146,`
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": 383,`
1102	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000383.pt",`
1103	`"window_count": 120,`
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": 388,`
1116	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000388.pt",`
1117	`"window_count": 131,`
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": 393,`
1130	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000393.pt",`
1131	`"window_count": 117,`
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": 398,`
1144	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000398.pt",`
1145	`"window_count": 130,`
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": 403,`
1158	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000403.pt",`
1159	`"window_count": 146,`
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": 408,`
1172	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000408.pt",`
1173	`"window_count": 155,`
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": 413,`
1186	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000413.pt",`
1187	`"window_count": 134,`
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": 418,`
1200	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000418.pt",`
1201	`"window_count": 135,`
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": 423,`
1214	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000423.pt",`
1215	`"window_count": 115,`
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": 428,`
1228	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000428.pt",`
1229	`"window_count": 152,`
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": 433,`
1242	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000433.pt",`
1243	`"window_count": 139,`
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": 438,`
1256	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000438.pt",`
1257	`"window_count": 141,`
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": 443,`
1270	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000443.pt",`
1271	`"window_count": 137,`
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": 448,`
1284	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000448.pt",`
1285	`"window_count": 139,`
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": 453,`
1298	`"episode_path": "/disk/rl/data/LIBERO_LeRobot_v3_cosmos_exact_window_shared_vae_v1/libero_object/episodes/episode_000453.pt",`
1299	`"window_count": 136,`
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": 3,`
1312	`"num_shards": 5`
1313	`}`
1314