From 8955da42bd3b66543ecea80f00ec979a6140f5bc Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Sat, 16 Jun 2018 13:40:59 +0200
Subject: deploy bundle

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

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

diff --git a/app/relay/modules/pix2pix.js b/app/relay/modules/pix2pix.js
index d6a4161..54fcdc3 100644
--- a/app/relay/modules/pix2pix.js
+++ b/app/relay/modules/pix2pix.js
@@ -60,6 +60,7 @@ const train = {
   params: (task) => {
     return [
       '--dataroot', path.join(cwd, 'datasets', task.module, task.dataset),
+      '--module-name', task.module,
       '--name', task.dataset,
       '--model', 'pix2pix',
       '--loadSize', task.opt.load_size || 264,
@@ -74,7 +75,6 @@ const train = {
       '--no_lsgan',
       '--norm', 'batch',
       '--pool_size', '0',
-      '--cortex_module', task.module,
     ]
   },
 }
@@ -84,6 +84,7 @@ const generate = {
   params: (task) => {
     return [
       '--dataroot', '/sequences/' + task.module + '/' + task.dataset,
+      '--module-name', task.module,
       '--name', task.dataset,
       '--start_img', '/sequences/' + task.module + '/' + task.dataset + '/frame_00001.png',
       '--how_many', 1000,
-- 
cgit v1.2.3-70-g09d2