serotiny.models.utils.mlflow module

serotiny.models.utils.mlflow.get_config(tracking_uri, run_id, tmp_dir, mode='train')

serotiny.models.utils.mlflow.load_model_from_checkpoint(tracking_uri, run_id, strict=True, path='checkpoints/last.ckpt')