Skip to content

Instantly share code, notes, and snippets.

@carpedm20
Created July 14, 2016 19:01
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 carpedm20/4df895e02f076254eb7f96b6dcd4c133 to your computer and use it in GitHub Desktop.
Save carpedm20/4df895e02f076254eb7f96b6dcd4c133 to your computer and use it in GitHub Desktop.
NAF-tensorflow writeup

NAF-tensorflow

First, clone the project:

git clone https://github.com/devsisters/NAF-tensorflow/
cd NAF-tensorflow

To train a model for Pendulum:

python main.py --env_name=Pendulum-v0 --is_train=True
python main.py --env_name=Pendulum-v0 --is_train=True --display=True

To test and record the screen with gym (the repo already contains checkpoint):

python main.py --env_name=Pendulum-v0 --is_train=False
python main.py --env_name=Pendulum-v0 --is_train=False --display=True

The best model can be obtained after 2K steps with:

python main.py --env_name=Pendulum-v0 --is_train=True
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment