Skip to content

Instantly share code, notes, and snippets.

@JoshBroomberg

JoshBroomberg/train.py

Last active October 1, 2021 21:39
Show Gist options
  • Save JoshBroomberg/840deb35ac4c0879e8ff96e2a1798063 to your computer and use it in GitHub Desktop.
Save JoshBroomberg/840deb35ac4c0879e8ff96e2a1798063 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
train.py
import mlflow
import os
REMOTE_SERVER_URI = "http://mlflow:5555"
S3_ENDPOINT_URL = "http://minio:9000"
# set the tracking server hostname/port
mlflow.set_tracking_uri(REMOTE_SERVER_URI)
# set the artifact backend to be the local minio backend
os.environ["MLFLOW_S3_ENDPOINT_URL"] = S3_ENDPOINT_URL
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment