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

backup wip

Browse Source
This commit is contained in:
Alexander Soare
2024-04-16 16:31:44 +01:00
parent 43a614c173
commit 23be5e1e7b
4 changed files with 24 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/3_train_policy.py
View File

@@ -55,7 +55,7 @@ while not done:
for batch in dataloader:
for k in batch:
batch[k] = batch[k].to(device, non_blocking=True)
info = policy(batch)
info = policy.update(batch)
if step % log_freq == 0:
num_samples = (step + 1) * cfg.batch_size
loss = info["loss"]