我正在尝试将 Hydra 与 MLFlow 一起使用,因此我编写了最基本的脚本来查看它们是否一起工作(导入等)。两者都可以单独工作,但是放在一起我会得到一个奇怪的结果。
我有下面的脚本:
import hydra
from omegaconf import DictConfig
from mlflow import log_metric, log_param, log_artifact,start_run
@hydra.main(config_path="config.yaml")
def my_app(cfg : DictConfig):
# print(cfg.pretty())
# print(cfg['coordinates']['x0'])
log_param("a",2)
log_metric("b",3)
if __name__ == "__main__":
my_app()
但是,运行时,我收到以下错误:
ilknull@nurmachine:~/Files/Code/Python/MLFlow_test$ python3 hydra_temp.py
Error in atexit._run_exitfuncs:
Traceback (most recent call last):
File "/home/ilknull/.local/lib/python3.7/site-packages/mlflow/tracking/fluent.py", line 164, in end_run
MlflowClient().set_terminated(_active_run_stack[-1].info.run_id, status)
File "/home/ilknull/.local/lib/python3.7/site-packages/mlflow/tracking/client.py", line 311, in set_terminated
self._tracking_client.set_terminated(run_id, status, end_time)
File "/home/ilknull/.local/lib/python3.7/site-packages/mlflow/tracking/_tracking_service/client.py", line 312, in set_terminated
end_time=end_time)
File "/home/ilknull/.local/lib/python3.7/site-packages/mlflow/store/tracking/file_store.py", line 377, in update_run_info
run_info = self._get_run_info(run_id)
File "/home/ilknull/.local/lib/python3.7/site-packages/mlflow/store/tracking/file_store.py", line 442, in _get_run_info
databricks_pb2.RESOURCE_DOES_NOT_EXIST)
mlflow.exceptions.MlflowException: Run '9066793c02604a6783d081ed965d5eff' not found
同样,它们在单独使用时工作得很好,但它们一起会导致这个错误。有任何想法吗?