def __init__(self, planes, args, phase=1, filters=192, board_size=15, model_dir="./value_net_models",
model_file=None,
device="gpu", gpu=1, optimizer="sgd", learn_rate=1e-6, distributed_train=False):
self.board_size = board_size
self.phase = phase
self.planes = planes
# init network
if distributed_train:
ps_device = "/job:ps/task:0/cpu:0"
worker_device = "/job:worker/task:%d/gpu:%d" % (args.task_index, args.gpu_id)
else:
ps_device = "/cpu:0"
if device == "cpu":
worker_device = "/cpu:0"
else:
worker_device = "/gpu:%d" % gpu
self.tf_var = dict()
self.tf_var["in"], self.tf_var["out"] = AI_net.create_value_network(
planes, ps_device, worker_device, filters=filters, board_size=self.board_size, name_prefix="value_net")
# super init
AI_net.SuperNetwork.__init__(self, model_dir=model_dir)
history_step = int(self.param_unserierlize(init_params={"global_step": 0})["global_step"])
with tf.device(ps_device):
self.global_step = tf.Variable(history_step)
# loss function
with tf.device(worker_device):
self.loss_function(optimizer, learn_rate, args.values_net_batch_size)
def __init__(self,
planes,
args,
phase=1,
filters=192,
board_size=15,
model_dir="./value_net_models",
model_file=None,
device="gpu",
gpu=1,
optimizer="sgd",
learn_rate=1e-6,
distributed_train=False):
self.board_size = board_size
self.phase = phase
self.planes = planes
# init network
if distributed_train:
ps_device = "/job:ps/task:0/cpu:0"
worker_device = "/job:worker/task:%d/gpu:%d" % (args.task_index,
args.gpu_id)
else:
ps_device = "/cpu:0"
if device == "cpu":
worker_device = "/cpu:0"
else:
worker_device = "/gpu:%d" % gpu
self.tf_var = dict()
self.tf_var["in"], self.tf_var["out"] = AI_net.create_value_network(
planes,
ps_device,
worker_device,
filters=filters,
board_size=self.board_size,
name_prefix="value_net")
# super init
AI_net.SuperNetwork.__init__(self, model_dir=model_dir)
history_step = int(
self.param_unserierlize(
init_params={"global_step": 0})["global_step"])
with tf.device(ps_device):
self.global_step = tf.Variable(history_step)
# loss function
with tf.device(worker_device):
self.loss_function(optimizer, learn_rate,
args.values_net_batch_size)
