fix online training

2024-04-19 23:48:43 +00:00
parent 85f1554da8
commit b2b5329683
1 changed files with 1 additions and 1 deletions
--- 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):