summaryrefslogtreecommitdiff
path: root/app/relay
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-06-16 13:40:59 +0200
committerJules Laplace <julescarbon@gmail.com>2018-06-16 13:40:59 +0200
commit8955da42bd3b66543ecea80f00ec979a6140f5bc (patch)
treebee6366fed6c6ba9298021bac8e4c57ba34913c5 /app/relay
parent8a1db5c3583818d9ed5d998edfe89b88a4553a1b (diff)
deploy bundle
Diffstat (limited to 'app/relay')
-rw-r--r--app/relay/modules/pix2pix.js3
1 files changed, 2 insertions, 1 deletions
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,