Zaktualizuj 'train-mlflow.py'

This commit is contained in:
Kornelia Girejko 2022-05-15 16:03:59 +02:00
parent 5a732c3b04
commit aaa89a679e

View File

@ -29,7 +29,7 @@ from urllib.parse import urlparse
from mlflow.models import infer_signature
mlflow.set_experiment("s478815")
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):