danijar / dreamerv2

Mastering Atari with Discrete World Models

Home Page:https://danijar.com/dreamerv2

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Render episodes

robjlyons opened this issue · comments

Is there a way to render eval episodes for the open ai Atari envs?

Yes, they are visualized in TensorBoard and they are also stored to logdir/eval_episodes. You can visualize the stored episodes via:

import imageio
import numpy as np

imageio.mimsave('video.mp4', np.load('episode.npz')['image'])