update dllib-mlflow.py
This commit is contained in:
parent
5804f0c3f0
commit
ccf638a9f1
@ -20,6 +20,7 @@ from sacred.observers import MongoObserver
|
||||
|
||||
# EPOCHS = int(sys.argv[1])
|
||||
|
||||
mlflow.set_tracking_uri("http://172.17.0.1:5000")
|
||||
mlflow.set_experiment("s444356")
|
||||
|
||||
class Model(nn.Module):
|
||||
@ -303,8 +304,6 @@ def my_main(epochs):
|
||||
|
||||
input_example = features_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
|
||||
# print(tracking_url_type_store)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user