Skip to content

Instantly share code, notes, and snippets.

@carpedm20

carpedm20/README.md

Last active June 20, 2016 03:42
Show Gist options
  • Save carpedm20/2035298fdc3313e16aae0c6bb690bada to your computer and use it in GitHub Desktop.
Save carpedm20/2035298fdc3313e16aae0c6bb690bada to your computer and use it in GitHub Desktop.
Download ZIP
DQN-tensorflow writeup
Raw
README.md

DQN-tensorflow

First, clone the project:

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

To train a model for Breakout:

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

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

python main.py --is_train=False
python main.py --is_train=False --display=True

The best model can be obtained for 10M iteration with:

python main.py --env_name=Breakout-v0 --is_train=True --model=m2
@carpedm20
Copy link
Author

@Sessa93 You need a bigger memory to run this model. My server has 32GB memory but 16GB should be enough. How much memory do you have?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment