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

add test interface for Network

Browse Source
This commit is contained in:
Dong Yan 2017-11-06 23:13:11 +08:00
parent a8030c95f2
commit 2fc87f7020
2 changed files with 28 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.gitignore vendored
View File

@ -1,4 +1,4 @@
.idea .idea
leela-zero leela-zero
.pyc *.pyc
parameters parameters

38
AlphaGo/Network.py
View File

@ -3,6 +3,7 @@ import numpy as np
import time import time
import multi_gpu import multi_gpu
import tensorflow.contrib.layers as layers import tensorflow.contrib.layers as layers
import sys
def residual_block(input, is_training): def residual_block(input, is_training):
normalizer_params = {'is_training': is_training, normalizer_params = {'is_training': is_training,
@ -96,15 +97,30 @@ def train():
def forward(board): def forward(board):
result_path = "./results/" result_path = "./results/"
with multi_gpu.create_session() as sess: itflag = False
sess.run(tf.global_variables_initializer()) res = None
ckpt_file = tf.train.latest_checkpoint(result_path) if board is None:
if ckpt_file is not None: board = np.load("/home/yama/tongzheng/AG/self_play_204/d7d7d552b7be4b51883de99d74a8e51b.npz")
print('Restoring model from {}...'.format(ckpt_file)) board = board["boards"][100].reshape(-1, 19, 19, 17)
saver.restore(sess, ckpt_file) result_path = "../parameters/checkpoints"
else: itflag = True
raise ValueError("No model loaded") with multi_gpu.create_session() as sess:
return sess.run([p,v], feed_dict={x:board}) sess.run(tf.global_variables_initializer())
ckpt_file = tf.train.latest_checkpoint(result_path)
if ckpt_file is not None:
print('Restoring model from {}...'.format(ckpt_file))
saver.restore(sess, ckpt_file)
else:
raise ValueError("No model loaded")
res = sess.run([tf.nn.softmax(p),v], feed_dict={x:board, is_training:itflag})
#res = sess.run([tf.nn.softmax(p),v], feed_dict={x:fix_board["boards"][300].reshape(-1, 19, 19, 17), is_training:False})
#res = sess.run([tf.nn.softmax(p),v], feed_dict={x:fix_board["boards"][50].reshape(-1, 19, 19, 17), is_training:True})
print(res)
#print(res[0].tolist()[0])
#print(np.argmax(res[0]))
return res
if __name__='main': if __name__=='__main__':
train() #train()
if sys.argv[1] == "test":
forward(None)