Skip to content

Commit 10584ff

Browse files
committed
removed tf1 and chainer implementations
1 parent 2ad461c commit 10584ff

File tree

16 files changed

+5
-1014
lines changed

16 files changed

+5
-1014
lines changed

README.md

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
# Alpha Zero General (any game, any framework!)
2-
A simplified, highly flexible, commented and (hopefully) easy to understand implementation of self-play based reinforcement learning based on the AlphaGo Zero paper (Silver et al). It is designed to be easy to adopt for any two-player turn-based adversarial game and any deep learning framework of your choice. A sample implementation has been provided for the game of Othello in PyTorch, Keras, TensorFlow and Chainer. An accompanying tutorial can be found [here](http://web.stanford.edu/~surag/posts/alphazero.html). We also have implementations for GoBang and TicTacToe.
2+
A simplified, highly flexible, commented and (hopefully) easy to understand implementation of self-play based reinforcement learning based on the AlphaGo Zero paper (Silver et al). It is designed to be easy to adopt for any two-player turn-based adversarial game and any deep learning framework of your choice. A sample implementation has been provided for the game of Othello in PyTorch and Keras. An accompanying tutorial can be found [here](http://web.stanford.edu/~surag/posts/alphazero.html). We also have implementations for many other games like GoBang and TicTacToe.
33

4-
To use a game of your choice, subclass the classes in ```Game.py``` and ```NeuralNet.py``` and implement their functions. Example implementations for Othello can be found in ```othello/OthelloGame.py``` and ```othello/{pytorch,keras,tensorflow,chainer}/NNet.py```.
4+
To use a game of your choice, subclass the classes in ```Game.py``` and ```NeuralNet.py``` and implement their functions. Example implementations for Othello can be found in ```othello/OthelloGame.py``` and ```othello/{pytorch,keras}/NNet.py```.
55

6-
```Coach.py``` contains the core training loop and ```MCTS.py``` performs the Monte Carlo Tree Search. The parameters for the self-play can be specified in ```main.py```. Additional neural network parameters are in ```othello/{pytorch,keras,tensorflow,chainer}/NNet.py``` (cuda flag, batch size, epochs, learning rate etc.).
6+
```Coach.py``` contains the core training loop and ```MCTS.py``` performs the Monte Carlo Tree Search. The parameters for the self-play can be specified in ```main.py```. Additional neural network parameters are in ```othello/{pytorch,keras}/NNet.py``` (cuda flag, batch size, epochs, learning rate etc.).
77

88
To start training a model for Othello:
99
```bash
@@ -27,7 +27,7 @@ We trained a PyTorch model for 6x6 Othello (~80 iterations, 100 episodes per ite
2727

2828
A concise description of our algorithm can be found [here](https://github.com/suragnair/alpha-zero-general/raw/master/pretrained_models/writeup.pdf).
2929

30-
### Citing
30+
### Citation
3131

3232
If you found this work useful, feel free to cite it as
3333

@@ -58,3 +58,4 @@ While the current code is fairly functional, we could benefit from the following
5858
* [Carlos Aguayo](https://github.com/carlos-aguayo) contributed rules and a trained model for Dots and Boxes along with a [JavaScript implementation](https://github.com/carlos-aguayo/carlos-aguayo.github.io/tree/master/alphazero).
5959
* [Robert Ronan](https://github.com/rlronan) contributed rules for Santorini.
6060

61+
Note: Chainer and TensorFlow v1 versions have been removed but can be found prior to commit [2ad461c](https://github.com/suragnair/alpha-zero-general/tree/2ad461c393ecf446e76f6694b613e394b8eb652f).

connect4/tensorflow/Connect4NNet.py

Lines changed: 0 additions & 56 deletions
This file was deleted.

connect4/tensorflow/NNet.py

Lines changed: 0 additions & 102 deletions
This file was deleted.

connect4/tensorflow/__init__.py

Whitespace-only changes.

gobang/tensorflow/GobangNNet.py

Lines changed: 0 additions & 57 deletions
This file was deleted.

gobang/tensorflow/NNet.py

Lines changed: 0 additions & 99 deletions
This file was deleted.

gobang/tensorflow/__init__.py

Whitespace-only changes.

0 commit comments

Comments
 (0)