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/relay/modules/pix2pixhd.js | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'app/relay')

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) {
-- 
cgit v1.2.3-70-g09d2