oop

1 files changed, 3 insertions, 3 deletions

diff --git a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
index 2e7cbbd..f3c5342 100644
--- a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
+++ b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
@@ -31,21 +31,21 @@ export const train_task = (dataset, folder_id, epochs=1) => dispatch => {
     epochs: epochs,
     opt: {
       folder_id: folder_id,
-      load_size: 264, // switch to 256 for pix2wav
     }
   }
   console.log(task)
   return actions.queue.add_task(task)
 }
 
-export const live_task = (sequence, checkpoint) => dispatch => {
+export const live_task = (sequence, checkpoint, opt) => dispatch => {
   const task = {
     module: module.name,
     activity: 'live',
     dataset: sequence,
     checkpoint,
     opt: {
-      poll_delay: 0.09,
+      ...opt,
+      poll_delay: 0.01,
     }
   }
   console.log(task)