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