在MLflow中继续停止运行

Question

我们在AWS现场实例上运行了我们的实验。有时实验会停止，我们更愿意继续记录到相同的运行。如何设置活动跑步的跑步id

类似这样的伪代码（不工作）：

if new:
    mlflow.start_run(experiment_id=1, run_name=x)
else:
    mlflow.set_run(run_id)

Answer 1

您可以将run_id直接传递给start_run：

mlflow.start_run(experiment_id=1,
                 run_name=x,
                 run_id=<run_id_of_interrupted_run> # pass None to start a new run
                 ) 

当然，您必须为此存储run_id。你可以用^{}得到它