1 files changed, 2 insertions, 3 deletions
diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
index ae46fdb..6aade36 100644
--- a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
+++ b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
@@ -37,7 +37,8 @@ class Pix2PixHDTrain extends Component {
   }
   componentDidMount(){
     const id = this.props.match.params.id || localStorage.getItem('pix2pixhd.last_id')
-    console.log('load dataset:', id)
+    const dataset = localStorage.getItem('pix2pixhd.last_dataset')
+    console.log('load dataset:', id, dataset)
     const { match, pix2pixhd, actions } = this.props
     if (id === 'new') return
     if (id) {
@@ -48,7 +49,6 @@ class Pix2PixHDTrain extends Component {
     } else {
       this.props.history.push('/pix2pixhd/new/')
     }
-    const dataset = localStorage.getItem('pix2pixhd.last_dataset')
     if (dataset) {
       this.setState({ checkpoint_name: dataset })
     }
@@ -126,7 +126,6 @@ class Pix2PixHDTrain extends Component {
           <Group title='Sequence Editor'>
             <SequenceEditor
               module={pix2pixhdModule}
-              dataset={this.state.checkpoint_name}
               checkpoint={this.props.pix2pixhd.checkpoint}
             />
           </Group>