Registred model.
This commit is contained in:
parent
63cb9a7d84
commit
8aa04f9fac
1
.gitignore
vendored
1
.gitignore
vendored
@ -7,3 +7,4 @@ plot.png
|
|||||||
my_runs
|
my_runs
|
||||||
mlruns
|
mlruns
|
||||||
my_model
|
my_model
|
||||||
|
1/
|
@ -29,7 +29,7 @@ if __name__ == "__main__":
|
|||||||
|
|
||||||
lr = args.lr
|
lr = args.lr
|
||||||
n_epochs = args.epochs
|
n_epochs = args.epochs
|
||||||
mlflow.set_tracking_uri("http://127.0.0.1:5000")
|
mlflow.set_tracking_uri("http://172.17.0.1:5000")
|
||||||
mlflow.set_experiment("s426206")
|
mlflow.set_experiment("s426206")
|
||||||
with mlflow.start_run():
|
with mlflow.start_run():
|
||||||
mlflow.log_param("lr", lr)
|
mlflow.log_param("lr", lr)
|
||||||
|
Loading…
Reference in New Issue
Block a user