From eed24b083a83a4fd3cdf0298c5ee70f7d387fdfa Mon Sep 17 00:00:00 2001 From: Remi Cadene Date: Wed, 6 Mar 2024 10:21:22 +0000 Subject: [PATCH] small fix --- lerobot/common/datasets/factory.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/common/datasets/factory.py b/lerobot/common/datasets/factory.py index c72da5e5..e054682e 100644 --- a/lerobot/common/datasets/factory.py +++ b/lerobot/common/datasets/factory.py @@ -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