diff options
Diffstat (limited to 'app')
| -rw-r--r-- | app/client/modules/pix2pixhd/views/pix2pixhd.train.js | 6 | ||||
| -rw-r--r-- | app/relay/modules/pix2pixhd.js | 1 |
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, |