Skip to content

Kedro AzureML Pipeline

Framework

stateful-y/kedro-azureml-pipeline

Framework¶

`kedro_azureml_pipeline.distributed.Framework` ¶

Bases: StrEnum

Supported distributed training frameworks.

Source Code ¶

Source code in src/kedro_azureml_pipeline/distributed/config.py

class Framework(StrEnum):
    """Supported distributed training frameworks."""

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