File size: 921 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 |
for dataset in cifar10 #ffhq afhqv2
do
for steps in 4 # 5 6 7 8 9 10
do
for solver in uni_pc # dpm_solver++ ipndm
do
python3 main.py \
--all_config configs/${dataset}.yml \
--data_dir train_data/train_data_${dataset}/uni_pc_NFE20_logSNR_seed0/ \
--main_valid_batch_size 25 \
--solver_name ${solver} \
--steps ${steps} \
--log_path all_logs/logs_${dataset}
python3 gen_data.py --learn \
--all_config configs/${dataset}.yml \
--sampling_batch_size 25 \
--solver_name ${solver} \
--steps ${steps} \
--total_samples 50000 \
--log_path all_logs/logs_${dataset} \
--data_dir sampling_data/sampling_data_${dataset} \
--save_png
done
done
done |