serotiny.loggers.mlflow module#

class serotiny.loggers.mlflow.MLFlowLogger(experiment_name: str = 'lightning_logs', run_name: str | None = None, tracking_uri: str | None = None, tags: Dict[str, Any] | None = None, save_dir: str | None = './mlruns', prefix: str = '', artifact_location: str | None = None, run_id: str | None = None)[source]#

Bases: MLFlowLogger

after_save_checkpoint(ckpt_callback: ModelCheckpoint) → None[source]#

Called after model checkpoint callback saves a new checkpoint.

log_hyperparams(params: Dict[str, Any] | Namespace, mode='train') → None[source]#