tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 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

2
lerobot/common/datasets/factory.py
View File

@@ -102,7 +102,7 @@ def make_offline_buffer(
offline_buffer.set_transform(transform) offline_buffer.set_transform(transform)
if not overwrite_sampler: 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) sampler.extend(index)
return offline_buffer return offline_buffer