search

danijar / dreamerv2

Mastering Atari with Discrete World Models
https://danijar.com/dreamerv2
MIT License
886 stars 195 forks source link

Render episodes #13

Closed robjlyons closed 3 years ago

robjlyons commented 3 years ago

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

danijar commented 3 years ago

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'])