File size: 524 Bytes
14ce5a9 |
1 2 3 4 5 6 7 8 9 10 11 |
#!/bin/bash
# Put your slurm commands here
accelerate launch --num_processes=1 {base_path}/main.py --batch_size {batch_size} --model_name {model_name} --split_name {split_name} --dataset_name {dataset_name} --output_dir {model_name}_results --n_take {n_take}
python {base_path}/evaluations/evaluate_images.py \
--original_dir {model_name}_results/original_images \
--reconstructed_dir {model_name}_results/reconstructed_images/ \
--metrics {metrics} \
--batch_size 16 \
--num_workers 8 | tee result.txt
|