1 files changed, 5 insertions, 3 deletions
diff --git a/mix.sh b/mix.sh
index 48e499b..484e750 100755
--- a/mix.sh
+++ b/mix.sh
@@ -1,14 +1,15 @@
 #!/bin/bash
 
-if [ "$#" -ne 4 ]; then
-    echo "Usage: $0 <results/exp\:experiment> render_name endpoint epoch"
+if [ "$#" -ne 5 ]; then
+    echo "Usage: $0 <results/exp\:experiment> render_name endpoint dataset epoch"
     exit
 fi
 
 dir=$1
 name=$2
 endpoint=$3
-epoch=$4
+dataset=$4
+epoch=$5
 now=`date +'%Y%m%d'`
 
 echo "rendering $name"
@@ -33,6 +34,7 @@ else
     -F "module=samplernn" \
     -F "activity=train" \
     -F "generated=true" \
+    -F "dataset=$dataset" \
     -F "epoch=$epoch" \
     -F "file=@./output/$name.mp3" \
     $endpoint