keras - MLflow 在每个 epoch 后保存权重

Question

我一直在使用 MLflow 跟踪测试一些小示例，但对于我的用例，我希望在每个 epoch 之后保存权重。有时我会在它们完全完成之前终止运行（我不能使用早期停止），但我现在的经验是权重没有保存到跟踪 ui 服务器。有没有办法在每个时代之后做到这一点？

Answer 1

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

mlflow.log_artifact()将本地文件或目录记录为工件，可选择使用 artifact_path 将其放置在运行的工件 URI 中。可以将运行工件组织到目录中，因此您可以通过这种方式将工件放置在目录中。

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