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

fix online training

Browse Source
This commit is contained in:
Cadene
2024-04-19 23:58:38 +00:00
parent b2b5329683
commit 06628ba059
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/scripts/eval.py
View File

@@ -223,8 +223,8 @@ def eval_policy(
if return_episode_data: if return_episode_data:
ep_dict = { ep_dict = {
"action": actions[ep_id, :num_frames], "action": actions[ep_id, :num_frames],
"episode_id": torch.tensor([ep_id] * num_frames), "episode_index": torch.tensor([ep_id] * num_frames),
"frame_id": torch.arange(0, num_frames, 1), "frame_index": torch.arange(0, num_frames, 1),
"timestamp": torch.arange(0, num_frames, 1) / fps, "timestamp": torch.arange(0, num_frames, 1) / fps,
"next.done": dones[ep_id, :num_frames], "next.done": dones[ep_id, :num_frames],
"next.reward": rewards[ep_id, :num_frames].type(torch.float32), "next.reward": rewards[ep_id, :num_frames].type(torch.float32),