From aa70e14033505f6eda6f9e0e4050843b5cbb5023 Mon Sep 17 00:00:00 2001 From: Jade Choghari Date: Sat, 22 Feb 2025 18:11:06 +0100 Subject: [PATCH] Update train.py --- 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 2b6f7d85..7dbefe2d 100644 --- a/lerobot/scripts/train.py +++ b/lerobot/scripts/train.py @@ -81,7 +81,7 @@ def custom_collate_fn(batch): ) # 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 for key, value in item.items():