python - 远程服务器上的工件存储和 MLFlow

Question

我正在尝试在本地网络中的另一台机器上运行 MLFlow，我想寻求一些帮助，因为我现在不知道该怎么做。

我有一个运行在服务器上的 mlflow 服务器。 mlflow 服务器在服务器上的我的用户下运行，并已像这样启动:

mlflow server --host 0.0.0.0 --port 9999 --default-artifact-root sftp://<MYUSERNAME>@<SERVER>:<PATH/TO/DIRECTORY/WHICH/EXISTS>

我应该将所有数据记录到 mlflow 服务器的程序如下所示:

from mlflow import log_metric, log_param, log_artifact, set_tracking_uri

if __name__ == "__main__":
    remote_server_uri = '<SERVER>' # this value has been replaced
    set_tracking_uri(remote_server_uri)
    # Log a parameter (key-value pair)
    log_param("param1", 5)

    # Log a metric; metrics can be updated throughout the run
    log_metric("foo", 1)
    log_metric("foo", 2)
    log_metric("foo", 3)

    # Log an artifact (output file)
    with open("output.txt", "w") as f:
        f.write("Hello world!")
    log_artifact("output.txt")

参数 get 和 metrics 会传输到服务器，但不会传输工件。为什么呢？

关于 SFTP 部分的注意事项:
我可以通过SFTP登录并且安装了pysftp包

Answer 1

我不知道我是否会得到我的问题的答案，但我确实是这样解决的。

在服务器上我创建了目录 /var/mlruns .我通过 --backend-store-uri file:///var/mlruns 将此目录传递给 mlflow

然后我通过例如挂载这个目录sshfs在我的本地机器上相同的路径下。

我不喜欢这个解决方案，但它现在已经足够好地解决了这个问题。