function run2 () {
  dataset=$1
  sample_rate=$2
  duration=$3
  let sample_length=$2*$3
  fs1=$4
  fs2=$5

  exp_name=space_2

  echo "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  echo "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  echo "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  echo ""
  echo ">> running $exp_name $4 $5"
  echo ""
  echo "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  echo "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  echo "@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@"
  python train.py \
    --exp $exp_name --dataset $dataset  \
    --frame_sizes $fs1 $fs2             \
    --n_rnn 2 --dim 1024 --q_levels 256 \
    --seq_len 1024 --batch_size 128     \
    --val_frac 0.1 --test_frac 0.1      \
    --sample_rate $sample_rate          \
    --sample_length $sample_length      \
    --keep_old_checkpoints False        \
    --epoch_limit 2                     \
    --resume True
}
function generate2 () {
  dataset=$1
  sample_rate=$2
  duration=$3
  let sample_length=$2*$3
  fs1=$4
  fs2=$5

  exp_name=space_2

  echo ">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>"
  echo ">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>"
  echo ">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>"
  echo ""
  echo ">> generating $exp_name $4 $5"
  echo ""
  echo ">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>"
  echo ">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>"
  echo ">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>"
  python generate.py \
    --exp $exp_name --dataset $dataset  \
    --frame_sizes $fs1 $fs2             \
    --n_rnn 2 --dim 1024 --q_levels 256 \
    --seq_len 1024 --batch_size 128     \
    --val_frac 0.1 --test_frac 0.1      \
    --sample_rate $sample_rate          \
    --sample_length $sample_length      \
    --keep_old_checkpoints False        \
    --epoch_limit 1                     \
    --n_samples 6                       \
    --resume True
}
function run3 () {
  dataset=$1
  sample_rate=$2
  duration=$3
  let sample_length=$2*$3
  fs1=$4
  fs2=$5
  fs3=$6

  exp_name=space_3

  echo ">> running $exp_name $4 $5 $6"
  python train.py \
    --exp $exp_name --dataset $dataset  \
    --frame_sizes $fs1 $fs2 $fs3        \
    --n_rnn 3 --dim 1024 --q_levels 256 \
    --seq_len 1024 --batch_size 128     \
    --val_frac 0.1 --test_frac 0.1      \
    --sample_rate $sample_rate          \
    --sample_length $sample_length      \
    --keep_old_checkpoints False        \
    --epoch_limit 1                     \
    --resume True
}

function run4 () {
  dataset=$1
  sample_rate=$2
  duration=$3
  let sample_length=$2*$3
  fs1=$4
  fs2=$5
  fs3=$6
  fs4=$7

  exp_name=space_4

  echo ">> running $exp_name $4 $5 $6 $7"
  python train.py \
    --exp $exp_name --dataset $dataset  \
    --frame_sizes $fs1 $fs2 $fs3 $fs4   \
    --n_rnn 4 --dim 1024 --q_levels 256 \
    --seq_len 1024 --batch_size 128     \
    --val_frac 0.1 --test_frac 0.1      \
    --sample_rate $sample_rate          \
    --sample_length $sample_length      \
    --keep_old_checkpoints False        \
    --epoch_limit 4                     \
    --resume True
}
function run5 () {
  dataset=$1
  sample_rate=$2
  duration=$3
  let sample_length=$2*$3
  fs1=$4
  fs2=$5
  fs3=$6
  fs4=$7
  fs5=$8

  exp_name=space_5

  echo ">> running $exp_name $4 $5 $6 $7 $8"
  python train.py \
    --exp $exp_name --dataset $dataset  \
    --frame_sizes $fs1 $fs2 $fs3 $fs4 $fs5 \
    --n_rnn 5 --dim 1024 --q_levels 256 \
    --seq_len 1024 --batch_size 128     \
    --val_frac 0.1 --test_frac 0.1      \
    --sample_rate $sample_rate          \
    --sample_length $sample_length      \
    --keep_old_checkpoints False        \
    --epoch_limit 4                     \
    --resume True
}
function run6 () {
  dataset=$1
  sample_rate=$2
  duration=$3
  let sample_length=$2*$3
  fs1=$4
  fs2=$5
  fs3=$6
  fs4=$7
  fs5=$8
  fs6=$9

  exp_name=space_6

  echo ">> running $exp_name $4 $5 $6 $7 $8 $9"
  python train.py \
    --exp $exp_name --dataset $dataset  \
    --frame_sizes $fs1 $fs2 $fs3 $fs4 $fs5 $fs6 \
    --n_rnn 5 --dim 1024 --q_levels 256 \
    --seq_len 1024 --batch_size 128     \
    --val_frac 0.1 --test_frac 0.1      \
    --sample_rate $sample_rate          \
    --sample_length $sample_length      \
    --keep_old_checkpoints False        \
    --epoch_limit 4                     \
    --resume True
}

run4 44k_glass_space1 44100 5 2 2 2 2
run5 44k_glass_space1 44100 5 2 2 2 2 2
run6 44k_glass_space1 44100 5 2 2 2 2 2 2

run4 44k_glass_space1 44100 5 8 2 2 2
run5 44k_glass_space1 44100 5 8 2 2 2 2
run6 44k_glass_space1 44100 5 8 2 2 2 2 2
#generate2 44k_glass_space1 44100 5 8 2
#run2 44k_glass_space1 44100 5 8 4
#generate2 44k_glass_space1 44100 5 8 4
#run2 44k_glass_space1 44100 5 8 1
#generate2 44k_glass_space1 44100 5 8 1
#run2 44k_glass_space1 44100 5 16 8
#run2 44k_glass_space1 44100 5 16 4
#run2 44k_glass_space1 44100 5 16 2
#generate2 44k_glass_space1 44100 5 16 2
#run2 44k_glass_space1 44100 5 16 1
#generate2 44k_glass_space1 44100 5 16 1

#run2 44k_glass_space1 44100 5 4 2
#run2 44k_glass_space1 44100 5 4 1
#run2 44k_glass_space1 44100 5 2 1

#run3 44k_glass_space1 44100 5 8 2 1
#run3 44k_glass_space1 44100 5 8 4 2
#run3 44k_glass_space1 44100 5 16 8 2
#run3 44k_glass_space1 44100 5 16 4 2
#run3 44k_glass_space1 44100 5 16 2 1

#run2 44k_glass_space1 44100 5 6 2
#run2 44k_glass_space1 44100 5 6 3
#run3 44k_glass_space1 44100 5 5 3 2
#run3 44k_glass_space1 44100 5 10 5 2
#run3 44k_glass_space1 44100 5 8 5 2