diff options
Diffstat (limited to 'app')
| -rw-r--r-- | app/relay/modules/pix2pixhd.js | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/app/relay/modules/pix2pixhd.js b/app/relay/modules/pix2pixhd.js index a90fc15..250fdd9 100644 --- a/app/relay/modules/pix2pixhd.js +++ b/app/relay/modules/pix2pixhd.js @@ -117,7 +117,7 @@ const augment = { const datasets_path = path.join(cwd, 'datasets', dataset) const checkpoints_path = path.join(cwd, 'checkpoints', dataset) // supply render_dir - return [ + let args = [ '--dataroot', datasets_path, '--results_dir', './recursive', '--module_name', task.module, @@ -126,9 +126,12 @@ const augment = { '--label_nc', 0, '--no_instance', '--augment-take', task.opt.augment_take, '--augment-make', task.opt.augment_make, - '--augment-name', task.opt.augment_name, - '--which_epoch', task.opt.epoch, + '--which_epoch', task.opt.epoch || "latest", ] + if (!!task.opt.augment_name) { + args.push('--augment-name', task.opt.augment_name) + } + return args }, } const clear_recursive = { |