diff options
Diffstat (limited to 'app/relay/modules/pix2pix.js')
| -rw-r--r-- | app/relay/modules/pix2pix.js | 3 |
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, |