hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

ready for review

Browse Source
This commit is contained in:
Alexander Soare
2024-04-12 11:36:52 +01:00
parent 5666ec3ec7
commit 6d0a45a97d
7 changed files with 11 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/scripts/train.py
View File

@@ -251,7 +251,7 @@ def train(cfg: dict, out_dir=None, job_name=None):
for key in batch:
batch[key] = batch[key].to(cfg.device, non_blocking=True)
train_info = policy.update(batch, step=step)
train_info = policy(batch, step=step)
# TODO(rcadene): is it ok if step_t=0 = 0 and not 1 as previously done?
if step % cfg.log_freq == 0: