MLFlowLogger

class composer.loggers.MLFlowLogger(experiment_name=None, run_name=None, tracking_uri=None, rank_zero_only=True)

Log to MLFlow.

Parameters

• experiment_name – (str, optional): MLFLow experiment name,

• run_name – (str, optional): MLFlow run name.

• tracking_uri (str | Path, optional) – MLFlow tracking uri, the URI to the remote or local endpoint where logs are stored (If none it is set to ./mlruns)

• rank_zero_only (bool, optional) – Whether to log only on the rank-zero process (default: True).