python - 如何在 mlflow.start_run() 中获取当前的 run_id？

Question

mlflow.active_run()什么都不返回所以我不能只使用current_rui_id = mlflow.active_run().info.run_id
我必须得到 run_id 在此构造内部，以便能够继续记录另一个块内的参数、指标和工件，但对于同一模型:

with mlflow.start_run(run_name="test_ololo"):

    """ 
       fitting a model here ...
    """

    for name, val in metrics:
        mlflow.log_metric(name, np.float(val))

    # Log our parameters into mlflow
    for k, v in params.items():
        mlflow.log_param(key=k, value=v)

    pytorch.log_model(learn.model, f'model')
    mlflow.log_artifact('./outputs/fig.jpg')

我必须得到当前 run_id 在同一次运行中继续训练

with mlflow.start_run(run_id="215d3a71925a4709a9b694c45012988a"):

    """
       fit again
       log_metrics
    """

    pytorch.log_model(learn.model, f'model')
    mlflow.log_artifact('./outputs/fig2.jpg')

Answer 1

with mlflow.start_run(run_name="test_ololo") as run:

    run_id = run.info.run_id
    mlflow.log_metric("metric_1", metric_val)