def my_model(): inputs = keras.Input(shape=(height,width,3)) x = layers.Conv2D(32,3)(inputs) x = layers.BatchNorma