diff options
Diffstat (limited to 'app/client/modules/pix2pixhd/views/pix2pixhd.train.js')
| -rw-r--r-- | app/client/modules/pix2pixhd/views/pix2pixhd.train.js | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js index 05ad638..06caa5a 100644 --- a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js +++ b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js @@ -52,6 +52,7 @@ class Pix2PixHDTrain extends Component { if (prevState.checkpoint_name !== this.state.checkpoint_name) { this.setState({ epoch: 'latest' }) this.props.actions.list_epochs(this.state.checkpoint_name) + this.props.actions.count_dataset(this.state.checkpoint_name) } } handleChange(name, value){ @@ -152,6 +153,7 @@ class Pix2PixHDTrain extends Component { </Group> <Group title='Augmentation Grid'> <AugmentationGrid + checkpoint={this.props.pix2pixhd.checkpoint} take={[1,2,3,4,5,10,15,20,25,50,75,100,200,300,400,500,1000]} make={[1,2,3,4,5,10,15,20,25,50,75,100,200,]} onAugment={(augment_take, augment_make) => { |