I have this kind of model:
def get_model2(mask_kind): epochs = 300 learning_rate = 0.1 learning_rate = learning_rate / epochs inp_1 = tf.keras.Input(shape=(74, 1