Ejemplos de init_ray en Python

Ejemplo n.º 1

 def _ray_function():
     init_ray()
     env = ray_function()
     params = {
         "actions": {
             "dtype": numpy.int64
         },
         "critic": {
             "dtype": numpy.float32
         }
     }
     states = StatesModel(state_dict=params, batch_size=N_WALKERS)
     return env, states

Ejemplo n.º 2

Archivo: test_distributed_export.py Proyecto: softmaxhuanchen/fragile

 def swarm_with_score(self, request):
     init_ray()
     swarm = swarm_dict.get(request.param, create_cartpole_swarm)()
     score = test_scores[request.param]
     request.addfinalizer(lambda: kill_swarm(swarm))
     return swarm, score

Ejemplo n.º 3

def export_swarm(request):
    init_ray()
    swarm = swarm_dict.get(request.param)()
    request.addfinalizer(lambda: kill_swarm(swarm))
    return swarm

Ejemplo n.º 4

Archivo: test_export_swarm.py Proyecto: Guillemdb/fragile

def export_swarm(request):
    init_ray()
    swarm = swarm_dict.get(request.param)()
    yield swarm
    kill_swarm(swarm)