7e9239e 95aa666 7e9239e
1
2
3
4
5
6
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