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¶