nvflare.fuel.utils.constants module

class FrameworkType(value)

Bases: str, Enum

Machine learning framework type.

Used to identify the ML framework for model handling, persistor selection, and checkpoint validation across both server and client components.

NUMPY = 'numpy'

PYTORCH = 'pytorch'

RAW = 'raw'

TENSORFLOW = 'tensorflow'

class Mode(value)

Bases: str, Enum

An enumeration.

ACTIVE = 'ACTIVE'

PASSIVE = 'PASSIVE'

class PipeChannelName

Bases: object

METRIC = 'metric'

TASK = 'task'