This commit is contained in:
s434695 2021-05-24 00:22:18 +02:00
parent 27743b754f
commit a024e32143

View File

@ -16,6 +16,8 @@ from tensorflow.keras.models import Sequential
import mlflow
mlflow.set_tracking_uri("http://172.17.0.1:5000")
mlflow.set_experiment("s434749")
def my_main(epochs, batch_size):