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