ssh_thru_server,ssh_password))
pworker.daemon = True
pworkers.append(pworker)
if start:
for pworker in pworkers:
pworker.start()
return pworkers
if __name__ == "__main__":
# endpoint:
endpoint = "tcp://localhost:6005"
bindpoint = "tcp://*:6005"
weights_dir = "/mnt/disk1/nutufts/kmason/sparsenet/ubdl/sparse_infill/sparse_infill/training/"
weights_files = {0:weights_dir+"/sparseinfill_uplane_test.tar",
1:weights_dir+"/sparseinfill_vplane_test.tar",
2:weights_dir+"/sparseinfill_yplane_test.tar"}
logging.basicConfig(level=logging.DEBUG)
pbroker = start_broker(bindpoint)
pworkers = startup_infill_workers(endpoint, weights, nplanes=[0,1,2])
print("[ENTER] to quit.")
raw_input()
weights_files = {
"y2u": weights_dir + "/devfiltered_larflow_y2u_832x512_32inplanes.tar",
"y2v": weights_dir + "/devfiltered_larflow_y2v_832x512_32inplanes.tar"
}
# set the device
device = args.device
# broker address
broker_address = args.brokeraddr
if "tcp://" not in broker_address:
broker_address = "tcp://" + broker_address
# if specified in arguments. start a broker
if args.launch_broker:
pbroker = start_broker(broker_address)
else:
pbroker = None
# get the ssh url and password if specified in arguments
if args.ssh_tunnel is not None:
if args.ssh_user is None:
raise ValueError("If using ssh tunnel, must provide user")
print "Using ssh, please provide password"
ssh_password = getpass.getpass()
ssh_url = "%s@%s" % (args.ssh_user, args.ssh_tunnel)
else:
ssh_url = None
ssh_password = None
# start the workers
