def main(unused_argv):
base_config = config_util.get_config(urdf_root=FLAGS.urdf_data_path)
config = config_util.generate_config(
base_config, current_time_string=FLAGS.current_time_string)
servers = []
server_creds = loas2.loas2_server_credentials()
port = FLAGS.port
if not config.run_on_borg:
port = 20000 + FLAGS.server_id
server = grpc.server(futures.ThreadPoolExecutor(max_workers=100),
ports=(port, ))
blackbox_object = config.blackbox_object_fn()
np.random.seed(FLAGS.server_id)
if config.algorithm == "zero_order":
if FLAGS.server_id < config.test_workers:
worker_mode = "Test"
task_ids = range(config.train_set_size,
config.train_set_size + config.test_set_size)
else:
worker_mode = "Train"
task_ids = range(config.train_set_size)
servicer = blackbox_maml_objects.GeneralMAMLBlackboxWorker(
worker_id=FLAGS.server_id,
blackbox_object=blackbox_object,
task_ids=task_ids,
task_batch_size=config.task_batch_size,
worker_mode=worker_mode)
zero_order_pb2_grpc.add_EvaluationServicer_to_server(servicer, server)
elif config.algorithm == "first_order":
tasks = [
config.make_task_fn(s)
for s in range(config.train_set_size + config.test_set_size)
]
servicer = blackbox_maml_objects.GradientMAMLWorker(
FLAGS.server_id, blackbox_object=blackbox_object, tasks=tasks)
first_order_pb2_grpc.add_EvaluationServicer_to_server(servicer, server)
server.add_secure_port("[::]:{}".format(port), server_creds)
servers.append(server)
server.start()
print("Start server {}".format(FLAGS.server_id))
# prevent the main thread from exiting
try:
while True:
time.sleep(_ONE_DAY_IN_SECONDS)
except KeyboardInterrupt:
for server in servers:
server.stop(0)
def main(unused_argv):
servers = []
server_creds = loas2.loas2_server_credentials()
port = FLAGS.port
if not FLAGS.run_on_borg:
port = 20000 + FLAGS.server_id
server = grpc.server(
futures.ThreadPoolExecutor(max_workers=10), ports=(port,))
servicer = ars_evaluation_service.ParameterEvaluationServicer(
FLAGS.config_name, worker_id=FLAGS.server_id)
ars_evaluation_service_pb2_grpc.add_EvaluationServicer_to_server(
servicer, server)
server.add_secure_port("[::]:{}".format(port), server_creds)
servers.append(server)
server.start()
print("Start server {}".format(FLAGS.server_id))
# prevent the main thread from exiting
try:
while True:
time.sleep(_ONE_DAY_IN_SECONDS)
except KeyboardInterrupt:
for server in servers:
server.stop(0)
def main(unused_argv):
servers = []
server_creds = loas2.loas2_server_credentials()
port = FLAGS.port
if not FLAGS.run_on_borg:
port = 20000 + FLAGS.server_id
server = grpc.server(futures.ThreadPoolExecutor(max_workers=10),
ports=(port, ))
servicer = ars_evaluation_service.ParameterEvaluationServicer(
FLAGS.config_name, worker_id=FLAGS.server_id)
ars_evaluation_service_pb2_grpc.add_EvaluationServicer_to_server(
servicer, server)
server.add_secure_port("[::]:{}".format(port), server_creds)
servers.append(server)
server.start()
print("Start server {}".format(FLAGS.server_id))
# prevent the main thread from exiting
try:
while True:
time.sleep(_ONE_DAY_IN_SECONDS)
except KeyboardInterrupt:
for server in servers:
server.stop(0)
def add_port(server):
server_credentials = loas2.loas2_server_credentials()
return server.add_secure_port(
_ADDRESS_FORMAT.format(FLAGS.env_service_port), server_credentials)
def add_port(server, port):
server_credentials = loas2.loas2_server_credentials()
return server.add_secure_port(
_ADDRESS_FORMAT.format(port), server_credentials)