CFG.eval_win_rate = arguments.eval_win_rate
CFG.load_model = arguments.load_model
CFG.human_play = arguments.human_play
CFG.resnet_blocks = arguments.resnet_blocks
CFG.record_loss = arguments.record_loss
CFG.loss_file = arguments.loss_file
# CFG.game = arguments.game
game = BlokusGame()
net = NeuralNetworkWrapper(game)
# Initialize the network with the best model.
if CFG.load_model:
file_path = CFG.model_directory + "best_model.meta"
if os.path.exists(file_path):
net.load_model("best_model")
else:
print("Trained model doesn't exist. Starting from scratch.")
else:
print("Trained model not loaded. Starting from scratch.")
# Play vs the AI as a human instead of training.
# DOESN'T WORK YET
if CFG.human_play:
human_player = Human_player(game, net)
human_player.play()
else:
train = Train(game, net)
train.start()
