diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-06-07 03:18:24 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-06-07 03:18:24 +0200 |
| commit | 7f0ce9a33f7e1bd473a038a86715e443ab225065 (patch) | |
| tree | 761292c548e47d30a4d1257f1aa6e0c05b2dd9a3 /app/relay/modules/pix2pix.js | |
| parent | 9ea790731e7454b69033d6b0c8e1cb4c1371f31a (diff) | |
train dataroot
Diffstat (limited to 'app/relay/modules/pix2pix.js')
| -rw-r--r-- | app/relay/modules/pix2pix.js | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/app/relay/modules/pix2pix.js b/app/relay/modules/pix2pix.js index 76a52fc..bb4226b 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', path.join(process.env.PIX2PIX_CWD, 'datasets', task.dataset), + '--dataroot', path.join(cwd, 'datasets', task.dataset), '--name', task.dataset, '--model', 'pix2pix', '--loadSize', opt.load_size || 264, @@ -92,8 +92,8 @@ const live = { params: (task) => { console.log(task) return [ - '--dataroot', path.join(process.env.PIX2PIX_CWD, 'sequences', task.module, task.dataset), - '--start_img', path.join(process.env.PIX2PIX_CWD, 'sequences', task.module, task.dataset, 'frame_00001.png'), + '--dataroot', path.join(cwd, 'sequences', task.module, task.dataset), + '--start_img', path.join(cwd, 'sequences', task.module, task.dataset, 'frame_00001.png'), '--experiment', task.checkpoint, '--name', task.checkpoint, '--module-name', task.module, |