python swim/train.py \
    data.root_dir=/cm/shared/ninhnq3/datasets/swim_data \
    data.img_size=256 \
    data.batch_size=64 \
    compile=true \
    model.gan_start=5000 \
    callbacks.model_checkpoint.dirpath=/cm/shared/ninhnq3/checkpoints/swim/autoencoder/simple