1 files changed, 3 insertions, 2 deletions
diff --git a/app/relay/modules/pix2pix.js b/app/relay/modules/pix2pix.js
index 0c6636a..72d5c7e 100644
--- a/app/relay/modules/pix2pix.js
+++ b/app/relay/modules/pix2pix.js
@@ -23,8 +23,8 @@ const fetch = {
   listen: (task, line, i) => {
     // here i need to bridge again... get the filename that comes back from youtube-dl
     // and tell the cortex that URL -> fn and add the filename!
-    if ( line.match(/^youtube-dl got fn, /) ) {
-      let filename = line.split(' => ')[1].trim()
+    if ( line.match(/^created dataset: /) ) {
+      let filename = line.split(': ')[1].trim()
       task.dataset = filename.split('.')[0]
       task.opt.filename = filename
       return { type: 'progress', action: 'resolve_dataset', task, }
@@ -64,6 +64,7 @@ const train = {
       '--no_lsgan',
       '--norm', 'batch',
       '--pool_size', '0',
+      '--cortex_module', task.module,
     ]
   },
 }