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, target=write_video,
args=( args=(
str(video_path), 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"], env.unwrapped.metadata["render_fps"],
), ),
) )