s434695/ium_434695
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix

Browse Source
This commit is contained in:
s434695 2021-05-24 00:51:33 +02:00
parent 84641091ce
commit 93a992d53d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vgsales-mlflow.py
View File

@ -18,8 +18,7 @@ from mlflow.tracking import MlflowClient
mlflow.set_tracking_uri("http://172.17.0.1:5000")
mlflow.set_experiment("s434695")
client = MlflowClient()
def my_main(epochs, batch_size):
@ -60,6 +59,7 @@ def my_main(epochs, batch_size):
epochs = int(sys.argv[1]) if len(sys.argv) > 1 else 15
batch_size = int(sys.argv[2]) if len(sys.argv) > 2 else 16
mlflow.set_experiment("s434695")
with mlflow.start_run():