MLflow 在每個 epoch 后保存權重

Question

我一直在使用 MLflow 跟蹤測試一些小示例，但對於我的用例，我希望在每個 epoch 之后保存權重。 有時我會在它們完全完成之前終止運行（我不能使用earlystopping），但我現在的經驗是權重沒有保存到跟蹤ui服務器。 有沒有辦法在每個時代之后做到這一點？

Answer 1

將權重保存到磁盤，然后將它們記錄為工件。 只要檢查點/權重保存到磁盤，您就可以使用mlflow_log_artifact()或mlflow_log_artifacts()記錄它們。 從文檔中，

mlflow.log_artifact()將本地文件或目錄記錄為工件，可選擇使用 artifact_path 將其放置在運行的工件 URI 中。 可以將運行工件組織到目錄中，因此您可以通過這種方式將工件放置在目錄中。

mlflow.log_artifacts()將給定目錄中的所有文件記錄為工件，再次采用可選的 artifact_path。