zoo.orca.learn.horovod package
Submodules
zoo.orca.learn.horovod.horovod_ray_runner module
-
class
zoo.orca.learn.horovod.horovod_ray_runner.HorovodRayRunner(ray_ctx, worker_cls=None, worker_param=None, workers_per_node=1)[source]
Bases: object
-
run(func)[source]
-
class
zoo.orca.learn.horovod.horovod_ray_runner.HorovodWorker[source]
Bases: object
-
ip_addr()[source]
-
run(env, func)[source]
-
set_gloo_iface()[source]
-
zoo.orca.learn.horovod.horovod_ray_runner.make_worker(worker_cls, HorovodWorker)[source]
Module contents