update dllib-mlflow.py
This commit is contained in:
parent
30a09f0f7d
commit
cfaaedf748
@ -20,7 +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_tracking_uri("http://172.17.0.1:5000")
|
||||||
mlflow.set_experiment("s444356")
|
mlflow.set_experiment("s444356")
|
||||||
|
|
||||||
class Model(nn.Module):
|
class Model(nn.Module):
|
||||||
|
Loading…
Reference in New Issue
Block a user