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')
