hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

start a random player if checkpoint path is not specified

Browse Source
This commit is contained in:
rtz19970824 2017-12-19 15:39:31 +08:00
parent 0991fef527
commit 4a2d8f0003
2 changed files with 22 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
AlphaGo/play.py
View File

@ -1,3 +1,4 @@
import argparse
import subprocess
import sys
import re
@ -11,14 +12,17 @@ if __name__ == '__main__':
Note that, this function requires the installation of the Pyro4 library.
"""
# TODO : we should set the network path in a more configurable way.
black_weight_path = "./checkpoints"
white_weight_path = "./checkpoints_origin"
if (not os.path.exists(black_weight_path)):
print "Can't not find the network weights for black player."
sys.exit()
if (not os.path.exists(white_weight_path)):
print "Can't not find the network weights for white player."
sys.exit()
parser = argparse.ArgumentParser()
parser.add_argument("--black_weight_path", type=str, default=None)
parser.add_argument("--white_weight_path", type=str, default=None)
args = parser.parse_args()
# black_weight_path = "./checkpoints"
# white_weight_path = "./checkpoints_origin"
if args.black_weight_path is not None and (not os.path.exists(args.black_weight_path)):
raise ValueError("Can't not find the network weights for black player.")
if args.white_weight_path is not None and (not os.path.exists(args.white_weight_path)):
raise ValueError("Can't not find the network weights for white player.")
# kill the old server
kill_old_server = subprocess.Popen(['killall', 'pyro4-ns'])
@ -31,14 +35,16 @@ if __name__ == '__main__':
time.sleep(1)
# start two different player with different network weights.
agent_v0 = subprocess.Popen(['python', '-u', 'player.py', '--role=black'],
stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
agent_v1 = subprocess.Popen(['python', '-u', 'player.py', '--role=white', '--checkpoint_path=./checkpoints_origin/'],
stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
agent_v0 = subprocess.Popen(['python', '-u', 'player.py', '--role=black', '--checkpoint_path=' + str(args.black_weight_path)],
stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
agent_v1 = subprocess.Popen(['python', '-u', 'player.py', '--role=white', '--checkpoint_path=' + str(args.white_weight_path)],
stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
server_list = ""
while ("black" not in server_list) or ("white" not in server_list):
server_list = subprocess.check_output(['pyro4-nsc', 'list'])
print "Waining for the server start..."
print "Waiting for the server start..."
time.sleep(1)
print server_list
print "Start black player at : " + str(agent_v0.pid)

4
AlphaGo/player.py
View File

@ -22,10 +22,12 @@ class Player(object):
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument("--checkpoint_path", type=str, default="./checkpoints/")
parser.add_argument("--checkpoint_path", type=str, default=None)
parser.add_argument("--role", type=str, default="unknown")
args = parser.parse_args()
if args.checkpoint_path == 'None':
args.checkpoint_path = None
game = Game(checkpoint_path=args.checkpoint_path)
engine = GTPEngine(game_obj=game, name='tianshou', version=0)