cyto_dl.loggers.mlflow module#

class cyto_dl.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, fault_tolerant=True)[source]#

Bases: MLFlowLogger

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