argz

1 files changed, 2 insertions, 1 deletions

diff --git a/app/relay/modules/pix2pixhd.js b/app/relay/modules/pix2pixhd.js
index e05f427..b4a206a 100644
--- a/app/relay/modules/pix2pixhd.js
+++ b/app/relay/modules/pix2pixhd.js
@@ -210,7 +210,7 @@ const render_recursive = {
       console.log('will not render mov')
       return 'CANCEL'
     }
-    const tag = task.checkpoint + '/' + task.dataset + '_' + pad(task.augment_take, 5) + '_' + pad(task.augment_make, 5)
+    const tag = task.dataset + '/' + pad(task.opt.epoch || 'latest', 2) + '_' + pad(task.opt.augment_take, 5) + '_' + pad(task.opt.augment_make, 5)
     console.log('rendering recursve path:', tag)
     return [
       '--path', 'recursive',
@@ -222,6 +222,7 @@ const render_recursive = {
 }
 
 function pad(num, size) {
+  if (isNaN(parseInt(num))) return num;
   var s = num + "";
   while (s.length < size) s = "0" + s;
   return s;