From 48922f830b4cdbffdb7e8b61ea869cd89328052f Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Thu, 7 Jun 2018 00:58:55 +0200
Subject: check callback

---
 app/relay/modules/pix2pix.js | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'app/relay/modules/pix2pix.js')

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,
     ]
   },
 }
-- 
cgit v1.2.3-70-g09d2