python swim/train.py \
    data.root_dir=/cm/shared/ninhnq3/datasets/swim_data \
    data.img_size=512 \
    data.batch_size=16 \
    compile=true \
    callbacks.model_checkpoint.dirpath=/cm/shared/ninhnq3/checkpoints/swim/autoencoder/simple