chao.wu/Lerobot
1
0
Fork 0
Code Pull Requests 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
Show Stats 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