From 2bcf2631b923818a1bab1c80a627f2f8759bae8a Mon Sep 17 00:00:00 2001 From: Remi Cadene Date: Mon, 4 Mar 2024 22:34:44 +0000 Subject: [PATCH] minor comment --- 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 e162a0f4..c169b49b 100644 --- a/lerobot/scripts/train.py +++ b/lerobot/scripts/train.py @@ -174,7 +174,7 @@ def train(cfg: dict, out_dir=None, job_name=None): logging.info(f"{num_learnable_params=} ({format_big_number(num_learnable_params)})") logging.info(f"{num_total_params=} ({format_big_number(num_total_params)})") - step = 0 # number of policy update + step = 0 # number of policy update (forward + backward + optim) is_offline = True for offline_step in range(cfg.offline_steps):