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 |
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 |