diff --git a/main.py b/main.py index 81f039e59f21acbb1cd571a48347a7dd96eb649c..6ae1cd1c6a3e4e3dda58c77c1525b9d9c5cc7610 100644 --- a/main.py +++ b/main.py @@ -124,8 +124,6 @@ def create_and_compile_model(): model.add(Conv2D(32, kernel_size=(3, 3), activation="relu", input_shape=(constants.N_MFCC, constants.MFCC_MAX_LEN, constants.CHANNELS))) model.add(Conv2D(32, kernel_size=(3, 3), activation="relu")) model.add(MaxPooling2D(pool_size=(2,2))) - model.add(Dropout(0.25)) - model.add(Conv2D(64, kernel_size=(3, 3), activation="relu")) model.add(Conv2D(64, kernel_size=(3, 3), activation="relu")) model.add(MaxPooling2D(pool_size=(2,2))) model.add(Dropout(0.5))