train dataroot

1 files changed, 4 insertions, 3 deletions

diff --git a/app/client/modules/pix2pix/pix2pix.tasks.js b/app/client/modules/pix2pix/pix2pix.tasks.js
index 9572418..db7d4b1 100644
--- a/app/client/modules/pix2pix/pix2pix.tasks.js
+++ b/app/client/modules/pix2pix/pix2pix.tasks.js
@@ -5,14 +5,15 @@ import types from '../../types'
 
 import actions from '../../actions'
 
-export const live_task = (dataset, checkpoint, epochs=1) => dispatch => {
+export const live_task = (sequence, checkpoint) => dispatch => {
   const task = {
     module: 'pix2pix',
     activity: 'live',
-    dataset: dataset.name,
-    checkpoint: checkpoint,
+    dataset: sequence,
+    checkpoint,
   }
   console.log(task)
+  console.log('add live task')
   return actions.queue.add_task(task)
 }