summaryrefslogtreecommitdiff
path: root/app/relay/modules/pix2pix.js
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-06-07 03:08:33 +0200
committerJules Laplace <julescarbon@gmail.com>2018-06-07 03:08:33 +0200
commiteda6b509f909a6ab5d9ab368bf8287fc8e64ba13 (patch)
treed6e0c6c08222be76d54f98ebbd2a878fa546f68b /app/relay/modules/pix2pix.js
parent798a9f835899103d79397a9461ff933d4000634c (diff)
train dataroot
Diffstat (limited to 'app/relay/modules/pix2pix.js')
-rw-r--r--app/relay/modules/pix2pix.js2
1 files changed, 1 insertions, 1 deletions
diff --git a/app/relay/modules/pix2pix.js b/app/relay/modules/pix2pix.js
index aba4312..c590379 100644
--- a/app/relay/modules/pix2pix.js
+++ b/app/relay/modules/pix2pix.js
@@ -47,7 +47,7 @@ const train = {
script: 'train.py',
params: (task) => {
return [
- '--dataroot', './datasets/' + task.dataset,
+ '--dataroot', path.join(process.env.PIX2PIX_CWD, './datasets/', task.dataset),
'--name', task.dataset,
'--model', 'pix2pix',
'--loadSize', opt.load_size || 264,