继续在 MLflow 中停止运行

Question

我们在 AWS Spot 实例上运行我们的实验。有时实验会停止，我们更愿意继续记录到相同的运行。如何设置活动运行的运行 ID？

类似这个伪代码的东西（不起作用）：

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