SamplerGAN/run_images.sh at main · HamidrezaGholamrezaei/SamplerGAN · GitHub

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
#!/bin/bash

datasets=("MNIST" "CIFAR10")
models=("SamplerGAN" "CGAN" "InfoGAN" "ACGAN")

for dataset in "${datasets[@]}"
do
    if [ "datasets" == "MNIST" ]
    then
        for model in "${models[@]}"
        do
            if [ "$model" == "InfoGAN" ]
            then
                python main.py --domain "image" --dataset "$dataset" --gan_model "$model" --batch_size 128 --noise_dim 62 --num_epochs 25
            else
                python main.py --domain "image" --dataset "$dataset" --gan_model "$model" --batch_size 128 --noise_dim 100 --num_epochs 25
            fi
        done
    elif [ "datasets" == "CIFAR10" ]
    then
        for model in "${models[@]}"
        do
            if [ "$model" == "InfoGAN" ]
            then
                python main.py --domain "image" --dataset "$dataset" --gan_model "$model" --batch_size 100 --noise_dim 62 --num_epochs 500
            else
                python main.py --domain "image" --dataset "$dataset" --gan_model "$model" --batch_size 100 --noise_dim 100 --num_epochs 500
            fi
        done
    fi
done