diff options
| -rw-r--r-- | app/relay/modules/pix2pixhd.js | 7 | ||||
| -rw-r--r-- | app/relay/remote.js | 2 |
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() |