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

add an args to intrigue training

Browse Source
This commit is contained in:
rtz19970824 2018-01-13 15:59:57 +08:00
parent 2e8662889f
commit 3b222f5edb
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
AlphaGo/play.py
View File

@ -90,8 +90,9 @@ if __name__ == '__main__':
parser.add_argument("--black_weight_path", type=str, default=None) parser.add_argument("--black_weight_path", type=str, default=None)
parser.add_argument("--white_weight_path", type=str, default=None) parser.add_argument("--white_weight_path", type=str, default=None)
parser.add_argument("--save_path", type=str, default="./go/") parser.add_argument("--save_path", type=str, default="./go/")
parser.add_argument("--debug", type=bool, default=False) parser.add_argument("--debug", action="store_true", default=False)
parser.add_argument("--game", type=str, default="go") parser.add_argument("--game", type=str, default="go")
parser.add_argument("--train", action="store_true", default=False)
args = parser.parse_args() args = parser.parse_args()
if not os.path.exists(args.data_path): if not os.path.exists(args.data_path):
@ -113,7 +114,8 @@ if __name__ == '__main__':
thread_train = threading.Thread(target=game.model.train, args=("file",), thread_train = threading.Thread(target=game.model.train, args=("file",),
kwargs={'data_path':args.data_path, 'batch_size':128, 'save_path':args.save_path}) kwargs={'data_path':args.data_path, 'batch_size':128, 'save_path':args.save_path})
thread_play = threading.Thread(target=play, args=(engine, args.data_path)) thread_play = threading.Thread(target=play, args=(engine, args.data_path))
thread_list.append(thread_train) if args.train:
thread_list.append(thread_train)
thread_list.append(thread_play) thread_list.append(thread_play)
for t in thread_list: for t in thread_list: