I have a build model function which change through a parameter the nature of a lambda layer inside the model itself:
def get_model2(mask_kind): epochs = 20