surface sequence/dataset count...

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) => {