seq name

2 files changed, 4 insertions, 3 deletions

diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
index df3a1f2..ace93e4 100644
--- a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
+++ b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
@@ -141,7 +141,7 @@ class Pix2PixHDTrain extends Component {
             <Button
               title="Augment dataset"
               value="Augment"
-              onClick={() => remote.augment_task(this.state.checkpoint_name, this.state)}
+              onClick={() => this.props.remote.augment_task(this.state.checkpoint_name, this.state)}
             />
           </Group>
 
@@ -149,7 +149,7 @@ class Pix2PixHDTrain extends Component {
             <Button
               title="Train one epoch"
               value="Train"
-              onClick={() => remote.train_task(this.state.checkpoint_name, pix2pixhd.folder_id, 1)}
+              onClick={() => this.props.remote.train_task(this.state.checkpoint_name, pix2pixhd.folder_id, 1)}
             />
           </Group>
 
@@ -157,7 +157,7 @@ class Pix2PixHDTrain extends Component {
             <Button
               title="Delete recursive frames"
               value="Clear"
-              onClick={() => remote.clear_recursive_task(this.state.checkpoint_name)}
+              onClick={() => this.props.remote.clear_recursive_task(this.state.checkpoint_name)}
             />
           </Group>
         </div>
diff --git a/app/relay/modules/pix2pixhd.js b/app/relay/modules/pix2pixhd.js
index 250fdd9..54f916d 100644
--- a/app/relay/modules/pix2pixhd.js
+++ b/app/relay/modules/pix2pixhd.js
@@ -122,6 +122,7 @@ const augment = {
       '--results_dir', './recursive',
       '--module_name', task.module,
       '--name', dataset,
+      '--sequence-name', dataset,
       '--model', 'pix2pixHD',
       '--label_nc', 0, '--no_instance',
       '--augment-take', task.opt.augment_take,