File size: 242 Bytes
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