1 files changed, 27 insertions, 0 deletions
diff --git a/app/relay/modules/pix2pixhd.js b/app/relay/modules/pix2pixhd.js
index 7a8c7f4..0391480 100644
--- a/app/relay/modules/pix2pixhd.js
+++ b/app/relay/modules/pix2pixhd.js
@@ -132,8 +132,12 @@ const augment = {
     if (!!task.opt.augment_name) {
       args.push('--augment-name', task.opt.augment_name)
     }
+    if (!!task.opt.no_symlinks) {
+      args.push('--no-symlinks')
+    }
     return args
   },
+  after: 'render_recursive',
 }
 const clear_recursive = {
   type: 'pytorch',
@@ -191,12 +195,34 @@ const render = {
   script: 'dir-to-movie.pl',
   params: (task) => {
     return [
+      '--path', 'results',
       '--tag', task.dataset,
       '--module', task.module,
       '--endpoint', process.env.API_REMOTE + '/api/folder/' + task.opt.folder_id + '/upload/',
     ]
   }
 }
+const render_recursive = {
+  type: 'perl',
+  script: 'dir-to-movie.pl',
+  params: (task) => {
+    if (!!task.opt.mov) {
+      return 'CANCEL'
+    }
+    return [
+      '--path', 'recursive',
+      '--tag', task.checkpoint + '/' + task.dataset + '_' + pad(task.augment_take, 5) + '_' + pad(task.augment_make, 5),
+      '--module', task.module,
+      '--endpoint', process.env.API_REMOTE + '/api/folder/' + task.opt.folder_id + '/upload/',
+    ]
+  }
+}
+
+function pad(num, size) {
+  var s = num + "";
+  while (s.length < size) s = "0" + s;
+  return s;
+}
 
 export default {
   name, cwd, env,
@@ -209,5 +235,6 @@ export default {
     clear_recursive,
     live,
     render,
+    render_recursive,
   }
 }