From eda6b509f909a6ab5d9ab368bf8287fc8e64ba13 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 7 Jun 2018 03:08:33 +0200 Subject: train dataroot --- app/relay/modules/pix2pix.js | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'app/relay/modules') 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, -- cgit v1.2.3-70-g09d2