add more changes

This commit is contained in:
Jade Choghari
2025-12-23 13:11:18 +00:00
parent 5781754c30
commit 9b5ac4387c
11 changed files with 79 additions and 35 deletions
+18
View File
@@ -0,0 +1,18 @@
import torch
from huggingface_hub import HfApi
import lerobot
from lerobot.datasets.lerobot_dataset import LeRobotDataset, LeRobotDatasetMetadata
dataset = LeRobotDataset(repo_id="lerobot/libero")
dataloader = torch.utils.data.DataLoader(
dataset,
num_workers=0,
batch_size=4,
shuffle=True,
)
batch = next(iter(dataloader))
print(batch.keys())
breakpoint()