Skip to content

Instantly share code, notes, and snippets.

@holmdk
Created July 22, 2020 11:57
Show Gist options
  • Save holmdk/2c3f4a8d8f811a6574ac04544a4e0b55 to your computer and use it in GitHub Desktop.
Save holmdk/2c3f4a8d8f811a6574ac04544a4e0b55 to your computer and use it in GitHub Desktop.
example_viz_lib
from absl import app
from dopamine.utils import example_viz_lib
def main(_):
example_viz_lib.run(agent='dqn',
game='Pong',
num_steps=100,
root_dir='C:/Users/Holm/Documents/dopamine',
restore_ckpt='C:/Users/Holm/Documents/dopamine/models/tf_ckpt-7',
use_legacy_checkpoint=False)
if __name__ == '__main__':
app.run(main)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment