From 06628ba059241301702e8964527bc957214522b2 Mon Sep 17 00:00:00 2001 From: Cadene Date: Fri, 19 Apr 2024 23:58:38 +0000 Subject: [PATCH] fix online training --- lerobot/scripts/eval.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lerobot/scripts/eval.py b/lerobot/scripts/eval.py index 15ea4f1b2..1528612d1 100644 --- a/lerobot/scripts/eval.py +++ b/lerobot/scripts/eval.py @@ -223,8 +223,8 @@ def eval_policy( if return_episode_data: ep_dict = { "action": actions[ep_id, :num_frames], - "episode_id": torch.tensor([ep_id] * num_frames), - "frame_id": torch.arange(0, num_frames, 1), + "episode_index": torch.tensor([ep_id] * num_frames), + "frame_index": torch.arange(0, num_frames, 1), "timestamp": torch.arange(0, num_frames, 1) / fps, "next.done": dones[ep_id, :num_frames], "next.reward": rewards[ep_id, :num_frames].type(torch.float32),