File size: 1,296 Bytes
d382778 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
for dataset in cifar10 afhqv2 ffhq
do
CUDA_VISIBLE_DEVICES=0 python3 gen_data.py \
--all_config configs/${dataset}.yml \
--total_samples 100 \
--sampling_batch_size 20 \
--steps 20 \
--solver_name uni_pc \
--skip_type logSNR \
--save_pt --save_png --data_dir train_data_test/train_data_${dataset}
done
CUDA_VISIBLE_DEVICES=0 python3 gen_data.py \
--all_config configs/latent_diff_LSUN.yml \
--total_samples 100 \
--sampling_batch_size 20 \
--steps 20 \
--solver_name uni_pc \
--skip_type time_uniform \
--save_pt --save_png --data_dir train_data_test/train_data_LSUN
CUDA_VISIBLE_DEVICES=3 python3 gen_data.py \
--all_config configs/latent_diff_imn.yml \
--total_samples 100 \
--sampling_batch_size 20 \
--steps 10 \
--solver_name uni_pc \
--skip_type time_quadratic \
--save_pt --save_png --data_dir train_data_test/train_data_imn
for step in {5..8}
do
CUDA_VISIBLE_DEVICES=0 python3 gen_data.py \
--all_config configs/stable_diff_v1-5.yml \
--total_samples 50 \
--num_samples_per_prompt 10 \
--sampling_batch_size 2 \
--steps ${step} \
--solver_name ipndm \
--skip_type gits \
--use_gits \
--save_pt --save_png --data_dir train_data_test2/train_data_stable_diff_v1-5 \
--prompt_path data/prompts.txt --low_gpu
done
|