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

Fix episode chunk

Browse Source
This commit is contained in:
Simon Alibert
2024-10-23 00:27:14 +02:00
parent c3c0141738
commit 9dca233d7e
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lerobot/common/datasets/lerobot_dataset.py
View File

@@ -305,10 +305,7 @@ class LeRobotDataset(torch.utils.data.Dataset):
return Path(fpath) return Path(fpath)
def get_episode_chunk(self, ep_index: int) -> int: def get_episode_chunk(self, ep_index: int) -> int:
ep_chunk = ep_index // self.chunks_size return ep_index // self.chunks_size
if ep_index > 0 and ep_index % self.chunks_size == 0:
ep_chunk -= 1
return ep_chunk
@property @property
def data_path(self) -> str: def data_path(self) -> str: