class composer.loggers.MLFlowLogger(experiment_name=None, run_name=None, tracking_uri=None, rank_zero_only=True)[source]#

Log to MLFlow.

  • 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).