docs: Updated readme to include group picture.

README.md

@@ -87,12 +87,13 @@ By adhering to the above guidelines, you'll be well-prepared to contribute to or
 
 -----
 
-<!--INSERT PICTURE OF TEAM-->
 <div align="center">
-<img src="./docs/pictures/sample_pic.jpg" style="width: 30vw;"></img>
+<img src="./docs/pictures/alphazero-group-image.jpg" style="width: 30vw;"></img>
 </div>
 
-Left to right: [@example](https://github.com/Jonrodtang)    [@example](https://github.com/Jonrodtang)    [@example](https://github.com/Jonrodtang)    [@example](https://github.com/Jonrodtang)  
+Back row left to right: [Nils Henrik Lund](https://github.com/Nilsthehacker), [Haagen Mæland Moe](https://github.com/Thesmund)
+
+Front row left to right: [Kristian Carlenius](https://github.com/kristiancarlenius), [Ludvig Øvrevik](https://github.com/ludvigovrevik), [Christian Fredrik Johnsen](https://github.com/ChristianFredrikJohnsen), [Brage Kvamme](https://github.com/BrageHK)
 
 #### Leaders
 

main.py

@@ -1,7 +1,5 @@
-import pyspiel
-import torch
+from argparse import ArgumentParser, ArgumentTypeError
 from torch import multiprocessing as mp
-from src.alphazero.agents.alphazero_training_agent import AlphaZero
 from src.neuralnet.neural_network import NeuralNetwork
 from src.neuralnet.neural_network_connect_four import NeuralNetworkConnectFour
 
@@ -24,8 +22,8 @@ def test_overfit(context: GameContext):
           context=context,
           num_games=3,
           num_simulations=100,
-          epochs=1,
-          batch_size=64
+          epochs=1000,
+          batch_size=256
      )
 
 def train_tic_tac_toe(context: GameContext):
@@ -51,7 +49,7 @@ def train_connect_four(context: GameContext):
                train_alphazero_model(
                     context=context,
                     num_games=20,
-                    num_simulations=200,
+                    num_simulations=100,
                     epochs=3,
                     batch_size=256,
                )
@@ -91,19 +89,57 @@ def play(context: GameContext, first: bool, mcts: bool = False):
      save_path=connect4_path
 )
 
+def str2bool(v: str) -> bool:
+    if isinstance(v, bool):
+        return v
+    if v.lower() in ('yes', 'true', 't', 'y', '1'):
+        return True
+    elif v.lower() in ('no', 'false', 'f', 'n', '0'):
+        return False
+    else:
+        raise ArgumentTypeError('Boolean value expected.')
+
+parser: ArgumentParser = ArgumentParser(description='Control the execution of the AlphaZero game playing system.')
+parser.add_argument('--test_overfit', action='store_true', help='Test overfitting on Connect Four.')
+parser.add_argument('--train_tic_tac_toe', action='store_true', help='Train AlphaZero on Tic Tac Toe.')
+parser.add_argument('--train_connect_four', action='store_true', help='Train AlphaZero on Connect Four for a long time.')
+
+parser.add_argument('--self_play_ttt', action='store_true', help='Run self-play on Tic Tac Toe.')
+parser.add_argument('--self_play_c4', action='store_true', help='Run self-play on Connect Four.')
+
+parser.add_argument('--play_ttt', action='store_true', help='Enable playing against AlphaZero on Tic Tac Toe.')
+parser.add_argument('--play_c4', action='store_true', help='Play against AlphaZero on Connect Four.')
+
+parser.add_argument('-f', '--first', action='store_true', help='Play first in the game.')
+parser.add_argument('-m', '--mcts', action='store_true', help='Replace human player with MCTS.')
+
+args = parser.parse_args()
+
+
 
 if __name__ == '__main__': # Needed for multiprocessing to work
 
+     if args.test_overfit:
+          test_overfit(overfit_context)
+
+     if args.train_tic_tac_toe:
+          train_tic_tac_toe(tic_tac_toe_context)
+
+     if args.train_connect_four:
+          train_connect_four(connect4_context)
+
+     if args.self_play_ttt:
+          self_play(tic_tac_toe_context)
 
+     if args.self_play_c4:
+          self_play(connect4_context)
 
-     # test_overfit(overfit_context)
-     # train_tic_tac_toe(tic_tac_toe_context)
-     # train_connect_four(connect4_context)
-     # self_play(tic_tac_toe_context)
-     # self_play(connect4_context)
-     # play(tic_tac_toe_context, first=False)
-     play(connect4_context, first=False)
-     # play(connect4_context, first=True, mcts=True)
+     if args.play_ttt:
+          play(tic_tac_toe_context, first=args.first)
+
+     if args.play_c4:
+          play(connect4_context, first=args.first, mcts=args.mcts)
 
+
      # create_tic_tac_toe_model("initial_test")
      # create_connect_four_model("overfit_nn")