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))