in src/model_def.py [0:0]
def encode(x):
x = layers.Conv2D(filters=32, kernel_size=3, strides=(2,2), padding='same', activation='relu')(x)
x = layers.Conv2D(filters=64, kernel_size=3, strides=(2,2), padding='same', activation='relu')(x)
x = layers.Flatten()(x)
return x