is_active

2 files changed, 6 insertions, 3 deletions

diff --git a/app/relay/modules/pix2pixhd.js b/app/relay/modules/pix2pixhd.js
index d5d6ada..ebfebf2 100644
--- a/app/relay/modules/pix2pixhd.js
+++ b/app/relay/modules/pix2pixhd.js
@@ -49,9 +49,10 @@ const train = {
     const datasets_path = path.join(cwd, 'datasets', dataset)
     const checkpoints_path = path.join(cwd, 'checkpoints', dataset)
     const iter_txt = path.join(checkpoints_path, 'iter.txt')
-    console.log(dataset, datasets_path, checkpoints_path)
-    if (fs.existsSync(checkpoints_path)) {
-      const iter = fs.readFileSync(iter_txt).split('\n')
+    console.log(dataset, iter_txt)
+    if (fs.existsSync(iter_txt)) {
+      const iter = fs.readFileSync(iter_txt).toString().split('\n');
+      console.log(iter)
       epoch = iter[0] || 0
       console.log(task.module, dataset, '=>', epoch, task.epochs)
     } else {
diff --git a/app/relay/remote.js b/app/relay/remote.js
index ebd56b5..8f72f03 100644
--- a/app/relay/remote.js
+++ b/app/relay/remote.js
@@ -65,6 +65,8 @@ remote.on('task', (data) => {
       queue.remove_task(data.task)
       break
     case 'start_queue':
+      queue = runner.get_queue(data.task)
+      console.log(queue)
       if (! queue.is_active()) {
         queue.activate()
         runner.start_queue()