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')