wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update train.py

This commit is contained in:
Jade Choghari 2025-02-22 18:11:06 +01:00 committed by GitHub
parent 6ca03b0dac
commit aa70e14033
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/scripts/train.py
View File

@ -81,7 +81,7 @@ def custom_collate_fn(batch):
) )
# stack frames for this video key and add directly to the item # stack frames for this video key and add directly to the item
item[vid_key] = torch.stack(frames) item[vid_key] = frames
# add item data (both video and non-video) to final_batch # add item data (both video and non-video) to final_batch
for key, value in item.items(): for key, value in item.items():