fix loading bars

1 files changed, 32 insertions, 0 deletions

diff --git a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
index f3c5342..92c0ff4 100644
--- a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
+++ b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
@@ -52,3 +52,35 @@ export const live_task = (sequence, checkpoint, opt) => dispatch => {
   console.log('add live task')
   return actions.queue.add_task(task)
 }
+
+export const augment_task = (opt) => dispatch => {
+  const task = {
+    module: module.name,
+    activity: 'augment',
+    dataset: sequence,
+    checkpoint,
+    opt: {
+      ...opt,
+      poll_delay: 0.01,
+    }
+  }
+  console.log(task)
+  console.log('add live task')
+  return actions.queue.add_task(task)
+}
+
+export const clear_recursive_task = (opt) => dispatch => {
+  const task = {
+    module: module.name,
+    activity: 'clear_recursive',
+    dataset: sequence,
+    checkpoint,
+    opt: {
+      ...opt,
+    }
+  }
+  console.log(task)
+  console.log('add live task')
+  return actions.queue.add_task(task)
+}
+