Skip to content

Framework

kedro_azureml_pipeline.distributed.Framework

Bases: StrEnum

Supported distributed training frameworks.

Source Code

View on GitHub

Show/Hide source 
 8
 9
10
11
12
13
class Framework(StrEnum):
    """Supported distributed training frameworks."""

    PyTorch = "PyTorch"
    TensorFlow = "TensorFlow"
    MPI = "MPI"