def encode()

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