1 files changed, 44 insertions, 0 deletions
diff --git a/train_drums.sh b/train_drums.sh
new file mode 100644
index 0000000..2157fc7
--- /dev/null
+++ b/train_drums.sh
@@ -0,0 +1,44 @@
+function generate () {
+  exp_name=$1
+  epoch_limit=$2
+  sample_rate=$3
+  sample_length=$4
+
+  echo ">> generating $exp_name"
+  python generate.py \
+    --exp $exp_name --dataset $exp_name \
+    --frame_sizes 8 2                   \
+    --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 $epoch_limit          \
+    --resume True
+}
+function run () {
+  exp_name=$1
+  epoch_limit=$2
+  sample_rate=$3
+  sample_length=$4
+
+  echo ">> running $exp_name"
+  python train.py \
+    --exp $exp_name --dataset $exp_name \
+    --frame_sizes 8 2                   \
+    --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 $epoch_limit          \
+    --resume True
+}
+
+#generate 44k_jlin3 64 44100 4134
+generate 44k_jlin4 64 44100 4134
+generate 44k_clouds2 64 44100 4134
+generate 44k_glassmix 64 44100 4134
+