glob

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 = {