hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Add root in example

Browse Source
This commit is contained in:
Cadene
2024-03-26 10:40:06 +00:00
parent 5a46b8a2a9
commit edacc1d2a0
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/1_visualize_dataset.py
View File

@@ -1,3 +1,5 @@
import os
from torchrl.data.replay_buffers import SamplerWithoutReplacement from torchrl.data.replay_buffers import SamplerWithoutReplacement
import lerobot import lerobot
@@ -10,7 +12,7 @@ print(lerobot.available_datasets)
# we use this sampler to sample 1 frame after the other # we use this sampler to sample 1 frame after the other
sampler = SamplerWithoutReplacement(shuffle=False) sampler = SamplerWithoutReplacement(shuffle=False)
dataset = AlohaDataset("aloha_sim_transfer_cube_human", sampler=sampler) dataset = AlohaDataset("aloha_sim_transfer_cube_human", sampler=sampler, root=os.environ.get("DATA_DIR"))
video_paths = render_dataset( video_paths = render_dataset(
dataset, dataset,