update dllib-mlflow.py
Some checks failed
s444356-evaluation/pipeline/head This commit looks good
s444356-training/pipeline/head There was a failure building this commit

This commit is contained in:
Maciej Czajka 2022-05-11 16:44:20 +02:00
parent 5804f0c3f0
commit ccf638a9f1

View File

@ -20,6 +20,7 @@ from sacred.observers import MongoObserver
# EPOCHS = int(sys.argv[1]) # EPOCHS = int(sys.argv[1])
mlflow.set_tracking_uri("http://172.17.0.1:5000")
mlflow.set_experiment("s444356") mlflow.set_experiment("s444356")
class Model(nn.Module): class Model(nn.Module):
@ -303,8 +304,6 @@ def my_main(epochs):
input_example = features_train_g input_example = features_train_g
siganture = infer_signature(features_train_g, labels_train_g) siganture = infer_signature(features_train_g, labels_train_g)
# mlflow.set_tracking_uri("http://172.17.0.1:5000")
mlflow.set_tracking_uri("http://172.17.0.1:5000/#/")
tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
# print(tracking_url_type_store) # print(tracking_url_type_store)