Esempio n. 1
0
def test_without_gui(clazz, model, rank, args):
    barrier = args
    print("rank %d os.pid=%d" % (rank, os.getpid()))
    env = clazz()
    env.set_model(model)
    env.set_rank(rank)

    if rank == 0:
        v = visualizer.Visualizer(DoubleA3CPPOEnv)
        v.env = env
        v.env.canvas = v.canvas
        v.visualize()

    gen = env.run(True)
    gen.send(None)
    while True:
        try:
            gen.send((None, ))
        except StopIteration:
            #barrier.wait()
            torch.save(env.a3c_model.state_dict(),
                       "./tmp/model_%d_%d" % (env.game_no, os.getpid()))
            #barrier.wait()
            gen = env.run(True)
            gen.send(None)
Esempio n. 2
0
def test():
    v = visualizer.Visualizer(A3CSpliterEnv)
    v.visualize()
def test():
    v = visualizer.Visualizer(DefaultGameEnv)
    v.visualize()
Esempio n. 4
0
def test():
    v = visualizer.Visualizer(ParallelA3CPPOEnv)
    v.visualize()
Esempio n. 5
0
def test():
    v = visualizer.Visualizer(DoubleA3CPPOEnv)
    v.visualize()
Esempio n. 6
0
def test():
    v = visualizer.Visualizer(A3CEnv)
    v.visualize()
Esempio n. 7
0
def test(model):
    v = visualizer.Visualizer(ParallelA3CPPOEnv)
    v.env.set_model(model)
    v.visualize()
def test():
    v = visualizer.Visualizer()
    v.visualize()