1 files changed, 13 insertions, 2 deletions
diff --git a/app/relay/modules/pix2pix.js b/app/relay/modules/pix2pix.js
index 520c4e8..fc612a8 100644
--- a/app/relay/modules/pix2pix.js
+++ b/app/relay/modules/pix2pix.js
@@ -30,7 +30,18 @@ const fetch = {
     return null
   }
 }
-
+const make_folds = {
+  type: 'pytorch',
+  script: 'datasets/combine_A_and_B.py',
+  params: (task) => {
+    return [
+      '--fold_A', task.module + '/a_b/' + task.dataset + '/A',
+      '--fold_B', task.module + '/a_b/' + task.dataset + '/B',
+      '--fold_AB', task.module + '/datasets/' + task.dataset,
+    ]
+  },
+  after: 'combine_folds',
+}
 const combine_folds = {
   type: 'pytorch',
   script: 'datasets/combine_A_and_B.py',
@@ -121,6 +132,6 @@ const live = {
 export default {
   name, cwd,
   activities: {
-    fetch, combine_folds, train, generate, live,
+    fetch, make_folds, combine_folds, train, generate, live,
   }
 }