From b2b53296839d172fae328c382160f67b8a170b7c Mon Sep 17 00:00:00 2001 From: Cadene Date: Fri, 19 Apr 2024 23:48:43 +0000 Subject: [PATCH] fix online training --- lerobot/scripts/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/scripts/train.py b/lerobot/scripts/train.py index 59fb199a..8a70a214 100644 --- a/lerobot/scripts/train.py +++ b/lerobot/scripts/train.py @@ -344,7 +344,7 @@ def train(cfg: dict, out_dir=None, job_name=None): sampler, hf_dataset=eval_info["episodes"]["hf_dataset"], episode_data_index=eval_info["episodes"]["episode_data_index"], - online_pc_sampling=cfg.get("demo_schedule", 0.5), + pc_online_samples=cfg.get("demo_schedule", 0.5), ) for _ in range(cfg.policy.utd):