admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-22 17:32:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

max 1m frames

Browse Source
This commit is contained in:
Pepijn
2026-02-21 17:43:58 +01:00
parent acae8417aa
commit de0663226a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/scripts/lerobot_train.py
+2 -2
View File
@@ -255,8 +255,8 @@ def train(cfg: TrainPipelineConfig, accelerator: Accelerator | None = None):
chunk_size = cfg.policy.chunk_size chunk_size = cfg.policy.chunk_size
hf = dataset.hf_dataset hf = dataset.hf_dataset
total_frames = len(hf) total_frames = len(hf)
max_samples = total_frames - chunk_size max_samples = min(1_000_000, total_frames - chunk_size)
indices = np.arange(max_samples) indices = np.random.choice(total_frames - chunk_size, max_samples, replace=False)
logging.info( logging.info(
f"use_delta_actions is enabled — computing delta action stats " f"use_delta_actions is enabled — computing delta action stats "
f"from {max_samples} chunk samples (chunk_size={chunk_size})" f"from {max_samples} chunk samples (chunk_size={chunk_size})"