Skip to content

Instantly share code, notes, and snippets.

@araffin

araffin/launch_tensorboard.sh

Created September 18, 2018 09:37
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save araffin/de84eb6685284a9453c1e31842a41d77 to your computer and use it in GitHub Desktop.
Save araffin/de84eb6685284a9453c1e31842a41d77 to your computer and use it in GitHub Desktop.
Download ZIP
Stable Baselines - Tensorboard integration for RL
Raw
launch_tensorboard.sh
tensorboard --logdir /tmp/a2c_cartpole_tensorboard/
Raw
tensorboard_rl.py
from stable_baselines import A2C
model = A2C('MlpPolicy', 'CartPole-v1', verbose=1, tensorboard_log="/tmp/a2c_cartpole_tensorboard/")
model.learn(total_timesteps=50000)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment