Zaktualizuj 'train-mlflow.py'

This commit is contained in:
Kornelia Girejko 2022-05-15 22:00:17 +02:00
parent dd6ae0f8f6
commit 6c1f78852c

View File

@ -29,7 +29,7 @@ from urllib.parse import urlparse
from mlflow.models import infer_signature
mlflow.set_experiment("478815")
mlflow.set_tracking_uri("http://172.17.0.1:5000")
#mlflow.set_tracking_uri("http://172.17.0.1:5000")
# Model
class Model(nn.Module):