Distributor¶

Distributor is the parent (abstract) class of TorchDistributor.

Creating Instance¶

Distributor takes the following to be created:

• Number of processes (default: 1)
• local_mode flag (default: True)
• use_gpu flag (default: True)

_get_num_tasks¶

_get_num_tasks(
  self) -> int

_get_num_tasks...FIXME

get_gpus_owned¶

get_gpus_owned(
  context: Union[SparkContext, BarrierTaskContext]) -> List[str]

get_gpus_owned...FIXME