diff options
| -rw-r--r-- | app/client/modules/pix2pixhd/views/pix2pixhd.train.js | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js index 1a23618..78def7a 100644 --- a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js +++ b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js @@ -101,7 +101,7 @@ class Pix2PixHDTrain extends Component { if (folder.name === 'results') return const datasets = folder.datasets.map(name => { const dataset = datasetLookup[name] - if (dataset.isBuilt && dataset.checkpoints.length) { + if (dataset.isBuilt) { // && dataset.checkpoints.length return name } return null @@ -112,6 +112,8 @@ class Pix2PixHDTrain extends Component { } }).filter(n => !!n && !!n.options.length).sort((a,b) => a.name.localeCompare(b.name)) + let epochs = this.props.pix2pixhd.epochs + if (!epochs || !epochs.length) epochs = ["latest"] // console.log('state', this.props.pix2pixhd.data.epochs) // console.log(this.state.dataset, this.state.epoch) // console.log(queue) @@ -134,7 +136,7 @@ class Pix2PixHDTrain extends Component { <Select title="Epoch" name="epoch" - options={this.props.pix2pixhd.epochs} + options={epochs} onChange={this.handleChange} value={this.state.epoch} /> |