Remove addtional frame from eval render

This commit is contained in:
Alexander Soare 2024-05-05 12:02:33 +01:00
parent d747195c57
commit bc424f6fb7
1 changed files with 1 additions and 1 deletions

View File

@ -338,7 +338,7 @@ def eval_policy(
target=write_video,
args=(
str(video_path),
stacked_frames[: done_index + 2], # + 2 to capture the observation frame after done
stacked_frames[: done_index + 1], # + 1 to capture the last observation
env.unwrapped.metadata["render_fps"],
),
)