def train(): num_timesteps = 1e9 seed = 1401 rank = MPI.COMM_WORLD.Get_rank() sess = U.single_threaded_session() sess.__enter__() workerseed = seed + 1000 * MPI.COMM_WORLD.Get_rank() rank = MPI.COMM_WORLD.Get_rank() set_global_seeds(workerseed) # if rank != 0: # logger.set_level(logger.DISABLED) # logger.log("rank %i" % MPI.COMM_WORLD.Get_rank()) world_group = MPI.COMM_WORLD.Get_group() mygroup = rank % 10 theta_group = world_group.Incl( [x for x in range(MPI.COMM_WORLD.size) if (x % 10 == mygroup)]) comm = MPI.COMM_WORLD.Create(theta_group) comm.Barrier() # comm = MPI.COMM_WORLD master.start(callback, args=args, workerseed=workerseed, rank=rank, comm=comm)
for h in range(cols): width = totalwidth/cols left = width * h right = left + width height = totalheight/rows top = height * w bot = top + height disp(toBounds(left, right ,top, bot), (left,top) ) big_display = (totalwidth, totalheight) mini_display = totalwidth/cols, totalheight/rows master = Process(target = master.setup, args = [ip, port, big_display, mini_display]) master.start() else: TOTAL_WIDTH = 1920 * 5 TOTAL_HEIGHT = 1200 * 3 mini_display = 1920, 1200 COLS = 5 ROWS = 3 name = platform.node() print not name.startswith('tile-') ip = "10.10.0.255" if name == 'master': master.setup(ip, port, (TOTAL_WIDTH, TOTAL_HEIGHT), mini_display) elif True: 'pong running on odd computer' else:
import time import master A_LONG_TIME = 999999 if __name__ == '__main__': server = master.start() try: time.sleep(A_LONG_TIME) except KeyboardInterrupt: print '[maplegym] CTRL + C detected, exiting...' server.stop()
from utils import find_column def generate_initial_waterfall(x, y, amount, map): # Find initial water position list_of_water_states = [] initial_state = map[:, :] position = find_column(initial_state, x, y) # Add new dimension for water level initial_state = np.vstack([initial_state, np.zeros(len(initial_state[0]))]) initial_state[3, position] = amount list_of_water_states.append(initial_state) return list_of_water_states if __name__ == '__main__': # mpirun -hostfile hostfile -np 1 python watefall.py # response = map_generator.request_osm_api() response = map_generator.load_request_from_file("samples/poland_s.json") print(str(response)) positions = map_generator.get_osm_positions_dict(response) map_array = map_generator.generate_map(positions) l_water = generate_initial_waterfall(48.0, 18.0, 1000, map_array) l_water = start(l_water) visualizer.save_animation(map_array, l_water, 'animation.mp4')
import slave import master import config mode = sys.argv[1] CURRENT_DIR = os.path.dirname(os.path.abspath(__file__)) sys.path.insert(0, os.path.dirname(CURRENT_DIR)) fmt = "%(asctime)s %(name)s %(message)s" logging.basicConfig(level=logging.WARNING, format=fmt) def on_term_handler(sig, frame): logging.warning('Recv SIGTERM') if mode == "slave": slave.terminate() elif mode == "master": master.terminate() signal.signal(signal.SIGTERM, on_term_handler) config.set_config('env', 'run') if mode: if mode == "master": master.start('jonas-lu') else: instance_id = sys.argv[2] if len(sys.argv) >= 3 else 0 slave.start(instance_id) else: sys.exit('Invalid arguments')
import master import config mode = sys.argv[1] CURRENT_DIR = os.path.dirname(os.path.abspath(__file__)) sys.path.insert(0, os.path.dirname(CURRENT_DIR)) fmt = "%(asctime)s %(name)s %(message)s" logging.basicConfig(level=logging.WARNING, format=fmt) def on_term_handler(sig, frame): logging.warning('Recv SIGTERM') if mode == "slave": slave.terminate() elif mode == "master": master.terminate() signal.signal(signal.SIGTERM, on_term_handler) config.set_config('env', 'run') if mode: if mode == "master": master.start('jonas-lu') else: instance_id = sys.argv[2] if len(sys.argv) >= 3 else 0 slave.start(instance_id) else: sys.exit('Invalid arguments')