admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-16 00:59:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

small fix

Browse Source
This commit is contained in:
Remi Cadene
2024-03-06 10:21:22 +00:00
parent f95ecd66fc
commit eed24b083a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lerobot/common/datasets/factory.py
+1 -1
View File
@@ -102,7 +102,7 @@ def make_offline_buffer(
offline_buffer.set_transform(transform)
if not overwrite_sampler:
index = torch.arange(0, offline_buffer.num_frames, 1)
index = torch.arange(0, offline_buffer.num_samples, 1)
sampler.extend(index)
return offline_buffer