每个时期后节省MLflow权重

Question

我一直在用MLflow跟踪测试一些小示例，但对于我的用例，我希望每个时期后都保存权重。 有时我会在运行未完全完成之前将其终止（我无法使用抢先停止），但是我现在遇到的是，权重没有保存到跟踪ui服务器中。 在每个时期之后有没有办法做到这一点？

Answer 1

将权重保存到磁盘，然后将其记录为工件。只要将检查点/权重保存到磁盘，就可以使用mlflow_log_artifact()或mlflow_log_artifacts()记录它们。在docs中，

  

mlflow.log_artifact（）将本地文件或目录记录为工件，   （可选）采用artifact_path将其放入跑步的   工件URI。可以将运行工件组织到目录中，以便您   可以通过这种方式将工件放置在目录中。

     

mlflow.log_artifacts（）将给定目录中的所有文件记录为   工件，再次采用可选的artifact_path。