From 4fc45ed8ef6201f91d818362878d3155dfe4295e Mon Sep 17 00:00:00 2001 From: patrycjalazna Date: Fri, 14 May 2021 20:58:03 +0200 Subject: [PATCH] minor fix --- sacred-training.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/sacred-training.py b/sacred-training.py index 85f518a..8825466 100644 --- a/sacred-training.py +++ b/sacred-training.py @@ -47,10 +47,6 @@ def prepare_model(): # kompilacja model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy']) - # model fit - epochs = int(sys.argv[1]) - batch_size = int(sys.argv[2]) - # trenowanie modelu model.fit(X_train, y_train, epochs=epochs, batch_size=batch_size, validation_data=(X_test, y_test))