diff options
Diffstat (limited to 'app/client/modules/pix2pixhd/views')
| -rw-r--r-- | app/client/modules/pix2pixhd/views/pix2pixhd.train.js | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js index cd6507f..df3a1f2 100644 --- a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js +++ b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js @@ -28,6 +28,7 @@ class Pix2PixHDTrain extends Component { this.state = { checkpoint_name: '', epoch: 'latest', + augment_name: '', augment_take: 100, augment_make: 20, } @@ -131,10 +132,16 @@ class Pix2PixHDTrain extends Component { min="1" max="1000" /> + <TextInput + name="augment_name" + title="Tag this epoch" + value={this.state.augment_name} + onChange={this.handleChange} + /> <Button title="Augment dataset" value="Augment" - onClick={() => remote.augment_task(dataset, pix2pixhd.folder_id, 1)} + onClick={() => remote.augment_task(this.state.checkpoint_name, this.state)} /> </Group> @@ -142,7 +149,7 @@ class Pix2PixHDTrain extends Component { <Button title="Train one epoch" value="Train" - onClick={() => remote.train_task(dataset, pix2pixhd.folder_id, 1)} + onClick={() => remote.train_task(this.state.checkpoint_name, pix2pixhd.folder_id, 1)} /> </Group> @@ -150,7 +157,7 @@ class Pix2PixHDTrain extends Component { <Button title="Delete recursive frames" value="Clear" - onClick={() => remote.clear_recursive_task(dataset, pix2pixhd.folder_id, 1)} + onClick={() => remote.clear_recursive_task(this.state.checkpoint_name)} /> </Group> </div> |