samplernn generate tasks

3 files changed, 24 insertions, 1 deletions

diff --git a/app/client/modules/samplernn/samplernn.tasks.js b/app/client/modules/samplernn/samplernn.tasks.js
index bb22791..55828b8 100644
--- a/app/client/modules/samplernn/samplernn.tasks.js
+++ b/app/client/modules/samplernn/samplernn.tasks.js
@@ -22,6 +22,21 @@ export const train_task = (dataset, folder_id, epochs=1) => dispatch => {
   console.log(task)
   return actions.queue.add_task(task)
 }
+export const generate_task = (dataset, folder_id, sample_length=5, n_samples=6, epoch=0) => dispatch => {
+  const task = {
+    module: 'samplernn',
+    activity: 'generate',
+    dataset: dataset.name,
+    epoch: epoch || (dataset.checkpoints.length && dataset.checkpoints[0].epoch) || 0,
+    folder_id: folder_id,
+    opt: {
+      sample_length: 44100 * sample_length,
+      n_samples,
+    }
+  }
+  console.log(task)
+  return actions.queue.add_task(task)
+}
 export const fetch_task = (url, folder_id, file_id, dataset) => dispatch => {
   if (! url) return console.log('input file inaccessible (no url)')
   const task = {
diff --git a/app/client/modules/samplernn/views/samplernn.show.js b/app/client/modules/samplernn/views/samplernn.show.js
index 7585e6c..6226fd6 100644
--- a/app/client/modules/samplernn/views/samplernn.show.js
+++ b/app/client/modules/samplernn/views/samplernn.show.js
@@ -93,6 +93,15 @@ class SampleRNNShow extends Component {
           <span class='link' onClick={() => remote.train_task(dataset, samplernn.folder_id, 6)}>6x</span>
           <span class='link' onClick={() => remote.train_task(dataset, samplernn.folder_id, 18)}>18x</span>
         </div>
+        <div class={'actions'}>
+          <span class='link' onClick={() => remote.generate_task(dataset, samplernn.folder_id, 5, 12)}>generate</span>
+          <span class='link' onClick={() => remote.generate_task(dataset, samplernn.folder_id, 0.5, 120)}>rhythm</span>
+          <span class='link' onClick={() => remote.generate_task(dataset, samplernn.folder_id, 2, 30)}>short</span>
+          <span class='link' onClick={() => remote.generate_task(dataset, samplernn.folder_id, 10, 6)}>medium</span>
+          <span class='link' onClick={() => remote.generate_task(dataset, samplernn.folder_id, 15, 4)}>long</span>
+          <span class='link' onClick={() => remote.generate_task(dataset, samplernn.folder_id, 30, 2)}>longer</span>
+          <span class='link' onClick={() => remote.generate_task(dataset, samplernn.folder_id, 60, 1)}>longest</span>
+        </div>
         {dataset.isBuilt
         ? <div class='subtext'>
             {'fetched '}
diff --git a/app/relay/modules/samplernn.js b/app/relay/modules/samplernn.js
index d25698f..b36aadf 100644
--- a/app/relay/modules/samplernn.js
+++ b/app/relay/modules/samplernn.js
@@ -62,7 +62,6 @@ const generate = {
       '--n_rnn', '2',
       '--sample_length', task.opt.sample_length || 44100 * 5,
       '--n_samples', task.opt.n_samples || 6,
-      '--keep_old_checkpoints', task.opt.keep_old_checkpoints ? 'True' : 'False',
     ]
   },
   after: 'publish',