2D_model.py

def get_model_mel():

    nclass = len(list_labels)

    inp = Input(shape=(157, 320, 1))

    incep_res = InceptionResNetV2(input_shape=(157, 320, 1), weights=None, include_top=False)

    x = incep_res(inp)

    x = GlobalMaxPool2D()(x)
    x = Dropout(rate=0.1)(x)

    x = Dense(128, activation=activations.relu)(x)
    x = Dense(nclass, activation=activations.softmax)(x)

    model = models.Model(inputs=inp, outputs=x)
    opt = optimizers.Adam()

    model.compile(optimizer=opt, loss=losses.sparse_categorical_crossentropy, metrics=['acc'])
    model.summary()
    return model