From ccc9f74102e4432ac8341395174b0004b1435e32 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Sun, 16 Sep 2018 16:46:37 +0200
Subject: surface sequence/dataset count...

---
 app/client/modules/pix2pixhd/views/pix2pixhd.train.js | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'app/client/modules/pix2pixhd/views')

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) => {
-- 
cgit v1.2.3-70-g09d2