From 58885f516b7447e7ef824b0e7a93ceca990ed584 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Mon, 24 Sep 2018 03:05:59 +0200
Subject: after train

---
 app/client/modules/pix2pixhd/pix2pixhd.tasks.js | 1 +
 app/relay/modules/pix2pixhd.js                  | 5 +++--
 app/server/util/api.js                          | 1 -
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
index 5d8e084..36c0f36 100644
--- a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
+++ b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
@@ -81,6 +81,7 @@ export const splice_task = (opt) => dispatch => {
     activity: 'splice',
     dataset: opt.title,
     folder_id: opt.folder_id,
+    epochs: 1,
     opt: {
       ...opt,
     }
diff --git a/app/relay/modules/pix2pixhd.js b/app/relay/modules/pix2pixhd.js
index 5f7b1c7..93f9c3a 100644
--- a/app/relay/modules/pix2pixhd.js
+++ b/app/relay/modules/pix2pixhd.js
@@ -64,7 +64,7 @@ const train = {
       '--name', dataset,
       '--model', 'pix2pixHD',
       '--label_nc', 0, '--no_instance',
-      '--niter', task.epochs,
+      '--niter', task.epochs || 1,
       '--niter_decay', 0,
       '--save_epoch_freq', 1,
     ]
@@ -243,7 +243,8 @@ const splice = {
       '--folder_id', task.opt.folder_id,
       '--endpoint', process.env.API_REMOTE + '/api/file/',
     ]
-  }
+  },
+  after: 'train'
 }
 
 function pad(num, size) {
diff --git a/app/server/util/api.js b/app/server/util/api.js
index b3f417d..65215f1 100644
--- a/app/server/util/api.js
+++ b/app/server/util/api.js
@@ -23,7 +23,6 @@ export function api (app, type) {
   // create
   app.post(type_uri, (req, res) => {
     console.log('create', type)
-    console.log(req.body)
     model.create(req.body).then( (data) => {
       res.json(data)
     }).catch( (err) => {
-- 
cgit v1.2.3-70-g09d2