3 files changed, 4 insertions, 2 deletions
diff --git a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
index 412f066..8fec652 100644
--- a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
+++ b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
@@ -62,6 +62,7 @@ export const augment_task = (dataset, opt) => dispatch => {
     module: module.name,
     activity: 'augment',
     dataset,
+    folder_id: opt.folder_id,
     opt: {
       ...opt,
     }
diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
index 98d5e88..9c8aacc 100644
--- a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
+++ b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
@@ -164,7 +164,7 @@ class Pix2PixHDTrain extends Component {
               <Button
                 title="Make a movie without augmenting"
                 value="Generate"
-                onClick={() => this.props.remote.augment_task(this.state.checkpoint_name, { ...this.state, no_symlinks: true, mov: true })}
+                onClick={() => this.props.remote.augment_task(this.state.checkpoint_name, { ...this.state, no_symlinks: true, mov: true, folder_id: this.props.pix2pixhd.data.resultsFolder.id })}
               />
             </Group>
             <Group title='Augmentation Presets'>
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;